add format button

src/App.tsx

@@ -1,10 +1,11 @@
-import { useRef, useState, useEffect } from 'react'
+import { useRef, useState, useEffect, useMemo } from 'react'
 import { Canvas } from '@react-three/fiber'
 import { Allotment } from 'allotment'
 import { OrbitControls, OrthographicCamera } from '@react-three/drei'
 import { lexer } from './lang/tokeniser'
 import { abstractSyntaxTree } from './lang/abstractSyntaxTree'
 import { executor } from './lang/executor'
+import { recast } from './lang/recast'
 import { BufferGeometry } from 'three'
 import CodeMirror from '@uiw/react-codemirror'
 import { javascript } from '@codemirror/lang-javascript'
@@ -38,6 +39,8 @@ function App() {
     code,
     setCode,
     setAst,
+    formatCode,
+    ast,
   } = useStore((s) => ({
     editorView: s.editorView,
     setEditorView: s.setEditorView,
@@ -51,7 +54,8 @@ function App() {
     setCode: s.setCode,
     ast: s.ast,
     setAst: s.setAst,
-    lastGuiMode: s.lastGuiMode
+    lastGuiMode: s.lastGuiMode,
+    formatCode: s.formatCode,
   }))
   // const onChange = React.useCallback((value: string, viewUpdate: ViewUpdate) => {
   const onChange = (value: string, viewUpdate: ViewUpdate) => {
@@ -116,11 +120,18 @@ function App() {
       addLog(e)
     }
   }, [code])
+  const shouldFormat = useMemo(() => {
+    if(!ast) return false
+    const recastedCode = recast(ast)
+    return recastedCode !== code
+  }, [code, ast])
   return (
     <div className="h-screen">
       <Allotment>
         <Logs />
-        <div className="bg-red h-full overflow-auto">
+        <div className="h-full flex flex-col items-start">
+          <button disabled={!shouldFormat} onClick={formatCode} className={`${!shouldFormat && "text-gray-300"}`}>format</button>
+          <div className="bg-red h-full w-full overflow-auto">
             <CodeMirror
               className="h-full"
               value={code}
@@ -130,6 +141,7 @@ function App() {
               onCreateEditor={(_editorView) => setEditorView(_editorView)}
             />
           </div>
+        </div>
         <div className="h-full">
           <Toolbar />
           <div className="border h-full border-gray-300 relative">

src/useStore.ts

@@ -1,7 +1,8 @@
 import create from 'zustand'
 import { addLineHighlight, EditorView } from './editor/highlightextension'
-import { Program } from './lang/abstractSyntaxTree'
+import { Program, abstractSyntaxTree } from './lang/abstractSyntaxTree'
 import { recast } from './lang/recast'
+import { lexer } from './lang/tokeniser'
 
 export type Range = [number, number]
 
@@ -42,6 +43,7 @@ interface StoreState {
   updateAst: (ast: Program) => void
   code: string
   setCode: (code: string) => void
+  formatCode: () => void
 }
 
 export const useStore = create<StoreState>()((set, get) => ({
@@ -98,5 +100,11 @@ export const useStore = create<StoreState>()((set, get) => ({
   code: '',
   setCode: (code) => {
     set({ code })
-  }
+  },
+  formatCode: () => {
+    const code = get().code
+    const ast = abstractSyntaxTree(lexer(code))
+    const newCode = recast(ast)
+    set({ code: newCode, ast })
+  },
 }))