Add tracking of operations for the feature tree

src/lang/errors.test.ts

@@ -9,13 +9,15 @@ describe('test kclErrToDiagnostic', () => {
         kind: 'semantic',
         msg: 'Semantic error',
         sourceRange: [0, 1, true],
+        operations: [],
       },
       {
         name: '',
         message: '',
         kind: 'type',
         msg: 'Type error',
         sourceRange: [4, 5, true],
+        operations: [],
       },
     ]
     const diagnostics = kclErrorsToDiagnostics(errors)

src/lang/errors.ts

@@ -6,85 +6,94 @@ import { Diagnostic as LspDiagnostic } from 'vscode-languageserver-protocol'
 import { Text } from '@codemirror/state'
 import { EditorView } from 'codemirror'
 import { SourceRange } from 'lang/wasm'
+import { Operation } from 'wasm-lib/kcl/bindings/Operation'
 
 type ExtractKind<T> = T extends { kind: infer K } ? K : never
 export class KCLError extends Error {
   kind: ExtractKind<RustKclError> | 'name'
   sourceRange: SourceRange
   msg: string
+  operations: Operation[]
 
   constructor(
     kind: ExtractKind<RustKclError> | 'name',
     msg: string,
-    sourceRange: SourceRange
+    sourceRange: SourceRange,
+    operations: Operation[]
   ) {
     super()
     this.kind = kind
     this.msg = msg
     this.sourceRange = sourceRange
+    this.operations = operations
     Object.setPrototypeOf(this, KCLError.prototype)
   }
 }
 
 export class KCLLexicalError extends KCLError {
-  constructor(msg: string, sourceRange: SourceRange) {
-    super('lexical', msg, sourceRange)
+  constructor(msg: string, sourceRange: SourceRange, operations: Operation[]) {
+    super('lexical', msg, sourceRange, operations)
     Object.setPrototypeOf(this, KCLSyntaxError.prototype)
   }
 }
 
 export class KCLInternalError extends KCLError {
-  constructor(msg: string, sourceRange: SourceRange) {
-    super('internal', msg, sourceRange)
+  constructor(msg: string, sourceRange: SourceRange, operations: Operation[]) {
+    super('internal', msg, sourceRange, operations)
     Object.setPrototypeOf(this, KCLSyntaxError.prototype)
   }
 }
 
 export class KCLSyntaxError extends KCLError {
-  constructor(msg: string, sourceRange: SourceRange) {
-    super('syntax', msg, sourceRange)
+  constructor(msg: string, sourceRange: SourceRange, operations: Operation[]) {
+    super('syntax', msg, sourceRange, operations)
     Object.setPrototypeOf(this, KCLSyntaxError.prototype)
   }
 }
 
 export class KCLSemanticError extends KCLError {
-  constructor(msg: string, sourceRange: SourceRange) {
-    super('semantic', msg, sourceRange)
+  constructor(msg: string, sourceRange: SourceRange, operations: Operation[]) {
+    super('semantic', msg, sourceRange, operations)
     Object.setPrototypeOf(this, KCLSemanticError.prototype)
   }
 }
 
 export class KCLTypeError extends KCLError {
-  constructor(msg: string, sourceRange: SourceRange) {
-    super('type', msg, sourceRange)
+  constructor(msg: string, sourceRange: SourceRange, operations: Operation[]) {
+    super('type', msg, sourceRange, operations)
     Object.setPrototypeOf(this, KCLTypeError.prototype)
   }
 }
 
 export class KCLUnimplementedError extends KCLError {
-  constructor(msg: string, sourceRange: SourceRange) {
-    super('unimplemented', msg, sourceRange)
+  constructor(msg: string, sourceRange: SourceRange, operations: Operation[]) {
+    super('unimplemented', msg, sourceRange, operations)
     Object.setPrototypeOf(this, KCLUnimplementedError.prototype)
   }
 }
 
 export class KCLUnexpectedError extends KCLError {
-  constructor(msg: string, sourceRange: SourceRange) {
-    super('unexpected', msg, sourceRange)
+  constructor(msg: string, sourceRange: SourceRange, operations: Operation[]) {
+    super('unexpected', msg, sourceRange, operations)
     Object.setPrototypeOf(this, KCLUnexpectedError.prototype)
   }
 }
 
 export class KCLValueAlreadyDefined extends KCLError {
-  constructor(key: string, sourceRange: SourceRange) {
-    super('name', `Key ${key} was already defined elsewhere`, sourceRange)
+  constructor(key: string, sourceRange: SourceRange, operations: Operation[]) {
+    super(
+      'name',
+      `Key ${key} was already defined elsewhere`,
+      sourceRange,
+      operations
+    )
     Object.setPrototypeOf(this, KCLValueAlreadyDefined.prototype)
   }
 }
 
 export class KCLUndefinedValueError extends KCLError {
-  constructor(key: string, sourceRange: SourceRange) {
-    super('name', `Key ${key} has not been defined`, sourceRange)
+  constructor(key: string, sourceRange: SourceRange, operations: Operation[]) {
+    super('name', `Key ${key} has not been defined`, sourceRange, operations)
     Object.setPrototypeOf(this, KCLUndefinedValueError.prototype)
   }
 }
@@ -100,11 +109,12 @@ export function lspDiagnosticsToKclErrors(
   return diagnostics
     .flatMap(
       ({ range, message }) =>
-        new KCLError('unexpected', message, [
-          posToOffset(doc, range.start)!,
-          posToOffset(doc, range.end)!,
-          true,
-        ])
+        new KCLError(
+          'unexpected',
+          message,
+          [posToOffset(doc, range.start)!, posToOffset(doc, range.end)!, true],
+          []
+        )
     )
     .sort((a, b) => {
       const c = a.sourceRange[0]

src/lang/executor.test.ts

@@ -480,7 +480,8 @@ const theExtrude = startSketchOn('XY')
       new KCLError(
         'undefined_value',
         'memory item key `myVarZ` is not defined',
-        [129, 135, true]
+        [129, 135, true],
+        []
       )
     )
   })

src/lang/wasm.ts

@@ -43,6 +43,7 @@ import { Node } from 'wasm-lib/kcl/bindings/Node'
 import { CompilationError } from 'wasm-lib/kcl/bindings/CompilationError'
 import { SourceRange as RustSourceRange } from 'wasm-lib/kcl/bindings/SourceRange'
 import { getAllCurrentSettings } from 'lib/settings/settingsUtils'
+import { KclErrorWithOutputs } from 'wasm-lib/kcl/bindings/KclErrorWithOutputs'
 
 export type { Configuration } from 'wasm-lib/kcl/bindings/Configuration'
 export type { Program } from '../wasm-lib/kcl/bindings/Program'
@@ -212,7 +213,8 @@ export const parse = (code: string | Error): ParseResult | Error => {
     return new KCLError(
       parsed.kind,
       parsed.msg,
-      sourceRangeFromRust(parsed.sourceRanges[0])
+      sourceRangeFromRust(parsed.sourceRanges[0]),
+      []
     )
   }
 }
@@ -531,11 +533,12 @@ export const _executor = async (
     return execStateFromRaw(execState)
   } catch (e: any) {
     console.log(e)
-    const parsed: RustKclError = JSON.parse(e.toString())
+    const parsed: KclErrorWithOutputs = JSON.parse(e.toString())
     const kclError = new KCLError(
-      parsed.kind,
-      parsed.msg,
-      sourceRangeFromRust(parsed.sourceRanges[0])
+      parsed.error.kind,
+      parsed.error.msg,
+      sourceRangeFromRust(parsed.error.sourceRanges[0]),
+      parsed.operations
     )
 
     return Promise.reject(kclError)
@@ -594,7 +597,8 @@ export const modifyAstForSketch = async (
     const kclError = new KCLError(
       parsed.kind,
       parsed.msg,
-      sourceRangeFromRust(parsed.sourceRanges[0])
+      sourceRangeFromRust(parsed.sourceRanges[0]),
+      []
     )
 
     console.log(kclError)
@@ -662,7 +666,8 @@ export function programMemoryInit(): ProgramMemory | Error {
     return new KCLError(
       parsed.kind,
       parsed.msg,
-      sourceRangeFromRust(parsed.sourceRanges[0])
+      sourceRangeFromRust(parsed.sourceRanges[0]),
+      []
     )
   }
 }

src/wasm-lib/derive-docs/src/lib.rs

@@ -38,6 +38,12 @@ struct StdlibMetadata {
     #[serde(default)]
     deprecated: bool,
 
+    /// Whether the function is displayed in the feature tree.
+    /// If true, calls to the function will be available for display.
+    /// If false, calls to the function will never be displayed.
+    #[serde(default)]
+    feature_tree_operation: bool,
+
     /// If true, expects keyword arguments.
     /// If false, expects positional arguments.
     #[serde(default)]
@@ -238,6 +244,12 @@ fn do_stdlib_inner(
         quote! { false }
     };
 
+    let feature_tree_operation = if metadata.feature_tree_operation {
+        quote! { true }
+    } else {
+        quote! { false }
+    };
+
     let uses_keyword_arguments = if metadata.keywords {
         quote! { true }
     } else {
@@ -451,6 +463,10 @@ fn do_stdlib_inner(
                 #deprecated
             }
 
+            fn feature_tree_operation(&self) -> bool {
+                #feature_tree_operation
+            }
+
             fn examples(&self) -> Vec<String> {
                 #code_blocks
             }

src/wasm-lib/derive-docs/tests/args_with_lifetime.gen

@@ -116,6 +116,10 @@ impl crate::docs::StdLibFn for SomeFn {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["someFn()"];
         code_blocks

src/wasm-lib/derive-docs/tests/args_with_refs.gen

@@ -116,6 +116,10 @@ impl crate::docs::StdLibFn for SomeFn {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["someFn()"];
         code_blocks

src/wasm-lib/derive-docs/tests/array.gen

@@ -154,6 +154,10 @@ impl crate::docs::StdLibFn for Show {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec![
             "This is another code block.\nyes sirrr.\nshow",

src/wasm-lib/derive-docs/tests/box.gen

@@ -117,6 +117,10 @@ impl crate::docs::StdLibFn for Show {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["This is code.\nIt does other shit.\nshow"];
         code_blocks

src/wasm-lib/derive-docs/tests/doc_comment_with_code.gen

@@ -155,6 +155,10 @@ impl crate::docs::StdLibFn for MyFunc {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec![
             "This is another code block.\nyes sirrr.\nmyFunc",

src/wasm-lib/derive-docs/tests/lineTo.gen

@@ -164,6 +164,10 @@ impl crate::docs::StdLibFn for LineTo {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec![
             "This is another code block.\nyes sirrr.\nlineTo",

src/wasm-lib/derive-docs/tests/min.gen

@@ -154,6 +154,10 @@ impl crate::docs::StdLibFn for Min {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec![
             "This is another code block.\nyes sirrr.\nmin",

src/wasm-lib/derive-docs/tests/option.gen

@@ -117,6 +117,10 @@ impl crate::docs::StdLibFn for Show {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["This is code.\nIt does other shit.\nshow"];
         code_blocks

src/wasm-lib/derive-docs/tests/option_input_format.gen

@@ -117,6 +117,10 @@ impl crate::docs::StdLibFn for Import {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["This is code.\nIt does other shit.\nimport"];
         code_blocks

src/wasm-lib/derive-docs/tests/return_vec_box_sketch.gen

@@ -117,6 +117,10 @@ impl crate::docs::StdLibFn for Import {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["This is code.\nIt does other shit.\nimport"];
         code_blocks

src/wasm-lib/derive-docs/tests/return_vec_sketch.gen

@@ -117,6 +117,10 @@ impl crate::docs::StdLibFn for Import {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["This is code.\nIt does other shit.\nimport"];
         code_blocks

src/wasm-lib/derive-docs/tests/show.gen

@@ -117,6 +117,10 @@ impl crate::docs::StdLibFn for Show {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["This is code.\nIt does other shit.\nshow"];
         code_blocks

src/wasm-lib/derive-docs/tests/test_args_with_exec_state.gen

@@ -110,6 +110,10 @@ impl crate::docs::StdLibFn for SomeFunction {
         false
     }
 
+    fn feature_tree_operation(&self) -> bool {
+        false
+    }
+
     fn examples(&self) -> Vec<String> {
         let code_blocks = vec!["someFunction()"];
         code_blocks