MikeCoder / hexo-blog-encrypt

'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!';

  }

  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;

    }

    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': () => fs.createReadStream(path.resolve(__dirname, 'lib/blog-encrypt.css')),

    'path': 'css/blog-encrypt.css',

  },

]);