Define Inputs corresponding definition to enable code localization.

CHANGELOG.md

@@ -24,6 +24,7 @@ Dates are in `YYYY-MM-DD` format and versions are in [semantic versioning](http:
 -   Fixed closing text delimiter localization.
 -   Hide comma separator when localizing names and docs.
 -   Permit comma separators between text literals, docs, and names, allowing line breaks for text.
+-   Define `Input`s corresponding definition to enable code localization.
 
 ## 0.10.4 2024-07-08
 

src/nodes/Context.ts

@@ -50,9 +50,11 @@ export default class Context {
     visit(node: Node) {
         this.stack.push(node);
     }
+
     unvisit() {
         this.stack.pop();
     }
+
     visited(node: Node) {
         return this.stack.includes(node);
     }

src/nodes/Input.ts

@@ -22,6 +22,7 @@ import SimpleExpression from './SimpleExpression';
 import Evaluate from './Evaluate';
 import Refer from '@edit/Refer';
 import ExpressionPlaceholder from './ExpressionPlaceholder';
+import type Definition from './Definition';
 
 export default class Input extends SimpleExpression {
     readonly name: Token;
@@ -122,6 +123,20 @@ export default class Input extends SimpleExpression {
         return this.value.evaluateTypeGuards(current, guard);
     }
 
+    /** Get the bind to which this input corresponds. */
+    getCorrespondingDefinition(context: Context): Definition | undefined {
+        const parent = context.getRoot(this)?.getParent(this);
+        if (parent instanceof Evaluate) {
+            const fun = parent.getFunction(context);
+            if (fun)
+                return fun.inputs.find((input) =>
+                    input.hasName(this.getName()),
+                );
+        }
+
+        return undefined;
+    }
+
     getName() {
         return this.name.getText();
     }