🐛 fix(vscode): 代码阅读预览记住浏览位置，编辑 / 预览互切位置对齐

vscode/webview-ui/src/codeReader/App.tsx

@@ -1,21 +1,30 @@
 import type { ReactElement } from 'react'
 import type { BlockTreeState } from './BlockTree'
 import type { DocumentMessage } from './types'
-import type { ViewState } from './vscode'
+import type { ViewState } from './viewState'
 import { TriangleAlert } from 'lucide-react'
 import { useCallback, useEffect, useMemo, useState } from 'react'
 import { SelectMenu } from '@/components/ui/select-menu'
 import { BlockLevel } from './BlockTree'
-import { loadViewState, onMessage, postMessage, saveViewState } from './vscode'
+import { useScrollPosition } from './useScrollPosition'
+import { onMessage, postMessage, viewState } from './vscode'
+
+type BlockModes = Pick<ViewState, 'editing' | 'collapsed'>
 
 function toggled(ids: string[], id: string): string[] {
   return ids.includes(id) ? ids.filter(x => x !== id) : [...ids, id]
 }
 
+function savedBlockModes(): BlockModes {
+  const { editing, collapsed } = viewState.get()
+  return { editing, collapsed }
+}
+
 export function App(): ReactElement | null {
   const [doc, setDoc] = useState<DocumentMessage | null>(null)
-  const [view, setView] = useState<ViewState>(loadViewState)
+  const [view, setView] = useState<BlockModes>(savedBlockModes)
   const [focusId, setFocusId] = useState<string | null>(null)
+  useScrollPosition(doc)
 
   useEffect(() => {
     const off = onMessage((msg) => {
@@ -27,7 +36,7 @@ export function App(): ReactElement | null {
   }, [])
 
   useEffect(() => {
-    saveViewState(view)
+    viewState.update(view)
   }, [view])
 
   const onFocused = useCallback(() => setFocusId(null), [])