MikeCoder / hexo-blog-encrypt

index.js

/* global hexo, __dirname */

'use strict';

const fs = require('hexo-fs');
const path = require('path');
const CryptoJS = require('crypto-js');
const log = require('hexo-log')({
  'debug': false,
  'silent': false,
});

hexo.extend.filter.register('after_post_render', function encrypt (data) {

  // Close the encrypt function
  if (!('encrypt' in hexo.config && hexo.config.encrypt && 'enable' in hexo.config.encrypt && hexo.config.encrypt.enable)) {

    return data;

  }
  if (!('default_template' in hexo.config.encrypt && hexo.config.encrypt.default_template)) { // No such template

    hexo.config.encrypt.default_template = fs.readFileSync(path.resolve(__dirname, './template.html'));

  }
  if (!('default_abstract' in hexo.config.encrypt && hexo.config.encrypt.default_abstract)) { // No read more info

    hexo.config.encrypt.default_abstract = 'The article has been encrypted, please enter your password to view.<br>';

  }
  if (!('default_message' in hexo.config.encrypt && hexo.config.encrypt.default_message)) { // No message

    hexo.config.encrypt.default_message = 'Please enter the password to read the blog.';

  }
  if (!('default_decryption_error' in hexo.config.encrypt && hexo.config.encrypt.default_decryption_error)) { // Wrong password

    hexo.config.encrypt.default_decryption_error = 'Incorrect Password!';

  }
  if (!('default_no_content_error' in hexo.config.encrypt && hexo.config.encrypt.default_no_content_error)) { // No content

    hexo.config.encrypt.default_no_content_error = 'No content to display!';

  }
  
  var tag1,tag2;
  if (!('password' in data && data.password) && ('tags' in hexo.config.encrypt)) {
    outer:
    for(var i in data.tags.data){

A for in loop automatically includes the property of any prototype object, consider checking the key using hasOwnProperty.

      tag1 = data.tags.data[i].name
      console.log(tag1)

console.log looks like debug code. Are you sure you do not want to remove it?

      for(var j in hexo.config.encrypt.tags){

A for in loop automatically includes the property of any prototype object, consider checking the key using hasOwnProperty.

        tag2 = hexo.config.encrypt.tags[j].name
        if (tag1==tag2){
          data.password = hexo.config.encrypt.tags[j].password;
          console.log(data.password)
          break outer;
        }
      }
    }
  }

  if ('password' in data && data.password) {

    // Use the blog's config first
    log.info(`Encrypted the blog: ${ data.title.trim() }`);

    // Store the origin data
    data.origin = data.content;
    data.encrypt = true;

    if (!('abstract' in data && data.abstract)) {

      data.abstract = hexo.config.encrypt.default_abstract;

    }
    if (!('template' in data && data.template)) {

      data.template = hexo.config.encrypt.default_template;

    }
    if (!('message' in data && data.message)) {

      data.message = hexo.config.encrypt.default_message;

    }
    if (!('decryptionError' in data && data.decryptionError)) {

      data.decryptionError = hexo.config.encrypt.default_decryption_error;

    }
    if (!('noContentError' in data && data.noContentError)) {

      data.noContentError = hexo.config.encrypt.default_no_content_error;

    }

    if (data.content.trim() === '') {

      log.warn('Warning: Your blog has no content, it may cause error when decrypting.');

    }

    data.content = escape(data.content);
    data.content = CryptoJS.enc.Utf8.parse(data.content);
    data.content = CryptoJS.enc.Base64.stringify(data.content);
    data.content = CryptoJS.AES.encrypt(data.content, String(data.password)).toString();

    data.template = data.template.replace('{{content}}', data.content);
    data.template = data.template.replace('{{message}}', data.message);
    data.template = data.template.replace('{{message}}', data.message);
    data.template = data.template.replace('{{decryptionError}}', data.decryptionError);
    data.template = data.template.replace('{{noContentError}}', data.noContentError);

    data.content = data.template;
    data.content += `<script src="${hexo.config.root}lib/crypto-js.js"></script>`;
    data.content += `<script src="${hexo.config.root}lib/blog-encrypt.js"></script>`;
    data.content += `<link href="${hexo.config.root}css/blog-encrypt.css" rel="stylesheet" type="text/css">`;

    data.more = data.abstract;
    data.excerpt = data.more;

  }

  return data;

});

hexo.extend.generator.register('blog-encrypt', () => [
  {
    'data': () => fs.createReadStream(path.resolve(path.dirname(require.resolve('crypto-js')), 'crypto-js.js')),
    'path': 'lib/crypto-js.js',
  }, {
    'data': function () {

      const Readable = require('stream').Readable;
      const stream = new Readable();
      stream.push(fs.readFileSync(path.resolve(__dirname, 'lib/blog-encrypt.js'))
        .replace('callBackReplaceHere', hexo.config.encrypt && hexo.config.encrypt.enable && hexo.config.encrypt.callback ? hexo.config.encrypt.callback : ''));
      stream.push(null); // Indicates the end of the stream
      return stream;

    },
    'path': 'lib/blog-encrypt.js',
  }, {
    'data': () => fs.createReadStream(path.resolve(__dirname, 'lib/blog-encrypt.css')),
    'path': 'css/blog-encrypt.css',
  },
]);