@shareai-lab/kode 1.0.70 → 1.0.71

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 (253) hide show
  1. package/README.md +202 -76
  2. package/README.zh-CN.md +246 -0
  3. package/cli.js +62 -0
  4. package/package.json +45 -25
  5. package/scripts/postinstall.js +56 -0
  6. package/src/ProjectOnboarding.tsx +180 -0
  7. package/src/Tool.ts +53 -0
  8. package/src/commands/approvedTools.ts +53 -0
  9. package/src/commands/bug.tsx +20 -0
  10. package/src/commands/clear.ts +43 -0
  11. package/src/commands/compact.ts +120 -0
  12. package/src/commands/config.tsx +19 -0
  13. package/src/commands/cost.ts +18 -0
  14. package/src/commands/ctx_viz.ts +209 -0
  15. package/src/commands/doctor.ts +24 -0
  16. package/src/commands/help.tsx +19 -0
  17. package/src/commands/init.ts +37 -0
  18. package/src/commands/listen.ts +42 -0
  19. package/src/commands/login.tsx +51 -0
  20. package/src/commands/logout.tsx +40 -0
  21. package/src/commands/mcp.ts +41 -0
  22. package/src/commands/model.tsx +40 -0
  23. package/src/commands/modelstatus.tsx +20 -0
  24. package/src/commands/onboarding.tsx +34 -0
  25. package/src/commands/pr_comments.ts +59 -0
  26. package/src/commands/refreshCommands.ts +54 -0
  27. package/src/commands/release-notes.ts +34 -0
  28. package/src/commands/resume.tsx +30 -0
  29. package/src/commands/review.ts +49 -0
  30. package/src/commands/terminalSetup.ts +221 -0
  31. package/src/commands.ts +136 -0
  32. package/src/components/ApproveApiKey.tsx +93 -0
  33. package/src/components/AsciiLogo.tsx +13 -0
  34. package/src/components/AutoUpdater.tsx +148 -0
  35. package/src/components/Bug.tsx +367 -0
  36. package/src/components/Config.tsx +289 -0
  37. package/src/components/ConsoleOAuthFlow.tsx +326 -0
  38. package/src/components/Cost.tsx +23 -0
  39. package/src/components/CostThresholdDialog.tsx +46 -0
  40. package/src/components/CustomSelect/option-map.ts +42 -0
  41. package/src/components/CustomSelect/select-option.tsx +52 -0
  42. package/src/components/CustomSelect/select.tsx +143 -0
  43. package/src/components/CustomSelect/use-select-state.ts +414 -0
  44. package/src/components/CustomSelect/use-select.ts +35 -0
  45. package/src/components/FallbackToolUseRejectedMessage.tsx +15 -0
  46. package/src/components/FileEditToolUpdatedMessage.tsx +66 -0
  47. package/src/components/Help.tsx +215 -0
  48. package/src/components/HighlightedCode.tsx +33 -0
  49. package/src/components/InvalidConfigDialog.tsx +113 -0
  50. package/src/components/Link.tsx +32 -0
  51. package/src/components/LogSelector.tsx +86 -0
  52. package/src/components/Logo.tsx +145 -0
  53. package/src/components/MCPServerApprovalDialog.tsx +100 -0
  54. package/src/components/MCPServerDialogCopy.tsx +25 -0
  55. package/src/components/MCPServerMultiselectDialog.tsx +109 -0
  56. package/src/components/Message.tsx +219 -0
  57. package/src/components/MessageResponse.tsx +15 -0
  58. package/src/components/MessageSelector.tsx +211 -0
  59. package/src/components/ModeIndicator.tsx +88 -0
  60. package/src/components/ModelConfig.tsx +301 -0
  61. package/src/components/ModelListManager.tsx +223 -0
  62. package/src/components/ModelSelector.tsx +3208 -0
  63. package/src/components/ModelStatusDisplay.tsx +228 -0
  64. package/src/components/Onboarding.tsx +274 -0
  65. package/src/components/PressEnterToContinue.tsx +11 -0
  66. package/src/components/PromptInput.tsx +710 -0
  67. package/src/components/SentryErrorBoundary.ts +33 -0
  68. package/src/components/Spinner.tsx +129 -0
  69. package/src/components/StructuredDiff.tsx +184 -0
  70. package/src/components/TextInput.tsx +246 -0
  71. package/src/components/TokenWarning.tsx +31 -0
  72. package/src/components/ToolUseLoader.tsx +40 -0
  73. package/src/components/TrustDialog.tsx +106 -0
  74. package/src/components/binary-feedback/BinaryFeedback.tsx +63 -0
  75. package/src/components/binary-feedback/BinaryFeedbackOption.tsx +111 -0
  76. package/src/components/binary-feedback/BinaryFeedbackView.tsx +172 -0
  77. package/src/components/binary-feedback/utils.ts +220 -0
  78. package/src/components/messages/AssistantBashOutputMessage.tsx +22 -0
  79. package/src/components/messages/AssistantLocalCommandOutputMessage.tsx +45 -0
  80. package/src/components/messages/AssistantRedactedThinkingMessage.tsx +19 -0
  81. package/src/components/messages/AssistantTextMessage.tsx +144 -0
  82. package/src/components/messages/AssistantThinkingMessage.tsx +40 -0
  83. package/src/components/messages/AssistantToolUseMessage.tsx +123 -0
  84. package/src/components/messages/UserBashInputMessage.tsx +28 -0
  85. package/src/components/messages/UserCommandMessage.tsx +30 -0
  86. package/src/components/messages/UserKodingInputMessage.tsx +28 -0
  87. package/src/components/messages/UserPromptMessage.tsx +35 -0
  88. package/src/components/messages/UserTextMessage.tsx +39 -0
  89. package/src/components/messages/UserToolResultMessage/UserToolCanceledMessage.tsx +12 -0
  90. package/src/components/messages/UserToolResultMessage/UserToolErrorMessage.tsx +36 -0
  91. package/src/components/messages/UserToolResultMessage/UserToolRejectMessage.tsx +31 -0
  92. package/src/components/messages/UserToolResultMessage/UserToolResultMessage.tsx +57 -0
  93. package/src/components/messages/UserToolResultMessage/UserToolSuccessMessage.tsx +35 -0
  94. package/src/components/messages/UserToolResultMessage/utils.tsx +56 -0
  95. package/src/components/permissions/BashPermissionRequest/BashPermissionRequest.tsx +121 -0
  96. package/src/components/permissions/FallbackPermissionRequest.tsx +155 -0
  97. package/src/components/permissions/FileEditPermissionRequest/FileEditPermissionRequest.tsx +182 -0
  98. package/src/components/permissions/FileEditPermissionRequest/FileEditToolDiff.tsx +75 -0
  99. package/src/components/permissions/FileWritePermissionRequest/FileWritePermissionRequest.tsx +164 -0
  100. package/src/components/permissions/FileWritePermissionRequest/FileWriteToolDiff.tsx +81 -0
  101. package/src/components/permissions/FilesystemPermissionRequest/FilesystemPermissionRequest.tsx +242 -0
  102. package/src/components/permissions/PermissionRequest.tsx +103 -0
  103. package/src/components/permissions/PermissionRequestTitle.tsx +69 -0
  104. package/src/components/permissions/hooks.ts +44 -0
  105. package/src/components/permissions/toolUseOptions.ts +59 -0
  106. package/src/components/permissions/utils.ts +23 -0
  107. package/src/constants/betas.ts +5 -0
  108. package/src/constants/claude-asterisk-ascii-art.tsx +238 -0
  109. package/src/constants/figures.ts +4 -0
  110. package/src/constants/keys.ts +3 -0
  111. package/src/constants/macros.ts +6 -0
  112. package/src/constants/models.ts +935 -0
  113. package/src/constants/oauth.ts +18 -0
  114. package/src/constants/product.ts +17 -0
  115. package/src/constants/prompts.ts +177 -0
  116. package/src/constants/releaseNotes.ts +7 -0
  117. package/src/context/PermissionContext.tsx +149 -0
  118. package/src/context.ts +278 -0
  119. package/src/cost-tracker.ts +84 -0
  120. package/src/entrypoints/cli.tsx +1498 -0
  121. package/src/entrypoints/mcp.ts +176 -0
  122. package/src/history.ts +25 -0
  123. package/src/hooks/useApiKeyVerification.ts +59 -0
  124. package/src/hooks/useArrowKeyHistory.ts +55 -0
  125. package/src/hooks/useCanUseTool.ts +138 -0
  126. package/src/hooks/useCancelRequest.ts +39 -0
  127. package/src/hooks/useDoublePress.ts +42 -0
  128. package/src/hooks/useExitOnCtrlCD.ts +31 -0
  129. package/src/hooks/useInterval.ts +25 -0
  130. package/src/hooks/useLogMessages.ts +16 -0
  131. package/src/hooks/useLogStartupTime.ts +12 -0
  132. package/src/hooks/useNotifyAfterTimeout.ts +65 -0
  133. package/src/hooks/usePermissionRequestLogging.ts +44 -0
  134. package/src/hooks/useSlashCommandTypeahead.ts +137 -0
  135. package/src/hooks/useTerminalSize.ts +49 -0
  136. package/src/hooks/useTextInput.ts +315 -0
  137. package/src/messages.ts +37 -0
  138. package/src/permissions.ts +268 -0
  139. package/src/query.ts +704 -0
  140. package/src/screens/ConfigureNpmPrefix.tsx +197 -0
  141. package/src/screens/Doctor.tsx +219 -0
  142. package/src/screens/LogList.tsx +68 -0
  143. package/src/screens/REPL.tsx +792 -0
  144. package/src/screens/ResumeConversation.tsx +68 -0
  145. package/src/services/browserMocks.ts +66 -0
  146. package/src/services/claude.ts +1947 -0
  147. package/src/services/customCommands.ts +683 -0
  148. package/src/services/fileFreshness.ts +377 -0
  149. package/src/services/mcpClient.ts +564 -0
  150. package/src/services/mcpServerApproval.tsx +50 -0
  151. package/src/services/notifier.ts +40 -0
  152. package/src/services/oauth.ts +357 -0
  153. package/src/services/openai.ts +796 -0
  154. package/src/services/sentry.ts +3 -0
  155. package/src/services/statsig.ts +171 -0
  156. package/src/services/statsigStorage.ts +86 -0
  157. package/src/services/systemReminder.ts +406 -0
  158. package/src/services/vcr.ts +161 -0
  159. package/src/tools/ArchitectTool/ArchitectTool.tsx +122 -0
  160. package/src/tools/ArchitectTool/prompt.ts +15 -0
  161. package/src/tools/AskExpertModelTool/AskExpertModelTool.tsx +505 -0
  162. package/src/tools/BashTool/BashTool.tsx +270 -0
  163. package/src/tools/BashTool/BashToolResultMessage.tsx +38 -0
  164. package/src/tools/BashTool/OutputLine.tsx +48 -0
  165. package/src/tools/BashTool/prompt.ts +174 -0
  166. package/src/tools/BashTool/utils.ts +56 -0
  167. package/src/tools/FileEditTool/FileEditTool.tsx +316 -0
  168. package/src/tools/FileEditTool/prompt.ts +51 -0
  169. package/src/tools/FileEditTool/utils.ts +58 -0
  170. package/src/tools/FileReadTool/FileReadTool.tsx +371 -0
  171. package/src/tools/FileReadTool/prompt.ts +7 -0
  172. package/src/tools/FileWriteTool/FileWriteTool.tsx +297 -0
  173. package/src/tools/FileWriteTool/prompt.ts +10 -0
  174. package/src/tools/GlobTool/GlobTool.tsx +119 -0
  175. package/src/tools/GlobTool/prompt.ts +8 -0
  176. package/src/tools/GrepTool/GrepTool.tsx +147 -0
  177. package/src/tools/GrepTool/prompt.ts +11 -0
  178. package/src/tools/MCPTool/MCPTool.tsx +106 -0
  179. package/src/tools/MCPTool/prompt.ts +3 -0
  180. package/src/tools/MemoryReadTool/MemoryReadTool.tsx +127 -0
  181. package/src/tools/MemoryReadTool/prompt.ts +3 -0
  182. package/src/tools/MemoryWriteTool/MemoryWriteTool.tsx +89 -0
  183. package/src/tools/MemoryWriteTool/prompt.ts +3 -0
  184. package/src/tools/MultiEditTool/MultiEditTool.tsx +366 -0
  185. package/src/tools/MultiEditTool/prompt.ts +45 -0
  186. package/src/tools/NotebookEditTool/NotebookEditTool.tsx +298 -0
  187. package/src/tools/NotebookEditTool/prompt.ts +3 -0
  188. package/src/tools/NotebookReadTool/NotebookReadTool.tsx +266 -0
  189. package/src/tools/NotebookReadTool/prompt.ts +3 -0
  190. package/src/tools/StickerRequestTool/StickerRequestTool.tsx +93 -0
  191. package/src/tools/StickerRequestTool/prompt.ts +19 -0
  192. package/src/tools/TaskTool/TaskTool.tsx +382 -0
  193. package/src/tools/TaskTool/constants.ts +1 -0
  194. package/src/tools/TaskTool/prompt.ts +56 -0
  195. package/src/tools/ThinkTool/ThinkTool.tsx +56 -0
  196. package/src/tools/ThinkTool/prompt.ts +12 -0
  197. package/src/tools/TodoWriteTool/TodoWriteTool.tsx +289 -0
  198. package/src/tools/TodoWriteTool/prompt.ts +63 -0
  199. package/src/tools/lsTool/lsTool.tsx +269 -0
  200. package/src/tools/lsTool/prompt.ts +2 -0
  201. package/src/tools.ts +63 -0
  202. package/src/types/PermissionMode.ts +120 -0
  203. package/src/types/RequestContext.ts +72 -0
  204. package/src/utils/Cursor.ts +436 -0
  205. package/src/utils/PersistentShell.ts +373 -0
  206. package/src/utils/agentStorage.ts +97 -0
  207. package/src/utils/array.ts +3 -0
  208. package/src/utils/ask.tsx +98 -0
  209. package/src/utils/auth.ts +13 -0
  210. package/src/utils/autoCompactCore.ts +223 -0
  211. package/src/utils/autoUpdater.ts +318 -0
  212. package/src/utils/betas.ts +20 -0
  213. package/src/utils/browser.ts +14 -0
  214. package/src/utils/cleanup.ts +72 -0
  215. package/src/utils/commands.ts +261 -0
  216. package/src/utils/config.ts +771 -0
  217. package/src/utils/conversationRecovery.ts +54 -0
  218. package/src/utils/debugLogger.ts +1123 -0
  219. package/src/utils/diff.ts +42 -0
  220. package/src/utils/env.ts +57 -0
  221. package/src/utils/errors.ts +21 -0
  222. package/src/utils/exampleCommands.ts +108 -0
  223. package/src/utils/execFileNoThrow.ts +51 -0
  224. package/src/utils/expertChatStorage.ts +136 -0
  225. package/src/utils/file.ts +402 -0
  226. package/src/utils/fileRecoveryCore.ts +71 -0
  227. package/src/utils/format.tsx +44 -0
  228. package/src/utils/generators.ts +62 -0
  229. package/src/utils/git.ts +92 -0
  230. package/src/utils/globalLogger.ts +77 -0
  231. package/src/utils/http.ts +10 -0
  232. package/src/utils/imagePaste.ts +38 -0
  233. package/src/utils/json.ts +13 -0
  234. package/src/utils/log.ts +382 -0
  235. package/src/utils/markdown.ts +213 -0
  236. package/src/utils/messageContextManager.ts +289 -0
  237. package/src/utils/messages.tsx +938 -0
  238. package/src/utils/model.ts +836 -0
  239. package/src/utils/permissions/filesystem.ts +118 -0
  240. package/src/utils/ripgrep.ts +167 -0
  241. package/src/utils/sessionState.ts +49 -0
  242. package/src/utils/state.ts +25 -0
  243. package/src/utils/style.ts +29 -0
  244. package/src/utils/terminal.ts +49 -0
  245. package/src/utils/theme.ts +122 -0
  246. package/src/utils/thinking.ts +144 -0
  247. package/src/utils/todoStorage.ts +431 -0
  248. package/src/utils/tokens.ts +43 -0
  249. package/src/utils/toolExecutionController.ts +163 -0
  250. package/src/utils/unaryLogging.ts +26 -0
  251. package/src/utils/user.ts +37 -0
  252. package/src/utils/validate.ts +165 -0
  253. package/cli.mjs +0 -1803
@@ -0,0 +1,120 @@
1
+ // Permission mode types based on original Claude Code implementation
2
+ export type PermissionMode =
3
+ | 'default'
4
+ | 'acceptEdits'
5
+ | 'plan'
6
+ | 'bypassPermissions'
7
+
8
+ export interface PermissionContext {
9
+ mode: PermissionMode
10
+ allowedTools: string[]
11
+ allowedPaths: string[]
12
+ restrictions: {
13
+ readOnly: boolean
14
+ requireConfirmation: boolean
15
+ bypassValidation: boolean
16
+ }
17
+ metadata: {
18
+ activatedAt?: string
19
+ previousMode?: PermissionMode
20
+ transitionCount: number
21
+ }
22
+ }
23
+
24
+ export interface ModeConfig {
25
+ name: PermissionMode
26
+ label: string
27
+ icon: string
28
+ color: string
29
+ description: string
30
+ allowedTools: string[]
31
+ restrictions: {
32
+ readOnly: boolean
33
+ requireConfirmation: boolean
34
+ bypassValidation: boolean
35
+ }
36
+ }
37
+
38
+ // Mode configuration based on original Claude Code
39
+ export const MODE_CONFIGS: Record<PermissionMode, ModeConfig> = {
40
+ default: {
41
+ name: 'default',
42
+ label: 'DEFAULT',
43
+ icon: '🔒',
44
+ color: 'blue',
45
+ description: 'Standard permission checking',
46
+ allowedTools: ['*'],
47
+ restrictions: {
48
+ readOnly: false,
49
+ requireConfirmation: true,
50
+ bypassValidation: false,
51
+ },
52
+ },
53
+ acceptEdits: {
54
+ name: 'acceptEdits',
55
+ label: 'ACCEPT EDITS',
56
+ icon: '✅',
57
+ color: 'green',
58
+ description: 'Auto-approve edit operations',
59
+ allowedTools: ['*'],
60
+ restrictions: {
61
+ readOnly: false,
62
+ requireConfirmation: false,
63
+ bypassValidation: false,
64
+ },
65
+ },
66
+ plan: {
67
+ name: 'plan',
68
+ label: 'PLAN MODE',
69
+ icon: '📝',
70
+ color: 'yellow',
71
+ description: 'Research and planning - read-only tools only',
72
+ allowedTools: [
73
+ 'Read',
74
+ 'Grep',
75
+ 'Glob',
76
+ 'LS',
77
+ 'WebSearch',
78
+ 'WebFetch',
79
+ 'NotebookRead',
80
+ 'exit_plan_mode',
81
+ ],
82
+ restrictions: {
83
+ readOnly: true,
84
+ requireConfirmation: true,
85
+ bypassValidation: false,
86
+ },
87
+ },
88
+ bypassPermissions: {
89
+ name: 'bypassPermissions',
90
+ label: 'BYPASS PERMISSIONS',
91
+ icon: '🔓',
92
+ color: 'red',
93
+ description: 'All permissions bypassed',
94
+ allowedTools: ['*'],
95
+ restrictions: {
96
+ readOnly: false,
97
+ requireConfirmation: false,
98
+ bypassValidation: true,
99
+ },
100
+ },
101
+ }
102
+
103
+ // Mode cycling function (based on original yg2 function)
104
+ export function getNextPermissionMode(
105
+ currentMode: PermissionMode,
106
+ isBypassAvailable: boolean = true,
107
+ ): PermissionMode {
108
+ switch (currentMode) {
109
+ case 'default':
110
+ return 'acceptEdits'
111
+ case 'acceptEdits':
112
+ return 'plan'
113
+ case 'plan':
114
+ return isBypassAvailable ? 'bypassPermissions' : 'default'
115
+ case 'bypassPermissions':
116
+ return 'default'
117
+ default:
118
+ return 'default'
119
+ }
120
+ }
@@ -0,0 +1,72 @@
1
+ // Request Context for perfect state isolation
2
+ // Based on official Kode patterns
3
+
4
+ export interface RequestContext {
5
+ id: string
6
+ abortController: AbortController
7
+ startTime: number
8
+ isActive: boolean
9
+ type: 'query' | 'tool' | 'koding'
10
+ }
11
+
12
+ export interface AbortBarrier {
13
+ requestId: string
14
+ checkAbort(): boolean
15
+ onAbort(callback: () => void): void
16
+ cleanup(): void
17
+ }
18
+
19
+ export function createRequestContext(
20
+ type: RequestContext['type'] = 'query',
21
+ ): RequestContext {
22
+ return {
23
+ id: crypto.randomUUID(),
24
+ abortController: new AbortController(),
25
+ startTime: Date.now(),
26
+ isActive: true,
27
+ type,
28
+ }
29
+ }
30
+
31
+ export function createAbortBarrier(
32
+ requestContext: RequestContext,
33
+ ): AbortBarrier {
34
+ let cleanupCallbacks: (() => void)[] = []
35
+
36
+ return {
37
+ requestId: requestContext.id,
38
+
39
+ checkAbort(): boolean {
40
+ // Only respond to aborts for THIS specific request
41
+ return (
42
+ requestContext.isActive && requestContext.abortController.signal.aborted
43
+ )
44
+ },
45
+
46
+ onAbort(callback: () => void): void {
47
+ if (requestContext.isActive) {
48
+ const abortHandler = () => {
49
+ if (requestContext.isActive) {
50
+ callback()
51
+ }
52
+ }
53
+ requestContext.abortController.signal.addEventListener(
54
+ 'abort',
55
+ abortHandler,
56
+ )
57
+ cleanupCallbacks.push(() => {
58
+ requestContext.abortController.signal.removeEventListener(
59
+ 'abort',
60
+ abortHandler,
61
+ )
62
+ })
63
+ }
64
+ },
65
+
66
+ cleanup(): void {
67
+ cleanupCallbacks.forEach(cleanup => cleanup())
68
+ cleanupCallbacks = []
69
+ requestContext.isActive = false
70
+ },
71
+ }
72
+ }
@@ -0,0 +1,436 @@
1
+ import wrapAnsi from 'wrap-ansi'
2
+
3
+ type WrappedText = string[]
4
+ type Position = {
5
+ line: number
6
+ column: number
7
+ }
8
+
9
+ export class Cursor {
10
+ readonly offset: number
11
+ constructor(
12
+ readonly measuredText: MeasuredText,
13
+ offset: number = 0,
14
+ readonly selection: number = 0,
15
+ ) {
16
+ // it's ok for the cursor to be 1 char beyond the end of the string
17
+ this.offset = Math.max(0, Math.min(this.measuredText.text.length, offset))
18
+ }
19
+
20
+ static fromText(
21
+ text: string,
22
+ columns: number,
23
+ offset: number = 0,
24
+ selection: number = 0,
25
+ ): Cursor {
26
+ // make MeasuredText on less than columns width, to account for cursor
27
+ return new Cursor(new MeasuredText(text, columns - 1), offset, selection)
28
+ }
29
+
30
+ render(cursorChar: string, mask: string, invert: (text: string) => string) {
31
+ const { line, column } = this.getPosition()
32
+ return this.measuredText
33
+ .getWrappedText()
34
+ .map((text, currentLine, allLines) => {
35
+ let displayText = text
36
+ if (mask && currentLine === allLines.length - 1) {
37
+ const lastSixStart = Math.max(0, text.length - 6)
38
+ displayText = mask.repeat(lastSixStart) + text.slice(lastSixStart)
39
+ }
40
+ // looking for the line with the cursor
41
+ if (line != currentLine) return displayText.trimEnd()
42
+
43
+ return (
44
+ displayText.slice(0, column) +
45
+ invert(displayText[column] || cursorChar) +
46
+ displayText.trimEnd().slice(column + 1)
47
+ )
48
+ })
49
+ .join('\n')
50
+ }
51
+
52
+ left(): Cursor {
53
+ return new Cursor(this.measuredText, this.offset - 1)
54
+ }
55
+
56
+ right(): Cursor {
57
+ return new Cursor(this.measuredText, this.offset + 1)
58
+ }
59
+
60
+ up(): Cursor {
61
+ const { line, column } = this.getPosition()
62
+ if (line == 0) {
63
+ return new Cursor(this.measuredText, 0, 0)
64
+ }
65
+
66
+ const newOffset = this.getOffset({ line: line - 1, column })
67
+ return new Cursor(this.measuredText, newOffset, 0)
68
+ }
69
+
70
+ down(): Cursor {
71
+ const { line, column } = this.getPosition()
72
+ if (line >= this.measuredText.lineCount - 1) {
73
+ return new Cursor(this.measuredText, this.text.length, 0)
74
+ }
75
+
76
+ const newOffset = this.getOffset({ line: line + 1, column })
77
+ return new Cursor(this.measuredText, newOffset, 0)
78
+ }
79
+
80
+ startOfLine(): Cursor {
81
+ const { line } = this.getPosition()
82
+ return new Cursor(
83
+ this.measuredText,
84
+ this.getOffset({
85
+ line,
86
+ column: 0,
87
+ }),
88
+ 0,
89
+ )
90
+ }
91
+
92
+ endOfLine(): Cursor {
93
+ const { line } = this.getPosition()
94
+ const column = this.measuredText.getLineLength(line)
95
+ const offset = this.getOffset({ line, column })
96
+ return new Cursor(this.measuredText, offset, 0)
97
+ }
98
+
99
+ nextWord(): Cursor {
100
+ // eslint-disable-next-line @typescript-eslint/no-this-alias
101
+ let nextCursor: Cursor = this
102
+ // If we're on a word, move to the next non-word
103
+ while (nextCursor.isOverWordChar() && !nextCursor.isAtEnd()) {
104
+ nextCursor = nextCursor.right()
105
+ }
106
+ // now move to the next word char
107
+ while (!nextCursor.isOverWordChar() && !nextCursor.isAtEnd()) {
108
+ nextCursor = nextCursor.right()
109
+ }
110
+ return nextCursor
111
+ }
112
+
113
+ prevWord(): Cursor {
114
+ // eslint-disable-next-line @typescript-eslint/no-this-alias
115
+ let cursor: Cursor = this
116
+
117
+ // if we are already at the beginning of a word, step off it
118
+ if (!cursor.left().isOverWordChar()) {
119
+ cursor = cursor.left()
120
+ }
121
+
122
+ // Move left over any non-word characters
123
+ while (!cursor.isOverWordChar() && !cursor.isAtStart()) {
124
+ cursor = cursor.left()
125
+ }
126
+
127
+ // If we're over a word character, move to the start of this word
128
+ if (cursor.isOverWordChar()) {
129
+ while (cursor.left().isOverWordChar() && !cursor.isAtStart()) {
130
+ cursor = cursor.left()
131
+ }
132
+ }
133
+
134
+ return cursor
135
+ }
136
+
137
+ private modifyText(end: Cursor, insertString: string = ''): Cursor {
138
+ const startOffset = this.offset
139
+ const endOffset = end.offset
140
+
141
+ const newText =
142
+ this.text.slice(0, startOffset) +
143
+ insertString +
144
+ this.text.slice(endOffset)
145
+
146
+ return Cursor.fromText(
147
+ newText,
148
+ this.columns,
149
+ startOffset + insertString.length,
150
+ )
151
+ }
152
+
153
+ insert(insertString: string): Cursor {
154
+ const newCursor = this.modifyText(this, insertString)
155
+ return newCursor
156
+ }
157
+
158
+ del(): Cursor {
159
+ if (this.isAtEnd()) {
160
+ return this
161
+ }
162
+ return this.modifyText(this.right())
163
+ }
164
+
165
+ backspace(): Cursor {
166
+ if (this.isAtStart()) {
167
+ return this
168
+ }
169
+
170
+ // Get the current position
171
+ const currentOffset = this.offset
172
+
173
+ // Create a new cursor at the position before the current one
174
+ const leftCursor = this.left()
175
+ const leftOffset = leftCursor.offset
176
+
177
+ // Create the new text by removing one character
178
+ const newText =
179
+ this.text.slice(0, leftOffset) + this.text.slice(currentOffset)
180
+
181
+ // Return a new cursor with the updated text and position
182
+ return Cursor.fromText(newText, this.columns, leftOffset)
183
+ }
184
+
185
+ deleteToLineStart(): Cursor {
186
+ return this.startOfLine().modifyText(this)
187
+ }
188
+
189
+ deleteToLineEnd(): Cursor {
190
+ // If cursor is on a newline character, delete just that character
191
+ if (this.text[this.offset] === '\n') {
192
+ return this.modifyText(this.right())
193
+ }
194
+
195
+ return this.modifyText(this.endOfLine())
196
+ }
197
+
198
+ deleteWordBefore(): Cursor {
199
+ if (this.isAtStart()) {
200
+ return this
201
+ }
202
+ return this.prevWord().modifyText(this)
203
+ }
204
+
205
+ deleteWordAfter(): Cursor {
206
+ if (this.isAtEnd()) {
207
+ return this
208
+ }
209
+
210
+ return this.modifyText(this.nextWord())
211
+ }
212
+
213
+ private isOverWordChar(): boolean {
214
+ const currentChar = this.text[this.offset] ?? ''
215
+ return /\w/.test(currentChar)
216
+ }
217
+
218
+ equals(other: Cursor): boolean {
219
+ return (
220
+ this.offset === other.offset && this.measuredText == other.measuredText
221
+ )
222
+ }
223
+
224
+ private isAtStart(): boolean {
225
+ return this.offset == 0
226
+ }
227
+ private isAtEnd(): boolean {
228
+ return this.offset == this.text.length
229
+ }
230
+
231
+ public get text(): string {
232
+ return this.measuredText.text
233
+ }
234
+
235
+ private get columns(): number {
236
+ return this.measuredText.columns + 1
237
+ }
238
+
239
+ private getPosition(): Position {
240
+ return this.measuredText.getPositionFromOffset(this.offset)
241
+ }
242
+
243
+ private getOffset(position: Position): number {
244
+ return this.measuredText.getOffsetFromPosition(position)
245
+ }
246
+ }
247
+
248
+ class WrappedLine {
249
+ constructor(
250
+ public readonly text: string,
251
+ public readonly startOffset: number,
252
+ public readonly isPrecededByNewline: boolean,
253
+ public readonly endsWithNewline: boolean = false,
254
+ ) {}
255
+
256
+ equals(other: WrappedLine): boolean {
257
+ return this.text === other.text && this.startOffset === other.startOffset
258
+ }
259
+
260
+ get length(): number {
261
+ return this.text.length + (this.endsWithNewline ? 1 : 0)
262
+ }
263
+ }
264
+
265
+ export class MeasuredText {
266
+ private wrappedLines: WrappedLine[]
267
+
268
+ constructor(
269
+ readonly text: string,
270
+ readonly columns: number,
271
+ ) {
272
+ this.wrappedLines = this.measureWrappedText()
273
+ }
274
+
275
+ private measureWrappedText(): WrappedLine[] {
276
+ const wrappedText = wrapAnsi(this.text, this.columns, {
277
+ hard: true,
278
+ trim: false,
279
+ })
280
+
281
+ const wrappedLines: WrappedLine[] = []
282
+ let searchOffset = 0
283
+ let lastNewLinePos = -1
284
+
285
+ const lines = wrappedText.split('\n')
286
+ for (let i = 0; i < lines.length; i++) {
287
+ const text = lines[i]!
288
+ const isPrecededByNewline = (startOffset: number) =>
289
+ i == 0 || (startOffset > 0 && this.text[startOffset - 1] === '\n')
290
+
291
+ if (text.length === 0) {
292
+ // For blank lines, find the next newline character after the last one
293
+ lastNewLinePos = this.text.indexOf('\n', lastNewLinePos + 1)
294
+
295
+ if (lastNewLinePos !== -1) {
296
+ const startOffset = lastNewLinePos
297
+ const endsWithNewline = true
298
+
299
+ wrappedLines.push(
300
+ new WrappedLine(
301
+ text,
302
+ startOffset,
303
+ isPrecededByNewline(startOffset),
304
+ endsWithNewline,
305
+ ),
306
+ )
307
+ } else {
308
+ // If we can't find another newline, this must be the end of text
309
+ const startOffset = this.text.length
310
+ wrappedLines.push(
311
+ new WrappedLine(
312
+ text,
313
+ startOffset,
314
+ isPrecededByNewline(startOffset),
315
+ false,
316
+ ),
317
+ )
318
+ }
319
+ } else {
320
+ // For non-blank lines
321
+ const startOffset = this.text.indexOf(text, searchOffset)
322
+ if (startOffset === -1) {
323
+ console.log('Debug: Failed to find wrapped line in original text')
324
+ console.log('Debug: Current text:', text)
325
+ console.log('Debug: Full original text:', this.text)
326
+ console.log('Debug: Search offset:', searchOffset)
327
+ console.log('Debug: Wrapped text:', wrappedText)
328
+ throw new Error('Failed to find wrapped line in original text')
329
+ }
330
+
331
+ searchOffset = startOffset + text.length
332
+
333
+ // Check if this line ends with a newline in the original text
334
+ const potentialNewlinePos = startOffset + text.length
335
+ const endsWithNewline =
336
+ potentialNewlinePos < this.text.length &&
337
+ this.text[potentialNewlinePos] === '\n'
338
+
339
+ if (endsWithNewline) {
340
+ lastNewLinePos = potentialNewlinePos
341
+ }
342
+
343
+ wrappedLines.push(
344
+ new WrappedLine(
345
+ text,
346
+ startOffset,
347
+ isPrecededByNewline(startOffset),
348
+ endsWithNewline,
349
+ ),
350
+ )
351
+ }
352
+ }
353
+
354
+ return wrappedLines
355
+ }
356
+
357
+ public getWrappedText(): WrappedText {
358
+ return this.wrappedLines.map(line =>
359
+ line.isPrecededByNewline ? line.text : line.text.trimStart(),
360
+ )
361
+ }
362
+
363
+ private getLine(line: number): WrappedLine {
364
+ return this.wrappedLines[
365
+ Math.max(0, Math.min(line, this.wrappedLines.length - 1))
366
+ ]!
367
+ }
368
+
369
+ public getOffsetFromPosition(position: Position): number {
370
+ const wrappedLine = this.getLine(position.line)
371
+ const startOffsetPlusColumn = wrappedLine.startOffset + position.column
372
+
373
+ // Handle blank lines specially
374
+ if (wrappedLine.text.length === 0 && wrappedLine.endsWithNewline) {
375
+ return wrappedLine.startOffset
376
+ }
377
+
378
+ // For normal lines
379
+ const lineEnd = wrappedLine.startOffset + wrappedLine.text.length
380
+ // Add 1 only if this line ends with a newline
381
+ const maxOffset = wrappedLine.endsWithNewline ? lineEnd + 1 : lineEnd
382
+
383
+ return Math.min(startOffsetPlusColumn, maxOffset)
384
+ }
385
+
386
+ public getLineLength(line: number): number {
387
+ const currentLine = this.getLine(line)
388
+ const nextLine = this.getLine(line + 1)
389
+ if (nextLine.equals(currentLine)) {
390
+ return this.text.length - currentLine.startOffset
391
+ }
392
+
393
+ return nextLine.startOffset - currentLine.startOffset - 1
394
+ }
395
+
396
+ public getPositionFromOffset(offset: number): Position {
397
+ const lines = this.wrappedLines
398
+ for (let line = 0; line < lines.length; line++) {
399
+ const currentLine = lines[line]!
400
+ const nextLine = lines[line + 1]
401
+ if (
402
+ offset >= currentLine.startOffset &&
403
+ (!nextLine || offset < nextLine.startOffset)
404
+ ) {
405
+ const leadingWhitepace = currentLine.isPrecededByNewline
406
+ ? 0
407
+ : currentLine.text.length - currentLine.text.trimStart().length
408
+ const column = Math.max(
409
+ 0,
410
+ Math.min(
411
+ offset - currentLine.startOffset - leadingWhitepace,
412
+ currentLine.text.length,
413
+ ),
414
+ )
415
+ return {
416
+ line,
417
+ column,
418
+ }
419
+ }
420
+ }
421
+
422
+ // If we're past the last character, return the end of the last line
423
+ const line = lines.length - 1
424
+ return {
425
+ line,
426
+ column: this.wrappedLines[line]!.text.length,
427
+ }
428
+ }
429
+
430
+ public get lineCount(): number {
431
+ return this.wrappedLines.length
432
+ }
433
+ equals(other: MeasuredText): boolean {
434
+ return this.text === other.text && this.columns === other.columns
435
+ }
436
+ }