@bubblelab/bubble-runtime 0.1.14 → 0.1.16

package/dist/extraction/BubbleParser.d.ts

@@ -1,25 +1,40 @@
 import type { TSESTree } from '@typescript-eslint/typescript-estree';
 import type { ScopeManager } from '@bubblelab/ts-scope-manager';
 import { BubbleFactory } from '@bubblelab/bubble-core';
-import type { ParsedBubbleWithInfo } from '@bubblelab/shared-schemas';
+import type { MethodInvocationInfo } from '../parse/BubbleScript';
+import type { ParsedBubbleWithInfo, ParsedWorkflow } from '@bubblelab/shared-schemas';
 export declare class BubbleParser {
     private bubbleScript;
+    private cachedAST;
+    private methodInvocationOrdinalMap;
+    private invocationBubbleCloneCache;
+    /**
+     * Track which call expressions have been assigned an invocation index.
+     * Key: `methodName:startOffset` (using AST range start position)
+     * Value: the assigned invocation index
+     * This prevents double-counting when the same call site is processed multiple times
+     * (e.g., once in .map() callback processing, again in Promise.all resolution)
+     */
+    private processedCallSiteIndexes;
+    /** Custom tool func ranges for marking bubbles inside custom tools */
+    private customToolFuncs;
     constructor(bubbleScript: string);
     /**
      * Parse bubble dependencies from an AST using the provided factory and scope manager
      */
     parseBubblesFromAST(bubbleFactory: BubbleFactory, ast: TSESTree.Program, scopeManager: ScopeManager): {
         bubbles: Record<number, ParsedBubbleWithInfo>;
-        handleMethodLocation: {
+        workflow: ParsedWorkflow;
+        instanceMethodsLocation: Record<string, {
             startLine: number;
             endLine: number;
             definitionStartLine: number;
             bodyStartLine: number;
-        } | null;
+            invocationLines: MethodInvocationInfo[];
+        }>;
     };
     private findDependenciesForBubble;
     private buildDependencyGraph;
-    private hashUniqueIdToVarId;
     /**
      * Build a JSON Schema object for the payload parameter of the top-level `handle` entrypoint.
      * Supports primitives, arrays, unions (anyOf), intersections (allOf), type literals, and
@@ -44,13 +59,26 @@ export declare class BubbleParser {
     /** Resolve in-file interface/type alias by name to JSON Schema */
     private resolveTypeNameToJson;
     /**
-     * Find the handle method location in the AST
+     * Find the main class that extends BubbleFlow
      */
-    private findHandleMethodLocation;
+    private findMainBubbleFlowClass;
     /**
-     * Find handle method within a class declaration
+     * Extract all instance methods from a class
      */
-    private findHandleMethodInClass;
+    private findAllInstanceMethods;
+    /**
+     * Find all method invocations in the AST with full details
+     */
+    private findMethodInvocations;
+    /**
+     * Check if a child node is in the condition/test part of a control flow statement
+     * Returns true if the child is the test/discriminant expression, false if it's in the body
+     */
+    private isNodeInConditionPart;
+    /**
+     * Helper to recursively visit child nodes for finding invocations
+     */
+    private visitChildNodesForInvocations;
     /**
      * Recursively visit AST nodes to find bubble instantiations
      */
@@ -83,5 +111,156 @@ export declare class BubbleParser {
      * Extract parameter value and type from an expression
      */
     private extractParameterValue;
+    /**
+     * Find custom tools in ai-agent bubbles and populate customToolFuncs.
+     * This scans the AST for ai-agent instantiations and extracts custom tool func locations.
+     */
+    private findCustomToolsInAIAgentBubbles;
+    /**
+     * Mark bubbles that are inside custom tool funcs with isInsideCustomTool flag.
+     */
+    private markBubblesInsideCustomTools;
+    /**
+     * Extract comment/description for a node by looking at preceding comments
+     **/
+    private extractCommentForNode;
+    /**
+     * Extract JSDoc info including description, @canBeFile, and @canBeGoogleFile tags from a node's preceding comments.
+     * The @canBeFile tag controls whether file upload is enabled for string fields in the UI.
+     * The @canBeGoogleFile tag enables Google Picker UI for Google Drive file/folder ID fields.
+     */
+    private extractJSDocForNode;
+    /**
+     * Check if a list of workflow nodes contains a terminating statement (return/throw)
+     * A branch terminates if its last statement is a return or throw
+     */
+    private branchTerminates;
+    /**
+     * Build hierarchical workflow structure from AST
+     */
+    private buildWorkflowTree;
+    /**
+     * Group consecutive nodes of the same type
+     * - Consecutive variable_declaration nodes → merge into one
+     * - Consecutive code_block nodes → merge into one
+     * - return nodes are NOT grouped (each is a distinct exit point)
+     */
+    private groupConsecutiveNodes;
+    /**
+     * Merge a group of nodes of the same type into a single node
+     */
+    private mergeGroup;
+    /**
+     * Build a workflow node from an AST statement
+     */
+    private buildWorkflowNodeFromStatement;
+    /**
+     * Build an if node from IfStatement
+     */
+    private buildIfNode;
+    /**
+     * Build a for node from ForStatement/ForInStatement/ForOfStatement
+     */
+    private buildForNode;
+    /**
+     * Build a while node from WhileStatement
+     */
+    private buildWhileNode;
+    /**
+     * Build a try-catch node from TryStatement
+     */
+    private buildTryCatchNode;
+    /**
+     * Build a code block node from a statement
+     */
+    private buildCodeBlockNode;
+    /**
+     * Find a bubble in an expression by checking if it matches any parsed bubble
+     */
+    private findBubbleInExpression;
+    /**
+     * Extract the NewExpression from an expression, handling await, .action(), etc.
+     */
+    private extractNewExpression;
+    /**
+     * Build a variable declaration node from a VariableDeclaration statement
+     */
+    private buildVariableDeclarationNode;
+    /**
+     * Build a return node from a ReturnStatement
+     */
+    private buildReturnNode;
+    /**
+     * Detect if an expression is Promise.all([...]) or Promise.all(variable)
+     */
+    private detectPromiseAll;
+    /**
+     * Detect if an expression is a function call
+     */
+    private detectFunctionCall;
+    /**
+     * Find a method definition in the class by name
+     */
+    private findMethodDefinition;
+    /**
+     * Check if a workflow node tree contains any bubbles (recursively)
+     */
+    private containsBubbles;
+    /**
+     * Build a function call node from a function call expression
+     */
+    private buildFunctionCallNode;
+    /**
+     * Extract the body of a callback function (arrow or regular function expression)
+     * Handles both block statements and concise arrow functions
+     */
+    private extractCallbackBody;
+    /**
+     * Find array elements from .push() calls or .map() callbacks
+     * Handles both patterns:
+     * - .push(): array.push(item1, item2, ...)
+     * - .map(): const promises = items.map(item => this.processItem(item))
+     */
+    private findArrayElements;
+    /**
+     * Extract expression from callback function
+     */
+    private extractCallbackExpression;
+    /**
+     * Get elements from source array (literal or variable)
+     */
+    private getSourceArrayElements;
+    /**
+     * Find all return statements in a block statement
+     */
+    private findReturnStatements;
+    /**
+     * Build a parallel execution node from Promise.all()
+     */
+    private buildParallelExecutionNode;
+    /**
+     * Get the invocation index for a method call.
+     * If the same call expression (identified by its AST range) has been processed before,
+     * return the same index to avoid double-counting.
+     *
+     * @param methodName - The name of the method being called
+     * @param callExprStartOffset - Optional start offset of the CallExpression in the source.
+     *                              Used to deduplicate when the same call is processed multiple times
+     *                              (e.g., .map() callback processing vs Promise.all resolution)
+     */
+    private getNextInvocationIndex;
+    private cloneWorkflowNodesForInvocation;
+    private cloneWorkflowNodeForInvocation;
+    private ensureClonedBubbleForInvocation;
+    private cloneBubbleForInvocation;
+    private cloneDependencyGraphNodeForInvocation;
+    /**
+     * Extract condition string from a test expression
+     */
+    private extractConditionString;
+    /**
+     * Extract location from a node
+     */
+    private extractLocation;
 }
 //# sourceMappingURL=BubbleParser.d.ts.map

package/dist/extraction/BubbleParser.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"BubbleParser.d.ts","sourceRoot":"","sources":["../../src/extraction/BubbleParser.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,sCAAsC,CAAC;AACrE,OAAO,KAAK,EAAS,YAAY,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EACL,aAAa,EAEd,MAAM,wBAAwB,CAAC;AAEhC,OAAO,KAAK,EACV,oBAAoB,EAMrB,MAAM,2BAA2B,CAAC;AAInC,qBAAa,YAAY;IACvB,OAAO,CAAC,YAAY,CAAS;gBAEjB,YAAY,EAAE,MAAM;IAGhC;;OAEG;IACH,mBAAmB,CACjB,aAAa,EAAE,aAAa,EAC5B,GAAG,EAAE,QAAQ,CAAC,OAAO,EACrB,YAAY,EAAE,YAAY,GACzB;QACD,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;QAC9C,oBAAoB,EAAE;YACpB,SAAS,EAAE,MAAM,CAAC;YAClB,OAAO,EAAE,MAAM,CAAC;YAChB,mBAAmB,EAAE,MAAM,CAAC;YAC5B,aAAa,EAAE,MAAM,CAAC;SACvB,GAAG,IAAI,CAAC;KACV;IAiFD,OAAO,CAAC,yBAAyB;IAkFjC,OAAO,CAAC,oBAAoB;IAwK5B,OAAO,CAAC,mBAAmB;IAW3B;;;;OAIG;IACI,oBAAoB,CACzB,GAAG,EAAE,QAAQ,CAAC,OAAO,GACpB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI;IAsDjC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IA6D9B,OAAO,CAAC,iBAAiB;IAezB,+EAA+E;IAC/E,OAAO,CAAC,gCAAgC;IAwDxC,OAAO,CAAC,2BAA2B;IAkBnC,sEAAsE;IACtE,OAAO,CAAC,+BAA+B;IAwDvC,2DAA2D;IAC3D,OAAO,CAAC,kBAAkB;IAwE1B,OAAO,CAAC,wBAAwB;IAMhC,OAAO,CAAC,sBAAsB;IA2B9B,OAAO,CAAC,gBAAgB;IA0DxB,kEAAkE;IAClE,OAAO,CAAC,qBAAqB;IA8B7B;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAuHhC;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAgC/B;;OAEG;IACH,OAAO,CAAC,SAAS;IA+IjB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA8B7B;;OAEG;IACH,OAAO,CAAC,iCAAiC;IA8BzC;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAwB/B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAsD5B;;OAEG;IACH,OAAO,CAAC,2BAA2B;IA6CnC;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAgHhC;;OAEG;IACH,OAAO,CAAC,qBAAqB;CAqF9B"}
+{"version":3,"file":"BubbleParser.d.ts","sourceRoot":"","sources":["../../src/extraction/BubbleParser.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,sCAAsC,CAAC;AAErE,OAAO,KAAK,EAAS,YAAY,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EACL,aAAa,EAEd,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAElE,OAAO,KAAK,EACV,oBAAoB,EAOpB,cAAc,EASf,MAAM,2BAA2B,CAAC;AAuBnC,qBAAa,YAAY;IACvB,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,SAAS,CAAiC;IAClD,OAAO,CAAC,0BAA0B,CAAkC;IACpE,OAAO,CAAC,0BAA0B,CACtB;IACZ;;;;;;OAMG;IACH,OAAO,CAAC,wBAAwB,CAAkC;IAClE,sEAAsE;IACtE,OAAO,CAAC,eAAe,CAA4B;gBAEvC,YAAY,EAAE,MAAM;IAGhC;;OAEG;IACH,mBAAmB,CACjB,aAAa,EAAE,aAAa,EAC5B,GAAG,EAAE,QAAQ,CAAC,OAAO,EACrB,YAAY,EAAE,YAAY,GACzB;QACD,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;QAC9C,QAAQ,EAAE,cAAc,CAAC;QACzB,uBAAuB,EAAE,MAAM,CAC7B,MAAM,EACN;YACE,SAAS,EAAE,MAAM,CAAC;YAClB,OAAO,EAAE,MAAM,CAAC;YAChB,mBAAmB,EAAE,MAAM,CAAC;YAC5B,aAAa,EAAE,MAAM,CAAC;YACtB,eAAe,EAAE,oBAAoB,EAAE,CAAC;SACzC,CACF,CAAC;KACH;IAgMD,OAAO,CAAC,yBAAyB;IAkFjC,OAAO,CAAC,oBAAoB;IAuK5B;;;;OAIG;IACI,oBAAoB,CACzB,GAAG,EAAE,QAAQ,CAAC,OAAO,GACpB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI;IAsDjC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IA6D9B,OAAO,CAAC,iBAAiB;IAezB,+EAA+E;IAC/E,OAAO,CAAC,gCAAgC;IAwDxC,OAAO,CAAC,2BAA2B;IAkBnC,sEAAsE;IACtE,OAAO,CAAC,+BAA+B;IAwDvC,2DAA2D;IAC3D,OAAO,CAAC,kBAAkB;IAwE1B,OAAO,CAAC,wBAAwB;IAMhC,OAAO,CAAC,sBAAsB;IA0C9B,OAAO,CAAC,gBAAgB;IAiDxB,kEAAkE;IAClE,OAAO,CAAC,qBAAqB;IA+B7B;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAmD/B;;OAEG;IACH,OAAO,CAAC,sBAAsB;IA8C9B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA6V7B;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAqC7B;;OAEG;IACH,OAAO,CAAC,6BAA6B;IA8BrC;;OAEG;IACH,OAAO,CAAC,SAAS;IAiKjB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA8B7B;;OAEG;IACH,OAAO,CAAC,iCAAiC;IA8BzC;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAwB/B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAsD5B;;OAEG;IACH,OAAO,CAAC,2BAA2B;IA6CnC;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAmHhC;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAsF7B;;;OAGG;IACH,OAAO,CAAC,+BAA+B;IAgKvC;;OAEG;IACH,OAAO,CAAC,4BAA4B;IAqBpC;;QAEI;IACJ,OAAO,CAAC,qBAAqB;IAkG7B;;;;OAIG;IACH,OAAO,CAAC,mBAAmB;IAmH3B;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IA0CxB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAqEzB;;;;;OAKG;IACH,OAAO,CAAC,qBAAqB;IA4E7B;;OAEG;IACH,OAAO,CAAC,UAAU;IAmFlB;;OAEG;IACH,OAAO,CAAC,8BAA8B;IAkNtC;;OAEG;IACH,OAAO,CAAC,WAAW;IAoFnB;;OAEG;IACH,OAAO,CAAC,YAAY;IA+EpB;;OAEG;IACH,OAAO,CAAC,cAAc;IAwCtB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAiDzB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAiC1B;;OAEG;IACH,OAAO,CAAC,sBAAsB;IA0B9B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IA0B5B;;OAEG;IACH,OAAO,CAAC,4BAA4B;IAkCpC;;OAEG;IACH,OAAO,CAAC,eAAe;IAyBvB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA8CxB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAsD1B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IA8C5B;;OAEG;IACH,OAAO,CAAC,eAAe;IA2BvB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAiQ7B;;;OAGG;IACH,OAAO,CAAC,mBAAmB;IAqB3B;;;;;OAKG;IACH,OAAO,CAAC,iBAAiB;IAwFzB;;OAEG;IACH,OAAO,CAAC,yBAAyB;IAoBjC;;OAEG;IACH,OAAO,CAAC,sBAAsB;IA2D9B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAkB5B;;OAEG;IACH,OAAO,CAAC,0BAA0B;IAqHlC;;;;;;;;;OASG;IACH,OAAO,CAAC,sBAAsB;IAyB9B,OAAO,CAAC,+BAA+B;IAgBvC,OAAO,CAAC,8BAA8B;IAsDtC,OAAO,CAAC,+BAA+B;IAuBvC,OAAO,CAAC,wBAAwB;IAkGhC,OAAO,CAAC,qCAAqC;IAsB7C;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAI9B;;OAEG;IACH,OAAO,CAAC,eAAe;CAgBxB"}