imgur.js

var rstream = require('./rstream.js');
var request = require('request');
require("dotenv").config();
const EventEmitter = require('events');

var db = require("./db.js");
const regex = /href=".*imgur.com.*"/g;
const baseurl = "https://api.imgur.com/3/album/";
const apikey = process.env.imgurkey;
const emitter = new EventEmitter();
var viablecount = 0, bscount = 0;
module.exports = emitter;

db.get("viable", (count) => {
    viablecount = count;
});
db.get("bs", (count) => {
    bscount = count;
});

rstream.on('post', function (post) {
    //console.log("post", post.title);
    var arr = [];
    arr.push({ url: post.url });
    checkAlbum(arr, function (directs) {
        if (parse(post, directs)) {
            emitter.emit('post', post);
        }
    });
});

rstream.on('comment', function (comment) {
    //console.log("comment", comment.body);
    var urls = comment.body_html.match(regex);
    if (urls) {
        for (var i = 0; i < urls.length; i++) {
            urls[i] = urls[i].split('"')[1];
            urls[i] = { url: urls[i] };
        }
        checkAlbum(urls, function (directs) {
            if (parse(comment, directs)) {
                emitter.emit('comment', comment);
            }
        });
    }
});

function parse(poc, directs) {
    if (directs.length !== 0) {
        poc.direct = directs;
        //console.log("single img album");
        bscount++;
        if (bscount % 25 === 0) {
            db.set("bs", bscount);
            db.set("viable", viablecount);
        }
        return true;
    } else {
        //console.log("viable album");
        viablecount++;
        return false;
    }
}

rstream.on('error', function (err) {
    emitter.emit('error', err);
});

function checkAlbum(urls, callback) {
    //console.log(urls);
    checkSingleRec(urls, 0, callback);
}

function checkSingleRec(urls, index, callback) {
    if (index >= urls.length) {
        callback(urls.filter(function (e) {
            return e.imgurdirect !== undefined;
        }));
    } else {
        callapi(urls[index].url, function (body) {
            if (body.data.images_count == 1) {
                var imageLink = body.data.images[0].link;

                // Replace gif links with gif video links
                if (imageLink.indexOf(".gif") > -1 && imageLink.indexOf(".gifv") == -1) {
                    imageLink = imageLink.replace(".gif", ".gifv");
                }

                // Add https protocol if http protocol was found
                if (imageLink.indexOf("http://") === 0) {
                    imageLink = "https" + imageLink.substring(4);
                }

                urls[index].imgurdirect = imageLink;
            }
            checkSingleRec(urls, index + 1, callback);
        });
    }
}

function callapi(url, callback) {
    var requrl = baseurl + url.split("/a/")[1];
    request(requrl, {
        headers: {
            "authorization": "Client-ID " + apikey
        }
    }, function (err, res, body) {
        try {
            body = JSON.parse(body);
        } catch (err) {
            //fs.writeFileSync("error"+Date.now()+".html");
            console.log("imgur request err:", body);
            return;
        }
        callback(body);
    });
}