48 lines
1.7 KiB
TypeScript
48 lines
1.7 KiB
TypeScript
import type { CharacterArcNode } from '../../shared/arc'
|
|
import { KnowledgeRepository } from '../db/repositories/knowledge.repo'
|
|
import { SettingRepository } from '../db/repositories/setting.repo'
|
|
import { ChapterRepository } from '../db/repositories/chapter.repo'
|
|
import type { SqliteDb } from '../db/types'
|
|
|
|
export class ArcService {
|
|
constructor(private db: SqliteDb) {}
|
|
|
|
getForCharacter(settingId: string): CharacterArcNode[] {
|
|
const setting = new SettingRepository(this.db).get(settingId)
|
|
if (!setting || setting.type !== 'character') return []
|
|
|
|
const knowledge = new KnowledgeRepository(this.db)
|
|
.list()
|
|
.filter((k) => k.type === 'character' && k.title.includes(setting.name))
|
|
const chapterRepo = new ChapterRepository(this.db)
|
|
const chapters = new Map(chapterRepo.list().map((c) => [c.id, c]))
|
|
|
|
const nodes: CharacterArcNode[] = knowledge
|
|
.map((k) => {
|
|
const chapterId = k.lastMentionChapterId ?? k.sourceChapterId
|
|
const chapter = chapterId ? chapters.get(chapterId) : undefined
|
|
return {
|
|
id: k.id,
|
|
title: k.title,
|
|
content: k.content,
|
|
chapterId: chapterId ?? null,
|
|
chapterTitle: chapter?.title ?? null,
|
|
updatedAt: k.updatedAt
|
|
}
|
|
})
|
|
.sort((a, b) => a.updatedAt.localeCompare(b.updatedAt))
|
|
|
|
if (nodes.length === 0 && setting.description.trim()) {
|
|
nodes.push({
|
|
id: `setting:${setting.id}`,
|
|
title: '角色设定',
|
|
content: setting.description.replace(/<[^>]+>/g, '').slice(0, 200),
|
|
chapterId: null,
|
|
chapterTitle: null,
|
|
updatedAt: setting.updatedAt
|
|
})
|
|
}
|
|
return nodes
|
|
}
|
|
}
|