hexo-butterfly-douban/index.js

Generate douban pages of books , movies and games for Hexo.

/* global hexo */

'use strict'
const fs = require('hexo-fs')
const path = require('path')
const log = require('hexo-log')({
  debug: false,
  silent: false
})

hexo.extend.generator.register('books', function (locals) {
  if (!this.config.douban || !this.config.douban.builtin) {
    return
  }
  return require('./lib/books-generator').call(this, locals)
})

hexo.extend.generator.register('movies', function (locals) {
  if (!this.config.douban || !this.config.douban.builtin) {
    return
  }
  return require('./lib/movies-generator').call(this, locals)
})

hexo.extend.generator.register('games', function (locals) {
  if (!this.config.douban || !this.config.douban.builtin) {
    return
  }
  return require('./lib/games-generator').call(this, locals)
})

const options = {
  options: [
    { name: '-b, --books', desc: 'Generate douban books only' },
    { name: '-m, --movies', desc: 'Generate douban movies only' },
    { name: '-g, --games', desc: 'Generate douban games only' }
  ]
}

hexo.extend.console.register('douban', 'Generate pages from douban', options, function (args) {
  const names = []

  if (args.b || args.books) {
    names.push('books')
  }
  if (args.m || args.movies) {
    names.push('movies')
  }

  if (args.g || args.games) {
    names.push('games')
  }
  if (names.length === 0) {
    names.push('books', 'movies', 'games')
  }

  const doubanLoadingPath = '/assets/douban-loading.gif'
  // Load static files
  hexo.extend.generator.register('douban-gif', function (locals) {
    return {
      path: doubanLoadingPath,
      data: function () {
        return fs.createReadStream(path.join(__dirname, '/lib/templates/douban-loading.gif'))
      }
    }
  })

  // Register route
  names.forEach(name => {
    hexo.extend.generator.register(name, require('./lib/' + name + '-generator'))
  })

  const self = this
  const publicDir = self.public_dir

  // Generate files
  self.load().then(function () {
    if (!fs.existsSync(publicDir)) {
      fs.mkdirSync(publicDir)
    }
    if (!fs.existsSync(path.join(publicDir, 'assets'))) {
      fs.mkdirSync(path.join(publicDir, 'assets'))
    }
    self.route.get(doubanLoadingPath)._data().then(function (stream) {
      stream.pipe(fs.createWriteStream(path.join(publicDir, doubanLoadingPath)))
    })
    names.forEach(name => {
      const id = name + '/index.html'
      self.route.get(id) && self.route.get(id)._data().then(function (contents) {
        fs.writeFile(path.join(publicDir, id), contents)
        log.info('Generated: %s', id)
      })
    })
  })
})