From ee6d20ddebea55b334f3c082b5155eaf797990eb Mon Sep 17 00:00:00 2001
From: garrettmills <shout@garrettmills.dev>
Date: Sat, 9 Apr 2022 21:10:17 -0500
Subject: [PATCH] Implement remove for var decls and expressions

---
 src/components/Statement.vue   | 23 +++++++-------
 src/components/TextBox.vue     |  4 +--
 src/pages/Editor.vue           | 56 +++++++++++++++++++++++++++++++---
 src/pages/ExpressionEditor.vue | 26 +++++++++++++---
 src/pages/VarDeclEditor.vue    | 28 ++++++++++++++---
 src/support/page.ts            |  5 +++
 src/support/parse.ts           | 17 +++++++++--
 7 files changed, 128 insertions(+), 31 deletions(-)

diff --git a/src/components/Statement.vue b/src/components/Statement.vue
index 05b4a2e..9812305 100644
--- a/src/components/Statement.vue
+++ b/src/components/Statement.vue
@@ -36,14 +36,6 @@ computed(() => value = getValueStatement())
   .sidebar {
     padding-left: 10px;
   }
-
-  .edit-button {
-    border: none;
-  }
-
-  .edit-button:hover {
-    cursor: pointer;
-  }
 </style>
 
 <template>
@@ -56,9 +48,18 @@ computed(() => value = getValueStatement())
       </div>
     </div>
     <div class="sidebar">
-      <button class="edit-button" @click="() => $emit('edit')" title="Edit this expression">
-        <img src="../assets/edit.svg" alt="Edit" height="16">
-      </button>
+      <q-btn color="grey-7" round flat icon="more_vert">
+        <q-menu cover auto-close>
+          <q-list>
+            <q-item clickable>
+              <q-item-section @click="() => $emit('edit')">Edit</q-item-section>
+            </q-item>
+            <q-item clickable>
+              <q-item-section @click="() => $emit('remove')">Remove</q-item-section>
+            </q-item>
+          </q-list>
+        </q-menu>
+      </q-btn>
     </div>
   </div>
 </template>
diff --git a/src/components/TextBox.vue b/src/components/TextBox.vue
index 80216ac..59bea62 100644
--- a/src/components/TextBox.vue
+++ b/src/components/TextBox.vue
@@ -25,12 +25,12 @@ const props = defineProps({
                   <q-list>
                     <q-item clickable>
                       <q-item-section @click="() => $emit('edit')"
-                        >Edit</q-item-section
+                      >Edit</q-item-section
                       >
                     </q-item>
                     <q-item clickable>
                       <q-item-section @click="() => $emit('remove')"
-                        >Remove</q-item-section
+                      >Remove</q-item-section
                       >
                     </q-item>
                   </q-list>
diff --git a/src/pages/Editor.vue b/src/pages/Editor.vue
index 5bc9b99..c782304 100644
--- a/src/pages/Editor.vue
+++ b/src/pages/Editor.vue
@@ -1,17 +1,15 @@
 <script setup lang="ts">
 import {onMounted, ref} from 'vue'
 import { v4 as uuidv4 } from 'uuid'
-import WrapperBox from '../components/WrapperBox.vue'
 import {MathPage} from '../support/page'
 import {MathStatement} from '../support/parse'
-import Katex from '../components/Katex.vue'
-import {EvaluationResult, hasOwnProperty, StatementID} from '../types'
+import {EvaluationResult, hasOwnProperty} from '../types'
 import Statement from '../components/Statement.vue'
 import VarDeclEditor from './VarDeclEditor.vue'
 import ExpressionEditor from './ExpressionEditor.vue'
 import TextBox from '../components/TextBox.vue'
-import {RichTextBox} from "../types.ts";
-import { stepX, stepY } from "../support/const.ts";
+import {RichTextBox} from '../types'
+import { stepX, stepY } from '../support/const'
 
 const math = new MathPage(uuidv4())
 const statements = ref<MathStatement[]>([])
@@ -49,6 +47,17 @@ const openNewExpressionModal = () => {
   newExpressionModalOpen.value = true
 }
 
+const editingStatement = ref<MathStatement|undefined>()
+const editExpressionModalOpen = ref(false)
+const openEditExpressionModal = () => {
+  editExpressionModalOpen.value = true
+}
+
+const editVarDeclModalOpen = ref(false)
+const openEditVarDeclModal = () => {
+  editVarDeclModalOpen.value = true
+}
+
 const updateStatements = () => {
   statements.value = math.getStatements()
   try {
@@ -92,6 +101,27 @@ const saveNewExpression = (stmt: MathStatement) => {
   updateStatements()
 }
 
+const editStatement = (stmt: MathStatement) => {
+  editingStatement.value = stmt
+  if ( stmt.isDeclaration() ) {
+    openEditVarDeclModal()
+  } else if ( stmt.isFunctionDeclaration() ) {
+
+  } else {
+    openEditExpressionModal()
+  }
+}
+
+const removeStatement = (stmt: MathStatement) => {
+  math.removeStatement(stmt.id)
+  updateStatements()
+}
+
+const finishEditStatement = () => {
+  editExpressionModalOpen.value = false
+  updateStatements()
+}
+
 /*
   Rich Text Stuff
 */
@@ -204,6 +234,8 @@ function richUpdateValue() {
             :key="statementsKey"
             :statement="statement"
             :evaluation="evaluation"
+            v-on:edit="() => editStatement(statement)"
+            v-on:remove="() => removeStatement(statement)"
           />
         </div>
 
@@ -222,6 +254,20 @@ function richUpdateValue() {
         />
       </q-dialog>
 
+      <q-dialog v-model="editExpressionModalOpen">
+        <ExpressionEditor
+          :statement="editingStatement"
+          v-on:save="() => finishEditStatement()"
+        />
+      </q-dialog>
+
+      <q-dialog v-model="editVarDeclModalOpen">
+        <VarDeclEditor
+          :statement="editingStatement"
+          v-on:save="() => finishEditStatement()"
+        />
+      </q-dialog>
+
       <q-page-sticky position="bottom-right" :offset="[32, 32]">
         <q-fab color="primary" icon="add" direction="left">
           <q-fab-action
diff --git a/src/pages/ExpressionEditor.vue b/src/pages/ExpressionEditor.vue
index 6fffc12..a334215 100644
--- a/src/pages/ExpressionEditor.vue
+++ b/src/pages/ExpressionEditor.vue
@@ -1,10 +1,14 @@
 <script setup lang="ts">
-import {defineEmits, ref} from 'vue'
+import {defineEmits, onMounted, ref} from 'vue'
 import {MathStatement} from '../support/parse'
 import Katex from '../components/Katex.vue'
 import { v4 as uuidv4 } from 'uuid'
 import {StatementID} from '../types'
 
+const props = defineProps<{
+  statement?: MathStatement,
+}>()
+
 const emit = defineEmits<{
   (eventName: 'save', statement: MathStatement): void
 }>()
@@ -39,11 +43,23 @@ const saveExpression = () => {
     return
   }
 
-  emit('save', new MathStatement(
-      uuidv4() as StatementID,
-      expressionValue.value
-  ))
+  if ( props.statement ) {
+    props.statement.raw = expressionValue.value
+    emit('save', props.statement)
+  } else {
+    emit('save', new MathStatement(
+        uuidv4() as StatementID,
+        expressionValue.value
+    ))
+  }
 }
+
+onMounted(() => {
+  if ( props.statement ) {
+    expressionValue.value = props.statement.raw
+    updateExpressionPreview()
+  }
+})
 </script>
 
 <template>
diff --git a/src/pages/VarDeclEditor.vue b/src/pages/VarDeclEditor.vue
index a8f545f..0d5f914 100644
--- a/src/pages/VarDeclEditor.vue
+++ b/src/pages/VarDeclEditor.vue
@@ -1,10 +1,14 @@
 <script setup lang="ts">
-import {ref} from 'vue'
+import {onMounted, ref} from 'vue'
 import {MathStatement} from '../support/parse'
 import {v4 as uuidv4} from 'uuid'
 import Katex from '../components/Katex.vue'
 import {StatementID} from '../types'
 
+const props = defineProps<{
+  statement?: MathStatement,
+}>()
+
 const emit = defineEmits<{
   (eventName: 'save', statement: MathStatement): void,
 }>()
@@ -48,11 +52,25 @@ const saveNewVariable = () => {
     return
   }
 
-  emit('save', new MathStatement(
-      uuidv4() as StatementID,
-      `${newVariableName.value} = ${newVariableValue.value}`
-  ))
+  if ( props.statement ) {
+    props.statement.raw = `${newVariableName.value} = ${newVariableValue.value}`
+    emit('save', props.statement)
+  } else {
+    emit('save', new MathStatement(
+        uuidv4() as StatementID,
+        `${newVariableName.value} = ${newVariableValue.value}`
+    ))
+  }
 }
+
+onMounted(() => {
+  if ( props.statement ) {
+    const [ define ] = props.statement.defines()
+    newVariableName.value = define.name
+    newVariableValue.value = props.statement.raw
+    updateNewVariablePreview()
+  }
+})
 </script>
 
 <template>
diff --git a/src/support/page.ts b/src/support/page.ts
index b4e60fa..1214755 100644
--- a/src/support/page.ts
+++ b/src/support/page.ts
@@ -21,6 +21,11 @@ export class MathPage {
         return Object.values(this.statements)
     }
 
+    /** Remove a statement from the math page. */
+    removeStatement(id: StatementID) {
+        delete this.statements[id]
+    }
+
     /** Get a statement by ID if it exists. */
     getStatement(id: StatementID): Maybe<MathStatement> {
         return this.statements[id]
diff --git a/src/support/parse.ts b/src/support/parse.ts
index b9cf433..08625f4 100644
--- a/src/support/parse.ts
+++ b/src/support/parse.ts
@@ -268,10 +268,13 @@ export class MathStatement {
         public readonly id: StatementID,
 
         /** The raw statement input by the user. */
-        public readonly raw: string,
+        public raw: string,
 
-        public x: Number = 0,
-        public y: Number = 0,
+        /** The x position in the grid. */
+        public x: number = 0,
+
+        /** The y position in the grid. */
+        public y: number = 0,
     ) {}
 
     /** Parse the raw statement to an AST. */
@@ -341,4 +344,12 @@ export class MathStatement {
 
         return true
     }
+
+    isDeclaration(): boolean {
+        return this.defines().length > 0
+    }
+
+    isFunctionDeclaration(): boolean {
+        return false
+    }
 }