Added support for Vidble (image) albums. Replaced extensions.json and reverseLookup with mime-types lib.

src/methods/methods.js

@@ -6,6 +6,8 @@ const redditVideo = require('./redditVideo.js');
 const redditPreview = require('./redditPreview.js');
 const imgurImage = require('./imgurImage.js');
 const imgurAlbum = require('./imgurAlbum.js');
+const vidbleImage = require('./vidbleImage.js');
+const vidbleAlbum = require('./vidbleAlbum.js');
 const gfycat = require('./gfycat.js');
 const eroshare = require('./eroshare.js');
 
@@ -16,6 +18,8 @@ module.exports = {
     redditPreview,
     imgurImage,
     imgurAlbum,
+    vidbleImage,
+    vidbleAlbum,
     gfycat,
     eroshare
 };

src/methods/redditImage.js

@@ -2,14 +2,7 @@
 
 const util = require('util');
 const config = require('config');
-const path = require('path');
-const fetch = require('node-fetch');
-
-const extensions = require('../extensions.json');
-
-function reverseLookup(extension) {
-    return Object.keys(extensions).find(mime => extensions[mime] === extension);
-};
+const mime = require('mime-types');
 
 function redditImage(post) {
     return Promise.resolve({
@@ -19,7 +12,7 @@ function redditImage(post) {
             url: post.url,
             title: post.title,
             datetime: post.datetime,
-            type: reverseLookup(path.extname(post.url)),
+            type: mime.lookup(post.url),
             original: post
         }]
     });

src/methods/redditPreview.js

@@ -4,14 +4,9 @@ const util = require('util');
 const config = require('config');
 const path = require('path');
 const fetch = require('node-fetch');
+const mime = require('mime-types');
 const urlPattern = require('url-pattern');
 
-const extensions = require('../extensions.json');
-
-function reverseLookup(extension) {
-    return Object.keys(extensions).find(mime => extensions[mime] === extension);
-};
-
 function redditPreview(post) {
     return Promise.resolve({
         album: post.preview.length > 1 ? {
@@ -22,14 +17,14 @@ function redditPreview(post) {
             original: post
         } : null,
         items: post.preview.map(image => {
-            const urlComponents = new urlPattern('http(s)\\://i.redditmedia.com/:id(.:ext)(?*)').match(image.url);
+            console.log(mime.lookup(image.url));
 
             return {
                 id: post.host.id || post.id,
                 url: image.url,
                 title: post.title,
                 datetime: post.datetime,
-                type: reverseLookup(`.${urlComponents.ext}`),
+                type: mime.lookup(image.url),
                 preview: true,
                 original: post
             };

src/methods/vidbleAlbum.js

@@ -0,0 +1,55 @@
+ 'use strict';
+
+const util = require('util');
+const config = require('config');
+const fetch = require('node-fetch');
+const urlPattern = require('url-pattern');
+const cheerio = require('cheerio');
+const mime = require('mime-types');
+
+const pattern = new urlPattern('https\\://(www.)vidble.com/:id(_med)(.:ext)');
+
+function vidbleAlbum(post) {
+    return fetch(`https://www.vidble.com/album/${post.host.id}`).then(res => {
+        if(res.status !== 200) {
+            throw new Error(`Could not fetch info for vidble album '${post.host.id}': '${res.error}'`);
+        }
+
+        return res.text();
+    }).then(res => {
+        const $ = cheerio.load(res);
+
+        const title = $('h2').text();
+        const imgUrls = $('img.img2').toArray().map(img => `https://vidble.com${img.attribs.src || img.attribs['data-original']}`);
+
+        const extract = config.library.album.extractSingleItem && imgUrls.length === 1;
+
+        if(extract) {
+            console.log('\x1b[36m%s\x1b[0m', `Extracting single item from album '${post.title}' - ${res.link}`);
+        }
+
+        return {
+            album: extract ? null : {
+                id: post.host.id,
+                url: post.url,
+                title: post.title,
+                datetime: post.datetime
+            },
+            items: imgUrls.map(url => {
+                const components = pattern.match(url);
+                const id = components.id.replace('_med', '');
+                const mimetype = mime.lookup(components.ext);
+
+                return {
+                    extracted: extract,
+                    id: id,
+                    url: `https://vidble.com/${id}.${components.ext}`,
+                    type: mimetype,
+                    datetime: post.datetime
+                };
+            })
+        };
+    });
+};
+
+module.exports = vidbleAlbum;

src/methods/vidbleImage.js

@@ -0,0 +1,26 @@
+ 'use strict';
+
+const util = require('util');
+const config = require('config');
+const path = require('path');
+const fetch = require('node-fetch');
+const $ = require('cheerio');
+
+function vidbleImage(post) {
+    return fetch(`https://vidble.com/${post.host.id}`).then(res => res.text()).then(res => {
+        console.log(res, $('img'));
+
+        if(res.status !== 200) {
+            throw new Error(`Could not fetch info for vidble album '${post.host.id}': '${res.data.error}'`);
+        }
+
+        const extract = config.library.album.extractSingleItem && res.data.images.length === 1;
+
+        if(extract) {
+            console.log('\x1b[36m%s\x1b[0m', `Extracting single item from album '${post.title}' - ${res.data.link}`);
+        }
+
+    });
+};
+
+module.exports = vidbleImage;