From dfa0d396c130428a63036253ec080c9ccd2fd4a0 Mon Sep 17 00:00:00 2001
From: apple <245524539+apples-kksk@users.noreply.github.com>
Date: Sun, 10 May 2026 15:17:56 +0800
Subject: [PATCH] Add custom Vim key mappings

---
 .../settings/UserPreferencesForm.tsx          | 108 ++++++++++++++++++
 src/cloud/lib/editor/CodeMirror.ts            |  45 ++++++++
 .../editor/components/CodeMirrorEditor.tsx    |   8 +-
 src/cloud/lib/i18n/enUS.ts                    |   5 +
 src/cloud/lib/i18n/fr.ts                      |   5 +
 src/cloud/lib/i18n/ja.ts                      |   5 +
 src/cloud/lib/i18n/types.ts                   |   3 +
 src/cloud/lib/i18n/zhCN.ts                    |   5 +
 src/cloud/lib/stores/settings/store.ts        |   1 +
 src/cloud/lib/stores/settings/types.ts        |   7 ++
 10 files changed, 191 insertions(+), 1 deletion(-)

diff --git a/src/cloud/components/settings/UserPreferencesForm.tsx b/src/cloud/components/settings/UserPreferencesForm.tsx
index b6cb5cf50f..1b88929243 100644
--- a/src/cloud/components/settings/UserPreferencesForm.tsx
+++ b/src/cloud/components/settings/UserPreferencesForm.tsx
@@ -6,6 +6,7 @@ import {
   CodeMirrorEditorTheme,
   codeMirrorKeyMap,
   CodeMirrorKeyMap,
+  CodeMirrorVimMapEntry,
   GeneralEditorIndentType,
   GeneralEditorIndentSize,
   GeneralLanguageOptions,
@@ -32,6 +33,10 @@ const UserPreferencesForm = () => {
   const [fontFamily, setFontFamily] = useState(
     settings['general.editorFontFamily']
   )
+  const [vimKeyMaps, setVimKeyMaps] = useState(() => {
+    return formatVimKeyMaps(settings['general.editorVimKeyMaps'])
+  })
+  const [vimKeyMapsError, setVimKeyMapsError] = useState<string | null>(null)
 
   const resetSettings = useCallback(() => {
     setSettings({})
@@ -162,6 +167,27 @@ const UserPreferencesForm = () => {
     [fontFamily, setSettings]
   )
 
+  const updateVimKeyMaps: ChangeEventHandler<HTMLTextAreaElement> = useCallback(
+    (event) => {
+      setVimKeyMaps(event.target.value)
+    },
+    [setVimKeyMaps]
+  )
+  useDebounce(
+    () => {
+      try {
+        setSettings({
+          'general.editorVimKeyMaps': parseVimKeyMaps(vimKeyMaps),
+        })
+        setVimKeyMapsError(null)
+      } catch (error) {
+        setVimKeyMapsError(t(lngKeys.SettingsEditorVimKeyMapsInvalid))
+      }
+    },
+    500,
+    [vimKeyMaps, setSettings, t]
+  )
+
   return (
     <Form
       fullWidth={true}
@@ -368,6 +394,45 @@ const UserPreferencesForm = () => {
             },
           ],
         },
+        {
+          title: t(lngKeys.SettingsEditorVimKeyMaps),
+          items: [
+            {
+              type: 'node',
+              element: (
+                <div>
+                  <textarea
+                    style={{
+                      width: '100%',
+                      minHeight: 110,
+                      resize: 'vertical',
+                      fontFamily: 'monospace',
+                    }}
+                    value={vimKeyMaps}
+                    placeholder={
+                      '[{"toKeys":"jk","keys":"<Esc>","context":"insert"}]'
+                    }
+                    onChange={updateVimKeyMaps}
+                  />
+                  <p style={{ margin: '4px 0 0', fontSize: 12 }}>
+                    {t(lngKeys.SettingsEditorVimKeyMapsDescription)}
+                  </p>
+                  {vimKeyMapsError != null && (
+                    <p
+                      style={{
+                        margin: '4px 0 0',
+                        color: '#f66',
+                        fontSize: 12,
+                      }}
+                    >
+                      {vimKeyMapsError}
+                    </p>
+                  )}
+                </div>
+              ),
+            },
+          ],
+        },
         {
           title: t(lngKeys.SettingsIndentType),
           items: [
@@ -444,3 +509,46 @@ const UserPreferencesForm = () => {
 }
 
 export default UserPreferencesForm
+
+function formatVimKeyMaps(vimKeyMaps: CodeMirrorVimMapEntry[]) {
+  return JSON.stringify(Array.isArray(vimKeyMaps) ? vimKeyMaps : [], null, 2)
+}
+
+function parseVimKeyMaps(value: string): CodeMirrorVimMapEntry[] {
+  const trimmedValue = value.trim()
+
+  if (trimmedValue === '') {
+    return []
+  }
+
+  const parsedValue = JSON.parse(trimmedValue)
+
+  if (!Array.isArray(parsedValue)) {
+    throw new Error('Vim key mappings must be an array')
+  }
+
+  return parsedValue.map((entry) => {
+    if (!isVimKeyMapEntry(entry)) {
+      throw new Error('Invalid Vim key mapping entry')
+    }
+
+    return {
+      toKeys: entry.toKeys,
+      keys: entry.keys,
+      context: entry.context,
+    }
+  })
+}
+
+function isVimKeyMapEntry(entry: unknown): entry is CodeMirrorVimMapEntry {
+  if (typeof entry !== 'object' || entry == null) {
+    return false
+  }
+
+  const vimKeyMapEntry = entry as CodeMirrorVimMapEntry
+  return (
+    typeof vimKeyMapEntry.toKeys === 'string' &&
+    typeof vimKeyMapEntry.keys === 'string' &&
+    typeof vimKeyMapEntry.context === 'string'
+  )
+}
diff --git a/src/cloud/lib/editor/CodeMirror.ts b/src/cloud/lib/editor/CodeMirror.ts
index 530b3ffbb5..f8c8f570df 100644
--- a/src/cloud/lib/editor/CodeMirror.ts
+++ b/src/cloud/lib/editor/CodeMirror.ts
@@ -13,11 +13,56 @@ import 'codemirror/keymap/sublime'
 import 'codemirror/keymap/emacs'
 import 'codemirror/addon/scroll/scrollpastend'
 import { loadMode } from '../../../design/lib/codemirror/util'
+import { CodeMirrorVimMapEntry } from '../stores/settings/types'
 
 loadMode(CodeMirror)
 
 export default CodeMirror
 
+interface CodeMirrorVimApi {
+  map: (toKeys: string, keys: string, context?: string) => void
+  unmap: (toKeys: string, context?: string) => void
+}
+
+let lastSerializedVimKeyMaps = ''
+let lastAppliedVimKeyMaps: CodeMirrorVimMapEntry[] = []
+
+export function syncCodeMirrorVimKeyMaps(
+  vimKeyMaps: CodeMirrorVimMapEntry[] = []
+) {
+  const normalizedVimKeyMaps = Array.isArray(vimKeyMaps)
+    ? vimKeyMaps.filter((entry) => {
+        return entry.toKeys.trim() !== '' && entry.keys.trim() !== ''
+      })
+    : []
+  const serializedVimKeyMaps = JSON.stringify(normalizedVimKeyMaps)
+
+  if (serializedVimKeyMaps === lastSerializedVimKeyMaps) {
+    return
+  }
+
+  const vimApi = (
+    CodeMirror as typeof CodeMirror & {
+      Vim?: CodeMirrorVimApi
+    }
+  ).Vim
+
+  if (vimApi == null) {
+    return
+  }
+
+  lastAppliedVimKeyMaps.forEach((entry) => {
+    vimApi.unmap(entry.toKeys, entry.context)
+  })
+
+  normalizedVimKeyMaps.forEach((entry) => {
+    vimApi.map(entry.toKeys, entry.keys, entry.context)
+  })
+
+  lastAppliedVimKeyMaps = normalizedVimKeyMaps
+  lastSerializedVimKeyMaps = serializedVimKeyMaps
+}
+
 export interface EditorPosition {
   line: number
   ch: number
diff --git a/src/cloud/lib/editor/components/CodeMirrorEditor.tsx b/src/cloud/lib/editor/components/CodeMirrorEditor.tsx
index ab594f5397..f62d104d0a 100644
--- a/src/cloud/lib/editor/components/CodeMirrorEditor.tsx
+++ b/src/cloud/lib/editor/components/CodeMirrorEditor.tsx
@@ -1,11 +1,12 @@
 import React, { useEffect, useRef } from 'react'
-import CodeMirror from '../../editor/CodeMirror'
+import CodeMirror, { syncCodeMirrorVimKeyMaps } from '../../editor/CodeMirror'
 import { CodeMirrorBinding } from 'y-codemirror'
 import { useEffectOnce } from 'react-use'
 import { WebsocketProvider } from 'y-websocket'
 import throttle from 'lodash.throttle'
 import { UndoManager, YEvent } from 'yjs'
 import { YText } from 'yjs/dist/src/internals'
+import { useSettings } from '../../stores/settings'
 
 interface EditorProps {
   config?: CodeMirror.EditorConfiguration
@@ -24,11 +25,16 @@ const CodeMirrorEditor = ({
   onLineScroll,
   onYTextChange,
 }: EditorProps) => {
+  const { settings } = useSettings()
   const editorRootRef = useRef<HTMLDivElement>(null)
   const editorRef = useRef<CodeMirror.Editor>()
   const onScrollLineRef = useRef(onLineScroll)
   const skipOnScrollRef = useRef(false)
 
+  useEffect(() => {
+    syncCodeMirrorVimKeyMaps(settings['general.editorVimKeyMaps'])
+  }, [settings])
+
   useEffect(() => {
     onScrollLineRef.current = onLineScroll
   }, [onLineScroll])
diff --git a/src/cloud/lib/i18n/enUS.ts b/src/cloud/lib/i18n/enUS.ts
index 3555a126ec..5712195417 100644
--- a/src/cloud/lib/i18n/enUS.ts
+++ b/src/cloud/lib/i18n/enUS.ts
@@ -58,6 +58,11 @@ const enTranslation: TranslationSource = {
   [lngKeys.SettingsEditorTheme]: 'Editor Theme',
   [lngKeys.SettingsCodeBlockTheme]: 'Code Block Theme',
   [lngKeys.SettingsEditorKeyMap]: 'Editor Keymap',
+  [lngKeys.SettingsEditorVimKeyMaps]: 'Editor Vim Mappings',
+  [lngKeys.SettingsEditorVimKeyMapsDescription]:
+    'JSON array of Vim mappings. Example: [{"toKeys":"jk","keys":"<Esc>","context":"insert"}]',
+  [lngKeys.SettingsEditorVimKeyMapsInvalid]:
+    'Invalid Vim mappings. Enter a JSON array of { "toKeys", "keys", "context" } entries.',
   [lngKeys.SettingsEditorFontSize]: 'Editor Font Size',
   [lngKeys.SettingsEditorFontFamily]: 'Editor Font Family',
   [lngKeys.SettingsLight]: 'Light',
diff --git a/src/cloud/lib/i18n/fr.ts b/src/cloud/lib/i18n/fr.ts
index fabf363f58..24ecf0bb3f 100644
--- a/src/cloud/lib/i18n/fr.ts
+++ b/src/cloud/lib/i18n/fr.ts
@@ -62,6 +62,11 @@ const frTranslation: TranslationSource = {
   [lngKeys.SettingsEditorTheme]: "Thème de l'éditeur",
   [lngKeys.SettingsCodeBlockTheme]: 'Thème des blocs de code',
   [lngKeys.SettingsEditorKeyMap]: "KeyMap pour l'éditeur",
+  [lngKeys.SettingsEditorVimKeyMaps]: "Mappages Vim de l'éditeur",
+  [lngKeys.SettingsEditorVimKeyMapsDescription]:
+    'Tableau JSON de mappages Vim. Exemple : [{"toKeys":"jk","keys":"<Esc>","context":"insert"}]',
+  [lngKeys.SettingsEditorVimKeyMapsInvalid]:
+    'Mappages Vim invalides. Entrez un tableau JSON avec des entrées { "toKeys", "keys", "context" }.',
   [lngKeys.SettingsEditorFontSize]: "Taille de la police de l'éditeur",
   [lngKeys.SettingsEditorFontFamily]: "Famille de polices de l'éditeur",
   [lngKeys.SettingsLight]: 'Clair',
diff --git a/src/cloud/lib/i18n/ja.ts b/src/cloud/lib/i18n/ja.ts
index ec0187cef9..5d0ea24184 100644
--- a/src/cloud/lib/i18n/ja.ts
+++ b/src/cloud/lib/i18n/ja.ts
@@ -60,6 +60,11 @@ const jpTranslation: TranslationSource = {
   [lngKeys.SettingsEditorTheme]: 'エディタテーマ',
   [lngKeys.SettingsCodeBlockTheme]: 'コードブロックテーマ',
   [lngKeys.SettingsEditorKeyMap]: 'エディタのキーマップ',
+  [lngKeys.SettingsEditorVimKeyMaps]: 'エディタの Vim マッピング',
+  [lngKeys.SettingsEditorVimKeyMapsDescription]:
+    'Vim マッピングの JSON 配列です。例: [{"toKeys":"jk","keys":"<Esc>","context":"insert"}]',
+  [lngKeys.SettingsEditorVimKeyMapsInvalid]:
+    'Vim マッピングが無効です。{ "toKeys", "keys", "context" } を含む JSON 配列を入力してください。',
   [lngKeys.SettingsEditorFontSize]: 'エディタのフォントサイズ',
   [lngKeys.SettingsEditorFontFamily]: 'エディタフォントファミリ',
   [lngKeys.SettingsLight]: 'ライト',
diff --git a/src/cloud/lib/i18n/types.ts b/src/cloud/lib/i18n/types.ts
index 5d52ebd00c..ce3c6fe696 100644
--- a/src/cloud/lib/i18n/types.ts
+++ b/src/cloud/lib/i18n/types.ts
@@ -130,6 +130,9 @@ export enum lngKeys {
   SettingsEditorTheme = 'settings.editorTheme',
   SettingsCodeBlockTheme = 'settings.codeblockTheme',
   SettingsEditorKeyMap = 'settings.editorKeyMap',
+  SettingsEditorVimKeyMaps = 'settings.editorVimKeyMaps',
+  SettingsEditorVimKeyMapsDescription = 'settings.editorVimKeyMaps.description',
+  SettingsEditorVimKeyMapsInvalid = 'settings.editorVimKeyMaps.invalid',
   SettingsEditorFontSize = 'settings.editorFontSize',
   SettingsEditorFontFamily = 'settings.editorFontFamily',
   SettingsLight = 'settings.light',
diff --git a/src/cloud/lib/i18n/zhCN.ts b/src/cloud/lib/i18n/zhCN.ts
index 7405ab8e72..fed9bbb4dc 100644
--- a/src/cloud/lib/i18n/zhCN.ts
+++ b/src/cloud/lib/i18n/zhCN.ts
@@ -59,6 +59,11 @@ const zhTranslation: TranslationSource = {
   [lngKeys.SettingsEditorTheme]: '编辑主题',
   [lngKeys.SettingsCodeBlockTheme]: '代码块主题',
   [lngKeys.SettingsEditorKeyMap]: '编辑器键映射',
+  [lngKeys.SettingsEditorVimKeyMaps]: '编辑器 Vim 映射',
+  [lngKeys.SettingsEditorVimKeyMapsDescription]:
+    'Vim 映射的 JSON 数组。例如：[{"toKeys":"jk","keys":"<Esc>","context":"insert"}]',
+  [lngKeys.SettingsEditorVimKeyMapsInvalid]:
+    'Vim 映射无效。请输入包含 { "toKeys", "keys", "context" } 的 JSON 数组。',
   [lngKeys.SettingsEditorFontSize]: '编辑器字体大小',
   [lngKeys.SettingsEditorFontFamily]: '编辑器字体系列',
   [lngKeys.SettingsLight]: 'Light',
diff --git a/src/cloud/lib/stores/settings/store.ts b/src/cloud/lib/stores/settings/store.ts
index 5bf5c6e0ce..5e21e8df1d 100644
--- a/src/cloud/lib/stores/settings/store.ts
+++ b/src/cloud/lib/stores/settings/store.ts
@@ -20,6 +20,7 @@ export const baseUserSettings: UserSettings = {
   'general.codeBlockTheme': 'default',
   'general.customBlockEditorTheme': 'dark',
   'general.editorKeyMap': 'default',
+  'general.editorVimKeyMaps': [],
   'general.editorIndentType': 'spaces',
   'general.editorIndentSize': 2,
   'general.editorFontSize': 15,
diff --git a/src/cloud/lib/stores/settings/types.ts b/src/cloud/lib/stores/settings/types.ts
index a03c6f964a..9b7510526e 100644
--- a/src/cloud/lib/stores/settings/types.ts
+++ b/src/cloud/lib/stores/settings/types.ts
@@ -3,6 +3,12 @@ export type GeneralEditorIndentType = 'spaces' | 'tab'
 export type GeneralEditorIndentSize = 2 | 4 | 8
 export type GeneralLanguageOptions = 'en-US' | 'ja' | 'fr' | 'kr' | 'zh-CN'
 
+export interface CodeMirrorVimMapEntry {
+  toKeys: string
+  keys: string
+  context: string
+}
+
 export interface UserSettings {
   // General
   'general.language': GeneralLanguageOptions
@@ -11,6 +17,7 @@ export interface UserSettings {
   'general.codeBlockTheme': CodeMirrorEditorTheme
   'general.customBlockEditorTheme': MonacoEditorTheme
   'general.editorKeyMap': CodeMirrorKeyMap
+  'general.editorVimKeyMaps': CodeMirrorVimMapEntry[]
   'general.editorIndentType': GeneralEditorIndentType
   'general.editorIndentSize': GeneralEditorIndentSize
   'general.editorFontSize': number