Cursors should stay after a code-mod (#113)

* setup to get path to nodes back from ast-mods

* fix cursor setting for constraint buttons that use transformSecondarySketchLinesTagFirst

* fix cursors for constraints that use transformAstSketchLines

src/components/Toolbar/SetAngleBetween.tsx

@@ -19,19 +19,20 @@ import {
 import { GetInfoModal } from '../SetHorVertDistanceModal'
 import { createVariableDeclaration } from '../../lang/modifyAst'
 import { removeDoubleNegatives } from '../AvailableVarsHelpers'
+import { updateCursors } from '../../lang/util'
 
 const getModalInfo = create(GetInfoModal as any)
 
 export const SetAngleBetween = () => {
-  const { guiMode, selectionRanges, ast, programMemory, updateAst } = useStore(
-    (s) => ({
+  const { guiMode, selectionRanges, ast, programMemory, updateAst, setCursor } =
+    useStore((s) => ({
       guiMode: s.guiMode,
       ast: s.ast,
       updateAst: s.updateAst,
       selectionRanges: s.selectionRanges,
       programMemory: s.programMemory,
-    })
-  )
+      setCursor: s.setCursor,
+    }))
   const [enable, setEnable] = useState(false)
   const [transformInfos, setTransformInfos] = useState<TransformInfo[]>()
   useEffect(() => {
@@ -83,63 +84,66 @@ export const SetAngleBetween = () => {
   return (
     <button
       onClick={async () => {
-        if (transformInfos && ast) {
-          const { modifiedAst, tagInfo, valueUsedInTransform } =
+        if (!(transformInfos && ast)) return
+        const { modifiedAst, tagInfo, valueUsedInTransform, pathToNodeMap } =
+          transformSecondarySketchLinesTagFirst({
+            ast: JSON.parse(JSON.stringify(ast)),
+            selectionRanges,
+            transformInfos,
+            programMemory,
+          })
+        const {
+          segName,
+          value,
+          valueNode,
+          variableName,
+          newVariableInsertIndex,
+          sign,
+        }: {
+          segName: string
+          value: number
+          valueNode: Value
+          variableName?: string
+          newVariableInsertIndex: number
+          sign: number
+        } = await getModalInfo({
+          segName: tagInfo?.tag,
+          isSegNameEditable: !tagInfo?.isTagExisting,
+          value: valueUsedInTransform,
+          initialVariableName: 'angle',
+        } as any)
+        if (segName === tagInfo?.tag && value === valueUsedInTransform) {
+          updateAst(modifiedAst, {
+            callBack: updateCursors(setCursor, selectionRanges, pathToNodeMap),
+          })
+        } else {
+          const finalValue = removeDoubleNegatives(
+            valueNode as BinaryPart,
+            sign,
+            variableName
+          )
+          // transform again but forcing certain values
+          const { modifiedAst: _modifiedAst, pathToNodeMap } =
             transformSecondarySketchLinesTagFirst({
-              ast: JSON.parse(JSON.stringify(ast)),
+              ast,
               selectionRanges,
               transformInfos,
               programMemory,
+              forceSegName: segName,
+              forceValueUsedInTransform: finalValue,
             })
-          const {
-            segName,
-            value,
-            valueNode,
-            variableName,
-            newVariableInsertIndex,
-            sign,
-          }: {
-            segName: string
-            value: number
-            valueNode: Value
-            variableName?: string
-            newVariableInsertIndex: number
-            sign: number
-          } = await getModalInfo({
-            segName: tagInfo?.tag,
-            isSegNameEditable: !tagInfo?.isTagExisting,
-            value: valueUsedInTransform,
-            initialVariableName: 'angle',
-          } as any)
-          if (segName === tagInfo?.tag && value === valueUsedInTransform) {
-            updateAst(modifiedAst)
-          } else {
-            const finalValue = removeDoubleNegatives(
-              valueNode as BinaryPart,
-              sign,
-              variableName
+          if (variableName) {
+            const newBody = [..._modifiedAst.body]
+            newBody.splice(
+              newVariableInsertIndex,
+              0,
+              createVariableDeclaration(variableName, valueNode)
             )
-            // transform again but forcing certain values
-            const { modifiedAst: _modifiedAst } =
-              transformSecondarySketchLinesTagFirst({
-                ast,
-                selectionRanges,
-                transformInfos,
-                programMemory,
-                forceSegName: segName,
-                forceValueUsedInTransform: finalValue,
-              })
-            if (variableName) {
-              const newBody = [..._modifiedAst.body]
-              newBody.splice(
-                newVariableInsertIndex,
-                0,
-                createVariableDeclaration(variableName, valueNode)
-              )
-              _modifiedAst.body = newBody
-            }
-            updateAst(_modifiedAst)
+            _modifiedAst.body = newBody
           }
+          updateAst(_modifiedAst, {
+            callBack: updateCursors(setCursor, selectionRanges, pathToNodeMap),
+          })
         }
       }}
       className={`border m-1 px-1 rounded text-xs ${