reigncode-app 1.3.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (300) hide show
  1. package/AGENTS.md +30 -0
  2. package/Dockerfile +21 -0
  3. package/README.md +51 -0
  4. package/bunfig.toml +3 -0
  5. package/create-effect-simplification-spec.md +515 -0
  6. package/e2e/AGENTS.md +226 -0
  7. package/e2e/actions.ts +1018 -0
  8. package/e2e/app/home.spec.ts +24 -0
  9. package/e2e/app/navigation.spec.ts +10 -0
  10. package/e2e/app/palette.spec.ts +20 -0
  11. package/e2e/app/server-default.spec.ts +58 -0
  12. package/e2e/app/session.spec.ts +16 -0
  13. package/e2e/app/titlebar-history.spec.ts +120 -0
  14. package/e2e/commands/input-focus.spec.ts +15 -0
  15. package/e2e/commands/panels.spec.ts +33 -0
  16. package/e2e/commands/tab-close.spec.ts +32 -0
  17. package/e2e/files/file-open.spec.ts +31 -0
  18. package/e2e/files/file-tree.spec.ts +56 -0
  19. package/e2e/files/file-viewer.spec.ts +156 -0
  20. package/e2e/fixtures.ts +154 -0
  21. package/e2e/models/model-picker.spec.ts +48 -0
  22. package/e2e/models/models-visibility.spec.ts +61 -0
  23. package/e2e/projects/project-edit.spec.ts +43 -0
  24. package/e2e/projects/projects-close.spec.ts +54 -0
  25. package/e2e/projects/projects-switch.spec.ts +116 -0
  26. package/e2e/projects/workspace-new-session.spec.ts +94 -0
  27. package/e2e/projects/workspaces.spec.ts +375 -0
  28. package/e2e/prompt/context.spec.ts +95 -0
  29. package/e2e/prompt/prompt-async.spec.ts +76 -0
  30. package/e2e/prompt/prompt-drop-file-uri.spec.ts +22 -0
  31. package/e2e/prompt/prompt-drop-file.spec.ts +30 -0
  32. package/e2e/prompt/prompt-history.spec.ts +184 -0
  33. package/e2e/prompt/prompt-mention.spec.ts +26 -0
  34. package/e2e/prompt/prompt-multiline.spec.ts +24 -0
  35. package/e2e/prompt/prompt-shell.spec.ts +62 -0
  36. package/e2e/prompt/prompt-slash-open.spec.ts +22 -0
  37. package/e2e/prompt/prompt-slash-share.spec.ts +64 -0
  38. package/e2e/prompt/prompt-slash-terminal.spec.ts +18 -0
  39. package/e2e/prompt/prompt.spec.ts +55 -0
  40. package/e2e/selectors.ts +75 -0
  41. package/e2e/session/session-child-navigation.spec.ts +37 -0
  42. package/e2e/session/session-composer-dock.spec.ts +530 -0
  43. package/e2e/session/session-model-persistence.spec.ts +359 -0
  44. package/e2e/session/session-review.spec.ts +426 -0
  45. package/e2e/session/session-undo-redo.spec.ts +233 -0
  46. package/e2e/session/session.spec.ts +174 -0
  47. package/e2e/settings/settings-keybinds.spec.ts +389 -0
  48. package/e2e/settings/settings-models.spec.ts +122 -0
  49. package/e2e/settings/settings-providers.spec.ts +136 -0
  50. package/e2e/settings/settings.spec.ts +519 -0
  51. package/e2e/sidebar/sidebar-popover-actions.spec.ts +118 -0
  52. package/e2e/sidebar/sidebar-session-links.spec.ts +30 -0
  53. package/e2e/sidebar/sidebar.spec.ts +40 -0
  54. package/e2e/status/status-popover.spec.ts +94 -0
  55. package/e2e/terminal/terminal-init.spec.ts +28 -0
  56. package/e2e/terminal/terminal-reconnect.spec.ts +46 -0
  57. package/e2e/terminal/terminal-tabs.spec.ts +168 -0
  58. package/e2e/terminal/terminal.spec.ts +18 -0
  59. package/e2e/thinking-level.spec.ts +25 -0
  60. package/e2e/tsconfig.json +9 -0
  61. package/e2e/utils.ts +63 -0
  62. package/happydom.ts +75 -0
  63. package/index.html +23 -0
  64. package/package.json +77 -0
  65. package/playwright.config.ts +45 -0
  66. package/public/_headers +17 -0
  67. package/public/oc-theme-preload.js +35 -0
  68. package/script/e2e-local.ts +180 -0
  69. package/src/addons/serialize.test.ts +319 -0
  70. package/src/addons/serialize.ts +634 -0
  71. package/src/app.tsx +308 -0
  72. package/src/components/debug-bar.tsx +443 -0
  73. package/src/components/dialog-connect-provider.tsx +617 -0
  74. package/src/components/dialog-custom-provider-form.ts +158 -0
  75. package/src/components/dialog-custom-provider.test.ts +80 -0
  76. package/src/components/dialog-custom-provider.tsx +329 -0
  77. package/src/components/dialog-edit-project.tsx +255 -0
  78. package/src/components/dialog-fork.tsx +108 -0
  79. package/src/components/dialog-manage-models.tsx +101 -0
  80. package/src/components/dialog-release-notes.tsx +144 -0
  81. package/src/components/dialog-select-directory.tsx +392 -0
  82. package/src/components/dialog-select-file.tsx +466 -0
  83. package/src/components/dialog-select-mcp.tsx +107 -0
  84. package/src/components/dialog-select-model-unpaid.tsx +137 -0
  85. package/src/components/dialog-select-model.tsx +220 -0
  86. package/src/components/dialog-select-provider.tsx +86 -0
  87. package/src/components/dialog-select-server.tsx +649 -0
  88. package/src/components/dialog-settings.tsx +73 -0
  89. package/src/components/file-tree.test.ts +78 -0
  90. package/src/components/file-tree.tsx +507 -0
  91. package/src/components/link.tsx +26 -0
  92. package/src/components/model-tooltip.tsx +91 -0
  93. package/src/components/prompt-input/attachments.test.ts +44 -0
  94. package/src/components/prompt-input/attachments.ts +201 -0
  95. package/src/components/prompt-input/build-request-parts.test.ts +312 -0
  96. package/src/components/prompt-input/build-request-parts.ts +175 -0
  97. package/src/components/prompt-input/context-items.tsx +88 -0
  98. package/src/components/prompt-input/drag-overlay.tsx +25 -0
  99. package/src/components/prompt-input/editor-dom.test.ts +99 -0
  100. package/src/components/prompt-input/editor-dom.ts +148 -0
  101. package/src/components/prompt-input/files.ts +66 -0
  102. package/src/components/prompt-input/history.test.ts +153 -0
  103. package/src/components/prompt-input/history.ts +256 -0
  104. package/src/components/prompt-input/image-attachments.tsx +58 -0
  105. package/src/components/prompt-input/paste.ts +24 -0
  106. package/src/components/prompt-input/placeholder.test.ts +48 -0
  107. package/src/components/prompt-input/placeholder.ts +15 -0
  108. package/src/components/prompt-input/slash-popover.tsx +141 -0
  109. package/src/components/prompt-input/submit.test.ts +346 -0
  110. package/src/components/prompt-input/submit.ts +579 -0
  111. package/src/components/prompt-input.tsx +1595 -0
  112. package/src/components/server/server-row.tsx +130 -0
  113. package/src/components/session/index.ts +5 -0
  114. package/src/components/session/session-context-breakdown.test.ts +61 -0
  115. package/src/components/session/session-context-breakdown.ts +132 -0
  116. package/src/components/session/session-context-format.ts +20 -0
  117. package/src/components/session/session-context-metrics.test.ts +101 -0
  118. package/src/components/session/session-context-metrics.ts +82 -0
  119. package/src/components/session/session-context-tab.tsx +339 -0
  120. package/src/components/session/session-header.tsx +486 -0
  121. package/src/components/session/session-new-view.tsx +91 -0
  122. package/src/components/session/session-sortable-tab.tsx +70 -0
  123. package/src/components/session/session-sortable-terminal-tab.tsx +193 -0
  124. package/src/components/session-context-usage.tsx +122 -0
  125. package/src/components/settings-general.tsx +585 -0
  126. package/src/components/settings-keybinds.tsx +453 -0
  127. package/src/components/settings-list.tsx +5 -0
  128. package/src/components/settings-models.tsx +137 -0
  129. package/src/components/settings-providers.tsx +251 -0
  130. package/src/components/status-popover.tsx +419 -0
  131. package/src/components/terminal.tsx +653 -0
  132. package/src/components/titlebar-history.test.ts +63 -0
  133. package/src/components/titlebar-history.ts +57 -0
  134. package/src/components/titlebar.tsx +312 -0
  135. package/src/constants/file-picker.ts +89 -0
  136. package/src/context/command-keybind.test.ts +69 -0
  137. package/src/context/command.test.ts +25 -0
  138. package/src/context/command.tsx +437 -0
  139. package/src/context/comments.test.ts +186 -0
  140. package/src/context/comments.tsx +243 -0
  141. package/src/context/file/content-cache.ts +88 -0
  142. package/src/context/file/path.test.ts +360 -0
  143. package/src/context/file/path.ts +151 -0
  144. package/src/context/file/tree-store.ts +170 -0
  145. package/src/context/file/types.ts +41 -0
  146. package/src/context/file/view-cache.ts +146 -0
  147. package/src/context/file/watcher.test.ts +149 -0
  148. package/src/context/file/watcher.ts +53 -0
  149. package/src/context/file-content-eviction-accounting.test.ts +65 -0
  150. package/src/context/file.tsx +280 -0
  151. package/src/context/global-sdk.tsx +232 -0
  152. package/src/context/global-sync/bootstrap.ts +206 -0
  153. package/src/context/global-sync/child-store.test.ts +38 -0
  154. package/src/context/global-sync/child-store.ts +281 -0
  155. package/src/context/global-sync/event-reducer.test.ts +552 -0
  156. package/src/context/global-sync/event-reducer.ts +359 -0
  157. package/src/context/global-sync/eviction.ts +28 -0
  158. package/src/context/global-sync/queue.ts +83 -0
  159. package/src/context/global-sync/session-cache.test.ts +102 -0
  160. package/src/context/global-sync/session-cache.ts +62 -0
  161. package/src/context/global-sync/session-load.ts +25 -0
  162. package/src/context/global-sync/session-prefetch.test.ts +96 -0
  163. package/src/context/global-sync/session-prefetch.ts +100 -0
  164. package/src/context/global-sync/session-trim.test.ts +59 -0
  165. package/src/context/global-sync/session-trim.ts +56 -0
  166. package/src/context/global-sync/types.ts +133 -0
  167. package/src/context/global-sync/utils.ts +25 -0
  168. package/src/context/global-sync.test.ts +122 -0
  169. package/src/context/global-sync.tsx +408 -0
  170. package/src/context/highlights.tsx +233 -0
  171. package/src/context/language.tsx +248 -0
  172. package/src/context/layout-scroll.test.ts +64 -0
  173. package/src/context/layout-scroll.ts +126 -0
  174. package/src/context/layout.test.ts +69 -0
  175. package/src/context/layout.tsx +937 -0
  176. package/src/context/local.tsx +422 -0
  177. package/src/context/model-variant.test.ts +86 -0
  178. package/src/context/model-variant.ts +52 -0
  179. package/src/context/models.tsx +163 -0
  180. package/src/context/notification.tsx +373 -0
  181. package/src/context/permission-auto-respond.test.ts +102 -0
  182. package/src/context/permission-auto-respond.ts +51 -0
  183. package/src/context/permission.tsx +277 -0
  184. package/src/context/platform.tsx +99 -0
  185. package/src/context/prompt.tsx +297 -0
  186. package/src/context/sdk.tsx +49 -0
  187. package/src/context/server.tsx +295 -0
  188. package/src/context/settings.tsx +241 -0
  189. package/src/context/sync-optimistic.test.ts +123 -0
  190. package/src/context/sync.tsx +618 -0
  191. package/src/context/terminal-title.ts +51 -0
  192. package/src/context/terminal.test.ts +82 -0
  193. package/src/context/terminal.tsx +437 -0
  194. package/src/entry.tsx +144 -0
  195. package/src/env.d.ts +18 -0
  196. package/src/hooks/use-providers.ts +44 -0
  197. package/src/i18n/ar.ts +855 -0
  198. package/src/i18n/br.ts +867 -0
  199. package/src/i18n/bs.ts +943 -0
  200. package/src/i18n/da.ts +937 -0
  201. package/src/i18n/de.ts +879 -0
  202. package/src/i18n/en.ts +948 -0
  203. package/src/i18n/es.ts +950 -0
  204. package/src/i18n/fr.ts +878 -0
  205. package/src/i18n/ja.ts +861 -0
  206. package/src/i18n/ko.ts +860 -0
  207. package/src/i18n/no.ts +944 -0
  208. package/src/i18n/parity.test.ts +32 -0
  209. package/src/i18n/pl.ts +865 -0
  210. package/src/i18n/ru.ts +946 -0
  211. package/src/i18n/th.ts +933 -0
  212. package/src/i18n/tr.ts +952 -0
  213. package/src/i18n/zh.ts +930 -0
  214. package/src/i18n/zht.ts +925 -0
  215. package/src/index.css +29 -0
  216. package/src/index.ts +6 -0
  217. package/src/pages/directory-layout.tsx +88 -0
  218. package/src/pages/error.tsx +327 -0
  219. package/src/pages/home.tsx +131 -0
  220. package/src/pages/layout/deep-links.ts +50 -0
  221. package/src/pages/layout/helpers.test.ts +211 -0
  222. package/src/pages/layout/helpers.ts +98 -0
  223. package/src/pages/layout/inline-editor.tsx +126 -0
  224. package/src/pages/layout/sidebar-items.tsx +437 -0
  225. package/src/pages/layout/sidebar-project.tsx +384 -0
  226. package/src/pages/layout/sidebar-shell.tsx +125 -0
  227. package/src/pages/layout/sidebar-workspace.tsx +504 -0
  228. package/src/pages/layout.tsx +2509 -0
  229. package/src/pages/session/composer/index.ts +2 -0
  230. package/src/pages/session/composer/session-composer-region.tsx +255 -0
  231. package/src/pages/session/composer/session-composer-state.test.ts +128 -0
  232. package/src/pages/session/composer/session-composer-state.ts +249 -0
  233. package/src/pages/session/composer/session-followup-dock.tsx +109 -0
  234. package/src/pages/session/composer/session-permission-dock.tsx +74 -0
  235. package/src/pages/session/composer/session-question-dock.tsx +449 -0
  236. package/src/pages/session/composer/session-request-tree.ts +52 -0
  237. package/src/pages/session/composer/session-revert-dock.tsx +99 -0
  238. package/src/pages/session/composer/session-todo-dock.tsx +330 -0
  239. package/src/pages/session/file-tab-scroll.test.ts +40 -0
  240. package/src/pages/session/file-tab-scroll.ts +67 -0
  241. package/src/pages/session/file-tabs.tsx +456 -0
  242. package/src/pages/session/handoff.ts +36 -0
  243. package/src/pages/session/helpers.test.ts +181 -0
  244. package/src/pages/session/helpers.ts +198 -0
  245. package/src/pages/session/message-gesture.test.ts +62 -0
  246. package/src/pages/session/message-gesture.ts +21 -0
  247. package/src/pages/session/message-id-from-hash.ts +6 -0
  248. package/src/pages/session/message-timeline.tsx +1013 -0
  249. package/src/pages/session/review-tab.tsx +170 -0
  250. package/src/pages/session/session-layout.ts +20 -0
  251. package/src/pages/session/session-model-helpers.test.ts +51 -0
  252. package/src/pages/session/session-model-helpers.ts +16 -0
  253. package/src/pages/session/session-side-panel.tsx +453 -0
  254. package/src/pages/session/terminal-label.ts +16 -0
  255. package/src/pages/session/terminal-panel.test.ts +25 -0
  256. package/src/pages/session/terminal-panel.tsx +326 -0
  257. package/src/pages/session/use-session-commands.tsx +495 -0
  258. package/src/pages/session/use-session-hash-scroll.test.ts +16 -0
  259. package/src/pages/session/use-session-hash-scroll.ts +197 -0
  260. package/src/pages/session.tsx +1841 -0
  261. package/src/sst-env.d.ts +12 -0
  262. package/src/testing/model-selection.ts +80 -0
  263. package/src/testing/prompt.ts +56 -0
  264. package/src/testing/session-composer.ts +84 -0
  265. package/src/testing/terminal.ts +118 -0
  266. package/src/theme-preload.test.ts +46 -0
  267. package/src/utils/agent.ts +23 -0
  268. package/src/utils/aim.ts +138 -0
  269. package/src/utils/base64.ts +10 -0
  270. package/src/utils/comment-note.ts +88 -0
  271. package/src/utils/id.ts +99 -0
  272. package/src/utils/notification-click.test.ts +27 -0
  273. package/src/utils/notification-click.ts +13 -0
  274. package/src/utils/persist.test.ts +115 -0
  275. package/src/utils/persist.ts +476 -0
  276. package/src/utils/prompt.test.ts +44 -0
  277. package/src/utils/prompt.ts +203 -0
  278. package/src/utils/runtime-adapters.test.ts +62 -0
  279. package/src/utils/runtime-adapters.ts +39 -0
  280. package/src/utils/same.ts +6 -0
  281. package/src/utils/scoped-cache.test.ts +69 -0
  282. package/src/utils/scoped-cache.ts +104 -0
  283. package/src/utils/server-errors.test.ts +131 -0
  284. package/src/utils/server-errors.ts +80 -0
  285. package/src/utils/server-health.test.ts +123 -0
  286. package/src/utils/server-health.ts +91 -0
  287. package/src/utils/server.ts +22 -0
  288. package/src/utils/solid-dnd.tsx +49 -0
  289. package/src/utils/sound.ts +117 -0
  290. package/src/utils/terminal-writer.test.ts +64 -0
  291. package/src/utils/terminal-writer.ts +65 -0
  292. package/src/utils/time.ts +22 -0
  293. package/src/utils/uuid.test.ts +78 -0
  294. package/src/utils/uuid.ts +12 -0
  295. package/src/utils/worktree.test.ts +46 -0
  296. package/src/utils/worktree.ts +73 -0
  297. package/sst-env.d.ts +10 -0
  298. package/tsconfig.json +26 -0
  299. package/vite.config.ts +15 -0
  300. package/vite.js +26 -0
@@ -0,0 +1,130 @@
1
+ import { Tooltip } from "@reign-labs/ui/tooltip"
2
+ import {
3
+ children,
4
+ createEffect,
5
+ createMemo,
6
+ createSignal,
7
+ type JSXElement,
8
+ onCleanup,
9
+ onMount,
10
+ type ParentProps,
11
+ Show,
12
+ } from "solid-js"
13
+ import { useLanguage } from "@/context/language"
14
+ import { type ServerConnection, serverName } from "@/context/server"
15
+ import type { ServerHealth } from "@/utils/server-health"
16
+
17
+ interface ServerRowProps extends ParentProps {
18
+ conn: ServerConnection.Any
19
+ status?: ServerHealth
20
+ class?: string
21
+ nameClass?: string
22
+ versionClass?: string
23
+ dimmed?: boolean
24
+ badge?: JSXElement
25
+ showCredentials?: boolean
26
+ }
27
+
28
+ export function ServerRow(props: ServerRowProps) {
29
+ const language = useLanguage()
30
+ const [truncated, setTruncated] = createSignal(false)
31
+ let nameRef: HTMLSpanElement | undefined
32
+ let versionRef: HTMLSpanElement | undefined
33
+ const name = createMemo(() => serverName(props.conn))
34
+
35
+ const check = () => {
36
+ const nameTruncated = nameRef ? nameRef.scrollWidth > nameRef.clientWidth : false
37
+ const versionTruncated = versionRef ? versionRef.scrollWidth > versionRef.clientWidth : false
38
+ setTruncated(nameTruncated || versionTruncated)
39
+ }
40
+
41
+ createEffect(() => {
42
+ name()
43
+ props.conn.http.url
44
+ props.status?.version
45
+ queueMicrotask(check)
46
+ })
47
+
48
+ onMount(() => {
49
+ check()
50
+ if (typeof ResizeObserver !== "function") return
51
+ const observer = new ResizeObserver(check)
52
+ if (nameRef) observer.observe(nameRef)
53
+ if (versionRef) observer.observe(versionRef)
54
+ onCleanup(() => observer.disconnect())
55
+ })
56
+
57
+ const tooltipValue = () => (
58
+ <span class="flex items-center gap-2">
59
+ <span>{serverName(props.conn, true)}</span>
60
+ <Show when={props.status?.version}>
61
+ <span class="text-text-invert-weak">v{props.status?.version}</span>
62
+ </Show>
63
+ </span>
64
+ )
65
+
66
+ const badge = children(() => props.badge)
67
+
68
+ return (
69
+ <Tooltip
70
+ class="flex-1 min-w-0"
71
+ value={tooltipValue()}
72
+ contentStyle={{ "max-width": "none", "white-space": "nowrap" }}
73
+ placement="top-start"
74
+ inactive={!truncated() && !props.conn.displayName}
75
+ >
76
+ <div class={props.class} classList={{ "opacity-50": props.dimmed }}>
77
+ <div class="flex flex-col items-start min-w-0 w-full">
78
+ <div class="flex flex-row items-center gap-2 min-w-0 w-full">
79
+ <span ref={nameRef} class={`${props.nameClass ?? "truncate"} min-w-0`}>
80
+ {name()}
81
+ </span>
82
+ <Show
83
+ when={badge()}
84
+ fallback={
85
+ <Show when={props.status?.version}>
86
+ <span
87
+ ref={versionRef}
88
+ class={`${props.versionClass ?? "text-text-weak text-14-regular truncate"} min-w-0`}
89
+ >
90
+ v{props.status?.version}
91
+ </span>
92
+ </Show>
93
+ }
94
+ >
95
+ {(badge) => badge()}
96
+ </Show>
97
+ </div>
98
+ <Show when={props.showCredentials && props.conn.type === "http" && props.conn}>
99
+ {(conn) => (
100
+ <div class="flex flex-row gap-3">
101
+ <span>
102
+ {conn().http.username ? (
103
+ <span class="text-text-weak">{conn().http.username}</span>
104
+ ) : (
105
+ <span class="text-text-weaker">{language.t("server.row.noUsername")}</span>
106
+ )}
107
+ </span>
108
+ {conn().http.password && <span class="text-text-weak">••••••••</span>}
109
+ </div>
110
+ )}
111
+ </Show>
112
+ </div>
113
+ {props.children}
114
+ </div>
115
+ </Tooltip>
116
+ )
117
+ }
118
+
119
+ export function ServerHealthIndicator(props: { health?: ServerHealth }) {
120
+ return (
121
+ <div
122
+ classList={{
123
+ "size-1.5 rounded-full shrink-0": true,
124
+ "bg-icon-success-base": props.health?.healthy === true,
125
+ "bg-icon-critical-base": props.health?.healthy === false,
126
+ "bg-border-weak-base": props.health === undefined,
127
+ }}
128
+ />
129
+ )
130
+ }
@@ -0,0 +1,5 @@
1
+ export { SessionHeader } from "./session-header"
2
+ export { SessionContextTab } from "./session-context-tab"
3
+ export { SortableTab, FileVisual } from "./session-sortable-tab"
4
+ export { SortableTerminalTab } from "./session-sortable-terminal-tab"
5
+ export { NewSessionView } from "./session-new-view"
@@ -0,0 +1,61 @@
1
+ import { describe, expect, test } from "bun:test"
2
+ import type { Message, Part } from "@reign-labs/sdk/v2/client"
3
+ import { estimateSessionContextBreakdown } from "./session-context-breakdown"
4
+
5
+ const user = (id: string) => {
6
+ return {
7
+ id,
8
+ role: "user",
9
+ time: { created: 1 },
10
+ } as unknown as Message
11
+ }
12
+
13
+ const assistant = (id: string) => {
14
+ return {
15
+ id,
16
+ role: "assistant",
17
+ time: { created: 1 },
18
+ } as unknown as Message
19
+ }
20
+
21
+ describe("estimateSessionContextBreakdown", () => {
22
+ test("estimates tokens and keeps remaining tokens as other", () => {
23
+ const messages = [user("u1"), assistant("a1")]
24
+ const parts = {
25
+ u1: [{ type: "text", text: "hello world" }] as unknown as Part[],
26
+ a1: [{ type: "text", text: "assistant response" }] as unknown as Part[],
27
+ }
28
+
29
+ const output = estimateSessionContextBreakdown({
30
+ messages,
31
+ parts,
32
+ input: 20,
33
+ systemPrompt: "system prompt",
34
+ })
35
+
36
+ const map = Object.fromEntries(output.map((segment) => [segment.key, segment.tokens]))
37
+ expect(map.system).toBe(4)
38
+ expect(map.user).toBe(3)
39
+ expect(map.assistant).toBe(5)
40
+ expect(map.other).toBe(8)
41
+ })
42
+
43
+ test("scales segments when estimates exceed input", () => {
44
+ const messages = [user("u1"), assistant("a1")]
45
+ const parts = {
46
+ u1: [{ type: "text", text: "x".repeat(400) }] as unknown as Part[],
47
+ a1: [{ type: "text", text: "y".repeat(400) }] as unknown as Part[],
48
+ }
49
+
50
+ const output = estimateSessionContextBreakdown({
51
+ messages,
52
+ parts,
53
+ input: 10,
54
+ systemPrompt: "z".repeat(200),
55
+ })
56
+
57
+ const total = output.reduce((sum, segment) => sum + segment.tokens, 0)
58
+ expect(total).toBeLessThanOrEqual(10)
59
+ expect(output.every((segment) => segment.width <= 100)).toBeTrue()
60
+ })
61
+ })
@@ -0,0 +1,132 @@
1
+ import type { Message, Part } from "@reign-labs/sdk/v2/client"
2
+
3
+ export type SessionContextBreakdownKey = "system" | "user" | "assistant" | "tool" | "other"
4
+
5
+ export type SessionContextBreakdownSegment = {
6
+ key: SessionContextBreakdownKey
7
+ tokens: number
8
+ width: number
9
+ percent: number
10
+ }
11
+
12
+ const estimateTokens = (chars: number) => Math.ceil(chars / 4)
13
+ const toPercent = (tokens: number, input: number) => (tokens / input) * 100
14
+ const toPercentLabel = (tokens: number, input: number) => Math.round(toPercent(tokens, input) * 10) / 10
15
+
16
+ const charsFromUserPart = (part: Part) => {
17
+ if (part.type === "text") return part.text.length
18
+ if (part.type === "file") return part.source?.text.value.length ?? 0
19
+ if (part.type === "agent") return part.source?.value.length ?? 0
20
+ return 0
21
+ }
22
+
23
+ const charsFromAssistantPart = (part: Part) => {
24
+ if (part.type === "text") return { assistant: part.text.length, tool: 0 }
25
+ if (part.type === "reasoning") return { assistant: part.text.length, tool: 0 }
26
+ if (part.type !== "tool") return { assistant: 0, tool: 0 }
27
+
28
+ const input = Object.keys(part.state.input).length * 16
29
+ if (part.state.status === "pending") return { assistant: 0, tool: input + part.state.raw.length }
30
+ if (part.state.status === "completed") return { assistant: 0, tool: input + part.state.output.length }
31
+ if (part.state.status === "error") return { assistant: 0, tool: input + part.state.error.length }
32
+ return { assistant: 0, tool: input }
33
+ }
34
+
35
+ const build = (
36
+ tokens: { system: number; user: number; assistant: number; tool: number; other: number },
37
+ input: number,
38
+ ) => {
39
+ return [
40
+ {
41
+ key: "system",
42
+ tokens: tokens.system,
43
+ },
44
+ {
45
+ key: "user",
46
+ tokens: tokens.user,
47
+ },
48
+ {
49
+ key: "assistant",
50
+ tokens: tokens.assistant,
51
+ },
52
+ {
53
+ key: "tool",
54
+ tokens: tokens.tool,
55
+ },
56
+ {
57
+ key: "other",
58
+ tokens: tokens.other,
59
+ },
60
+ ]
61
+ .filter((x) => x.tokens > 0)
62
+ .map((x) => ({
63
+ key: x.key,
64
+ tokens: x.tokens,
65
+ width: toPercent(x.tokens, input),
66
+ percent: toPercentLabel(x.tokens, input),
67
+ })) as SessionContextBreakdownSegment[]
68
+ }
69
+
70
+ export function estimateSessionContextBreakdown(args: {
71
+ messages: Message[]
72
+ parts: Record<string, Part[] | undefined>
73
+ input: number
74
+ systemPrompt?: string
75
+ }) {
76
+ if (!args.input) return []
77
+
78
+ const counts = args.messages.reduce(
79
+ (acc, msg) => {
80
+ const parts = args.parts[msg.id] ?? []
81
+ if (msg.role === "user") {
82
+ const user = parts.reduce((sum, part) => sum + charsFromUserPart(part), 0)
83
+ return { ...acc, user: acc.user + user }
84
+ }
85
+
86
+ if (msg.role !== "assistant") return acc
87
+ const assistant = parts.reduce(
88
+ (sum, part) => {
89
+ const next = charsFromAssistantPart(part)
90
+ return {
91
+ assistant: sum.assistant + next.assistant,
92
+ tool: sum.tool + next.tool,
93
+ }
94
+ },
95
+ { assistant: 0, tool: 0 },
96
+ )
97
+ return {
98
+ ...acc,
99
+ assistant: acc.assistant + assistant.assistant,
100
+ tool: acc.tool + assistant.tool,
101
+ }
102
+ },
103
+ {
104
+ system: args.systemPrompt?.length ?? 0,
105
+ user: 0,
106
+ assistant: 0,
107
+ tool: 0,
108
+ },
109
+ )
110
+
111
+ const tokens = {
112
+ system: estimateTokens(counts.system),
113
+ user: estimateTokens(counts.user),
114
+ assistant: estimateTokens(counts.assistant),
115
+ tool: estimateTokens(counts.tool),
116
+ }
117
+ const estimated = tokens.system + tokens.user + tokens.assistant + tokens.tool
118
+
119
+ if (estimated <= args.input) {
120
+ return build({ ...tokens, other: args.input - estimated }, args.input)
121
+ }
122
+
123
+ const scale = args.input / estimated
124
+ const scaled = {
125
+ system: Math.floor(tokens.system * scale),
126
+ user: Math.floor(tokens.user * scale),
127
+ assistant: Math.floor(tokens.assistant * scale),
128
+ tool: Math.floor(tokens.tool * scale),
129
+ }
130
+ const total = scaled.system + scaled.user + scaled.assistant + scaled.tool
131
+ return build({ ...scaled, other: Math.max(0, args.input - total) }, args.input)
132
+ }
@@ -0,0 +1,20 @@
1
+ import { DateTime } from "luxon"
2
+
3
+ export function createSessionContextFormatter(locale: string) {
4
+ return {
5
+ number(value: number | null | undefined) {
6
+ if (value === undefined) return "—"
7
+ if (value === null) return "—"
8
+ return value.toLocaleString(locale)
9
+ },
10
+ percent(value: number | null | undefined) {
11
+ if (value === undefined) return "—"
12
+ if (value === null) return "—"
13
+ return value.toLocaleString(locale) + "%"
14
+ },
15
+ time(value: number | undefined) {
16
+ if (!value) return "—"
17
+ return DateTime.fromMillis(value).setLocale(locale).toLocaleString(DateTime.DATETIME_MED)
18
+ },
19
+ }
20
+ }
@@ -0,0 +1,101 @@
1
+ import { describe, expect, test } from "bun:test"
2
+ import type { Message } from "@reign-labs/sdk/v2/client"
3
+ import { getSessionContextMetrics } from "./session-context-metrics"
4
+
5
+ const assistant = (
6
+ id: string,
7
+ tokens: { input: number; output: number; reasoning: number; read: number; write: number },
8
+ cost: number,
9
+ providerID = "openai",
10
+ modelID = "gpt-4.1",
11
+ ) => {
12
+ return {
13
+ id,
14
+ role: "assistant",
15
+ providerID,
16
+ modelID,
17
+ cost,
18
+ tokens: {
19
+ input: tokens.input,
20
+ output: tokens.output,
21
+ reasoning: tokens.reasoning,
22
+ cache: {
23
+ read: tokens.read,
24
+ write: tokens.write,
25
+ },
26
+ },
27
+ time: { created: 1 },
28
+ } as unknown as Message
29
+ }
30
+
31
+ const user = (id: string) => {
32
+ return {
33
+ id,
34
+ role: "user",
35
+ cost: 0,
36
+ time: { created: 1 },
37
+ } as unknown as Message
38
+ }
39
+
40
+ describe("getSessionContextMetrics", () => {
41
+ test("computes totals and usage from latest assistant with tokens", () => {
42
+ const messages = [
43
+ user("u1"),
44
+ assistant("a1", { input: 0, output: 0, reasoning: 0, read: 0, write: 0 }, 0.5),
45
+ assistant("a2", { input: 300, output: 100, reasoning: 50, read: 25, write: 25 }, 1.25),
46
+ ]
47
+ const providers = [
48
+ {
49
+ id: "openai",
50
+ name: "OpenAI",
51
+ models: {
52
+ "gpt-4.1": {
53
+ name: "GPT-4.1",
54
+ limit: { context: 1000 },
55
+ },
56
+ },
57
+ },
58
+ ]
59
+
60
+ const metrics = getSessionContextMetrics(messages, providers)
61
+
62
+ expect(metrics.totalCost).toBe(1.75)
63
+ expect(metrics.context?.message.id).toBe("a2")
64
+ expect(metrics.context?.total).toBe(500)
65
+ expect(metrics.context?.usage).toBe(50)
66
+ expect(metrics.context?.providerLabel).toBe("OpenAI")
67
+ expect(metrics.context?.modelLabel).toBe("GPT-4.1")
68
+ })
69
+
70
+ test("preserves fallback labels and null usage when model metadata is missing", () => {
71
+ const messages = [assistant("a1", { input: 40, output: 10, reasoning: 0, read: 0, write: 0 }, 0.1, "p-1", "m-1")]
72
+ const providers = [{ id: "p-1", models: {} }]
73
+
74
+ const metrics = getSessionContextMetrics(messages, providers)
75
+
76
+ expect(metrics.context?.providerLabel).toBe("p-1")
77
+ expect(metrics.context?.modelLabel).toBe("m-1")
78
+ expect(metrics.context?.limit).toBeUndefined()
79
+ expect(metrics.context?.usage).toBeNull()
80
+ })
81
+
82
+ test("recomputes when message array is mutated in place", () => {
83
+ const messages = [assistant("a1", { input: 10, output: 10, reasoning: 10, read: 10, write: 10 }, 0.25)]
84
+ const providers = [{ id: "openai", models: {} }]
85
+
86
+ const one = getSessionContextMetrics(messages, providers)
87
+ messages.push(assistant("a2", { input: 100, output: 20, reasoning: 0, read: 0, write: 0 }, 0.75))
88
+ const two = getSessionContextMetrics(messages, providers)
89
+
90
+ expect(one.context?.message.id).toBe("a1")
91
+ expect(two.context?.message.id).toBe("a2")
92
+ expect(two.totalCost).toBe(1)
93
+ })
94
+
95
+ test("returns empty metrics when inputs are undefined", () => {
96
+ const metrics = getSessionContextMetrics(undefined, undefined)
97
+
98
+ expect(metrics.totalCost).toBe(0)
99
+ expect(metrics.context).toBeUndefined()
100
+ })
101
+ })
@@ -0,0 +1,82 @@
1
+ import type { AssistantMessage, Message } from "@reign-labs/sdk/v2/client"
2
+
3
+ type Provider = {
4
+ id: string
5
+ name?: string
6
+ models: Record<string, Model | undefined>
7
+ }
8
+
9
+ type Model = {
10
+ name?: string
11
+ limit: {
12
+ context: number
13
+ }
14
+ }
15
+
16
+ type Context = {
17
+ message: AssistantMessage
18
+ provider?: Provider
19
+ model?: Model
20
+ providerLabel: string
21
+ modelLabel: string
22
+ limit: number | undefined
23
+ input: number
24
+ output: number
25
+ reasoning: number
26
+ cacheRead: number
27
+ cacheWrite: number
28
+ total: number
29
+ usage: number | null
30
+ }
31
+
32
+ type Metrics = {
33
+ totalCost: number
34
+ context: Context | undefined
35
+ }
36
+
37
+ const tokenTotal = (msg: AssistantMessage) => {
38
+ return msg.tokens.input + msg.tokens.output + msg.tokens.reasoning + msg.tokens.cache.read + msg.tokens.cache.write
39
+ }
40
+
41
+ const lastAssistantWithTokens = (messages: Message[]) => {
42
+ for (let i = messages.length - 1; i >= 0; i--) {
43
+ const msg = messages[i]
44
+ if (msg.role !== "assistant") continue
45
+ if (tokenTotal(msg) <= 0) continue
46
+ return msg
47
+ }
48
+ }
49
+
50
+ const build = (messages: Message[] = [], providers: Provider[] = []): Metrics => {
51
+ const totalCost = messages.reduce((sum, msg) => sum + (msg.role === "assistant" ? msg.cost : 0), 0)
52
+ const message = lastAssistantWithTokens(messages)
53
+ if (!message) return { totalCost, context: undefined }
54
+
55
+ const provider = providers.find((item) => item.id === message.providerID)
56
+ const model = provider?.models[message.modelID]
57
+ const limit = model?.limit.context
58
+ const total = tokenTotal(message)
59
+
60
+ return {
61
+ totalCost,
62
+ context: {
63
+ message,
64
+ provider,
65
+ model,
66
+ providerLabel: provider?.name ?? message.providerID,
67
+ modelLabel: model?.name ?? message.modelID,
68
+ limit,
69
+ input: message.tokens.input,
70
+ output: message.tokens.output,
71
+ reasoning: message.tokens.reasoning,
72
+ cacheRead: message.tokens.cache.read,
73
+ cacheWrite: message.tokens.cache.write,
74
+ total,
75
+ usage: limit ? Math.round((total / limit) * 100) : null,
76
+ },
77
+ }
78
+ }
79
+
80
+ export function getSessionContextMetrics(messages: Message[] = [], providers: Provider[] = []) {
81
+ return build(messages, providers)
82
+ }