bot.js

var bigimage = require('./userscript.user.js');
var sizeOf = require('image-size');
var probe = require('probe-image-size');
var http = require('http');
var https = require('https');
var url = require('url');
const NodeCache = require( "node-cache" );
var fs = require("fs");

var blacklist_json = JSON.parse(fs.readFileSync("./blacklist.json"));
//var env_json = JSON.parse(fs.readFileSync("./.env.json"));
var env_json = {};

require('dotenv').config();
env_json.user_agent = process.env.USERAGENT;
env_json.client_id = process.env.CLIENT_ID;
env_json.client_secret = process.env.CLIENT_SECRET;
env_json.refresh_token = process.env.REFRESH_TOKEN;
env_json.access_token = process.env.ACCESS_TOKEN;
//env_json.username = process.env.REDDIT_USER;
//env_json.password = process.env.REDDIT_PASS;

//console.dir(env_json);

var thresh_px = 200;

const Snoowrap = require('snoowrap');
const Snoostorm = require('snoostorm');

const r = new Snoowrap(env_json);

r.config({requestDelay: 1001});
const client = new Snoostorm(r);

var blacklist = [
  // Posts that would be insensitive to comment on
  "killed",
  "died",
  "death",
  "murdered",
  "murder",

  // Posts in-between the first and second category
  "embarrassed",
  "embarrassing",
  "cringe",
  "cringiest",
  "cringefest",
  "shame",
  "shaming",

  // Posts that people commonly dislike the bot commenting on
  "trump",
  "hillary",
  "punch",
  "punchable",
  "ugly",
  "fat"
];

function inblacklist(x) {
  var black = false;
  x.toLowerCase().split(" ").forEach((word) => {
    word = word
      .replace(/^[^a-z]*/, "")
      .replace(/[^a-z]*$/, "");
    if (blacklist.indexOf(word) >= 0) {
      black = true;
      return;
    }
  });

  return black;
}


/*function getimagesize(imgUrl, olddata) {
  var options = url.parse(imgUrl);
  return new Promise((resolve, reject) => {
    var getter = http;
    if (options.protocol === "https:")
      getter = https;
    getter.get(options, function(response) {
      if (response.statusCode !== 200) {
        reject({
          "status": response.statusCode
        });
        return;
      }

      var finish = function() {
        var buffer = Buffer.concat(chunks);
        response.destroy();
        try {
          var dimensions = sizeOf(buffer);
          resolve({
            length: length,
            width: dimensions.width,
            height: dimensions.height
          });
          return;
        } catch (e) {
          reject(e);
          return;
        }
      };

      var length = response.getHeader('content-length');
      if (length === olddata.length) {
        reject({
          "identical_length": length
        });
        return;
      }

      var chunks = [];
      var size = 0;
      response.on('data', function (chunk) {
        chunks.push(chunk);
        size += chunk.length;
        if (size > 2048) {
          finish();
        }
      }).on('end',function() {
        finish();
      });
    });
  });
  }*/

function getimagesize(url) {
  if (typeof(url) === "string") {
    return probe(url);
  }

  if (typeof(url.url) === "string") {
    return getimagesize(url.url);
  }

  return new Promise((resolve, reject) => {
    var do_getimage = function(urls, err) {
      if (urls.length === 0) {
        reject(err);
        return;
      }

      getimagesize(urls[0]).then(
        (data) => {
          resolve(data);
        },
        (err) => {
          do_getimage(urls.slice(1), err);
        }
      );
    };

    do_getimage(url.url);
  });
}

function dourl(url, post) {
  var big = bigimage(url, {fill_object:true});

  if (big.url instanceof Array) {
    if (big.url.indexOf(url) >= 0) {
      return;
    }
  } else if (big.url === url) {
    return;
  }

  /*if (big === url) {
    return;
  }*/

  if (post && inblacklist(post.title)) {
    console.log("Post blacklisted:\n" + post.title + "\n" + post.permalink + "\n" + post.url + "\n=====\n\n");
    return;
  }

  console.log(url);
  console.log(big);
  console.log("---");

  getimagesize(url).then(
    (data) => {
      getimagesize(big).then(
        (newdata) => {
          var wr = newdata.width / data.width;
          var hr = newdata.height / data.height;
          var r = (wr + hr) / 2;
          if (r >= 1.1 && (((newdata.width - data.width) > thresh_px &&
                            newdata.height > data.height) ||
                           ((newdata.height - data.height) > thresh_px &&
                            newdata.width > data.width))) {
            var times = "" + r.toFixed(1) + "x";
            if (r < 1.995) {
              times = "" + ((r-1) * 100).toFixed(0) + "%";
            }

            var filesize_text = "";
            var mbs = newdata.length / 1024 / 1024;
            if (mbs > 5) {
              filesize_text = ", " + mbs.toFixed(1) + "MB";
            }
            var comment = times + " larger (" + parseInt(newdata.width) + "x" + parseInt(newdata.height) + filesize_text + ") version of linked image:\n\n" + newdata.url + "\n\n";
            comment += "*****\n\n";
            comment += "^[source&nbsp;code](https://github.com/qsniyg/maxurl)&nbsp;|&nbsp;[website](https://qsniyg.github.io/maxurl/)&nbsp;/&nbsp;[userscript](https://greasyfork.org/en/scripts/36662-image-max-url)&nbsp;(finds&nbsp;larger&nbsp;images)";
            console.log(comment);
            if (post) {
              post.reply(comment).then((comment_data) => {
                comment_data.edit(
                  comment + "&nbsp;|&nbsp;[remove](https://www.reddit.com/message/compose/?to=MaxImageBot&subject=delete:+" + comment_data.id + "&message=delete)"
                );
              });
            }
          } else {
            console.log("Ratio too small: " + wr + ", " + hr);
          }
          console.log("========");
        },
        (err) => {
          console.dir(err);
          return;
        }
      );
    },
    (err) => {
      console.dir(err);
      return;
    }
  );
}

const links = new NodeCache({ stdTTL: 600, checkperiod: 1000 });


// large image
//dourl("https://i.guim.co.uk/img/media/856021cc9b024ee18480297110f6a9f38923b4ee/0_0_15637_9599/master/15637.jpg?w=1920&q=55&auto=format&usm=12&fit=max&s=774b471892a2a1870261227f29e9d77a");

//console.dir(blacklist_json.disallowed);
if (true) {
  var submissionStream = client.SubmissionStream({
    "subreddit": "all",
    "results": 100,
    "pollTime": 2000
  });

  setInterval(() => {
    r.getInbox({"filter":"messages"}).then((inbox) => {
      inbox.forEach((message_data) => {
        if (message_data.subject.indexOf("delete:") !== 0 ||
            message_data.subject.length >= 50 ||
            !message_data["new"]) {
          return;
        }

        var comment = message_data.subject.replace(/.*:[ +]*([A-Za-z0-9_]+).*/, "$1");
        if (comment === message_data.subject)
          return;
        console.log(comment);

        r.getComment(comment).fetch().then((comment_data) => {
          if (comment_data.author.name.toLowerCase() !== "maximagebot")
            return;

          r.getComment(comment_data.parent_id).fetch().then((post_data) => {
            if (post_data.author.name.toLowerCase() !== message_data.author.name.toLowerCase()) {
              return;
            }

            console.log("Deleting " + comment);
            comment_data.delete();
            message_data.deleteFromInbox();
          });
        });
      });
    });
  }, 10*1000);

  submissionStream.on("submission", function(post) {
    if (post.domain.startsWith("self.") || (post.over_18 && false)) {
      return;
    }

    if (post.subreddit.display_name) {
      if (blacklist_json.disallowed.indexOf(post.subreddit.display_name.toLowerCase()) >= 0 ||
          blacklist_json.users.indexOf(post.author.name.toLowerCase()) >= 0) {
        //console.log(post.subreddit);
        return;
      }
    }

    if (links.get(post.permalink) === true) {
      //console.log("Already processed " + post.permalink + ", skipping");
      return;
    }

    links.set(post.permalink, true);

    var url = post.url;
    dourl(url, post);
  });
}