使用Node.js實(shí)現(xiàn)一個(gè)文章生成器

更新時(shí)間：2024年07月05日 08:55:23 作者：愛吃土豆絲呦

本文將從零開始,講解如何使用Node.js來(lái)實(shí)現(xiàn)一個(gè)文章生成器,node里面有很多優(yōu)秀的模塊,現(xiàn)在我們就借助node的fs模塊來(lái)操控文本,來(lái)實(shí)現(xiàn)我們想要的效果,感興趣的小伙伴跟著小編一起來(lái)看看吧

前言

本文將從零開始，講解如何使用Node.js來(lái)實(shí)現(xiàn)一個(gè)文章生成器

node里面有很多優(yōu)秀的模塊，現(xiàn)在我們就借助node的fs模塊來(lái)操控文本，來(lái)實(shí)現(xiàn)我們想要的效果

效果展示

體驗(yàn) fs

首先我們先創(chuàng)建一個(gè)json文件

里面放一些內(nèi)容

接下來(lái)我們書寫代碼

import fs from "fs";
const data = fs.readFileSync("./cuepus/data.json", "utf-8", (err, data) => {
  if (!err) {
    // console.log(data);
  } else {
    console.log(err);
  }
});

可以看到我們就能讀取到文件里的數(shù)據(jù)，這里我們體驗(yàn)了fs里面強(qiáng)大的功能

在這里我們放了文章的一些必須點(diǎn)，如文章內(nèi)容、標(biāo)題等，通過(guò)隨機(jī)拿取數(shù)據(jù)生成一個(gè)文章

這里我們通過(guò)

import { fileURLToPath } from "url";
import { dirname,resolve } from "path";

去操控獲取絕對(duì)路徑

const url = import.meta.url;// 獲取當(dāng)前文件的絕對(duì)路徑 
console.log(url);
console.log(fileURLToPath(url));
console.log(dirname(fileURLToPath(url)));
// 獲取文件的絕對(duì)路徑
const path = resolve(dirname(fileURLToPath(url)), "./cuepus/data.json"); console.log(path);

const url = import.meta.url; ：使用了import.meta 對(duì)象來(lái)獲取當(dāng)前模塊文件的絕對(duì) URL 路徑，并將其賦值給變量 url 。
console.log(fileURLToPath(url)); ：將url路徑轉(zhuǎn)換為文件系統(tǒng)路徑
console.log(dirname(fileURLToPath(url))); ：將url轉(zhuǎn)換為文件系統(tǒng)路徑，然后通過(guò) dirname 函數(shù)獲取該路徑的目錄部分
const path = resolve(dirname(fileURLToPath(url)), "./cuepus/data.json"); ：將前面獲取到的目錄部分和相對(duì)路徑 "./cuepus/data.json" 進(jìn)行組合，生成一個(gè)完整的文件絕對(duì)路徑，并將其賦值給變量 path 。

接下來(lái)我們代碼就可以寫成以下的形式

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

const url = import.meta.url;// 獲取當(dāng)前文件的絕對(duì)路徑
// 獲取文件的絕對(duì)路徑
const path = resolve(dirname(fileURLToPath(url)), "./cuepus/data.json");
const data = fs.readFileSync(path, "utf-8", (err, data) => {
  if (!err) {
    // console.log(data);
  } else {
    console.log(err);
  }
});

console.log(JSON.parse(data));

接下來(lái)我們寫封裝兩個(gè)函數(shù)

random.js

export function randomInt(min, max) {
  const n = Math.random()
  return Math.floor(min * (1 - n) + max * n)
}

export function createRandomPicker(arr) {
  arr = [...arr]

  function randomPick() { 
    const len = arr.length - 1
    const index = randomInt(0, len);
    [arr[index], arr[len]] = [arr[len], arr[index]]
    return arr[index]
  }
  randomPick()  // 放棄第一次結(jié)果
  return randomPick
}

首先，定義了一個(gè)名為 randomInt 的導(dǎo)出函數(shù)，它接受兩個(gè)參數(shù) min 和 max ，用于生成指定范圍內(nèi)的隨機(jī)整數(shù)。

然后，定義了一個(gè)名為 createRandomPicker 的導(dǎo)出函數(shù)，它接受一個(gè)數(shù)組 arr 作為參數(shù)。

在函數(shù)內(nèi)部，先對(duì)傳入的數(shù)組進(jìn)行了復(fù)制，以避免修改原始數(shù)組。

接著，定義了一個(gè)內(nèi)部函數(shù) randomPick ，它的功能是：

計(jì)算數(shù)組的有效長(zhǎng)度。
生成一個(gè)在有效范圍內(nèi)的隨機(jī)索引。
通過(guò)交換將隨機(jī)索引處的元素與最后一個(gè)元素交換位置。
返回交換后的隨機(jī)索引處的元素。

在函數(shù)體中，首先調(diào)用了一次 randomPick 函數(shù)，但放棄了這次的結(jié)果。

最后，返回 randomPick 函數(shù)，以便外部可以調(diào)用它來(lái)獲取隨機(jī)抽取的數(shù)組元素。

generator.js

import { randomInt, createRandomPicker } from './random.js'

// 替換文本
function sentence(pick, replacer) {
  let ret = pick()  // 得到一個(gè)句子
  for (const key in replacer) {
    ret = ret.replace(
      new RegExp(`{{${key}}}`, 'g'), 
      typeof replacer[key] === 'function' ? replacer[key]() : replacer[key]
    ) 
  }
  return ret
}


export function generate(title, { corpus, min = 500, max = 800 }) {
  const articleLenth = randomInt(min, max)
  const { famous, bosh_before, bosh, conclude, said } = corpus
  const [pickFamous, pickBoshBefore, pickBosh, pickConclude, pickSaid] 
  = [famous, bosh_before, bosh, conclude, said].map(createRandomPicker)

  const article = []
  let totalLength = 0

  while (totalLength < articleLenth) {  // 生成文章
    let section = ''
    const sectionLength = randomInt(100, 300)
    while (section.length < sectionLength) {  // 生成段落
      const n = randomInt(0, 100)
      if (n < 20) {
        section += sentence(pickFamous, {said: pickSaid, conclude: pickConclude})
      } else if (n < 50) {
        section += sentence(pickBoshBefore, { title }) + sentence(pickBosh, { title })
      } else {
        section += sentence(pickBosh, { title })
      }
    }
    totalLength += section.length
    article.push(section)
  }
  return article
}

從 './random.js' 模塊導(dǎo)入了 randomInt 和 createRandomPicker 兩個(gè)函數(shù)。

然后，定義了 sentence 函數(shù)，它接受兩個(gè)參數(shù) pick 和 replacer 。

pick 函數(shù)用于獲取一個(gè)句子，然后通過(guò)遍歷 replacer 對(duì)象中的鍵，使用正則表達(dá)式匹配并替換句子中特定的模板字符串（形如 {{key}} ）。如果對(duì)應(yīng)的值是函數(shù)，則調(diào)用函數(shù)獲取結(jié)果進(jìn)行替換，否則直接使用值進(jìn)行替換。

接下來(lái)，定義了 generate 函數(shù)，它接受三個(gè)參數(shù)：title 、一個(gè)包含配置信息的對(duì)象（包含 corpus 以及可選的 min 和 max ）。

在函數(shù)內(nèi)部：

通過(guò) randomInt 生成一個(gè)在 min 和 max 范圍內(nèi)的文章長(zhǎng)度 articleLenth 。
從 corpus 對(duì)象中解構(gòu)出幾個(gè)部分，并對(duì)每個(gè)部分使用 createRandomPicker 函數(shù)創(chuàng)建隨機(jī)選擇器。
通過(guò)一個(gè)循環(huán)生成文章內(nèi)容，每次循環(huán)生成一個(gè)段落。在生成段落的過(guò)程中，根據(jù)隨機(jī)數(shù)決定使用不同的句子生成方式，并通過(guò) sentence 函數(shù)進(jìn)行處理和組合。
不斷累加段落的長(zhǎng)度，直到總長(zhǎng)度達(dá)到 articleLenth ，最終將生成的文章段落組成一個(gè)數(shù)組返回。

最后調(diào)用即可，修改index.js

import fs from 'fs'
import { fileURLToPath } from 'url'
import { dirname, resolve } from 'path'
import { generate } from './lib/generator.js'
import { createRandomPicker } from './lib/random.js'


function loadCorpus(src) {
  const url = import.meta.url  // 讀取當(dāng)前腳本的絕對(duì)路徑
  const path = resolve(dirname(fileURLToPath(url)), src)
  const data = fs.readFileSync(path, {encoding: 'utf8'})
  return JSON.parse(data)
}
const corpus = loadCorpus('curpus/data.json')
const pickTitle = createRandomPicker(corpus.title)
const title = pickTitle()

const article = generate(title, { corpus })

fs.writeFileSync(`./output/${title}.txt`, article.join(''));