hexo-plugin-aurora/lib/generators/index.js

A plugin for Hexo Aurora theme

/**
 * ObsidiaNext API generator
 * @author TriDiamond <code.tridiamond@gmail.com>
 *
 * - Added the photos fields.
 * - Added the `features` post data.
 * - Added truncate for filtering html tags.
 * - Added statistics.
 * - Added post unique ID property used for gitalk.
 * - Added previous and next post.
 * - Enhanced performance by reducing the number of unnecessary iterations.
 * - Added multi-author support.
 * - Enhanced the performance of the post list script.
 */

const SiteGenerator = require('./site');
const CategoryGenerator = require('./category');
const TagGenerator = require('./tag');
const PostGenerator = require('./post');
const PageGenerator = require('./page');
const StatisticGenerator = require('./statistic');
const SearchGenerator = require('./search');

module.exports = function (hexo) {
  let apiData = [];
  const defaultPages = ['Tags', 'Archives', 'Links'];

  // Remove hexo default generators
  ['page', 'archive', 'category', 'tag', 'links'].forEach(
    (name) => delete hexo.extend.generator.store[name]
  );

  hexo.extend.generator.register('aurora-page', function (site) {
    const pageData = [];
    const themeConfig = hexo.theme.config;

    // Generating default pages
    defaultPages.forEach(function (page) {
      if (themeConfig.menu[page]) {
        pageData.push({
          path: `${page.toLocaleLowerCase()}/index.html`,
          data: {},
          layout: ['index']
        });
      }
    });

    site.pages.forEach(function (page) {
      // About page need to be generated to the root of `public` folder.
      if (page.type === 'about') {
        pageData.push({
          path: page.path,
          data: {},
          layout: ['index']
        });
      } else {
        // All other custom pages are generated into `page` folder.
        pageData.push({
          path: `page/${page.path}`,
          data: {},
          layout: ['index']
        });
      }
    });
    // Generate the page for tag search.
    pageData.push({
      path: 'tags/search/index.html',
      data: {},
      layout: ['index']
    });

    return pageData;
  });

  hexo.extend.generator.register('aurora-api', function (site) {
    return generator(
      Object.assign({}, hexo.config, {
        theme_config: hexo.theme.config
      }),
      site
    );
  });

  function generator(configs, site) {
    const posts = new PostGenerator(site.posts, configs);
    apiData = posts.addPaginationPost(apiData);
    apiData = posts.addArchivesPaginationPost(apiData);
    apiData = posts.addArticles(apiData);
    apiData = posts.addFeatures(apiData);
    apiData = posts.addAuthorPost(apiData);
    // Updated feature status.
    configs.theme_config.theme.feature = posts.isFeature;

    const categories = new CategoryGenerator(site.categories, site.posts, configs);
    apiData = categories.addCategories(apiData);

    const tags = new TagGenerator(site.tags, site.posts, configs);
    apiData = tags.addTags(apiData);

    const pages = new PageGenerator(site.pages);
    apiData = pages.addPages(apiData);

    const statistic = new StatisticGenerator(
      {
        posts: posts.count(),
        categories: categories.count(),
        tags: tags.count()
      },
      posts.data
    );
    apiData = statistic.addStatistics(apiData);

    const search = new SearchGenerator(posts.data, configs);
    apiData = search.addSearchIndex(apiData);

    const siteG = new SiteGenerator(configs);
    apiData = siteG.addSiteConfig(apiData);

    return apiData;
  }
};