@clayroach/unplugin 0.1.0-source-trace.2 → 0.1.0-source-trace.4

package/dist/esm/transformers/annotateEffects.js

@@ -0,0 +1,140 @@
+const PURE_ANNOTATION = "#__PURE__";
+/**
+ * Checks if a node already has a PURE annotation.
+ */
+function isPureAnnotated(node) {
+    const leadingComments = node.leadingComments;
+    if (!leadingComments) {
+        return false;
+    }
+    return leadingComments.some((comment) => /[@#]__PURE__/.test(comment.value));
+}
+/**
+ * Adds a PURE annotation comment to a node.
+ */
+function annotateAsPure(path) {
+    if (isPureAnnotated(path.node)) {
+        return;
+    }
+    path.addComment("leading", PURE_ANNOTATION);
+}
+/**
+ * Checks if the parent is a CallExpression or NewExpression.
+ */
+function hasCallableParent(path) {
+    const parentPath = path.parentPath;
+    if (!parentPath)
+        return false;
+    return parentPath.isCallExpression() || parentPath.isNewExpression();
+}
+/**
+ * Checks if this node is used as a callee (e.g., `foo()` where foo is the callee).
+ */
+function isUsedAsCallee(path) {
+    if (!hasCallableParent(path)) {
+        return false;
+    }
+    const parentPath = path.parentPath;
+    return parentPath.get("callee") === path;
+}
+/**
+ * Checks if this node is inside a callee chain (e.g., `foo()()` or `foo.bar()`).
+ */
+function isInCallee(path) {
+    let current = path;
+    do {
+        current = current.parentPath;
+        if (!current)
+            return false;
+        if (isUsedAsCallee(current)) {
+            return true;
+        }
+    } while (!current.isStatement() && !current.isFunction());
+    return false;
+}
+/**
+ * Checks if this expression is executed during module initialization
+ * (not inside a function that isn't immediately invoked).
+ */
+function isExecutedDuringInitialization(path) {
+    let functionParent = path.getFunctionParent();
+    while (functionParent) {
+        if (!isUsedAsCallee(functionParent)) {
+            return false;
+        }
+        functionParent = functionParent.getFunctionParent();
+    }
+    return true;
+}
+/**
+ * Checks if this expression is in an assignment context
+ * (variable declaration, assignment expression, or class).
+ */
+function isInAssignmentContext(path) {
+    const statement = path.getStatementParent();
+    if (!statement)
+        return false;
+    let currentPath = path;
+    do {
+        currentPath = currentPath.parentPath;
+        if (!currentPath)
+            return false;
+        if (currentPath.isVariableDeclaration() ||
+            currentPath.isAssignmentExpression() ||
+            currentPath.isClass()) {
+            return true;
+        }
+    } while (currentPath !== statement);
+    return false;
+}
+/**
+ * Visitor function for CallExpression and NewExpression nodes.
+ */
+function callableExpressionVisitor(path, _state) {
+    // Skip if this is used as a callee (e.g., foo()())
+    if (isUsedAsCallee(path)) {
+        return;
+    }
+    // Skip if this is inside a callee chain
+    if (isInCallee(path)) {
+        return;
+    }
+    // Skip if not executed during initialization
+    if (!isExecutedDuringInitialization(path)) {
+        return;
+    }
+    // Must be in assignment context or export default
+    const statement = path.getStatementParent();
+    if (!isInAssignmentContext(path) && !statement?.isExportDefaultDeclaration()) {
+        return;
+    }
+    annotateAsPure(path);
+}
+/**
+ * Creates a Babel visitor that adds PURE annotations to call expressions.
+ */
+export function createAnnotateEffectsVisitor(filename, _options) {
+    return {
+        Program: {
+            enter(_path, state) {
+                state.filename = filename;
+            }
+        },
+        CallExpression(path, state) {
+            callableExpressionVisitor(path, state);
+        },
+        NewExpression(path, state) {
+            callableExpressionVisitor(path, state);
+        }
+    };
+}
+/**
+ * Creates the annotate effects transformer plugin.
+ */
+export function annotateEffectsTransformer(options) {
+    return {
+        name: "effect-annotate-pure-calls",
+        visitor: createAnnotateEffectsVisitor("", options)
+    };
+}
+//# sourceMappingURL=annotateEffects.js.map

package/dist/esm/transformers/annotateEffects.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"annotateEffects.js","sourceRoot":"","sources":["../../../src/transformers/annotateEffects.ts"],"names":[],"mappings":"AAgCA,MAAM,eAAe,GAAG,WAAW,CAAA;AAEnC;;GAEG;AACH,SAAS,eAAe,CAAC,IAAY;IACnC,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC5C,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,KAAK,CAAA;IACd,CAAC;IACD,OAAO,eAAe,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAA;AAC9E,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAkD;IACxE,IAAI,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/B,OAAM;IACR,CAAC;IACD,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,eAAe,CAAC,CAAA;AAC7C,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,IAAc;IACvC,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAA;IAClC,IAAI,CAAC,UAAU;QAAE,OAAO,KAAK,CAAA;IAC7B,OAAO,UAAU,CAAC,gBAAgB,EAAE,IAAI,UAAU,CAAC,eAAe,EAAE,CAAA;AACtE,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAc;IACpC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;QAC7B,OAAO,KAAK,CAAA;IACd,CAAC;IACD,MAAM,UAAU,GAAG,IAAI,CAAC,UAA0D,CAAA;IAClF,OAAO,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAA;AAC1C,CAAC;AAED;;GAEG;AACH,SAAS,UAAU,CAAC,IAAc;IAChC,IAAI,OAAO,GAAoB,IAAI,CAAA;IACnC,GAAG,CAAC;QACF,OAAO,GAAG,OAAO,CAAC,UAAU,CAAA;QAC5B,IAAI,CAAC,OAAO;YAAE,OAAO,KAAK,CAAA;QAC1B,IAAI,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAA;QACb,CAAC;IACH,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,EAAC;IACzD,OAAO,KAAK,CAAA;AACd,CAAC;AAED;;;GAGG;AACH,SAAS,8BAA8B,CAAC,IAAc;IACpD,IAAI,cAAc,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAA;IAC7C,OAAO,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,EAAE,CAAC;YACpC,OAAO,KAAK,CAAA;QACd,CAAC;QACD,cAAc,GAAG,cAAc,CAAC,iBAAiB,EAAE,CAAA;IACrD,CAAC;IACD,OAAO,IAAI,CAAA;AACb,CAAC;AAED;;;GAGG;AACH,SAAS,qBAAqB,CAAC,IAAc;IAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAA;IAC3C,IAAI,CAAC,SAAS;QAAE,OAAO,KAAK,CAAA;IAE5B,IAAI,WAAW,GAAoB,IAAI,CAAA;IACvC,GAAG,CAAC;QACF,WAAW,GAAG,WAAW,CAAC,UAAU,CAAA;QACpC,IAAI,CAAC,WAAW;YAAE,OAAO,KAAK,CAAA;QAE9B,IACE,WAAW,CAAC,qBAAqB,EAAE;YACnC,WAAW,CAAC,sBAAsB,EAAE;YACpC,WAAW,CAAC,OAAO,EAAE,EACrB,CAAC;YACD,OAAO,IAAI,CAAA;QACb,CAAC;IACH,CAAC,QAAQ,WAAW,KAAK,SAAS,EAAC;IAEnC,OAAO,KAAK,CAAA;AACd,CAAC;AAED;;GAEG;AACH,SAAS,yBAAyB,CAChC,IAAkD,EAClD,MAAsB;IAEtB,mDAAmD;IACnD,IAAI,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACzB,OAAM;IACR,CAAC;IAED,wCAAwC;IACxC,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACrB,OAAM;IACR,CAAC;IAED,6CAA6C;IAC7C,IAAI,CAAC,8BAA8B,CAAC,IAAI,CAAC,EAAE,CAAC;QAC1C,OAAM;IACR,CAAC;IAED,kDAAkD;IAClD,MAAM,SAAS,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAA;IAC3C,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,0BAA0B,EAAE,EAAE,CAAC;QAC7E,OAAM;IACR,CAAC;IAED,cAAc,CAAC,IAAI,CAAC,CAAA;AACtB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,4BAA4B,CAC1C,QAAgB,EAChB,QAAiC;IAEjC,OAAO;QACL,OAAO,EAAE;YACP,KAAK,CAAC,KAAK,EAAE,KAAK;gBAChB,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAA;YAC3B,CAAC;SACF;QAED,cAAc,CAAC,IAAgC,EAAE,KAAK;YACpD,yBAAyB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QACxC,CAAC;QAED,aAAa,CAAC,IAA+B,EAAE,KAAK;YAClD,yBAAyB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QACxC,CAAC;KACF,CAAA;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,0BAA0B,CAAC,OAAgC;IAIzE,OAAO;QACL,IAAI,EAAE,4BAA4B;QAClC,OAAO,EAAE,4BAA4B,CAAC,EAAE,EAAE,OAAO,CAAC;KACnD,CAAA;AACH,CAAC"}

package/dist/esm/transformers/withSpanTrace.js

@@ -0,0 +1,160 @@
+import * as t from "@babel/types";
+import { createHoistingState } from "../utils/hoisting.js";
+/**
+ * Checks if a CallExpression is a withSpan call.
+ * Matches: Effect.withSpan(...), _.withSpan(...), or standalone withSpan(...)
+ */
+function isWithSpanCall(node) {
+    const callee = node.callee;
+    // Match Effect.withSpan(...) or _.withSpan(...)
+    if (callee.type === "MemberExpression" &&
+        callee.property.type === "Identifier" &&
+        callee.property.name === "withSpan") {
+        return true;
+    }
+    // Match standalone withSpan(...)
+    if (callee.type === "Identifier" && callee.name === "withSpan") {
+        return true;
+    }
+    return false;
+}
+/**
+ * Determines if this is a data-first call (effect as first arg) or data-last (name as first arg).
+ * Data-first: withSpan(effect, "name", options?)
+ * Data-last: withSpan("name", options?)
+ */
+function isDataFirstCall(node) {
+    // If first argument is a string literal, it's data-last
+    if (node.arguments.length > 0 && t.isStringLiteral(node.arguments[0])) {
+        return false;
+    }
+    // If second argument exists and is a string literal, it's data-first
+    if (node.arguments.length > 1 && t.isStringLiteral(node.arguments[1])) {
+        return true;
+    }
+    // Default to data-last pattern
+    return false;
+}
+/**
+ * Creates the attributes object with source location.
+ */
+function createSourceAttributes(filepath, line, column) {
+    return t.objectExpression([
+        t.objectProperty(t.stringLiteral("code.filepath"), t.stringLiteral(filepath)),
+        t.objectProperty(t.stringLiteral("code.lineno"), t.numericLiteral(line)),
+        t.objectProperty(t.stringLiteral("code.column"), t.numericLiteral(column))
+    ]);
+}
+/**
+ * Merges source attributes into an existing options object or creates a new one.
+ */
+function mergeOrCreateOptions(existingOptions, sourceAttrs) {
+    if (!existingOptions || t.isSpreadElement(existingOptions) || t.isArgumentPlaceholder(existingOptions)) {
+        // No existing options, create new object with attributes
+        return t.objectExpression([
+            t.objectProperty(t.identifier("attributes"), sourceAttrs)
+        ]);
+    }
+    if (t.isObjectExpression(existingOptions)) {
+        // Check if there's an existing attributes property
+        const existingAttrsIndex = existingOptions.properties.findIndex((prop) => t.isObjectProperty(prop) &&
+            ((t.isIdentifier(prop.key) && prop.key.name === "attributes") ||
+                (t.isStringLiteral(prop.key) && prop.key.value === "attributes")));
+        if (existingAttrsIndex >= 0) {
+            // Merge with existing attributes using spread
+            const existingAttrsProp = existingOptions.properties[existingAttrsIndex];
+            const mergedAttrs = t.objectExpression([
+                t.spreadElement(existingAttrsProp.value),
+                ...sourceAttrs.properties
+            ]);
+            // Clone the options and replace the attributes property
+            const newProperties = [...existingOptions.properties];
+            newProperties[existingAttrsIndex] = t.objectProperty(t.identifier("attributes"), mergedAttrs);
+            return t.objectExpression(newProperties);
+        }
+        else {
+            // Add new attributes property to existing object
+            return t.objectExpression([
+                ...existingOptions.properties,
+                t.objectProperty(t.identifier("attributes"), sourceAttrs)
+            ]);
+        }
+    }
+    // If it's a variable reference, spread it and add attributes
+    return t.objectExpression([
+        t.spreadElement(existingOptions),
+        t.objectProperty(t.identifier("attributes"), sourceAttrs)
+    ]);
+}
+/**
+ * Creates a Babel visitor that injects source location attributes into Effect.withSpan calls.
+ */
+export function createWithSpanTraceVisitor(filename, _options) {
+    return {
+        Program: {
+            enter(_path, state) {
+                state.filename = filename;
+                state.hoisting = createHoistingState();
+            },
+            exit(path, state) {
+                // Prepend all hoisted statements to the program body
+                if (state.hoisting.statements.length > 0) {
+                    path.unshiftContainer("body", state.hoisting.statements);
+                }
+            }
+        },
+        CallExpression(path, state) {
+            const node = path.node;
+            if (!isWithSpanCall(node)) {
+                return;
+            }
+            // Get source location
+            const loc = node.loc;
+            if (!loc)
+                return;
+            const line = loc.start.line;
+            const column = loc.start.column;
+            // Create source attributes
+            const sourceAttrs = createSourceAttributes(state.filename, line, column);
+            const isDataFirst = isDataFirstCall(node);
+            if (isDataFirst) {
+                // Data-first: withSpan(effect, "name", options?)
+                // Options is at index 2
+                const optionsArg = node.arguments[2];
+                const newOptions = mergeOrCreateOptions(optionsArg, sourceAttrs);
+                if (node.arguments.length >= 3) {
+                    // Replace existing options
+                    node.arguments[2] = newOptions;
+                }
+                else {
+                    // Add options as third argument
+                    node.arguments.push(newOptions);
+                }
+            }
+            else {
+                // Data-last: withSpan("name", options?)
+                // Options is at index 1
+                const optionsArg = node.arguments[1];
+                const newOptions = mergeOrCreateOptions(optionsArg, sourceAttrs);
+                if (node.arguments.length >= 2) {
+                    // Replace existing options
+                    node.arguments[1] = newOptions;
+                }
+                else {
+                    // Add options as second argument
+                    node.arguments.push(newOptions);
+                }
+            }
+        }
+    };
+}
+/**
+ * Creates the withSpan trace transformer plugin.
+ */
+export function withSpanTraceTransformer(options) {
+    return {
+        name: "effect-withspan-trace",
+        visitor: createWithSpanTraceVisitor("", options)
+    };
+}
+//# sourceMappingURL=withSpanTrace.js.map

package/dist/esm/transformers/withSpanTrace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"withSpanTrace.js","sourceRoot":"","sources":["../../../src/transformers/withSpanTrace.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,CAAC,MAAM,cAAc,CAAA;AACjC,OAAO,EACL,mBAAmB,EAEpB,MAAM,sBAAsB,CAAA;AAoB7B;;;GAGG;AACH,SAAS,cAAc,CAAC,IAAsB;IAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;IAE1B,gDAAgD;IAChD,IACE,MAAM,CAAC,IAAI,KAAK,kBAAkB;QAClC,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,YAAY;QACrC,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,UAAU,EACnC,CAAC;QACD,OAAO,IAAI,CAAA;IACb,CAAC;IAED,iCAAiC;IACjC,IAAI,MAAM,CAAC,IAAI,KAAK,YAAY,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAC/D,OAAO,IAAI,CAAA;IACb,CAAC;IAED,OAAO,KAAK,CAAA;AACd,CAAC;AAED;;;;GAIG;AACH,SAAS,eAAe,CAAC,IAAsB;IAC7C,wDAAwD;IACxD,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,OAAO,KAAK,CAAA;IACd,CAAC;IACD,qEAAqE;IACrE,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,OAAO,IAAI,CAAA;IACb,CAAC;IACD,+BAA+B;IAC/B,OAAO,KAAK,CAAA;AACd,CAAC;AAED;;GAEG;AACH,SAAS,sBAAsB,CAC7B,QAAgB,EAChB,IAAY,EACZ,MAAc;IAEd,OAAO,CAAC,CAAC,gBAAgB,CAAC;QACxB,CAAC,CAAC,cAAc,CACd,CAAC,CAAC,aAAa,CAAC,eAAe,CAAC,EAChC,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,CAC1B;QACD,CAAC,CAAC,cAAc,CACd,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,EAC9B,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,CACvB;QACD,CAAC,CAAC,cAAc,CACd,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,EAC9B,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CACzB;KACF,CAAC,CAAA;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB,CAC3B,eAAmF,EACnF,WAA+B;IAE/B,IAAI,CAAC,eAAe,IAAI,CAAC,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,eAAe,CAAC,EAAE,CAAC;QACvG,yDAAyD;QACzD,OAAO,CAAC,CAAC,gBAAgB,CAAC;YACxB,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,WAAW,CAAC;SAC1D,CAAC,CAAA;IACJ,CAAC;IAED,IAAI,CAAC,CAAC,kBAAkB,CAAC,eAAe,CAAC,EAAE,CAAC;QAC1C,mDAAmD;QACnD,MAAM,kBAAkB,GAAG,eAAe,CAAC,UAAU,CAAC,SAAS,CAC7D,CAAC,IAAI,EAAE,EAAE,CACP,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC;YACxB,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,YAAY,CAAC;gBAC3D,CAAC,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,KAAK,YAAY,CAAC,CAAC,CACtE,CAAA;QAED,IAAI,kBAAkB,IAAI,CAAC,EAAE,CAAC;YAC5B,8CAA8C;YAC9C,MAAM,iBAAiB,GAAG,eAAe,CAAC,UAAU,CAAC,kBAAkB,CAAqB,CAAA;YAC5F,MAAM,WAAW,GAAG,CAAC,CAAC,gBAAgB,CAAC;gBACrC,CAAC,CAAC,aAAa,CAAC,iBAAiB,CAAC,KAAqB,CAAC;gBACxD,GAAG,WAAW,CAAC,UAAU;aAC1B,CAAC,CAAA;YAEF,wDAAwD;YACxD,MAAM,aAAa,GAAG,CAAC,GAAG,eAAe,CAAC,UAAU,CAAC,CAAA;YACrD,aAAa,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,cAAc,CAClD,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,EAC1B,WAAW,CACZ,CAAA;YACD,OAAO,CAAC,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAA;QAC1C,CAAC;aAAM,CAAC;YACN,iDAAiD;YACjD,OAAO,CAAC,CAAC,gBAAgB,CAAC;gBACxB,GAAG,eAAe,CAAC,UAAU;gBAC7B,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,WAAW,CAAC;aAC1D,CAAC,CAAA;QACJ,CAAC;IACH,CAAC;IAED,6DAA6D;IAC7D,OAAO,CAAC,CAAC,gBAAgB,CAAC;QACxB,CAAC,CAAC,aAAa,CAAC,eAAe,CAAC;QAChC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,WAAW,CAAC;KAC1D,CAAC,CAAA;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACxC,QAAgB,EAChB,QAA+B;IAE/B,OAAO;QACL,OAAO,EAAE;YACP,KAAK,CAAC,KAAK,EAAE,KAAK;gBAChB,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAA;gBACzB,KAAK,CAAC,QAAQ,GAAG,mBAAmB,EAAE,CAAA;YACxC,CAAC;YACD,IAAI,CAAC,IAAI,EAAE,KAAK;gBACd,qDAAqD;gBACrD,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAA;gBAC1D,CAAC;YACH,CAAC;SACF;QAED,cAAc,CAAC,IAAgC,EAAE,KAAK;YACpD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;YAEtB,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC1B,OAAM;YACR,CAAC;YAED,sBAAsB;YACtB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAA;YACpB,IAAI,CAAC,GAAG;gBAAE,OAAM;YAEhB,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,CAAA;YAC3B,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAA;YAE/B,2BAA2B;YAC3B,MAAM,WAAW,GAAG,sBAAsB,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAA;YAExE,MAAM,WAAW,GAAG,eAAe,CAAC,IAAI,CAAC,CAAA;YAEzC,IAAI,WAAW,EAAE,CAAC;gBAChB,iDAAiD;gBACjD,wBAAwB;gBACxB,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;gBACpC,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAU,EAAE,WAAW,CAAC,CAAA;gBAEhE,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;oBAC/B,2BAA2B;oBAC3B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,UAAU,CAAA;gBAChC,CAAC;qBAAM,CAAC;oBACN,gCAAgC;oBAChC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;gBACjC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,wCAAwC;gBACxC,wBAAwB;gBACxB,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;gBACpC,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAU,EAAE,WAAW,CAAC,CAAA;gBAEhE,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;oBAC/B,2BAA2B;oBAC3B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,UAAU,CAAA;gBAChC,CAAC;qBAAM,CAAC;oBACN,iCAAiC;oBACjC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;gBACjC,CAAC;YACH,CAAC;QACH,CAAC;KACF,CAAA;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAA8B;IAIrE,OAAO;QACL,IAAI,EAAE,uBAAuB;QAC7B,OAAO,EAAE,0BAA0B,CAAC,EAAE,EAAE,OAAO,CAAC;KACjD,CAAA;AACH,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@clayroach/unplugin",
-  "version": "0.1.0-source-trace.2",
+  "version": "0.1.0-source-trace.4",
   "description": "Universal bundler plugin for Effect transformations",
   "license": "MIT",
   "repository": {
@@ -18,7 +18,7 @@
     "@babel/generator": "^7.26.0"
   },
   "peerDependencies": {
-    "@clayroach/effect": "^3.19.14-source-trace.2"
+    "@clayroach/effect": "^3.19.14-source-trace.3"
   },
   "publishConfig": {
     "provenance": true

package/src/index.ts

@@ -28,7 +28,9 @@ import type { Scope, TraverseOptions, Visitor } from "@babel/traverse"
 import _traverse from "@babel/traverse"
 import type { Node } from "@babel/types"
 import { createUnplugin, type UnpluginFactory, type UnpluginInstance } from "unplugin"
+import { createAnnotateEffectsVisitor } from "./transformers/annotateEffects.js"
 import { createSourceTraceVisitor } from "./transformers/sourceTrace.js"
+import { createWithSpanTraceVisitor } from "./transformers/withSpanTrace.js"
 import type { HoistingState } from "./utils/hoisting.js"
 
 // Define function types for Babel packages (they export differently in ESM vs CJS)
@@ -76,6 +78,14 @@ export interface EffectPluginOptions {
    */
   readonly sourceTrace?: boolean
 
+  /**
+   * Enable source location attributes injection into Effect.withSpan() calls.
+   * When enabled, spans will include code.filepath, code.lineno, and code.column attributes.
+   *
+   * @default true
+   */
+  readonly spanTrace?: boolean
+
   /**
    * Enable @__PURE__ annotations for tree-shaking.
    *
@@ -142,10 +152,11 @@ function shouldTransform(
  */
 const unpluginFactory: UnpluginFactory<EffectPluginOptions | undefined> = (options = {}) => {
   const {
-    annotateEffects: _annotateEffects = false,
+    annotateEffects = false,
     exclude = DEFAULT_EXCLUDE,
     include = DEFAULT_INCLUDE,
-    sourceTrace = true
+    sourceTrace = true,
+    spanTrace = true
   } = options
 
   return {
@@ -169,8 +180,18 @@ const unpluginFactory: UnpluginFactory<EffectPluginOptions | undefined> = (optio
         visitors.push(createSourceTraceVisitor(id))
       }
 
+      if (spanTrace) {
+        visitors.push(createWithSpanTraceVisitor(id))
+      }
+
+      // Run annotateEffects separately (it has a simpler state)
+      if (annotateEffects) {
+        const annotateState = { filename: id }
+        traverse(ast, createAnnotateEffectsVisitor(id), undefined, annotateState)
+      }
+
       // Combine visitors
-      if (visitors.length === 0) {
+      if (visitors.length === 0 && !annotateEffects) {
         return null
       }
 

package/src/transformers/annotateEffects.ts

@@ -0,0 +1,197 @@
+/**
+ * Pure call annotation transformer.
+ *
+ * This transformer adds `#__PURE__` comments to call expressions for better tree-shaking.
+ * It replicates the behavior of `babel-plugin-annotate-pure-calls` but integrated into
+ * the Effect unplugin.
+ *
+ * The `#__PURE__` annotation tells bundlers (Webpack, Rollup, esbuild) that a function call
+ * has no side effects and can be safely removed if the result is unused.
+ *
+ * @since 0.1.0
+ */
+import type { NodePath, Visitor } from "@babel/traverse"
+import * as t from "@babel/types"
+
+/**
+ * Options for the annotate effects transformer.
+ */
+export interface AnnotateEffectsOptions {
+  /**
+   * Filter function to determine if a file should be transformed.
+   */
+  readonly filter?: (filename: string) => boolean
+}
+
+/**
+ * State passed through the transformer.
+ */
+interface TransformState {
+  filename: string
+}
+
+const PURE_ANNOTATION = "#__PURE__"
+
+/**
+ * Checks if a node already has a PURE annotation.
+ */
+function isPureAnnotated(node: t.Node): boolean {
+  const leadingComments = node.leadingComments
+  if (!leadingComments) {
+    return false
+  }
+  return leadingComments.some((comment) => /[@#]__PURE__/.test(comment.value))
+}
+
+/**
+ * Adds a PURE annotation comment to a node.
+ */
+function annotateAsPure(path: NodePath<t.CallExpression | t.NewExpression>): void {
+  if (isPureAnnotated(path.node)) {
+    return
+  }
+  path.addComment("leading", PURE_ANNOTATION)
+}
+
+/**
+ * Checks if the parent is a CallExpression or NewExpression.
+ */
+function hasCallableParent(path: NodePath): boolean {
+  const parentPath = path.parentPath
+  if (!parentPath) return false
+  return parentPath.isCallExpression() || parentPath.isNewExpression()
+}
+
+/**
+ * Checks if this node is used as a callee (e.g., `foo()` where foo is the callee).
+ */
+function isUsedAsCallee(path: NodePath): boolean {
+  if (!hasCallableParent(path)) {
+    return false
+  }
+  const parentPath = path.parentPath as NodePath<t.CallExpression | t.NewExpression>
+  return parentPath.get("callee") === path
+}
+
+/**
+ * Checks if this node is inside a callee chain (e.g., `foo()()` or `foo.bar()`).
+ */
+function isInCallee(path: NodePath): boolean {
+  let current: NodePath | null = path
+  do {
+    current = current.parentPath
+    if (!current) return false
+    if (isUsedAsCallee(current)) {
+      return true
+    }
+  } while (!current.isStatement() && !current.isFunction())
+  return false
+}
+
+/**
+ * Checks if this expression is executed during module initialization
+ * (not inside a function that isn't immediately invoked).
+ */
+function isExecutedDuringInitialization(path: NodePath): boolean {
+  let functionParent = path.getFunctionParent()
+  while (functionParent) {
+    if (!isUsedAsCallee(functionParent)) {
+      return false
+    }
+    functionParent = functionParent.getFunctionParent()
+  }
+  return true
+}
+
+/**
+ * Checks if this expression is in an assignment context
+ * (variable declaration, assignment expression, or class).
+ */
+function isInAssignmentContext(path: NodePath): boolean {
+  const statement = path.getStatementParent()
+  if (!statement) return false
+
+  let currentPath: NodePath | null = path
+  do {
+    currentPath = currentPath.parentPath
+    if (!currentPath) return false
+
+    if (
+      currentPath.isVariableDeclaration() ||
+      currentPath.isAssignmentExpression() ||
+      currentPath.isClass()
+    ) {
+      return true
+    }
+  } while (currentPath !== statement)
+
+  return false
+}
+
+/**
+ * Visitor function for CallExpression and NewExpression nodes.
+ */
+function callableExpressionVisitor(
+  path: NodePath<t.CallExpression | t.NewExpression>,
+  _state: TransformState
+): void {
+  // Skip if this is used as a callee (e.g., foo()())
+  if (isUsedAsCallee(path)) {
+    return
+  }
+
+  // Skip if this is inside a callee chain
+  if (isInCallee(path)) {
+    return
+  }
+
+  // Skip if not executed during initialization
+  if (!isExecutedDuringInitialization(path)) {
+    return
+  }
+
+  // Must be in assignment context or export default
+  const statement = path.getStatementParent()
+  if (!isInAssignmentContext(path) && !statement?.isExportDefaultDeclaration()) {
+    return
+  }
+
+  annotateAsPure(path)
+}
+
+/**
+ * Creates a Babel visitor that adds PURE annotations to call expressions.
+ */
+export function createAnnotateEffectsVisitor(
+  filename: string,
+  _options?: AnnotateEffectsOptions
+): Visitor<TransformState> {
+  return {
+    Program: {
+      enter(_path, state) {
+        state.filename = filename
+      }
+    },
+
+    CallExpression(path: NodePath<t.CallExpression>, state) {
+      callableExpressionVisitor(path, state)
+    },
+
+    NewExpression(path: NodePath<t.NewExpression>, state) {
+      callableExpressionVisitor(path, state)
+    }
+  }
+}
+
+/**
+ * Creates the annotate effects transformer plugin.
+ */
+export function annotateEffectsTransformer(options?: AnnotateEffectsOptions): {
+  visitor: Visitor<TransformState>
+  name: string
+} {
+  return {
+    name: "effect-annotate-pure-calls",
+    visitor: createAnnotateEffectsVisitor("", options)
+  }
+}