@cloud411716/fancy-webnovel 1.0.46 → 3.0.0

package/plugins/fancy-story-builder/storydb/context.js

@@ -0,0 +1,288 @@
+/**
+ * Token-bounded context packs.
+ *
+ * The LLM should never read a whole stage file. `chapterContext` returns only
+ * the chapter, its neighbours, the entities that matter, and the pending work.
+ */
+
+import { recordLiteral } from './ids.js';
+import { getEntity, requirementsFor } from './store.js';
+
+const STORY = 'story:main';
+
+function tail(text, chars) {
+  const value = String(text || '');
+  return value.length <= chars ? value : value.slice(-chars);
+}
+
+function uniqBy(rows, key) {
+  const seen = new Set();
+  const out = [];
+  for (const row of rows) {
+    const id = typeof key === 'function' ? key(row) : row[key];
+    if (seen.has(id)) continue;
+    seen.add(id);
+    out.push(row);
+  }
+  return out;
+}
+
+export async function chapterContext(sdb, displayNo, opts = {}) {
+  const chapter = await sdb.one(
+    `SELECT * FROM chapter WHERE story = ${STORY} AND display_no = $n LIMIT 1;`,
+    { n: Number(displayNo) },
+  );
+  if (!chapter) throw new Error('未找到第 ' + displayNo + ' 章');
+  const chapterLit = recordLiteral('chapter', chapter.chapter_id);
+
+  const [outline, castEdges, locationEdges, requirements, foreshadowReq, payoffReq, foreshadowActions, payoffActions, infoActions, climaxEdges, itemEdges, knowledgeEvents] = await Promise.all([
+    sdb.one(
+      `SELECT * FROM outline WHERE story = ${STORY} AND chapter_id = $id LIMIT 1;`,
+      { id: chapter.chapter_id },
+    ),
+    sdb.rows(
+      `SELECT out, role FROM chapter_cast WHERE story = ${STORY} AND in = ${chapterLit};`,
+    ),
+    sdb.rows(
+      `SELECT out FROM chapter_location WHERE story = ${STORY} AND in = ${chapterLit};`,
+    ),
+    requirementsFor(sdb, 'chapter:' + chapter.chapter_id),
+    sdb.rows(
+      `SELECT * FROM requirement WHERE story = ${STORY} AND status = 'pending' AND string::starts_with(subject, 'foreshadow:') ORDER BY due_chapter ASC LIMIT 20;`,
+    ),
+    sdb.rows(
+      `SELECT * FROM requirement WHERE story = ${STORY} AND status = 'pending' AND string::starts_with(subject, 'payoff:') ORDER BY due_chapter ASC LIMIT 20;`,
+    ),
+    sdb.rows(
+      `SELECT 'foreshadow' AS kind, record::id(out) AS entity_id, action, chapter_no, detail
+       FROM foreshadow_action WHERE story = ${STORY} AND in = ${chapterLit};`,
+    ),
+    sdb.rows(
+      `SELECT 'payoff' AS kind, record::id(out) AS entity_id, action, chapter_no, detail
+       FROM payoff_action WHERE story = ${STORY} AND in = ${chapterLit};`,
+    ),
+    sdb.rows(
+      `SELECT 'information' AS kind, record::id(out) AS entity_id, action, chapter_no, detail
+       FROM info_action WHERE story = ${STORY} AND in = ${chapterLit};`,
+    ),
+    sdb.rows(
+      `SELECT record::id(out) AS entity_id FROM chapter_climax WHERE story = ${STORY} AND in = ${chapterLit};`,
+    ),
+    sdb.rows(
+      `SELECT record::id(out) AS entity_id, role FROM chapter_item WHERE story = ${STORY} AND in = ${chapterLit};`,
+    ),
+    sdb.rows(
+      `SELECT subject_type, subject_id, character_id, change, coverage, knowledge, how, source_character
+       FROM knowledge_event WHERE story = ${STORY} AND chapter = ${chapterLit}
+       ORDER BY subject_type, subject_id, character_id;`,
+    ),
+  ]);
+
+  const castRecords = castEdges.map((row) => row.out);
+  const cast = [];
+  for (const row of castEdges) {
+    const ref = String(row.out);
+    const id = ref.split(':').pop().replace(/[⟨⟩]/g, '');
+    const entity = await sdb.one(`SELECT * FROM ${recordLiteral('character', id)};`);
+    if (entity) cast.push({ id: entity.entity_id || id, name: entity.name, role: row.role });
+  }
+
+  const locations = [];
+  for (const row of locationEdges) {
+    const ref = String(row.out);
+    const id = ref.split(':').pop().replace(/[⟨⟩]/g, '');
+    const entity = await sdb.one(`SELECT * FROM ${recordLiteral('location', id)};`);
+    if (entity) locations.push({ id: entity.entity_id || id, name: entity.name });
+  }
+
+  const climaxes = [];
+  for (const edge of climaxEdges || []) {
+    const id = String(edge.entity_id || '').replace(/[⟨⟩]/g, '');
+    const entity = await sdb.one(`SELECT * FROM ${recordLiteral('climax', id)};`);
+    if (entity) climaxes.push({ id, name: entity.name, type: entity.type, description: entity.description });
+  }
+
+  const items = [];
+  for (const edge of itemEdges || []) {
+    const id = String(edge.entity_id || '').replace(/[⟨⟩]/g, '');
+    const entity = await sdb.one(`SELECT * FROM ${recordLiteral('item', id)};`);
+    if (entity) items.push({ id, name: entity.name, category: entity.category, role: edge.role || '' });
+  }
+
+  const scenes = outline
+    ? await sdb.rows(
+      `SELECT * FROM scene WHERE story = ${STORY} AND outline_id = $id ORDER BY scene_order;`,
+      { id: chapter.chapter_id },
+    )
+    : [];
+
+  const knowledge = [];
+  for (const c of cast) {
+    const rows = await sdb.rows(
+      `SELECT subject_type, subject_id, state FROM knowledge_state
+       WHERE story = ${STORY} AND character = ${recordLiteral('character', c.id)}
+       ORDER BY subject_type, subject_id;`,
+    );
+    for (const row of rows) knowledge.push({ character: c.id, ...row });
+  }
+
+  const prevCount = Math.max(0, Math.min(Number(opts.prev) || 0, 6));
+  const tailChars = Math.max(200, Math.min(Number(opts.tailChars) || 600, 2000));
+  const previous = [];
+  if (prevCount) {
+    const rows = await sdb.rows(
+      `SELECT chapter_id, display_no, title FROM chapter
+       WHERE story = ${STORY} AND display_no < $n ORDER BY display_no DESC LIMIT $k;`,
+      { n: Number(displayNo), k: prevCount },
+    );
+    rows.reverse();
+    for (const row of rows) {
+      const [handoff, text] = await Promise.all([
+        sdb.one(`SELECT * FROM handoff WHERE story = ${STORY} AND chapter_id = $id LIMIT 1;`, { id: row.chapter_id }),
+        sdb.one(`SELECT content FROM chapter_text WHERE story = ${STORY} AND chapter_id = $id LIMIT 1;`, { id: row.chapter_id }),
+      ]);
+      previous.push({
+        chapter_id: row.chapter_id,
+        display_no: row.display_no,
+        title: row.title,
+        handoff: handoff || null,
+        tail: text ? tail(text.content, tailChars) : '',
+      });
+    }
+  }
+
+  const search = [];
+  if (opts.search) {
+    const hits = await sdb.rows(
+      `SELECT chapter_id, display_no, content, search::score(1) AS score
+       FROM chapter_text WHERE content @1@ $q ORDER BY score DESC LIMIT 5;`,
+      { q: String(opts.search) },
+    );
+    for (const hit of hits) {
+      search.push({
+        chapter_id: hit.chapter_id,
+        display_no: hit.display_no,
+        score: hit.score,
+        snippet: String(hit.content || '').slice(0, 300),
+      });
+    }
+  }
+
+  const foreshadowPending = [];
+  for (const req of foreshadowReq) {
+    const ref = String(req.subject).split(':');
+    if (ref[0] !== 'foreshadow') continue;
+    const entity = await sdb.one(`SELECT * FROM ${recordLiteral('foreshadow', ref[1])};`);
+    foreshadowPending.push({
+      id: ref[1],
+      name: entity && entity.name,
+      pending: req.kind,
+      due_chapter: req.due_chapter,
+      hard: Boolean(req.hard),
+    });
+  }
+  const payoffPending = [];
+  for (const req of payoffReq) {
+    const ref = String(req.subject).split(':');
+    if (ref[0] !== 'payoff') continue;
+    const entity = await sdb.one(`SELECT * FROM ${recordLiteral('payoff', ref[1])};`);
+    payoffPending.push({
+      id: ref[1],
+      name: entity && entity.name,
+      pending: req.kind,
+      due_chapter: req.due_chapter,
+      hard: Boolean(req.hard),
+    });
+  }
+
+  return {
+    chapter: {
+      chapter_id: chapter.chapter_id,
+      display_no: chapter.display_no,
+      title: chapter.title,
+      volume: chapter.volume,
+      act: chapter.act,
+      pov: chapter.pov,
+      time: chapter.time,
+      why: chapter.why,
+      how: chapter.how,
+      emotion: chapter.emotion,
+      hook: chapter.hook,
+      word_target: chapter.word_target,
+      status: chapter.status,
+    },
+    outline: outline
+      ? {
+        chapter_id: outline.chapter_id,
+        must_write: outline.must_write,
+        must_not_write: outline.must_not_write,
+        continuity: outline.continuity,
+        coverage: outline.coverage,
+        ending: outline.ending,
+        status: outline.status,
+      }
+      : null,
+    scenes,
+    cast,
+    locations,
+    climaxes,
+    items,
+    knowledge,
+    requirements,
+    actions: [...(foreshadowActions || []), ...(payoffActions || []), ...(infoActions || [])]
+      .sort((a, b) => Number(a.chapter_no) - Number(b.chapter_no)),
+    knowledge_events: knowledgeEvents || [],
+    foreshadows_due: uniqBy(foreshadowPending, (x) => x.id),
+    payoffs_due: uniqBy(payoffPending, (x) => x.id),
+    previous,
+    search,
+  };
+}
+
+export async function entityContext(sdb, kind, id) {
+  const entity = await getEntity(sdb, kind, id);
+  const requirements = await requirementsFor(sdb, kind + ':' + id);
+  const decisions = await sdb.rows(
+    `SELECT text, source, reference, reason, status FROM decision
+     WHERE story = ${STORY} AND subject = $subject AND status = 'active' ORDER BY created_at DESC;`,
+    { subject: kind + ':' + id },
+  );
+
+  if (kind === 'foreshadow') {
+    const actions = await sdb.rows(
+      `SELECT in.display_no AS chapter_no, action, detail
+       FROM foreshadow_action WHERE story = ${STORY} AND out = ${recordLiteral('foreshadow', id)} ORDER BY chapter_no;`,
+    );
+    const payoffs = await sdb.rows(
+      `SELECT out.entity_id AS payoff, out.name AS name FROM foreshadow_payoff
+       WHERE story = ${STORY} AND in = ${recordLiteral('foreshadow', id)};`,
+    );
+    const knowledge = await sdb.rows(
+      `SELECT chapter_no, character_id, change, coverage FROM knowledge_event
+       WHERE story = ${STORY} AND subject_type = 'foreshadow' AND subject_id = $id ORDER BY chapter_no;`,
+      { id: String(id) },
+    );
+    return { entity, requirements, decisions, actions, payoffs, knowledge };
+  }
+
+  if (kind === 'character') {
+    const relations = await sdb.rows(
+      `SELECT from_id, to_id, kind, initial_state FROM relation
+       WHERE story = ${STORY} AND (from_id = $id OR to_id = $id);`,
+      { id: String(id) },
+    );
+    const appearances = await sdb.rows(
+      `SELECT in.display_no AS chapter_no, in.title AS title, role FROM chapter_cast
+       WHERE story = ${STORY} AND out = ${recordLiteral('character', id)} ORDER BY chapter_no;`,
+    );
+    const knowledge = await sdb.rows(
+      `SELECT subject_type, subject_id, state FROM knowledge_state
+       WHERE story = ${STORY} AND character = ${recordLiteral('character', id)} ORDER BY subject_type, subject_id;`,
+    );
+    return { entity, requirements, decisions, relations, appearances, knowledge };
+  }
+
+  return { entity, requirements, decisions };
+}
+
+export { tail };