mirror of https://github.com/dunwu/db-tutorial.git
93 lines
3.0 KiB
JavaScript
93 lines
3.0 KiB
JavaScript
/**
|
||
* 批量添加和修改front matter ，需要配置 ./config.yml 文件。
|
||
*/
|
||
const fs = require('fs'); // 文件模块
|
||
const path = require('path'); // 路径模块
|
||
const matter = require('gray-matter'); // front matter解析器 https://github.com/jonschlinkert/gray-matter
|
||
const jsonToYaml = require('json2yaml')
|
||
const yamlToJs = require('yamljs')
|
||
const inquirer = require('inquirer') // 命令行操作
|
||
const chalk = require('chalk') // 命令行打印美化
|
||
const readFileList = require('./modules/readFileList');
|
||
const { type, repairDate} = require('./modules/fn');
|
||
const log = console.log
|
||
|
||
const configPath = path.join(__dirname, 'config.yml') // 配置文件的路径
|
||
|
||
main();
|
||
|
||
/**
|
||
* 主体函数
|
||
*/
|
||
async function main() {
|
||
|
||
const promptList = [{
|
||
type: "confirm",
|
||
message: chalk.yellow('批量操作frontmatter有修改数据的风险，确定要继续吗？'),
|
||
name: "edit",
|
||
}];
|
||
let edit = true;
|
||
|
||
await inquirer.prompt(promptList).then(answers => {
|
||
edit = answers.edit
|
||
})
|
||
|
||
if(!edit) { // 退出操作
|
||
return
|
||
}
|
||
|
||
const config = yamlToJs.load(configPath) // 解析配置文件的数据转为js对象
|
||
|
||
if (type(config.path) !== 'array') {
|
||
log(chalk.red('路径配置有误，path字段应该是一个数组'))
|
||
return
|
||
}
|
||
|
||
if (config.path[0] !== 'docs') {
|
||
log(chalk.red("路径配置有误，path数组的第一个成员必须是'docs'"))
|
||
return
|
||
}
|
||
|
||
const filePath = path.join(__dirname, '..', ...config.path); // 要批量修改的文件路径
|
||
const files = readFileList(filePath); // 读取所有md文件数据
|
||
|
||
files.forEach(file => {
|
||
let dataStr = fs.readFileSync(file.filePath, 'utf8');// 读取每个md文件的内容
|
||
const fileMatterObj = matter(dataStr) // 解析md文件的front Matter。 fileMatterObj => {content:'剔除frontmatter后的文件内容字符串', data:{<frontmatter对象>}, ...}
|
||
let matterData = fileMatterObj.data; // 得到md文件的front Matter
|
||
|
||
let mark = false
|
||
// 删除操作
|
||
if (config.delete) {
|
||
if( type(config.delete) !== 'array' ) {
|
||
log(chalk.yellow('未能完成删除操作，delete字段的值应该是一个数组！'))
|
||
} else {
|
||
config.delete.forEach(item => {
|
||
if (matterData[item]) {
|
||
delete matterData[item]
|
||
mark = true
|
||
}
|
||
})
|
||
|
||
}
|
||
}
|
||
|
||
// 添加、修改操作
|
||
if (type(config.data) === 'object') {
|
||
Object.assign(matterData, config.data) // 将配置数据合并到front Matter对象
|
||
mark = true
|
||
}
|
||
|
||
// 有操作时才继续
|
||
if (mark) {
|
||
if(matterData.date && type(matterData.date) === 'date') {
|
||
matterData.date = repairDate(matterData.date) // 修复时间格式
|
||
}
|
||
const newData = jsonToYaml.stringify(matterData).replace(/\n\s{2}/g,"\n").replace(/"/g,"") + '---\r\n' + fileMatterObj.content;
|
||
fs.writeFileSync(file.filePath, newData); // 写入
|
||
log(chalk.green(`update frontmatter：${file.filePath} `))
|
||
}
|
||
|
||
})
|
||
}
|