Supermemory

1function returnedFacts(contexts, tag) {
2  const facts = [];
3  for (const context of contexts || []) {
4    if (typeof context !== 'string') continue;
5    const block = context.match(
6      new RegExp(`<${tag}>([\\s\\S]*?)<\\/${tag}>`),
7    )?.[1];
8    if (!block) continue;
9    let current = -1;
10    for (const line of block.split('\n')) {
11      if (
12        line.startsWith('## User Profile') ||
13        line.startsWith('## Recent Context')
14      ) {
15        current = -1;
16        continue;
17      }
18      const text = line.match(/^- ◪ (.+)$/)?.[1];
19      if (text) {
20        current = facts.length;
21        facts.push(text);
22      } else if (tag === 'supermemory-context' && current >= 0) {
23        facts[current] += `\n${line}`;
24      }
25    }
26  }
27  return facts.map((fact) => fact.trimEnd());
28}
29
30const headlines = [
31  (count, narrow) =>
32    narrow
33      ? `clarified ${count}`
34      : `supermemory clarified ${count} ${count === 1 ? 'thing' : 'things'}`,
35  (count, narrow) =>
36    narrow
37      ? `surfaced ${count}`
38      : `supermemory surfaced ${count} ${count === 1 ? 'detail' : 'details'}`,
39  (count, narrow) =>
40    narrow
41      ? `${count} brought back`
42      : `supermemory brought back ${count} ${count === 1 ? 'memory' : 'memories'}`,
43  (count, narrow) =>
44    narrow
45      ? `found ${count}`
46      : `supermemory found ${count} ${count === 1 ? 'useful detail' : 'useful details'}`,
47  (count, narrow) =>
48    narrow
49      ? `uncovered ${count}`
50      : `supermemory uncovered ${count} ${count === 1 ? 'detail' : 'details'}`,
51  (count, narrow) =>
52    narrow
53      ? `remembers ${count}`
54      : `supermemory remembered ${count} ${count === 1 ? 'thing' : 'things'}`,
55];
56
57export function register(on) {
58  let facts = [];
59  let expanded = false;
60  let active = 0;
61  let headlineIndex = Math.floor(Math.random() * headlines.length) - 1;
62
63  on('classic.SessionStart', async ($, e, next) => {
64    facts = [];
65    expanded = false;
66    active = 0;
67    try {
68      const result = await next(e);
69      facts = returnedFacts(result.additionalContext, 'supermemory-context');
70      if (facts.length) headlineIndex = (headlineIndex + 1) % headlines.length;
71      return result;
72    } finally {
73      $.ui.invalidate('ui.render');
74    }
75  });
76
77  on('classic.UserPromptSubmit', async ($, e, next) => {
78    try {
79      facts = [];
80      expanded = false;
81      active = 0;
82      const result = await next(e);
83      facts = returnedFacts(result.additionalContext, 'supermemory-recall');
84      if (facts.length) headlineIndex = (headlineIndex + 1) % headlines.length;
85      return result;
86    } finally {
87      $.ui.invalidate('ui.render');
88    }
89  });
90
91  on(
92    'ui.render',
93    { component: 'AbovePrompt', surface: 'terminal' },
94    async ($, e, next) => {
95      const original = await next(e);
96      if (e.props.hasSurvey || facts.length === 0) return original;
97
98      const { Box, Text, Button } = $.ui.resolve(e);
99      const width = Math.min(
100        e.props.bodyColumns,
101        Math.max(1, Math.min(64, e.props.bodyColumns - 4)),
102      );
103      const label = `◪ ${headlines[headlineIndex](facts.length, e.props.bodyColumns < 32)}`;
104
105      return Box({
106        flexDirection: 'column',
107        children: [
108          original,
109          Button({
110            key: 'recall-details',
111            label: `${label} ${expanded ? '▴' : '▾'}`,
112            plain: true,
113            dimColor: true,
114            onPress: () => {
115              expanded = !expanded;
116              if (expanded) active = 0;
117              $.ui.invalidate('ui.render');
118            },
119          }),
120          expanded
121            ? Box({
122                paddingX: 1,
123                width,
124                flexDirection: 'column',
125                children: [
126                  Text({ wrap: 'wrap', children: facts[active] }),
127                  facts.length > 1
128                    ? Box({
129                        flexDirection: 'row',
130                        children: [
131                          Button({
132                            key: 'recall-previous',
133                            label: '‹',
134                            plain: true,
135                            dimColor: true,
136                            onPress: () => {
137                              active =
138                                (active - 1 + facts.length) % facts.length;
139                              $.ui.invalidate('ui.render');
140                            },
141                          }),
142                          Text({
143                            dimColor: true,
144                            children: ` ${active + 1}/${facts.length} `,
145                          }),
146                          Button({
147                            key: 'recall-next',
148                            label: '›',
149                            plain: true,
150                            dimColor: true,
151                            onPress: () => {
152                              active = (active + 1) % facts.length;
153                              $.ui.invalidate('ui.render');
154                            },
155                          }),
156                        ],
157                      })
158                    : null,
159                ],
160              })
161            : null,
162        ],
163      });
164    },
165  );
166}
167