@cloud411716/fancy-webnovel 2.2.0 → 3.0.0

package/plugins/fancy-material/genres.js

@@ -0,0 +1,256 @@
+/**
+ * Genre discovery for the material library.
+ *
+ * Two sources:
+ *   1. A topic_decision file: use the genres listed under its "选题方向" section.
+ *   2. Fanqie scan data: the `## <genre> — N 本` headings contain the most
+ *      complete genre list available in the workspace.
+ *
+ * If neither exists (for example when printing the prompt outside a workspace),
+ * fall back to the known Fanqie genre list. That keeps `prompt` usable as a
+ * standalone all-genres prompt.
+ */
+
+import fs from 'fs';
+import path from 'path';
+import { workspacePaths } from '#workspace';
+
+export const FALLBACK_GENRES = [
+  '西方奇幻',
+  '东方仙侠',
+  '科幻末世',
+  '都市日常',
+  '都市修真',
+  '都市高武',
+  '历史古代',
+  '战神赘婿',
+  '都市种田',
+  '传统玄幻',
+  '历史脑洞',
+  '悬疑脑洞',
+  '都市脑洞',
+  '玄幻脑洞',
+  '悬疑灵异',
+  '抗战谍战',
+  '游戏体育',
+  '动漫衍生',
+  '男频衍生',
+  '古风世情',
+  '女频衍生',
+  '玄幻言情',
+  '种田',
+  '年代',
+  '现言脑洞',
+  '宫斗宅斗',
+  '古言脑洞',
+  '快穿',
+  '青春甜宠',
+  '星光璀璨',
+  '女频悬疑',
+  '职场婚恋',
+  '豪门总裁',
+  '民国言情',
+];
+
+function unique(values) {
+  const seen = new Set();
+  const out = [];
+  for (const value of values) {
+    const text = String(value || '').trim();
+    if (!text || seen.has(text)) continue;
+    seen.add(text);
+    out.push(text);
+  }
+  return out;
+}
+
+function cleanGenre(value) {
+  return String(value || '')
+    .replace(/\*\*/g, '')
+    .replace(/^[\s\-*|]+/, '')
+    .replace(/[\s\-*|]+$/, '')
+    .replace(/[（(].*?[）)]/g, '')
+    .trim();
+}
+
+function splitGenres(value) {
+  return unique(String(value || '')
+    .split(/[、，,/；;]/)
+    .map(cleanGenre)
+    .filter((item) => item && !['全部', '通用', '其他', '未知'].includes(item)));
+}
+
+/**
+ * List all topic_decision_*.md files in <workspace>/扫榜结果, newest first.
+ *
+ * The material command lets the user pick one when several reports exist;
+ * it never merges the genre lists of multiple reports.
+ *
+ * @param {string} root workspace directory
+ * @returns {string[]} absolute file paths
+ */
+export function listTopicDecisionFiles(root) {
+  const dir = workspacePaths(root).scan;
+  let names;
+  try {
+    names = fs.readdirSync(dir);
+  } catch {
+    return [];
+  }
+  return names
+    .filter((name) => /^topic_decision_.*\.md$/i.test(name))
+    .sort()
+    .reverse()
+    .map((name) => path.join(dir, name))
+    .filter((file) => {
+      try {
+        const stat = fs.lstatSync(file);
+        return stat.isFile() && !stat.isSymbolicLink();
+      } catch {
+        return false;
+      }
+    });
+}
+
+/**
+ * Find the latest topic_decision_*.md in <workspace>/扫榜结果.
+ *
+ * @param {string} root workspace directory
+ * @returns {string|null}
+ */
+export function latestTopicDecision(root) {
+  return listTopicDecisionFiles(root)[0] || null;
+}
+
+/**
+ * Extract genres from the "选题方向" section of a topic_decision report.
+ *
+ * Supports both the current markdown table (`方向 | 题材 | ...`) and simple
+ * bullet lines such as `- **题材**：仙侠、都市高武`.
+ *
+ * @param {string} text
+ * @returns {string[]}
+ */
+export function parseTopicDecisionGenres(text) {
+  const source = String(text || '');
+  const lines = source.split(/\r?\n/);
+  let start = -1;
+  for (let i = 0; i < lines.length; i += 1) {
+    if (/^##\s*(?:[一二三四五六七八九十]+、)?选题方向(?:建议)?\s*$/.test(lines[i].trim())) {
+      start = i;
+      break;
+    }
+  }
+  if (start === -1) return [];
+
+  let end = lines.length;
+  for (let i = start + 1; i < lines.length; i += 1) {
+    if (/^##\s+/.test(lines[i])) {
+      end = i;
+      break;
+    }
+  }
+  const section = lines.slice(start + 1, end);
+  const genres = [];
+
+  // Markdown table: find the header that contains 题材.
+  let header = null;
+  let genreIndex = -1;
+  for (const line of section) {
+    const trimmed = line.trim();
+    if (!trimmed.startsWith('|')) continue;
+    const cells = trimmed.split('|').slice(1, -1).map((cell) => cell.trim());
+    if (!cells.length) continue;
+    const idx = cells.findIndex((cell) => cell.replace(/\*\*/g, '').trim() === '题材');
+    if (idx !== -1) {
+      header = cells;
+      genreIndex = idx;
+      break;
+    }
+  }
+  if (header && genreIndex !== -1) {
+    let inBody = false;
+    for (const line of section) {
+      const trimmed = line.trim();
+      if (!trimmed.startsWith('|')) continue;
+      const cells = trimmed.split('|').slice(1, -1).map((cell) => cell.trim());
+      if (!inBody) {
+        if (cells.every((cell) => /^:?-{2,}:?$/.test(cell.replace(/\s/g, '')))) {
+          inBody = true;
+        }
+        continue;
+      }
+      if (cells.length <= genreIndex) continue;
+      genres.push(...splitGenres(cells[genreIndex]));
+    }
+  }
+
+  // Bullet fallback, including lines like `- **题材**：仙侠`.
+  for (const line of section) {
+    const match = line.match(/^\s*[-*]\s*(?:\*\*)?题材(?:\*\*)?\s*[：:]\s*(.+)$/);
+    if (match) genres.push(...splitGenres(match[1]));
+  }
+
+  return unique(genres);
+}
+
+/**
+ * Read the genre headings from Fanqie scan files.
+ *
+ * @param {string} root workspace directory
+ * @returns {string[]}
+ */
+export function parseFanqieGenres(root) {
+  const dir = workspacePaths(root).scan;
+  let names;
+  try {
+    names = fs.readdirSync(dir);
+  } catch {
+    return [...FALLBACK_GENRES];
+  }
+  const files = names
+    .filter((name) => /^番茄.*\.md$/i.test(name))
+    .sort();
+  const genres = [];
+  for (const name of files) {
+    let text;
+    try {
+      text = fs.readFileSync(path.join(dir, name), 'utf8');
+    } catch {
+      continue;
+    }
+    for (const line of text.split(/\r?\n/)) {
+      const match = line.match(/^##\s+(.+?)\s+—\s+\d+\s+本\s*$/);
+      if (match) genres.push(...splitGenres(match[1]));
+    }
+  }
+  return genres.length ? unique(genres) : [...FALLBACK_GENRES];
+}
+
+/**
+ * Resolve the genres for a generation run.
+ *
+ * @param {string} root workspace directory
+ * @param {{topicFile?: string|null}} [options] explicit report selected by the user
+ * @returns {{mode: 'topic_decision'|'all_genres', genres: string[], topicFile: string|null}}
+ */
+export function resolveMaterialGenres(root, options = {}) {
+  const topicFile = options.topicFile || latestTopicDecision(root);
+  if (topicFile) {
+    try {
+      const genres = parseTopicDecisionGenres(fs.readFileSync(topicFile, 'utf8'));
+      if (genres.length) return { mode: 'topic_decision', genres, topicFile };
+    } catch { /* fall through to all genres */ }
+  }
+  return { mode: 'all_genres', genres: parseFanqieGenres(root), topicFile: null };
+}
+
+/**
+ * All known genres, used by the `prompt` subcommand.
+ *
+ * @param {string|null} root
+ * @returns {string[]}
+ */
+export function allKnownGenres(root) {
+  return root ? parseFanqieGenres(root) : [...FALLBACK_GENRES];
+}