@lwys-pkg-releaser/node-bullshit-generator/src/dict/english/process.js

Port of menzi11/BullshitGenerator@github.com to JavaScript

let roster = {};

/**
 * @param {string} theme
 * @returns {string}
 */
roster.onTheme = function onTheme(theme) {
    return theme.toLowerCase();
};

let prefix2Machine = null;
let examplesMachine = null;
let addingsMachine = null;
let contrastsMachine = null;

/**
 * @param {any[]} arr
 * @param {Object} opt
 * @param {Object} rng
 * @returns {Obsject}
 */
function generateIterator(arr, opt, rng) {
    return opt["menzi11#175"]
        ? rng.洗牌遍历(arr)
        : {
              next: function () {
                  return {
                      value: rng.瞎选一个(arr),
                      done: false,
                  };
              },
          };
}

/**
 * @param {string} seg
 * @param {Object} dict
 * @param {Object} opt
 * @param {Object} _
 * @param {Object} rng
 * @param {string} type
 * @returns {string}
 */
roster.onSegment = function onSegment(seg, dict, opt, _, rng, type) {
    if (type == "名人名言") {
        if (!examplesMachine)
            examplesMachine = generateIterator(dict["examples"], opt, rng);
        // if not random.random() > 0.3: - there is no weighted random here
        if (seg.includes("{prefix_2}")) {
            if (!prefix2Machine)
                prefix2Machine = generateIterator(dict["prefix_2"], opt, rng);
            return (
                examplesMachine.next().value +
                prefix2Machine.next().value +
                seg.replace(/\{prefix_2\}/g, "")
            );
        }
        return examplesMachine.next().value + seg;
    } else if (type == "废话" && (!opt.v3语法改进 || rng.取随机小数() > 0.3)) {
        if (rng.取随机数(100) - 20 <= 45) {
            if (!addingsMachine)
                addingsMachine = generateIterator(dict["addings"], opt, rng);
            return addingsMachine.next().value + seg;
        }
        if (!contrastsMachine)
            contrastsMachine = generateIterator(dict["contrasts"], opt, rng);
        return contrastsMachine.next().value + seg;
    }
    return seg;
};

/**
 * @param {string} str
 * @returns {string[]}
 */
function sentences(str) {
    return str.match(/([^\.!\?]+[\.!\?]+)|([^\.!\?]+$)/g);
}
/**
 * @param {string} str
 * @returns {string}
 */
function clean(str) {
    return str
        .replace(/  /g, " ")
        .replace(/\. \./g, ".")
        .replace(/\? \./g, "?")
        .replace(/, \./g, ",")
        .replace(/\.\./g, ".")
        .replace(/[a-z]/i, (x) => x.toUpperCase())
        .trim();
}

/**
 * @param {string[]} article
 * @returns {string[]}
 */
roster.onArticle = function onArticle(article) {
    return article.map((part) =>
        sentences(part)
            .map((i) => clean(i))
            .join(" ")
    );
};

module.exports = roster;