Register unpublish cli command to the console

lib/hexo/post.js

@@ -306,6 +306,59 @@ class Post {
     });
   }
 
+  unpublish(data, replace, callback) {
+    if (!callback && typeof replace === 'function') {
+      callback = replace;
+      replace = false;
+    }
+
+    if (data.layout === 'post') data.layout = 'draft';
+
+    const ctx = this.context;
+    const { config } = ctx;
+    const postDir = join(ctx.source_dir, '_posts');
+    const slug = slugize(data.slug.toString(), { transform: config.filename_case });
+    data.slug = slug;
+    const regex = new RegExp(`^${escapeRegExp(slug)}(?:[^\\/\\\\]+)`);
+    let src = '';
+    const result = {};
+
+    data.layout = (data.layout || config.default_layout).toLowerCase();
+
+    // Find the original post
+    return listDir(postDir).then(list => {
+      const item = list.find(item => regex.test(item));
+      if (!item) throw new Error(`Post "${slug}" does not exist.`);
+
+      // Read the post content
+      src = join(postDir, item);
+      return readFile(src);
+    }).then(content => {
+      // Create draft
+      Object.assign(data, yfmParse(content));
+      data.content = data._content;
+      data._content = undefined;
+
+      return this.create(data, replace);
+    }).then(post => {
+      result.path = post.path;
+      result.content = post.content;
+      return unlink(src);
+    }).then(() => { // Remove the original post file
+      if (!config.post_asset_folder) return;
+
+      // Copy assets
+      const assetSrc = removeExtname(src);
+      const assetDest = removeExtname(result.path);
+
+      return exists(assetSrc).then(exist => {
+        if (!exist) return;
+
+        return copyDir(assetSrc, assetDest).then(() => rmdir(assetSrc));
+      });
+    }).thenReturn(result).asCallback(callback);
+  }
+
   publish(data, replace, callback) {
     if (!callback && typeof replace === 'function') {
       callback = replace;

lib/plugins/console/index.js

@@ -67,6 +67,14 @@ module.exports = function(ctx) {
     ]
   }, require('./publish'));
 
+  console.register('unpublish', 'Moves a published post from _posts to _drafts folder.', {
+    usage: '[layout] <filename>',
+    arguments: [
+      {name: 'layout', desc: 'Post layout. Use post, page, draft or whatever you want.'},
+      {name: 'filename', desc: 'Post filename. "hello-world" for example.'}
+    ]
+  }, require('./publish'));
+
   console.register('render', 'Render files with renderer plugins.', {
     init: true,
     desc: 'Render files with renderer plugins (e.g. Markdown) and save them at the specified path.',

lib/plugins/console/unpublish.js

@@ -0,0 +1,20 @@
+'use strict';
+
+const tildify = require('tildify');
+const { magenta } = require('picocolors');
+
+function unPublishConsole(args) {
+  // Display help message if user didn't input any arguments
+  if (!args._.length) {
+    return this.call('help', {_: ['unpublish']});
+  }
+
+  return this.post.unpublish({
+    slug: args._.pop(),
+    layout: args._.length ? args._[0] : this.config.default_layout
+  }, args.r || args.replace).then(post => {
+    this.log.info('Unpublished: %s', magenta(tildify(post.path)));
+  });
+}
+
+module.exports = unPublishConsole;

test/scripts/console/index.js

@@ -8,6 +8,7 @@ describe('Console', () => {
   require('./migrate');
   require('./new');
   require('./publish');
+  require('./unpublish');
   require('./render');
   require('./list');
   require('./list_post');

test/scripts/console/unpublish.js

@@ -0,0 +1,139 @@
+'use strict';
+
+const { exists, mkdirs, readFile, rmdir, unlink } = require('hexo-fs');
+const moment = require('moment');
+const { join } = require('path');
+const Promise = require('bluebird');
+const { useFakeTimers, spy } = require('sinon');
+
+describe('unpublish', () => {
+  const Hexo = require('../../../lib/hexo');
+  const hexo = new Hexo(join(__dirname, 'unpublish_test'), {silent: true});
+  const unpublish = require('../../../lib/plugins/console/unpublish').bind(hexo);
+  const post = hexo.post;
+  const now = Date.now();
+  let clock;
+
+  before(async () => {
+    clock = useFakeTimers(now);
+
+    await mkdirs(hexo.base_dir);
+    await hexo.init();
+    await hexo.scaffold.set('post', [
+      '---',
+      'title: {{ title }}',
+      'date: {{ date }}',
+      'tags:',
+      '---'
+    ].join('\n'));
+    await hexo.scaffold.set('draft', [
+      '---',
+      'title: {{ title }}',
+      'tags:',
+      '---'
+    ].join('\n'));
+  });
+
+  after(() => {
+    clock.restore();
+    return rmdir(hexo.base_dir);
+  });
+
+  beforeEach(() => post.create({
+    title: 'Hello World',
+    layout: 'post',
+    date: moment(now).format('YYYY-MM-DD HH:mm:ss')
+  }));
+
+  it('slug', async () => {
+    const postPath = join(hexo.source_dir, '_posts', 'Hello-World.md');
+    const path = join(hexo.source_dir, '_drafts', 'Hello-World.md');
+
+    const content = [
+      '---',
+      'title: Hello World',
+      'tags:',
+      '---'
+    ].join('\n') + '\n';
+
+    await unpublish({
+      _: ['Hello-World']
+    });
+
+    const exist = await exists(postPath);
+    const data = await readFile(path);
+
+    exist.should.be.false;
+    data.should.eql(content);
+
+    await unlink(path);
+  });
+
+  it('no args', async () => {
+    const hexo = new Hexo(join(__dirname, 'unpublish_test'), {silent: true});
+    hexo.call = spy();
+    const unpublish = require('../../../lib/plugins/console/unpublish').bind(hexo);
+
+    await unpublish({_: []});
+
+    hexo.call.calledOnce.should.be.true;
+    hexo.call.args[0][0].should.eql('help');
+    hexo.call.args[0][1]._[0].should.eql('unpublish');
+  });
+
+  it('layout', async () => {
+    const path = join(hexo.source_dir, '_posts', 'Hello-World.md');
+    const date = moment(now);
+
+    const content = [
+      '---',
+      'title: Hello World',
+      'date: ' + date.format('YYYY-MM-DD HH:mm:ss'),
+      'tags:',
+      '---'
+    ].join('\n') + '\n';
+
+    await unpublish({
+      _: ['photo', 'Hello-World']
+    });
+    const data = await readFile(path);
+    data.should.eql(content);
+
+    await unlink(path);
+  });
+
+  it('rename if target existed', async () => {
+    const path = join(hexo.source_dir, '_posts', 'Hello-World.md');
+
+    await post.create({
+      title: 'Hello World'
+    });
+    await unpublish({
+      _: ['Hello-World']
+    });
+
+    const exist = await exists(path);
+    exist.should.be.true;
+
+    await Promise.all([
+      unlink(path),
+      unlink(join(hexo.source_dir, '_posts', 'Hello-World.md'))
+    ]);
+  });
+
+  it('replace existing target', async () => {
+    const path = join(hexo.source_dir, '_posts', 'Hello-World.md');
+
+    await post.create({
+      title: 'Hello World'
+    });
+    await unpublish({
+      _: ['Hello-World'],
+      replace: true
+    });
+    const exist = await exists(join(hexo.source_dir, '_drafts', 'Hello-World-1.md'));
+    exist.should.be.false;
+
+    await unlink(path);
+  });
+});