@tonyclaw/agent-inspector 4.0.1 → 4.0.3

package/src/proxy/groupStore.ts

@@ -1,336 +0,0 @@
-import assert from "node:assert/strict";
-import { randomUUID } from "node:crypto";
-import { existsSync, readFileSync, rmSync, writeFileSync } from "node:fs";
-import { dirname, isAbsolute, join, relative, resolve } from "node:path";
-import { z } from "zod";
-import {
-  AddInspectorGroupSessionInputSchema,
-  CreateInspectorGroupInputSchema,
-  DeleteInspectorGroupOptionsSchema,
-  InspectorGroupSchema,
-  UpdateInspectorGroupInputSchema,
-  type AddInspectorGroupSessionInput,
-  type CreateInspectorGroupInput,
-  type DeleteInspectorGroupOptions,
-  type DeleteInspectorGroupResponse,
-  type DeleteInspectorGroupsResponse,
-  type InspectorGroup,
-  type InspectorGroupEvidence,
-  type InspectorGroupMember,
-  type UpdateInspectorGroupInput,
-} from "../lib/groupContract";
-import { getDataDir } from "./dataDir";
-import { readJsonFile } from "../lib/jsonFile";
-import {
-  ensurePrivateDirectorySync,
-  PRIVATE_FILE_MODE,
-  securePrivateFileSync,
-} from "./privateDataPath";
-
-const GroupsFileSchema = z.object({
-  groups: z.array(InspectorGroupSchema),
-});
-
-function groupsFilePath(): string {
-  return join(getDataDir(), "groups.json");
-}
-
-function nowIso(): string {
-  return new Date().toISOString();
-}
-
-function stableId(input: string): string {
-  return input
-    .trim()
-    .replace(/[^A-Za-z0-9_.-]+/g, "-")
-    .replace(/^-+|-+$/g, "")
-    .slice(0, 80);
-}
-
-function createGroupId(): string {
-  return `grp_${Date.now().toString(36)}_${randomUUID().slice(0, 8)}`;
-}
-
-function createMemberId(): string {
-  return `mem_${Date.now().toString(36)}_${randomUUID().slice(0, 8)}`;
-}
-
-function normalizeGroupId(input: string | undefined): string {
-  if (input === undefined) return createGroupId();
-  const normalized = stableId(input);
-  return normalized.length > 0 ? normalized : createGroupId();
-}
-
-function normalizeMemberId(input: string | undefined, sessionId: string): string {
-  if (input === undefined) {
-    const normalizedSession = stableId(sessionId);
-    return normalizedSession.length > 0 ? `mem_${normalizedSession}` : createMemberId();
-  }
-  const normalized = stableId(input);
-  return normalized.length > 0 ? normalized : createMemberId();
-}
-
-function nullableText(value: string | null | undefined): string | null {
-  if (value === undefined || value === null) return null;
-  const trimmed = value.trim();
-  return trimmed.length > 0 ? trimmed : null;
-}
-
-function readGroups(): InspectorGroup[] {
-  const result = readJsonFile(groupsFilePath(), GroupsFileSchema);
-  return result.ok ? result.data.groups : [];
-}
-
-function writeGroups(groups: readonly InspectorGroup[]): void {
-  const filePath = groupsFilePath();
-  assert(ensurePrivateDirectorySync(dirname(filePath)), "Could not secure groups directory");
-  writeFileSync(filePath, `${JSON.stringify({ groups }, null, 2)}\n`, {
-    encoding: "utf8",
-    mode: PRIVATE_FILE_MODE,
-  });
-  assert(securePrivateFileSync(filePath), "Could not secure groups file");
-}
-
-function sortedGroups(groups: readonly InspectorGroup[]): InspectorGroup[] {
-  return [...groups].sort((left, right) => right.updatedAt.localeCompare(left.updatedAt));
-}
-
-function isPathInside(parentPath: string, childPath: string): boolean {
-  const relativePath = relative(parentPath, childPath);
-  return (
-    relativePath === "" ||
-    (relativePath.length > 0 && !relativePath.startsWith("..") && !isAbsolute(relativePath))
-  );
-}
-
-function evidenceDirectoryFor(group: InspectorGroup): string | null {
-  if (group.evidence === null) return null;
-
-  const root = resolve(getDataDir(), "evidence", "groups");
-  const evidenceDir = resolve(dirname(group.evidence.jsonPath));
-  return isPathInside(root, evidenceDir) ? evidenceDir : null;
-}
-
-function removeEvidenceDirectory(evidencePath: string | null, deleteEvidence: boolean): boolean {
-  if (!deleteEvidence || evidencePath === null || !existsSync(evidencePath)) return false;
-
-  try {
-    rmSync(evidencePath, { recursive: true, force: true });
-    return !existsSync(evidencePath);
-  } catch {
-    return false;
-  }
-}
-
-export function listGroups(): InspectorGroup[] {
-  return sortedGroups(readGroups());
-}
-
-export function getGroup(groupId: string): InspectorGroup | null {
-  return readGroups().find((group) => group.id === groupId) ?? null;
-}
-
-export function createGroup(input: CreateInspectorGroupInput): InspectorGroup {
-  const parsed = CreateInspectorGroupInputSchema.parse(input);
-  const groupId = normalizeGroupId(parsed?.groupId);
-  const existing = getGroup(groupId);
-  if (existing !== null) return existing;
-
-  const timestamp = nowIso();
-  const group: InspectorGroup = {
-    id: groupId,
-    title: parsed?.title ?? `Inspector Group ${groupId}`,
-    kind: parsed?.kind ?? "evaluation",
-    task: parsed?.task ?? null,
-    project: parsed?.project ?? null,
-    status: parsed?.status ?? "created",
-    tags: parsed?.tags ?? [],
-    metadata: parsed?.metadata ?? {},
-    evidence: null,
-    members: [],
-    createdAt: timestamp,
-    updatedAt: timestamp,
-  };
-
-  writeGroups([...readGroups(), group]);
-  return group;
-}
-
-export function updateGroup(
-  groupId: string,
-  input: UpdateInspectorGroupInput,
-): InspectorGroup | null {
-  const parsed = UpdateInspectorGroupInputSchema.parse(input);
-  const groups = readGroups();
-  let updatedGroup: InspectorGroup | null = null;
-  const updatedAt = nowIso();
-
-  const updatedGroups = groups.map((group) => {
-    if (group.id !== groupId) return group;
-    updatedGroup = {
-      ...group,
-      title: parsed.title ?? group.title,
-      kind: parsed.kind ?? group.kind,
-      task: parsed.task === undefined ? group.task : parsed.task,
-      project: parsed.project === undefined ? group.project : parsed.project,
-      status: parsed.status ?? group.status,
-      tags: parsed.tags ?? group.tags,
-      metadata:
-        parsed.metadata === undefined ? group.metadata : { ...group.metadata, ...parsed.metadata },
-      updatedAt,
-    };
-    return updatedGroup;
-  });
-
-  if (updatedGroup === null) return null;
-  writeGroups(updatedGroups);
-  return updatedGroup;
-}
-
-export function deleteGroup(
-  groupId: string,
-  options: DeleteInspectorGroupOptions = {},
-): DeleteInspectorGroupResponse | null {
-  const parsed = DeleteInspectorGroupOptionsSchema.parse(options);
-  const groups = readGroups();
-  const group = groups.find((candidate) => candidate.id === groupId);
-  if (group === undefined) return null;
-
-  const evidencePath = evidenceDirectoryFor(group);
-  writeGroups(groups.filter((candidate) => candidate.id !== groupId));
-
-  return {
-    groupId: group.id,
-    deleted: true,
-    evidenceDeleted: removeEvidenceDirectory(evidencePath, parsed.deleteEvidence === true),
-    evidencePath,
-  };
-}
-
-export function deleteGroups(
-  options: DeleteInspectorGroupOptions = {},
-): DeleteInspectorGroupsResponse {
-  const parsed = DeleteInspectorGroupOptionsSchema.parse(options);
-  const groups = readGroups();
-  const deletedGroups = groups.map((group) => {
-    const evidencePath = evidenceDirectoryFor(group);
-    return {
-      groupId: group.id,
-      deleted: true,
-      evidenceDeleted: removeEvidenceDirectory(evidencePath, parsed.deleteEvidence === true),
-      evidencePath,
-    };
-  });
-
-  writeGroups([]);
-
-  return {
-    deleted: deletedGroups.length,
-    groups: deletedGroups,
-  };
-}
-
-function matchesExistingMember(
-  member: InspectorGroupMember,
-  memberId: string,
-  sessionId: string,
-  runId: string | null,
-): boolean {
-  if (member.id === memberId) return true;
-  if (member.sessionId !== sessionId) return false;
-  return member.runId === runId;
-}
-
-function upsertMember(
-  group: InspectorGroup,
-  input: AddInspectorGroupSessionInput,
-  updatedAt: string,
-): InspectorGroup {
-  const parsed = AddInspectorGroupSessionInputSchema.parse(input);
-  const sessionId = parsed.sessionId.trim();
-  const runId = nullableText(parsed.runId);
-  const memberId = normalizeMemberId(parsed.memberId, sessionId);
-  let matched = false;
-
-  const members = group.members.map((member) => {
-    if (!matchesExistingMember(member, memberId, sessionId, runId)) return member;
-    matched = true;
-    return {
-      ...member,
-      sessionId,
-      runId,
-      label: parsed.label === undefined ? member.label : nullableText(parsed.label),
-      agent: parsed.agent === undefined ? member.agent : nullableText(parsed.agent),
-      provider: parsed.provider === undefined ? member.provider : nullableText(parsed.provider),
-      model: parsed.model === undefined ? member.model : nullableText(parsed.model),
-      status: parsed.status === undefined ? member.status : parsed.status,
-      metadata:
-        parsed.metadata === undefined
-          ? member.metadata
-          : { ...member.metadata, ...parsed.metadata },
-      updatedAt,
-    };
-  });
-
-  if (!matched) {
-    members.push({
-      id: memberId,
-      groupId: group.id,
-      sessionId,
-      runId,
-      label: nullableText(parsed.label),
-      agent: nullableText(parsed.agent),
-      provider: nullableText(parsed.provider),
-      model: nullableText(parsed.model),
-      status: parsed.status ?? null,
-      metadata: parsed.metadata ?? {},
-      addedAt: updatedAt,
-      updatedAt,
-    });
-  }
-
-  return { ...group, members, updatedAt };
-}
-
-export function addGroupSession(
-  groupId: string,
-  input: AddInspectorGroupSessionInput,
-): InspectorGroup | null {
-  const groups = readGroups();
-  let updatedGroup: InspectorGroup | null = null;
-  const updatedAt = nowIso();
-
-  const updatedGroups = groups.map((group) => {
-    if (group.id !== groupId) return group;
-    updatedGroup = upsertMember(group, input, updatedAt);
-    return updatedGroup;
-  });
-
-  if (updatedGroup === null) return null;
-  writeGroups(updatedGroups);
-  return updatedGroup;
-}
-
-export function updateGroupEvidence(
-  groupId: string,
-  evidence: InspectorGroupEvidence,
-): InspectorGroup | null {
-  const groups = readGroups();
-  let updatedGroup: InspectorGroup | null = null;
-  const updatedGroups = groups.map((group) => {
-    if (group.id !== groupId) return group;
-    updatedGroup = {
-      ...group,
-      evidence,
-      updatedAt: evidence.exportedAt,
-    };
-    return updatedGroup;
-  });
-  if (updatedGroup === null) return null;
-  writeGroups(updatedGroups);
-  return updatedGroup;
-}
-
-export function _resetGroupsForTests(): void {
-  writeGroups([]);
-}