Function createMemoryStore

mcp/src/server/store.ts:67–264

()

// -----------------------------------------------------------------------------

function createMemoryStore(): AFSStore {
  const sessions = new Map<string, Session>();
  const annotations = new Map<string, Annotation>();
  const events: AFSEvent[] = [];

  function generateId(): string {
    return `${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
  }

  return {
    createSession(url: string, projectId?: string): Session {
      const session: Session = {
        id: generateId(),
        url,
        status: "active",
        createdAt: new Date().toISOString(),
        projectId,
      };
      sessions.set(session.id, session);

      const event = eventBus.emit("session.created", session.id, session);
      events.push(event);

      return session;
    },

    getSession(id: string): Session | undefined {
      return sessions.get(id);
    },

    getSessionWithAnnotations(id: string): SessionWithAnnotations | undefined {
      const session = sessions.get(id);
      if (!session) return undefined;

      const sessionAnnotations = Array.from(annotations.values()).filter(
        (a) => a.sessionId === id
      );

      return {
        ...session,
        annotations: sessionAnnotations,
      };
    },

    updateSessionStatus(id: string, status: SessionStatus): Session | undefined {
      const session = sessions.get(id);
      if (!session) return undefined;

      session.status = status;
      session.updatedAt = new Date().toISOString();

      const eventType = status === "closed" ? "session.closed" : "session.updated";
      const event = eventBus.emit(eventType, id, session);
      events.push(event);

      return session;
    },