8文章生成器如何用fs模块保存文件

export function generate(title, {corpus, min = 6000, max = 10000}) {
  ...
}

{
  "title": [
    "一天掉多少根头发",
    "中午吃什么",
    ...
  ],
  "famous":[
    "爱迪生{{said}}，天才是百分之一的勤奋加百分之九十九的汗水。{{conclude}}",
    "查尔斯·史{{said}}，一个人几乎可以在任何他怀有无限热忱的事情上成功。{{conclude}}",
    "培根说过，深窥自己的心，而后发觉一切的奇迹在你自己。{{conclude}}",
    ...
  ],
  "bosh_before": [
    "既然如此，",
    "那么，",
    "我认为，",
    ...
  ],
  "bosh":[
    "{{title}}的发生，到底需要如何做到，不{{title}}的发生，又会如何产生。 ",
    "而这些并不是完全重要，更加重要的问题是，",
    "{{title}}，到底应该如何实现。 ",
    ...
  ],
  "conclude":[
    "这不禁令我深思。 ",
    "带着这句话，我们还要更加慎重的审视这个问题: ",
    "这启发了我。",
    ...
  ],
  "said":[
    "曾经说过",
    "在不经意间这样说过",
    "说过一句著名的话",
    ...
  ]
}

const pickFamous = createRandomPicker(corpus.famous);
const pickBosh = createRandomPicker(corpus.bosh);

pickFamous(); // 随机取出一条名人名言

pickBosh(); // 随机取出一条废话

"歌德曾经{{said}}，流水在碰到底处时才会释放活力。{{conclude}}" // 名人名言

"{{title}}的发生，到底需要如何做到，不{{title}}的发生，又会如何产生。 " // 废话

function sentence(pick, replacer) {
  let ret = pick(); // 返回一个句子文本
  for(const key in replacer) { // replacer是一个对象，存放替换占位符的规则
    // 如果 replacer[key] 是一个 pick 函数，那么执行它随机取一条替换占位符，否则将它直接替换占位符
    ret = ret.replace(new RegExp(`{{${key}}}`, 'g'),
      typeof replacer[key] === 'function' ? replacer[key]() : replacer[key]);
  }
  return ret;
}

const {famous, bosh_before, bosh, said, conclude} = corpus;

const [pickFamous, pickBoshBefore, pickBosh, pickSaid, pickConclude] = [famous, bosh_before, bosh, said, conclude].map((item) => {
  return createRandomPicker(item);
});

sentence(pickFamous, {said: pickSaid, conclude: pickConclude}); // 生成一条名人名言

sentence(pickBosh, {title});  // 生成一条废话

const articleLength = randomInt(min, max);

while(totalLength < articleLength) {
  // 如果文章内容的字数未超过文章总字数 继续生成段落
  let section = ''; // 添加段落
  const sectionLength = randomInt(200, 500); // 将段落长度设为200到500字之间
  // 如果当前段落字数小于段落长度，或者当前段落不是以句号。和问号？结尾
  while(section.length < sectionLength || !/[。？]$/.test(section)) {
    // 取一个 0~100 的随机数
    const n = randomInt(0, 100);
    if(n < 20) { 
      添加名人名言
    } else if(n < 50) {
      添加带前置从句的废话
    } else {
      添加不带前置从句的废话
    }
  }
  // 段落结束，更新总长度
  totalLength += section.length;
  // 将段落存放到文章列表中
  article.push(section);
}

"带着这句话，我们还要更加慎重的审视这个问题： ",

export function generate(title, {
  corpus,
  min = 6000, // 文章最少字数
  max = 10000, // 文章最多字数
} = {}) {
  // 将文章长度设置为 min 到 max之间的随机数
  const articleLength = randomInt(min, max);

  const {famous, bosh_before, bosh, said, conclude} = corpus;
  const [pickFamous, pickBoshBefore, pickBosh, pickSaid, pickConclude] = [famous, bosh_before, bosh, said, conclude].map((item) => {
    return createRandomPicker(item);
});

const article = [];
let totalLength = 0;

while(totalLength < articleLength) {
  // 如果文章内容的字数未超过文章总字数
  let section = ''; // 添加段落
  const sectionLength = randomInt(200, 500); // 将段落长度设为200到500字之间
  // 如果当前段落字数小于段落长度，或者当前段落不是以句号。和问号？结尾
  while(section.length < sectionLength || !/[。？]$/.test(section)) {
    // 取一个 0~100 的随机数
    const n = randomInt(0, 100);
    if(n < 20) { // 如果 n 小于 20，生成一条名人名言，也就是文章中有百分之二十的句子是名人名言
      section += sentence(pickFamous, {said: pickSaid, conclude: pickConclude});
    } else if(n < 50) {
      // 如果 n 小于 50，生成一个带有前置从句的废话
      section += sentence(pickBoshBefore, {title}) + sentence(pickBosh, {title});
    } else {
      // 否则生成一个不带有前置从句的废话
      section += sentence(pickBosh, {title});
    }
  }
  // 段落结束，更新总长度
  totalLength += section.length;
  // 将段落存放到文章列表中
  article.push(section);
}

// 将文章返回，文章是段落数组形式
return article;
}

import {readFileSync} from 'fs';
import {fileURLToPath} from 'url';
import {dirname, resolve} from 'path';

import {generate} from './lib/generator.js';
import {createRandomPicker} from './lib/random.js';

const __dirname = dirname(fileURLToPath(import.meta.url));

function loadCorpus(src) {
  const path = resolve(__dirname, src);
  const data = readFileSync(path, {encoding: 'utf-8'});
  return JSON.parse(data);
}

const corpus = loadCorpus('corpus/data.json');

const pickTitle = createRandomPicker(corpus.title);
const title = pickTitle();

const article = generate(title, {corpus});
console.log(`${title}\n\n    ${article.join('\n    ')}`);

function saveCorpus(title, article) {
  const outputDir = resolve(__dirname, 'output');
  const outputFile = resolve(outputDir, `${title}.txt`);

  // 检查outputDir是否存在，没有则创建一个
  if(!existsSync(outputDir)) {
    mkdirSync(outputDir);
  }

  const text = `${title}\n\n    ${article.join('\n    ')}`;
  writeFileSync(outputFile, text); // 将text写入outputFile文件中

  return outputFile;
}

$ npm install moment --save

function saveCorpus(title, article) {
  const outputDir = resolve(__dirname, 'output');
  const time = moment().format('|YYYY-MM-DD|HH:mm:ss');
  const outputFile = resolve(outputDir, `${title}${time}.txt`);

  if(!existsSync(outputDir)) {
    mkdirSync(outputDir);
  }

  const text = `${title}\n\n    ${article.join('\n    ')}`;
  writeFileSync(outputFile, text);

  return outputFile;
}