index.js

/*global hexo: true */

const bodyParser = require('body-parser');
const fs = require('fs');
const path = require('path');
const devicons = JSON.parse(fs.readFileSync(path.resolve(__dirname, './devicon/devicon.json')));
const {remove: removeDiacritics} = require('diacritics');

const sanitize = title => removeDiacritics(title).match(/[a-z ]/gi).join('').replace(/ /g, '-').toLowerCase();
const levenshtein = (a, b) => {
    let t = [];
    let m = a.length;
    let n = b.length;
    if (!m) return n;
    if (!n) return m;
    for (let j = 0; j <= n; j++) {
        t[j] = j;
    }
    let u;
    for (let i = 1; i <= m; i++) {
        u = [i];
        for (let j = 1; j <= n; j++) {
            u[j] = a[i - 1] === b[j - 1] ? t[j - 1] : Math.min(t[j - 1], t[j], u[j - 1]) + 1;
        }
        t = u;
    }
    return u[n];
};


hexo.extend.filter.register('server_middleware', (app) => {
    app.use(hexo.config.root + 'api/', bodyParser.json({
        limit: '50mb'
    }));
    app.use(hexo.config.root + 'api/posts/list', (req, res) => {
        res.end(
            JSON.stringify(
                hexo.model('Post').toArray().map(
                    post => ({
                        title: post.title,
                        author: post.author,
                        date: post.date,
                        tags: post.tags.data.map(c => c.name),
                        id: post._id,
                        uri: '/article/' + post._id.substr(-3) + '/' + sanitize(post.title)
                    })
                )
            )
        );
    });
    app.use(hexo.config.root + 'api/post/content/by-id', (req, res) => {
        res.end(
            JSON.stringify(
                hexo.model('Post').toArray()
                    .filter(post => post._id === req.body.id)
                    .map(
                        post => ({
                            title: post.title,
                            author: post.author,
                            date: post.date,
                            tags: post.tags.data.map(c => c.name),
                            id: post._id,
                            uri: '/article/' + post._id.substr(-3) + '/' + sanitize(post.title),
                            content: post.content
                        })
                    )[0]
            )
        );
    });
    app.use(hexo.config.root + 'api/post/content/by-uri/', (req, res) => {
        res.end(
            JSON.stringify(
                hexo.model('Post').toArray()
                    .filter(post => post._id.endsWith(req.body.id_suffix))
                    .map(post => Object.assign(post, {
                        sanitized: sanitize(post.title),
                        levenshtein: levenshtein(req.body.sanitized_title, sanitize(post.title))
                    }))
                    .sort((a, b) => a.levenshtein-b.levenshtein)
                    .map(
                        post => ({
                            title: post.title,
                            author: post.author,
                            date: post.date,
                            tags: post.tags.data.map(c => c.name),
                            id: post._id,
                            uri: '/article/' + post._id.substr(-3) + '/' + sanitize(post.title),
                            content: post.content
                        })
                    )[0]
            )
        );
    });
    app.use(hexo.config.root + 'api/icon/by-language/', (req, res) => {
        const language = devicons.find(e => e.name === req.url.slice(1));
        const variation = (language && language.versions.svg[0]);
        if (!variation) {
            res.statusCode = 404;
            res.end('');
            return;
        }
        fs.readFile(path.resolve(__dirname, `./devicon/icons/${language.name}/${language.name}-${variation}.svg`), (err, content) => {
            res.setHeader('Cache-Control', 'max-age=2678400');
            res.setHeader('Content-Type', 'image/svg+xml');
            res.end(content.toString());
        });
    });
});