feat: bootstrap BoxBrain framework

src/memory.ts

@@ -0,0 +1,178 @@
+import { IdentityDB, type Fact as IdentityFact } from 'identitydb';
+import type { BoxBrainMemoryStore, FactDraft, MemorySpace, ScheduleEntry, StoredFact } from './types';
+
+function normalizeTopics(topics: string[]): string[] {
+  return [...new Set(topics.map((topic) => topic.trim()).filter(Boolean))];
+}
+
+function includesAnyTopic(fact: StoredFact, topics: string[]): boolean {
+  const normalized = new Set(normalizeTopics(topics).map((topic) => topic.toLowerCase()));
+  return fact.topics.some((topic) => normalized.has(topic.toLowerCase()));
+}
+
+export class InMemoryMemoryStore implements BoxBrainMemoryStore {
+  readonly spaces = new Map<string, MemorySpace>();
+  readonly facts = new Map<string, StoredFact[]>();
+  readonly schedules = new Map<string, ScheduleEntry[]>();
+
+  async createSpace(input: { displayName: string; seedMessage: string; now: string }): Promise<MemorySpace> {
+    const slug = input.displayName.toLowerCase().replace(/[^a-z0-9가-힣]+/gi, '-').replace(/^-|-$/g, '') || 'persona';
+    const space: MemorySpace = {
+      id: `persona-${slug}-${crypto.randomUUID()}`,
+      displayName: input.displayName,
+      createdAt: input.now,
+      metadata: { seedMessage: input.seedMessage },
+    };
+    this.spaces.set(space.id, space);
+    return space;
+  }
+
+  async getSpace(spaceId: string): Promise<MemorySpace | null> {
+    return this.spaces.get(spaceId) ?? null;
+  }
+
+  async addFact(spaceId: string, fact: FactDraft): Promise<StoredFact> {
+    const stored: StoredFact = {
+      id: crypto.randomUUID(),
+      statement: fact.statement,
+      topics: normalizeTopics(fact.topics),
+      createdAt: new Date().toISOString(),
+      ...(fact.confidence === undefined ? {} : { confidence: fact.confidence }),
+      ...(fact.source === undefined ? {} : { source: fact.source }),
+      ...(fact.metadata === undefined ? {} : { metadata: fact.metadata }),
+    };
+    const existing = this.facts.get(spaceId) ?? [];
+    existing.push(stored);
+    this.facts.set(spaceId, existing);
+    return stored;
+  }
+
+  async findFacts(spaceId: string, topics: string[]): Promise<StoredFact[]> {
+    const facts = this.facts.get(spaceId) ?? [];
+    if (topics.length === 0) return [...facts];
+    return facts.filter((fact) => includesAnyTopic(fact, topics));
+  }
+
+  async saveScheduleEntries(spaceId: string, entries: ScheduleEntry[]): Promise<void> {
+    const existing = (this.schedules.get(spaceId) ?? []).filter((entry) => !entries.some((incoming) => incoming.id === entry.id));
+    this.schedules.set(spaceId, [...existing, ...entries].sort((a, b) => a.startAt.localeCompare(b.startAt)));
+  }
+
+  async listScheduleEntries(spaceId: string, fromInclusive: string, toExclusive: string): Promise<ScheduleEntry[]> {
+    return (this.schedules.get(spaceId) ?? [])
+      .filter((entry) => entry.startAt < toExclusive && entry.endAt > fromInclusive)
+      .sort((a, b) => a.startAt.localeCompare(b.startAt));
+  }
+
+  async deleteScheduleEntriesBefore(spaceId: string, cutoffExclusive: string): Promise<number> {
+    const entries = this.schedules.get(spaceId) ?? [];
+    const kept = entries.filter((entry) => entry.endAt > cutoffExclusive);
+    this.schedules.set(spaceId, kept);
+    return entries.length - kept.length;
+  }
+}
+
+export interface IdentityDbMemoryStoreOptions {
+  db: IdentityDB;
+}
+
+export class IdentityDbMemoryStore implements BoxBrainMemoryStore {
+  constructor(private readonly options: IdentityDbMemoryStoreOptions) {}
+
+  async createSpace(input: { displayName: string; seedMessage: string; now: string }): Promise<MemorySpace> {
+    const slug = input.displayName.toLowerCase().replace(/[^a-z0-9가-힣]+/gi, '-').replace(/^-|-$/g, '') || 'persona';
+    const spaceName = `persona-${slug}-${crypto.randomUUID()}`;
+    const space = await this.options.db.upsertSpace({
+      name: spaceName,
+      description: `BoxBrain persona space for ${input.displayName}`,
+      metadata: {
+        boxbrainType: 'persona-space',
+        displayName: input.displayName,
+        seedMessage: input.seedMessage,
+        createdAt: input.now,
+      },
+    });
+    return {
+      id: space.name,
+      displayName: input.displayName,
+      createdAt: space.createdAt,
+      metadata: { seedMessage: input.seedMessage, identityDbSpaceId: space.id },
+    };
+  }
+
+  async getSpace(spaceId: string): Promise<MemorySpace | null> {
+    const space = await this.options.db.getSpaceByName(spaceId);
+    if (!space) return null;
+    const metadata = typeof space.metadata === 'object' && space.metadata !== null && !Array.isArray(space.metadata) ? space.metadata as Record<string, unknown> : {};
+    const displayName = typeof metadata['displayName'] === 'string' ? metadata['displayName'] : space.name;
+    return { id: space.name, displayName, createdAt: space.createdAt, metadata };
+  }
+
+  async addFact(spaceId: string, fact: FactDraft): Promise<StoredFact> {
+    const stored = await this.options.db.addFact({
+      spaceName: spaceId,
+      statement: fact.statement,
+      confidence: fact.confidence ?? null,
+      source: fact.source ?? null,
+      metadata: (fact.metadata ?? null) as never,
+      topics: normalizeTopics(fact.topics).map((topic) => ({ name: topic, category: 'entity' as const, granularity: 'concrete' as const })),
+    });
+    return this.fromIdentityFact(stored);
+  }
+
+  async findFacts(spaceId: string, topics: string[]): Promise<StoredFact[]> {
+    const uniqueTopics = normalizeTopics(topics);
+    const collected = new Map<string, StoredFact>();
+    for (const topic of uniqueTopics) {
+      const facts = await this.options.db.getTopicFacts(topic, { spaceName: spaceId });
+      for (const fact of facts) {
+        collected.set(fact.id, this.fromIdentityFact(fact));
+      }
+    }
+    return [...collected.values()].sort((a, b) => a.createdAt.localeCompare(b.createdAt));
+  }
+
+  async saveScheduleEntries(spaceId: string, entries: ScheduleEntry[]): Promise<void> {
+    for (const entry of entries) {
+      await this.addFact(spaceId, {
+        statement: `${entry.title} from ${entry.startAt} to ${entry.endAt}.`,
+        topics: ['schedule', entry.startAt.slice(0, 10), entry.activity, 'persona'],
+        source: 'boxbrain.schedule',
+        metadata: { ...entry.metadata, scheduleEntry: entry },
+      });
+    }
+  }
+
+  async listScheduleEntries(spaceId: string, fromInclusive: string, toExclusive: string): Promise<ScheduleEntry[]> {
+    const facts = await this.findFacts(spaceId, ['schedule']);
+    return facts
+      .map((fact) => fact.metadata?.['scheduleEntry'])
+      .filter((value): value is ScheduleEntry => typeof value === 'object' && value !== null && !Array.isArray(value))
+      .filter((entry) => entry.startAt < toExclusive && entry.endAt > fromInclusive)
+      .sort((a, b) => a.startAt.localeCompare(b.startAt));
+  }
+
+  async deleteScheduleEntriesBefore(_spaceId: string, _cutoffExclusive: string): Promise<number> {
+    // IdentityDB is append-oriented at the public API level. Record schedule deletion as a fact at the Persona layer.
+    return 0;
+  }
+
+  private fromIdentityFact(fact: IdentityFact): StoredFact {
+    const metadata = typeof fact.metadata === 'object' && fact.metadata !== null && !Array.isArray(fact.metadata) ? fact.metadata as Record<string, unknown> : undefined;
+    return {
+      id: fact.id,
+      statement: fact.statement,
+      topics: fact.topics.map((topic) => topic.name),
+      createdAt: fact.createdAt,
+      ...(fact.confidence === null ? {} : { confidence: fact.confidence }),
+      ...(fact.source === null ? {} : { source: fact.source }),
+      ...(metadata === undefined ? {} : { metadata }),
+    };
+  }
+}
+
+export async function createSqliteIdentityMemoryStore(filename: string): Promise<IdentityDbMemoryStore> {
+  const db = await IdentityDB.connect({ client: 'sqlite', filename });
+  await db.initialize();
+  return new IdentityDbMemoryStore({ db });
+}