summaryrefslogtreecommitdiff
path: root/makima/frontend/src/hooks/useDirectiveMemories.ts
blob: b2a87d09911ba3746ba97451f9088fe6d132d846 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import { useState, useEffect, useCallback } from "react";
import {
  type DirectiveMemory,
  type MemoryCategory,
  type CreateDirectiveMemoryRequest,
  listDirectiveMemories,
  createDirectiveMemory,
  deleteDirectiveMemory,
} from "../lib/api";

export function useDirectiveMemories(directiveId: string | undefined) {
  const [memories, setMemories] = useState<DirectiveMemory[]>([]);
  const [loading, setLoading] = useState(false);
  const [error, setError] = useState<string | null>(null);

  const refreshMemories = useCallback(async () => {
    if (!directiveId) return;
    try {
      setLoading(true);
      setError(null);
      const response = await listDirectiveMemories(directiveId);
      setMemories(response.memories);
    } catch (e) {
      setError(e instanceof Error ? e.message : "Failed to load memories");
    } finally {
      setLoading(false);
    }
  }, [directiveId]);

  useEffect(() => {
    refreshMemories();
  }, [refreshMemories]);

  const add = useCallback(async (req: CreateDirectiveMemoryRequest) => {
    if (!directiveId) return;
    try {
      setError(null);
      await createDirectiveMemory(directiveId, req);
      await refreshMemories();
    } catch (e) {
      setError(e instanceof Error ? e.message : "Failed to add memory");
    }
  }, [directiveId, refreshMemories]);

  const remove = useCallback(async (memoryId: string) => {
    if (!directiveId) return;
    try {
      setError(null);
      await deleteDirectiveMemory(directiveId, memoryId);
      await refreshMemories();
    } catch (e) {
      setError(e instanceof Error ? e.message : "Failed to delete memory");
    }
  }, [directiveId, refreshMemories]);

  const clearAll = useCallback(async () => {
    if (!directiveId) return;
    try {
      setError(null);
      await Promise.all(memories.map((m) => deleteDirectiveMemory(directiveId, m.id)));
      setMemories([]);
    } catch (e) {
      setError(e instanceof Error ? e.message : "Failed to clear memories");
    }
  }, [directiveId, memories]);

  /** Group entries by category */
  const grouped = memories.reduce<Record<MemoryCategory, DirectiveMemory[]>>(
    (acc, entry) => {
      acc[entry.category].push(entry);
      return acc;
    },
    { decision: [], context: [], preference: [], learning: [], issue: [], progress: [], other: [] },
  );

  return {
    memories,
    grouped,
    loading,
    error,
    refresh: refreshMemories,
    add,
    remove,
    clearAll,
  };
}