nodebb-plugin-smoothshorts/lib/controller.js

Seamless short URLs for topic and post links in NodeBB.

'use strict';

var async = require.main.require('async');
var nconf = require.main.require('nconf');
var winston = require.main.require('winston');
var db = require.main.require('./src/database');
var PluginSocket = require.main.require('./src/socket.io/plugins');
var AdminSocket = require.main.require('./src/socket.io/admin');

var sockets = require('./sockets');
var settings = require('./settings');
var hashing = require('./hashing');
// var services = require('./services');

var SmoothShorts = {
  admin: {},
  purgePost: hashing.purgePost,
  purgeTopic: hashing.purgeTopic,
  shortenPost: hashing.shortenPost,
  shortenTopic: hashing.shortenTopic
};

SmoothShorts.init = function(app, cb) {
  settings.load(function(err) {
    if (err) {
      return cb(err);
    }
    setRoutes(app.router, app.middleware);
    openSockets();
    return cb(null, app);
  });
};

SmoothShorts.admin.addMenuItem = function(custom_header, cb) {
  custom_header.plugins.push({
    route: '/plugins/smoothshorts',
    icon: 'fa-location-arrow',
    name: 'SmoothShorts'
  });
  cb(null, custom_header);
};

function openSockets() {
  PluginSocket.SmoothShorts = sockets.plugin;
  AdminSocket.plugins.SmoothShorts = sockets.admin;
}

function setRoutes(router, middleware) {
  var format = settings.local.shortFormat;
  var path = !format ? '/ss/:hash' : format.slice(format.indexOf('/'));

  router.get(path, validateHash, hashing.resolveHash);
  winston.info('[plugins:SmoothShorts] Listening for short URLs on:', path);

  router.get('/admin/plugins/smoothshorts', middleware.applyCSRF, middleware.admin.buildHeader, renderAdmin);
  router.get('/api/admin/plugins/SmoothShorts', middleware.applyCSRF, renderAdmin);
}

function validateHash(req, res, next) {
  return next(/^[\dabcdef]{7,8}$/.test(req.params.hash) ? void 0 : 'route');
}

function renderAdmin(req, res, next) {
  async.parallel({
    postCount: function postCount(next) {
      db.sortedSetCard('posts:pid', next);
    },
    postHashCount: function postHashCount(next) {
      db.sortedSetCard('posts:smoothshorts', next);
    },
    topicCount: function topicCount(next) {
      db.sortedSetCard('topics:tid', next);
    },
    topicHashCount: function topicHashCount(next) {
      db.sortedSetCard('topics:smoothshorts', next);
    }
  }, function(err, result) {
    if (err) {
      return next(err);
    }
    var data = {
      // left side, "Settings"
      useModKey: settings.local.useModKey,
      modKey: {},
      shortFormat: settings.local.shortFormat,
      shortFormatDefault: nconf.get('url').replace(/http[s]?:\/\//, '') + '/ss/:hash',
      copyButtonClass: settings.local.copyButtonClass,
      // right side, "Status"
      postCount: result.postCount,
      postHashCount: result.postHashCount,
      topicCount: result.topicCount,
      topicHashCount: result.topicHashCount,
      postStatus: result.postCount !== result.postHashCount ? 'cout-warn' : '',
      topicStatus: result.topicCount !== result.topicHashCount ? 'cout-warn' : ''
    };
    data.modKey[settings.local.modKey] = true;
    res.render('admin/plugins/smoothshorts/index', data);
  });
}

module.exports = SmoothShorts;