fix(#1013): 修复代码块换行问题

Author: codexu

src/app/core/main/editor/markdown/code-block-extension.ts

@@ -0,0 +1,37 @@
+'use client'
+
+import CodeBlockLowlight from '@tiptap/extension-code-block-lowlight'
+
+export const StableCodeBlockLowlight = CodeBlockLowlight.extend({
+  addKeyboardShortcuts() {
+    const parentShortcuts = this.parent?.() ?? {}
+
+    return {
+      ...parentShortcuts,
+      Enter: ({ editor }) => {
+        const { selection } = editor.state
+        const { $from, empty } = selection
+
+        if (!empty || $from.parent.type !== this.type) {
+          return false
+        }
+
+        const isAtEnd = $from.parentOffset === $from.parent.nodeSize - 2
+        const endsWithDoubleNewline = $from.parent.textContent.endsWith('\n\n')
+        const action = isAtEnd && endsWithDoubleNewline ? 'exit' : 'newline'
+
+        if (action === 'exit') {
+          return editor.chain()
+            .command(({ tr }) => {
+              tr.delete($from.pos - 2, $from.pos)
+              return true
+            })
+            .exitCode()
+            .run()
+        }
+
+        return editor.commands.insertContent('\n')
+      },
+    }
+  },
+})

src/app/core/main/editor/markdown/tiptap-editor.tsx

@@ -6,7 +6,6 @@ import Placeholder from '@tiptap/extension-placeholder'
 import Link from '@tiptap/extension-link'
 import TaskList from '@tiptap/extension-task-list'
 import TaskItem from '@tiptap/extension-task-item'
-import CodeBlockLowlight from '@tiptap/extension-code-block-lowlight'
 import CharacterCount from '@tiptap/extension-character-count'
 import Highlight from '@tiptap/extension-highlight'
 import Underline from '@tiptap/extension-underline'
@@ -52,6 +51,7 @@ import { AISuggestionFloating } from './ai-suggestion-floating'
 import emitter from '@/lib/emitter'
 import { QuoteMark } from './quote-mark'
 import { MarkdownParagraph } from './markdown-paragraph'
+import { StableCodeBlockLowlight } from './code-block-extension'
 import useSettingStore from '@/stores/setting'
 import useChatStore from '@/stores/chat'
 import { Loader2, X } from 'lucide-react'
@@ -315,7 +315,7 @@ export function TipTapEditor({
       TaskItem.configure({
         nested: true,
       }),
-      CodeBlockLowlight.configure({
+      StableCodeBlockLowlight.configure({
         lowlight,
       }),
       CharacterCount,