attocode 0.1.0

package/dist/src/integrations/cancellation.js

@@ -0,0 +1,339 @@
+/**
+ * Cancellation Token Integration
+ *
+ * Provides graceful cancellation support for long-running agent operations.
+ * Based on the .NET CancellationToken pattern, adapted from tricks/cancellation.ts.
+ *
+ * Usage:
+ *   const cts = createCancellationTokenSource();
+ *   agent.run(task, { cancellationToken: cts.token });
+ *   // Later: cts.cancel('User requested cancellation');
+ */
+// =============================================================================
+// IMPLEMENTATION
+// =============================================================================
+class CancellationTokenImpl {
+    _cancelled = false;
+    _reason;
+    _callbacks = new Set();
+    _promise;
+    _resolve;
+    constructor() {
+        this._promise = new Promise((r) => {
+            this._resolve = r;
+        });
+    }
+    get isCancellationRequested() {
+        return this._cancelled;
+    }
+    get cancellationReason() {
+        return this._reason;
+    }
+    get onCancellationRequested() {
+        return this._promise;
+    }
+    register(callback) {
+        if (this._cancelled) {
+            // Already cancelled, invoke immediately
+            try {
+                callback(this._reason);
+            }
+            catch {
+                // Ignore callback errors
+            }
+            return { dispose: () => { } };
+        }
+        this._callbacks.add(callback);
+        return { dispose: () => this._callbacks.delete(callback) };
+    }
+    throwIfCancellationRequested() {
+        if (this._cancelled) {
+            throw new CancellationError(this._reason);
+        }
+    }
+    /** @internal */
+    _cancel(reason) {
+        if (this._cancelled)
+            return;
+        this._cancelled = true;
+        this._reason = reason;
+        this._resolve(reason);
+        for (const cb of this._callbacks) {
+            try {
+                cb(reason);
+            }
+            catch {
+                // Ignore callback errors to ensure all callbacks run
+            }
+        }
+        this._callbacks.clear();
+    }
+}
+class CancellationTokenSourceImpl {
+    _token = new CancellationTokenImpl();
+    _timeoutId;
+    _disposed = false;
+    get token() {
+        return this._token;
+    }
+    get isCancellationRequested() {
+        return this._token.isCancellationRequested;
+    }
+    cancel(reason) {
+        if (this._disposed)
+            return;
+        this._token._cancel(reason);
+    }
+    cancelAfter(ms) {
+        if (this._disposed || this._token.isCancellationRequested)
+            return this;
+        this._timeoutId = setTimeout(() => this.cancel('Operation timed out'), ms);
+        return this;
+    }
+    dispose() {
+        if (this._disposed)
+            return;
+        this._disposed = true;
+        if (this._timeoutId) {
+            clearTimeout(this._timeoutId);
+            this._timeoutId = undefined;
+        }
+        // Don't cancel on dispose - the operation completed successfully
+    }
+}
+// =============================================================================
+// ERROR
+// =============================================================================
+/**
+ * Error thrown when an operation is cancelled.
+ */
+export class CancellationError extends Error {
+    isCancellation = true;
+    reason;
+    constructor(reason) {
+        super(reason || 'Operation cancelled');
+        this.name = 'CancellationError';
+        this.reason = reason;
+    }
+}
+/**
+ * Check if an error is a cancellation error.
+ */
+export function isCancellationError(error) {
+    return (error instanceof CancellationError ||
+        (error instanceof Error && 'isCancellation' in error && error.isCancellation === true));
+}
+// =============================================================================
+// FACTORY FUNCTIONS
+// =============================================================================
+/**
+ * Create a new cancellation token source.
+ */
+export function createCancellationTokenSource() {
+    return new CancellationTokenSourceImpl();
+}
+/**
+ * Create a token source that auto-cancels after timeout.
+ */
+export function createTimeoutToken(ms) {
+    return createCancellationTokenSource().cancelAfter(ms);
+}
+/**
+ * Create a linked token that cancels when any source cancels.
+ */
+export function createLinkedToken(...sources) {
+    const linked = createCancellationTokenSource();
+    for (const source of sources) {
+        if (source.isCancellationRequested) {
+            linked.cancel(source.token.cancellationReason);
+            break;
+        }
+        source.token.register((reason) => linked.cancel(reason));
+    }
+    return linked;
+}
+/**
+ * Create a "none" token that is never cancelled.
+ * Useful as a default when no cancellation is needed.
+ */
+export const CancellationToken = {
+    None: {
+        isCancellationRequested: false,
+        cancellationReason: undefined,
+        onCancellationRequested: new Promise(() => { }),
+        register: () => ({ dispose: () => { } }),
+        throwIfCancellationRequested: () => { },
+    },
+};
+// =============================================================================
+// UTILITIES
+// =============================================================================
+/**
+ * Run a function with cancellation support.
+ */
+export async function withCancellation(fn, options = {}) {
+    const { token, timeout } = options;
+    let timeoutSource;
+    let effectiveToken;
+    if (timeout && !token) {
+        timeoutSource = createTimeoutToken(timeout);
+        effectiveToken = timeoutSource.token;
+    }
+    else if (timeout && token) {
+        // Link user token with timeout
+        timeoutSource = createTimeoutToken(timeout);
+        const linkedSource = createLinkedToken({ token, isCancellationRequested: token.isCancellationRequested, cancel: () => { }, cancelAfter: () => ({ token, isCancellationRequested: false, cancel: () => { }, cancelAfter: () => ({}), dispose: () => { } }), dispose: () => { } }, timeoutSource);
+        effectiveToken = linkedSource.token;
+    }
+    else {
+        effectiveToken = token || CancellationToken.None;
+    }
+    try {
+        effectiveToken.throwIfCancellationRequested();
+        return await fn(effectiveToken);
+    }
+    finally {
+        timeoutSource?.dispose();
+    }
+}
+/**
+ * Sleep with cancellation support.
+ */
+export function sleep(ms, token) {
+    return new Promise((resolve, reject) => {
+        if (token?.isCancellationRequested) {
+            reject(new CancellationError(token.cancellationReason));
+            return;
+        }
+        const id = setTimeout(resolve, ms);
+        const registration = token?.register((reason) => {
+            clearTimeout(id);
+            reject(new CancellationError(reason));
+        });
+        // Cleanup registration after timeout completes
+        if (registration) {
+            setTimeout(() => registration.dispose(), ms + 1);
+        }
+    });
+}
+/**
+ * Race a promise against cancellation.
+ */
+export function race(promise, token) {
+    if (token.isCancellationRequested) {
+        return Promise.reject(new CancellationError(token.cancellationReason));
+    }
+    return Promise.race([
+        promise,
+        token.onCancellationRequested.then((reason) => {
+            throw new CancellationError(reason);
+        }),
+    ]);
+}
+/**
+ * Create an AbortSignal from a CancellationToken.
+ * Useful for integrating with fetch and other AbortSignal-based APIs.
+ */
+export function toAbortSignal(token) {
+    const controller = new AbortController();
+    if (token.isCancellationRequested) {
+        controller.abort(token.cancellationReason);
+    }
+    else {
+        token.register((reason) => controller.abort(reason));
+    }
+    return controller.signal;
+}
+// =============================================================================
+// CANCELLATION MANAGER (for agent integration)
+// =============================================================================
+/**
+ * Manager that handles cancellation for agent operations.
+ * Provides a clean interface for the ProductionAgent to use.
+ */
+export class CancellationManager {
+    currentSource = null;
+    eventListeners = new Set();
+    /**
+     * Create a new cancellation context for an operation.
+     * Disposes the previous context if one exists.
+     */
+    createContext(timeout) {
+        // Dispose previous context
+        this.disposeContext();
+        // Create new source
+        this.currentSource = timeout ? createTimeoutToken(timeout) : createCancellationTokenSource();
+        return this.currentSource.token;
+    }
+    /**
+     * Get the current cancellation token.
+     */
+    get token() {
+        return this.currentSource?.token || CancellationToken.None;
+    }
+    /**
+     * Check if cancellation has been requested.
+     */
+    get isCancelled() {
+        return this.currentSource?.isCancellationRequested ?? false;
+    }
+    /**
+     * Request cancellation of the current operation.
+     */
+    cancel(reason) {
+        if (!this.currentSource)
+            return;
+        const effectiveReason = reason || 'User requested cancellation';
+        this.emit({ type: 'cancellation.requested', reason: effectiveReason, timestamp: new Date() });
+        this.currentSource.cancel(effectiveReason);
+    }
+    /**
+     * Dispose the current cancellation context.
+     */
+    disposeContext() {
+        if (this.currentSource) {
+            this.currentSource.dispose();
+            this.currentSource = null;
+        }
+    }
+    /**
+     * Check if there's an active cancellation context.
+     */
+    hasActiveContext() {
+        return this.currentSource !== null;
+    }
+    /**
+     * Subscribe to cancellation events.
+     */
+    subscribe(listener) {
+        this.eventListeners.add(listener);
+        return () => this.eventListeners.delete(listener);
+    }
+    /**
+     * Emit a cancellation event.
+     */
+    emit(event) {
+        for (const listener of this.eventListeners) {
+            try {
+                listener(event);
+            }
+            catch {
+                // Ignore listener errors
+            }
+        }
+    }
+    /**
+     * Cleanup all resources.
+     */
+    cleanup() {
+        this.disposeContext();
+        this.eventListeners.clear();
+    }
+}
+/**
+ * Create a cancellation manager.
+ */
+export function createCancellationManager() {
+    return new CancellationManager();
+}
+//# sourceMappingURL=cancellation.js.map

package/dist/src/integrations/cancellation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cancellation.js","sourceRoot":"","sources":["../../../src/integrations/cancellation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAyDH,gFAAgF;AAChF,iBAAiB;AACjB,gFAAgF;AAEhF,MAAM,qBAAqB;IACjB,UAAU,GAAG,KAAK,CAAC;IACnB,OAAO,CAAU;IACjB,UAAU,GAAG,IAAI,GAAG,EAA6B,CAAC;IAClD,QAAQ,CAAyB;IACjC,QAAQ,CAA6B;IAE7C;QACE,IAAI,CAAC,QAAQ,GAAG,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YAChC,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,uBAAuB;QACzB,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,IAAI,kBAAkB;QACpB,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,uBAAuB;QACzB,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,QAAQ,CAAC,QAAmC;QAC1C,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,wCAAwC;YACxC,IAAI,CAAC;gBACH,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACzB,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC;QAC/B,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9B,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;IAC7D,CAAC;IAED,4BAA4B;QAC1B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IAED,gBAAgB;IAChB,OAAO,CAAC,MAAe;QACrB,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QAC5B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACtB,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,EAAE,CAAC,MAAM,CAAC,CAAC;YACb,CAAC;YAAC,MAAM,CAAC;gBACP,qDAAqD;YACvD,CAAC;QACH,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;CACF;AAED,MAAM,2BAA2B;IACvB,MAAM,GAAG,IAAI,qBAAqB,EAAE,CAAC;IACrC,UAAU,CAAiC;IAC3C,SAAS,GAAG,KAAK,CAAC;IAE1B,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,IAAI,uBAAuB;QACzB,OAAO,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC;IAC7C,CAAC;IAED,MAAM,CAAC,MAAe;QACpB,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,WAAW,CAAC,EAAU;QACpB,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,uBAAuB;YAAE,OAAO,IAAI,CAAC;QACvE,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,EAAE,EAAE,CAAC,CAAC;QAC3E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO;QACL,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC9B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC9B,CAAC;QACD,iEAAiE;IACnE,CAAC;CACF;AAED,gFAAgF;AAChF,QAAQ;AACR,gFAAgF;AAEhF;;GAEG;AACH,MAAM,OAAO,iBAAkB,SAAQ,KAAK;IACjC,cAAc,GAAG,IAAI,CAAC;IACtB,MAAM,CAAU;IAEzB,YAAY,MAAe;QACzB,KAAK,CAAC,MAAM,IAAI,qBAAqB,CAAC,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,mBAAmB,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAc;IAChD,OAAO,CACL,KAAK,YAAY,iBAAiB;QAClC,CAAC,KAAK,YAAY,KAAK,IAAI,gBAAgB,IAAI,KAAK,IAAK,KAA2B,CAAC,cAAc,KAAK,IAAI,CAAC,CAC9G,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,oBAAoB;AACpB,gFAAgF;AAEhF;;GAEG;AACH,MAAM,UAAU,6BAA6B;IAC3C,OAAO,IAAI,2BAA2B,EAAE,CAAC;AAC3C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,EAAU;IAC3C,OAAO,6BAA6B,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;AACzD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,GAAG,OAAkC;IACrE,MAAM,MAAM,GAAG,6BAA6B,EAAE,CAAC;IAE/C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,MAAM,CAAC,uBAAuB,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAC/C,MAAM;QACR,CAAC;QACD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,IAAI,EAAE;QACJ,uBAAuB,EAAE,KAAK;QAC9B,kBAAkB,EAAE,SAAS;QAC7B,uBAAuB,EAAE,IAAI,OAAO,CAAO,GAAG,EAAE,GAAE,CAAC,CAAC;QACpD,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC;QACvC,4BAA4B,EAAE,GAAG,EAAE,GAAE,CAAC;KAClB;CACvB,CAAC;AAEF,gFAAgF;AAChF,YAAY;AACZ,gFAAgF;AAEhF;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,EAA4C,EAC5C,UAA8B,EAAE;IAEhC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEnC,IAAI,aAAkD,CAAC;IACvD,IAAI,cAAiC,CAAC;IAEtC,IAAI,OAAO,IAAI,CAAC,KAAK,EAAE,CAAC;QACtB,aAAa,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAC5C,cAAc,GAAG,aAAa,CAAC,KAAK,CAAC;IACvC,CAAC;SAAM,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;QAC5B,+BAA+B;QAC/B,aAAa,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAC5C,MAAM,YAAY,GAAG,iBAAiB,CACpC,EAAE,KAAK,EAAE,uBAAuB,EAAE,KAAK,CAAC,uBAAuB,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,uBAAuB,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAA8B,CAAA,EAAE,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,EAC/P,aAAa,CACd,CAAC;QACF,cAAc,GAAG,YAAY,CAAC,KAAK,CAAC;IACtC,CAAC;SAAM,CAAC;QACN,cAAc,GAAG,KAAK,IAAI,iBAAiB,CAAC,IAAI,CAAC;IACnD,CAAC;IAED,IAAI,CAAC;QACH,cAAc,CAAC,4BAA4B,EAAE,CAAC;QAC9C,OAAO,MAAM,EAAE,CAAC,cAAc,CAAC,CAAC;IAClC,CAAC;YAAS,CAAC;QACT,aAAa,EAAE,OAAO,EAAE,CAAC;IAC3B,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,KAAK,CAAC,EAAU,EAAE,KAAyB;IACzD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,KAAK,EAAE,uBAAuB,EAAE,CAAC;YACnC,MAAM,CAAC,IAAI,iBAAiB,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,EAAE,GAAG,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACnC,MAAM,YAAY,GAAG,KAAK,EAAE,QAAQ,CAAC,CAAC,MAAM,EAAE,EAAE;YAC9C,YAAY,CAAC,EAAE,CAAC,CAAC;YACjB,MAAM,CAAC,IAAI,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,+CAA+C;QAC/C,IAAI,YAAY,EAAE,CAAC;YACjB,UAAU,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,IAAI,CAAI,OAAmB,EAAE,KAAwB;IACnE,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;QAClC,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,iBAAiB,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,OAAO,OAAO,CAAC,IAAI,CAAC;QAClB,OAAO;QACP,KAAK,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAC5C,MAAM,IAAI,iBAAiB,CAAC,MAAgB,CAAC,CAAC;QAChD,CAAC,CAAC;KACH,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,KAAwB;IACpD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IAEzC,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;QAClC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IAC7C,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,UAAU,CAAC,MAAM,CAAC;AAC3B,CAAC;AAED,gFAAgF;AAChF,+CAA+C;AAC/C,gFAAgF;AAEhF;;;GAGG;AACH,MAAM,OAAO,mBAAmB;IACtB,aAAa,GAAmC,IAAI,CAAC;IACrD,cAAc,GAAmC,IAAI,GAAG,EAAE,CAAC;IAEnE;;;OAGG;IACH,aAAa,CAAC,OAAgB;QAC5B,2BAA2B;QAC3B,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtB,oBAAoB;QACpB,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,6BAA6B,EAAE,CAAC;QAE7F,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,iBAAiB,CAAC,IAAI,CAAC;IAC7D,CAAC;IAED;;OAEG;IACH,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,aAAa,EAAE,uBAAuB,IAAI,KAAK,CAAC;IAC9D,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,MAAe;QACpB,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAO;QAEhC,MAAM,eAAe,GAAG,MAAM,IAAI,6BAA6B,CAAC;QAChE,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,MAAM,EAAE,eAAe,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;QAC9F,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,KAAK,IAAI,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,QAAmC;QAC3C,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACK,IAAI,CAAC,KAAwB;QACnC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,yBAAyB;IACvC,OAAO,IAAI,mBAAmB,EAAE,CAAC;AACnC,CAAC"}

package/dist/src/integrations/codebase-context.d.ts

@@ -0,0 +1,319 @@
+/**
+ * Codebase Context Integration
+ *
+ * Provides intelligent code selection for context management.
+ * Analyzes repository structure and selects relevant code based on:
+ * - Task relevance (semantic matching)
+ * - Architectural importance (entry points, core modules)
+ * - Token budget constraints
+ *
+ * Key features:
+ * - Budget-aware context building (select code that fits)
+ * - Importance scoring (imports, exports, modification frequency)
+ * - Incremental expansion (start minimal, expand as needed)
+ * - Semantic relevance matching (keyword + structure analysis)
+ */
+/**
+ * A chunk of code with metadata for context selection.
+ */
+export interface CodeChunk {
+    /** Unique identifier */
+    id: string;
+    /** Relative file path from repo root */
+    filePath: string;
+    /** The actual code content */
+    content: string;
+    /** Estimated token count */
+    tokenCount: number;
+    /** Importance score (0-1) */
+    importance: number;
+    /** Type of code chunk */
+    type: CodeChunkType;
+    /** Symbols defined in this chunk */
+    symbols: string[];
+    /** Symbols imported/used by this chunk */
+    dependencies: string[];
+    /** Last modified timestamp */
+    lastModified?: Date;
+}
+export type CodeChunkType = 'entry_point' | 'core_module' | 'utility' | 'types' | 'config' | 'test' | 'documentation' | 'other';
+/**
+ * Repository structure map.
+ */
+export interface RepoMap {
+    /** Root directory of the repo */
+    root: string;
+    /** All discovered code chunks */
+    chunks: Map<string, CodeChunk>;
+    /** Entry point files */
+    entryPoints: string[];
+    /** Core module directories */
+    coreModules: string[];
+    /** Dependency graph (file -> imported files) */
+    dependencyGraph: Map<string, Set<string>>;
+    /** Reverse dependency graph (file -> files that import it) */
+    reverseDependencyGraph: Map<string, Set<string>>;
+    /** Total estimated tokens */
+    totalTokens: number;
+    /** Analysis timestamp */
+    analyzedAt: Date;
+}
+/**
+ * Configuration for codebase context.
+ */
+export interface CodebaseContextConfig {
+    /** Root directory to analyze */
+    root?: string;
+    /** File patterns to include (glob) */
+    includePatterns?: string[];
+    /** File patterns to exclude (glob) */
+    excludePatterns?: string[];
+    /** Maximum file size in bytes */
+    maxFileSize?: number;
+    /** Estimate tokens per character ratio */
+    tokensPerChar?: number;
+    /** Entry point file patterns */
+    entryPointPatterns?: string[];
+    /** Core module directory patterns */
+    coreModulePatterns?: string[];
+    /** Enable dependency graph analysis */
+    analyzeDependencies?: boolean;
+    /** Cache analysis results */
+    cacheResults?: boolean;
+    /** Cache TTL in milliseconds */
+    cacheTTL?: number;
+}
+/**
+ * Options for selecting relevant code.
+ */
+export interface SelectionOptions {
+    /** Maximum token budget */
+    maxTokens: number;
+    /** Task description for relevance scoring */
+    task?: string;
+    /** Specific files to prioritize */
+    priorityFiles?: string[];
+    /** Minimum importance threshold */
+    minImportance?: number;
+    /** Include types/interfaces */
+    includeTypes?: boolean;
+    /** Include tests */
+    includeTests?: boolean;
+    /** Strategy for selection */
+    strategy?: SelectionStrategy;
+}
+export type SelectionStrategy = 'importance_first' | 'relevance_first' | 'breadth_first' | 'depth_first';
+/**
+ * Result of code selection.
+ */
+export interface SelectionResult {
+    /** Selected code chunks */
+    chunks: CodeChunk[];
+    /** Total tokens used */
+    totalTokens: number;
+    /** Budget remaining */
+    budgetRemaining: number;
+    /** Files that were excluded due to budget */
+    excluded: string[];
+    /** Selection statistics */
+    stats: {
+        filesConsidered: number;
+        filesSelected: number;
+        coveragePercent: number;
+        averageImportance: number;
+    };
+}
+/**
+ * Events emitted by the codebase context manager.
+ */
+export type CodebaseContextEvent = {
+    type: 'analysis.started';
+    root: string;
+} | {
+    type: 'analysis.completed';
+    totalFiles: number;
+    totalTokens: number;
+} | {
+    type: 'analysis.error';
+    error: Error;
+} | {
+    type: 'selection.completed';
+    selected: number;
+    budget: number;
+} | {
+    type: 'cache.hit';
+    root: string;
+} | {
+    type: 'cache.miss';
+    root: string;
+};
+export type CodebaseContextEventListener = (event: CodebaseContextEvent) => void;
+/**
+ * Manages intelligent code selection for context building.
+ *
+ * @example
+ * ```typescript
+ * const codebase = createCodebaseContext({ root: './my-project' });
+ *
+ * // Analyze the repository
+ * const repoMap = await codebase.analyze();
+ *
+ * // Select relevant code for a task within budget
+ * const selection = await codebase.selectRelevantCode({
+ *   task: 'implement user authentication',
+ *   maxTokens: 8000,
+ *   strategy: 'relevance_first',
+ * });
+ *
+ * // Use selected code in context
+ * const context = selection.chunks.map(c => c.content).join('\n\n');
+ * ```
+ */
+export declare class CodebaseContextManager {
+    private config;
+    private repoMap;
+    private cache;
+    private listeners;
+    constructor(config?: CodebaseContextConfig);
+    /**
+     * Analyze a repository and build its structure map.
+     */
+    analyze(root?: string): Promise<RepoMap>;
+    /**
+     * Discover files matching include patterns.
+     */
+    private discoverFiles;
+    /**
+     * Process a single file into a code chunk.
+     */
+    private processFile;
+    /**
+     * Determine the type of a code chunk.
+     */
+    private determineChunkType;
+    /**
+     * Calculate base importance for a chunk type.
+     */
+    private calculateBaseImportance;
+    /**
+     * Extract exported symbols from code.
+     */
+    private extractSymbols;
+    /**
+     * Extract dependency file paths from imports.
+     */
+    private extractDependencies;
+    /**
+     * Extract dependency names (for symbol matching).
+     */
+    private extractDependencyNames;
+    /**
+     * Adjust importance based on how many files depend on each chunk.
+     */
+    private adjustImportanceByConnectivity;
+    /**
+     * Check if a path matches any of the given patterns.
+     */
+    private matchesPatterns;
+    /**
+     * Simple glob matching (supports * and **).
+     */
+    private matchesGlob;
+    /**
+     * Select relevant code chunks for a task within a token budget.
+     */
+    selectRelevantCode(options: SelectionOptions): Promise<SelectionResult>;
+    /**
+     * Calculate task relevance for a chunk.
+     */
+    private calculateRelevance;
+    /**
+     * Sort candidates by selection strategy.
+     */
+    private sortByStrategy;
+    /**
+     * Expand context by adding related files to current selection.
+     */
+    expandContext(currentChunks: CodeChunk[], options: {
+        maxTokensToAdd: number;
+        direction: 'dependencies' | 'dependents' | 'both';
+        query?: string;
+    }): Promise<CodeChunk[]>;
+    /**
+     * Get a chunk by file path.
+     */
+    getChunk(filePath: string): CodeChunk | undefined;
+    /**
+     * Get all chunks.
+     */
+    getAllChunks(): CodeChunk[];
+    /**
+     * Get entry points.
+     */
+    getEntryPoints(): CodeChunk[];
+    /**
+     * Estimate tokens for a string.
+     */
+    estimateTokens(content: string): number;
+    /**
+     * Get the current repo map.
+     */
+    getRepoMap(): RepoMap | null;
+    /**
+     * Clear cached data.
+     */
+    clearCache(): void;
+    /**
+     * Subscribe to events.
+     */
+    on(listener: CodebaseContextEventListener): () => void;
+    private emit;
+}
+/**
+ * Create a codebase context manager.
+ *
+ * @example
+ * ```typescript
+ * const codebase = createCodebaseContext({
+ *   root: './my-project',
+ *   includePatterns: ['src/**\/*.ts'],
+ *   excludePatterns: ['node_modules/**'],
+ * });
+ *
+ * // Analyze repo
+ * await codebase.analyze();
+ *
+ * // Select code for task
+ * const selection = await codebase.selectRelevantCode({
+ *   task: 'fix authentication bug',
+ *   maxTokens: 10000,
+ *   strategy: 'relevance_first',
+ * });
+ *
+ * console.log(`Selected ${selection.chunks.length} files`);
+ * console.log(`Using ${selection.totalTokens} tokens`);
+ * ```
+ */
+export declare function createCodebaseContext(config?: CodebaseContextConfig): CodebaseContextManager;
+/**
+ * Minimal chunk interface for building context.
+ * Only requires the properties actually used by buildContextFromChunks.
+ */
+export interface MinimalCodeChunk {
+    filePath: string;
+    content: string;
+    tokenCount: number;
+}
+/**
+ * Build context string from selected chunks.
+ */
+export declare function buildContextFromChunks(chunks: MinimalCodeChunk[], options?: {
+    includeFilePaths?: boolean;
+    includeSeparators?: boolean;
+    maxTotalTokens?: number;
+}): string;
+/**
+ * Create a summary of the repo structure.
+ */
+export declare function summarizeRepoStructure(repoMap: RepoMap): string;
+//# sourceMappingURL=codebase-context.d.ts.map

package/dist/src/integrations/codebase-context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"codebase-context.d.ts","sourceRoot":"","sources":["../../../src/integrations/codebase-context.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AASH;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,wBAAwB;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,wCAAwC;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,8BAA8B;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,4BAA4B;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,6BAA6B;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,yBAAyB;IACzB,IAAI,EAAE,aAAa,CAAC;IACpB,oCAAoC;IACpC,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,0CAA0C;IAC1C,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,8BAA8B;IAC9B,YAAY,CAAC,EAAE,IAAI,CAAC;CACrB;AAED,MAAM,MAAM,aAAa,GACrB,aAAa,GACb,aAAa,GACb,SAAS,GACT,OAAO,GACP,QAAQ,GACR,MAAM,GACN,eAAe,GACf,OAAO,CAAC;AAEZ;;GAEG;AACH,MAAM,WAAW,OAAO;IACtB,iCAAiC;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,iCAAiC;IACjC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC/B,wBAAwB;IACxB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,8BAA8B;IAC9B,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,gDAAgD;IAChD,eAAe,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IAC1C,8DAA8D;IAC9D,sBAAsB,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACjD,6BAA6B;IAC7B,WAAW,EAAE,MAAM,CAAC;IACpB,yBAAyB;IACzB,UAAU,EAAE,IAAI,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,gCAAgC;IAChC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,sCAAsC;IACtC,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B,sCAAsC;IACtC,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B,iCAAiC;IACjC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,0CAA0C;IAC1C,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gCAAgC;IAChC,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC9B,qCAAqC;IACrC,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC9B,uCAAuC;IACvC,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,6BAA6B;IAC7B,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,gCAAgC;IAChC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,2BAA2B;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,6CAA6C;IAC7C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,mCAAmC;IACnC,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,mCAAmC;IACnC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,+BAA+B;IAC/B,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,oBAAoB;IACpB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,6BAA6B;IAC7B,QAAQ,CAAC,EAAE,iBAAiB,CAAC;CAC9B;AAED,MAAM,MAAM,iBAAiB,GACzB,kBAAkB,GAClB,iBAAiB,GACjB,eAAe,GACf,aAAa,CAAC;AAElB;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,2BAA2B;IAC3B,MAAM,EAAE,SAAS,EAAE,CAAC;IACpB,wBAAwB;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,uBAAuB;IACvB,eAAe,EAAE,MAAM,CAAC;IACxB,6CAA6C;IAC7C,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,2BAA2B;IAC3B,KAAK,EAAE;QACL,eAAe,EAAE,MAAM,CAAC;QACxB,aAAa,EAAE,MAAM,CAAC;QACtB,eAAe,EAAE,MAAM,CAAC;QACxB,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;CACH;AAED;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAC5B;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAC1C;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAAE,GACvE;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,GACxC;IAAE,IAAI,EAAE,qBAAqB,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GACjE;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACnC;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC;AAEzC,MAAM,MAAM,4BAA4B,GAAG,CAAC,KAAK,EAAE,oBAAoB,KAAK,IAAI,CAAC;AAyCjF;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,qBAAa,sBAAsB;IACjC,OAAO,CAAC,MAAM,CAAkC;IAChD,OAAO,CAAC,OAAO,CAAwB;IACvC,OAAO,CAAC,KAAK,CAA6D;IAC1E,OAAO,CAAC,SAAS,CAAsC;gBAE3C,MAAM,GAAE,qBAA0B;IAQ9C;;OAEG;IACG,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IA0G9C;;OAEG;YACW,aAAa;IA6B3B;;OAEG;YACW,WAAW;IAoCzB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IA0D1B;;OAEG;IACH,OAAO,CAAC,uBAAuB;IA0B/B;;OAEG;IACH,OAAO,CAAC,cAAc;IAyCtB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IA6B3B;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAwB9B;;OAEG;IACH,OAAO,CAAC,8BAA8B;IAkBtC;;OAEG;IACH,OAAO,CAAC,eAAe;IASvB;;OAEG;IACH,OAAO,CAAC,WAAW;IAgBnB;;OAEG;IACG,kBAAkB,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,eAAe,CAAC;IAsF7E;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAmC1B;;OAEG;IACH,OAAO,CAAC,cAAc;IAkEtB;;OAEG;IACG,aAAa,CACjB,aAAa,EAAE,SAAS,EAAE,EAC1B,OAAO,EAAE;QACP,cAAc,EAAE,MAAM,CAAC;QACvB,SAAS,EAAE,cAAc,GAAG,YAAY,GAAG,MAAM,CAAC;QAClD,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GACA,OAAO,CAAC,SAAS,EAAE,CAAC;IAqEvB;;OAEG;IACH,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAIjD;;OAEG;IACH,YAAY,IAAI,SAAS,EAAE;IAI3B;;OAEG;IACH,cAAc,IAAI,SAAS,EAAE;IAO7B;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAIvC;;OAEG;IACH,UAAU,IAAI,OAAO,GAAG,IAAI;IAI5B;;OAEG;IACH,UAAU,IAAI,IAAI;IAKlB;;OAEG;IACH,EAAE,CAAC,QAAQ,EAAE,4BAA4B,GAAG,MAAM,IAAI;IAQtD,OAAO,CAAC,IAAI;CASb;AAMD;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,wBAAgB,qBAAqB,CACnC,MAAM,GAAE,qBAA0B,GACjC,sBAAsB,CAExB;AAED;;;GAGG;AACH,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,MAAM,EAAE,gBAAgB,EAAE,EAC1B,OAAO,GAAE;IACP,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;CACpB,GACL,MAAM,CA4BR;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CA8B/D"}