You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
2.8 KiB
103 lines
2.8 KiB
2 years ago
|
'use strict';
|
||
|
|
|
||
|
|
const { Cache } = require('hexo-util');
|
||
|
|
|
||
|
|
class Injector {
|
||
|
|
constructor() {
|
||
|
|
this.store = {
|
||
|
|
head_begin: {},
|
||
|
|
head_end: {},
|
||
|
|
body_begin: {},
|
||
|
|
body_end: {}
|
||
|
|
};
|
||
|
|
|
||
|
|
this.cache = new Cache();
|
||
|
|
}
|
||
|
|
|
||
|
|
list() {
|
||
|
|
return this.store;
|
||
|
|
}
|
||
|
|
|
||
|
|
get(entry, to = 'default') {
|
||
|
|
return Array.from(this.store[entry][to] || []);
|
||
|
|
}
|
||
|
|
|
||
|
|
getText(entry, to = 'default') {
|
||
|
|
const arr = this.get(entry, to);
|
||
|
|
if (!arr || !arr.length) return '';
|
||
|
|
return arr.join('');
|
||
|
|
}
|
||
|
|
|
||
|
|
getSize(entry) {
|
||
|
|
return this.cache.apply(`${entry}-size`, Object.keys(this.store[entry]).length);
|
||
|
|
}
|
||
|
|
|
||
|
|
register(entry, value, to = 'default') {
|
||
|
|
if (!entry) throw new TypeError('entry is required');
|
||
|
|
if (typeof value === 'function') value = value();
|
||
|
|
|
||
|
|
const entryMap = this.store[entry] || this.store.head_end;
|
||
|
|
const valueSet = entryMap[to] || new Set();
|
||
|
|
valueSet.add(value);
|
||
|
|
entryMap[to] = valueSet;
|
||
|
|
}
|
||
|
|
|
||
|
|
_getPageType(pageLocals) {
|
||
|
|
let currentType = 'default';
|
||
|
|
if (pageLocals.__index) currentType = 'home';
|
||
|
|
if (pageLocals.__post) currentType = 'post';
|
||
|
|
if (pageLocals.__page) currentType = 'page';
|
||
|
|
if (pageLocals.archive) currentType = 'archive';
|
||
|
|
if (pageLocals.category) currentType = 'category';
|
||
|
|
if (pageLocals.tag) currentType = 'tag';
|
||
|
|
if (pageLocals.layout) currentType = pageLocals.layout;
|
||
|
|
|
||
|
|
return currentType;
|
||
|
|
}
|
||
|
|
|
||
|
|
_injector(input, pattern, flag, isBegin = true, currentType) {
|
||
|
|
if (input.includes(`hexo injector ${flag}`)) return input;
|
||
|
|
|
||
|
|
const code = this.cache.apply(`${flag}-${currentType}-code`, () => {
|
||
|
|
const content = currentType === 'default' ? this.getText(flag, 'default') : this.getText(flag, currentType) + this.getText(flag, 'default');
|
||
|
|
|
||
|
|
if (!content.length) return '';
|
||
|
|
return '<!-- hexo injector ' + flag + ' start -->' + content + '<!-- hexo injector ' + flag + ' end -->';
|
||
|
|
});
|
||
|
|
|
||
|
|
// avoid unnecessary replace() for better performance
|
||
|
|
if (!code.length) return input;
|
||
|
|
|
||
|
|
return input.replace(pattern, str => { return isBegin ? str + code : code + str; });
|
||
|
|
}
|
||
|
|
|
||
|
|
exec(data, locals = { page: {} }) {
|
||
|
|
const { page } = locals;
|
||
|
|
const currentType = this._getPageType(page);
|
||
|
|
|
||
|
|
if (this.getSize('head_begin') !== 0) {
|
||
|
|
// Inject head_begin
|
||
|
|
data = this._injector(data, /<head.*?>/, 'head_begin', true, currentType);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (this.getSize('head_end') !== 0) {
|
||
|
|
// Inject head_end
|
||
|
|
data = this._injector(data, '</head>', 'head_end', false, currentType);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (this.getSize('body_begin') !== 0) {
|
||
|
|
// Inject body_begin
|
||
|
|
data = this._injector(data, /<body.*?>/, 'body_begin', true, currentType);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (this.getSize('body_end') !== 0) {
|
||
|
|
// Inject body_end
|
||
|
|
data = this._injector(data, '</body>', 'body_end', false, currentType);
|
||
|
|
}
|
||
|
|
|
||
|
|
return data;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = Injector;
|