feat(CodeBlock): add custom tab key press handling in the codeblock

gravity-ui · Oct 25, 2023 · fb3284c · fb3284c
1 parent 67bad99
commit fb3284c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/extensions/markdown/CodeBlock/CodeBlockSpecs/index.ts b/src/extensions/markdown/CodeBlock/CodeBlockSpecs/index.ts
@@ -68,6 +68,16 @@ export const CodeBlockSpecs: ExtensionAuto<CodeBlockSpecsOptions> = (builder, op
             throw new Error('Unexpected toYfm() call on fence node');
         },
     }));
+    builder.addKeymap(() => ({
+        Tab: (state, dispatch) => {
+            const {$anchor, $head} = state.selection;
+            if ($anchor.sameParent($head) && $anchor.parent.type.name === codeBlockNodeName) {
+                dispatch?.(state.tr.replaceSelectionWith(state.schema.text('\t')).scrollIntoView());
+                return true;
+            }
+            return false;
+        },
+    }));
 };
 
 function removeNewLineAtEnd(content: string): string {