@bastani/atomic 0.9.11-alpha.7 → 0.9.11-alpha.9

package/dist/core/extensions/loader-runtime.js

@@ -51,7 +51,9 @@ export function createExtensionRuntime() {
         flagOwnerOrigins: new Map(),
         pendingProviderRegistrations: [],
         canRegisterResource: () => true,
-        beginResourceRegistrationBatch: () => { batchDepth += 1; },
+        beginResourceRegistrationBatch: () => {
+            batchDepth += 1;
+        },
         endResourceRegistrationBatch: () => {
             batchDepth -= 1;
             if (batchDepth !== 0)
@@ -64,8 +66,9 @@ export function createExtensionRuntime() {
             for (const pending of pendingFlags.values()) {
                 pending.extension.flags.set(pending.name, pending.registration);
                 const ownerOrigin = runtime.flagOwnerOrigins?.get(pending.name);
-                if (ownerOrigin === pending.extension.sourceInfo.configurationOrigin
-                    && pending.defaultValue !== undefined && !runtime.flagValues.has(pending.name)) {
+                if (ownerOrigin === pending.extension.sourceInfo.configurationOrigin &&
+                    pending.defaultValue !== undefined &&
+                    !runtime.flagValues.has(pending.name)) {
                     runtime.flagValues.set(pending.name, pending.defaultValue);
                 }
             }
@@ -101,8 +104,10 @@ export function createExtensionRuntime() {
             const key = registrationKey(extension, name);
             const firstDefault = pendingFlags.get(key)?.defaultValue;
             pendingFlags.set(key, { extension, name, registration, defaultValue: firstDefault ?? defaultValue });
-            const owners = runtime.flagOwners ??= new Map();
-            const ownerOrigins = runtime.flagOwnerOrigins ??= new Map();
+            runtime.flagOwners ??= new Map();
+            const owners = runtime.flagOwners;
+            runtime.flagOwnerOrigins ??= new Map();
+            const ownerOrigins = runtime.flagOwnerOrigins;
             if (!owners.has(name)) {
                 owners.set(name, extension.path);
                 ownerOrigins.set(name, extension.sourceInfo.configurationOrigin);
@@ -141,8 +146,7 @@ export function createExtensionRuntime() {
         getPendingFlagDefault: (ownerPath, name) => {
             if (!pendingFlags.has(`${ownerPath}\0${name}`))
                 return undefined;
-            return [...pendingFlags.values()]
-                .find((pending) => pending.name === name && pending.defaultValue !== undefined)?.defaultValue;
+            return [...pendingFlags.values()].find((pending) => pending.name === name && pending.defaultValue !== undefined)?.defaultValue;
         },
         getAllToolsAfterRegistration: (extension) => {
             const tools = runtime.getAllTools();
@@ -221,15 +225,23 @@ export function createExtensionRuntime() {
         },
         assertActive,
         invalidate: (message) => {
-            state.staleMessage ??= message
-                ?? "This extension ctx is stale after session replacement or reload. Do not use a captured pi or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload().";
+            state.staleMessage ??=
+                message ??
+                    "This extension ctx is stale after session replacement or reload. Do not use a captured pi or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload().";
         },
         registerProvider: (nameOrProvider, configOrPath, extensionPath = "<unknown>") => {
             if (typeof nameOrProvider === "string") {
-                runtime.pendingProviderRegistrations.push({ name: nameOrProvider, config: configOrPath, extensionPath: extensionPath });
+                runtime.pendingProviderRegistrations.push({
+                    name: nameOrProvider,
+                    config: configOrPath,
+                    extensionPath: extensionPath,
+                });
             }
             else {
-                runtime.pendingProviderRegistrations.push({ provider: nameOrProvider, extensionPath: typeof configOrPath === "string" ? configOrPath : extensionPath });
+                runtime.pendingProviderRegistrations.push({
+                    provider: nameOrProvider,
+                    extensionPath: typeof configOrPath === "string" ? configOrPath : extensionPath,
+                });
             }
         },
         unregisterProvider: (name) => {

package/dist/core/extensions/loader-runtime.js.map

@@ -1 +1 @@
-{"version":3,"file":"loader-runtime.js","sourceRoot":"","sources":["../../../src/core/extensions/loader-runtime.ts"],"names":[],"mappings":"AASA,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAI,OAAyB,EAAE,GAAqB;IACpG,IAAI,CAAC,OAAO,CAAC,8BAA8B,IAAI,CAAC,OAAO,CAAC,4BAA4B;QAAE,OAAO,GAAG,EAAE,CAAC;IACnG,OAAO,CAAC,8BAA8B,EAAE,CAAC;IACzC,IAAI,CAAC;QACH,OAAO,MAAM,GAAG,EAAE,CAAC;IACrB,CAAC;YAAS,CAAC;QACT,OAAO,CAAC,4BAA4B,EAAE,CAAC;IACzC,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,SAAoB,EAAE,IAAY;IACzD,OAAO,GAAG,SAAS,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;AACtC,CAAC;AAED,+DAA+D;AAC/D,MAAM,UAAU,sBAAsB;IACpC,MAAM,cAAc,GAAG,GAAG,EAAE;QAC1B,MAAM,IAAI,KAAK,CAAC,8FAA8F,CAAC,CAAC;IAClH,CAAC,CAAC;IACF,MAAM,KAAK,GAA8B,EAAE,CAAC;IAC5C,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,MAAM,YAAY,GAAG,IAAI,GAAG,EAAgF,CAAC;IAC7G,MAAM,eAAe,GAAG,IAAI,GAAG,EAAmF,CAAC;IACnH,MAAM,YAAY,GAAG,IAAI,GAAG,EAAgH,CAAC;IAC7I,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAmF,CAAC;IACpH,MAAM,WAAW,GAAG,CAAC,SAAoB,EAAE,EAAE,CAAC,UAAU,GAAG,CAAC,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc,CAAC;IAC5H,IAAI,sBAA4C,CAAC;IACjD,MAAM,YAAY,GAAG,GAAG,EAAE;QACxB,IAAI,KAAK,CAAC,YAAY;YAAE,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC9D,CAAC,CAAC;IAEF,MAAM,OAAO,GAAqB;QAChC,WAAW,EAAE,cAAc;QAC3B,YAAY,EAAE,cAAc;QAC5B,eAAe,EAAE,cAAc;QAC/B,WAAW,EAAE,cAAc;QAC3B,cAAc,EAAE,cAAc;QAC9B,cAAc,EAAE,cAAc;QAC9B,QAAQ,EAAE,cAAc;QACxB,cAAc,EAAE,cAAc;QAC9B,WAAW,EAAE,cAAc;QAC3B,cAAc,EAAE,cAAc;QAC9B,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;QACtB,WAAW,EAAE,cAAc;QAC3B,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAC9E,gBAAgB,EAAE,cAAc;QAChC,gBAAgB,EAAE,cAAc;QAChC,UAAU,EAAE,IAAI,GAAG,EAAE;QACrB,iBAAiB,EAAE,IAAI,GAAG,EAAE;QAC5B,UAAU,EAAE,IAAI,GAAG,EAAE;QACrB,gBAAgB,EAAE,IAAI,GAAG,EAAE;QAC3B,4BAA4B,EAAE,EAAE;QAChC,mBAAmB,EAAE,GAAG,EAAE,CAAC,IAAI;QAC/B,8BAA8B,EAAE,GAAG,EAAE,GAAG,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1D,4BAA4B,EAAE,GAAG,EAAE;YACjC,UAAU,IAAI,CAAC,CAAC;YAChB,IAAI,UAAU,KAAK,CAAC;gBAAE,OAAO;YAC7B,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,GAAG,CAAC,CAAC;YAC3C,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE;gBAAE,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YAC7G,KAAK,MAAM,OAAO,IAAI,eAAe,CAAC,MAAM,EAAE;gBAAE,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YACnH,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC5C,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;gBAChE,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChE,IAAI,WAAW,KAAK,OAAO,CAAC,SAAS,CAAC,UAAU,CAAC,mBAAmB;uBAC/D,OAAO,CAAC,YAAY,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;oBACjF,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;gBAC7D,CAAC;YACH,CAAC;YACD,KAAK,MAAM,OAAO,IAAI,gBAAgB,CAAC,MAAM,EAAE;gBAAE,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,IAAa,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YAC9H,YAAY,CAAC,KAAK,EAAE,CAAC;YACrB,eAAe,CAAC,KAAK,EAAE,CAAC;YACxB,YAAY,CAAC,KAAK,EAAE,CAAC;YACrB,gBAAgB,CAAC,KAAK,EAAE,CAAC;YACzB,IAAI,YAAY;gBAAE,OAAO,CAAC,YAAY,EAAE,CAAC;YACzC,IAAI,sBAAsB;gBAAE,OAAO,CAAC,cAAc,CAAC,sBAAsB,CAAC,CAAC;YAC3E,sBAAsB,GAAG,SAAS,CAAC;QACrC,CAAC;QACD,qBAAqB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE;YACvD,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YACtF,IAAI,sBAAsB,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAAE,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxG,OAAO,IAAI,CAAC;QACd,CAAC;QACD,wBAAwB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE;YAC1D,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,eAAe,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YACzF,OAAO,IAAI,CAAC;QACd,CAAC;QACD,qBAAqB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE;YACrE,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,MAAM,GAAG,GAAG,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC7C,MAAM,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,YAAY,CAAC;YACzD,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,IAAI,YAAY,EAAE,CAAC,CAAC;YACrG,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC;YAChD,MAAM,YAAY,GAAG,OAAO,CAAC,gBAAgB,KAAK,IAAI,GAAG,EAAE,CAAC;YAC5D,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACtB,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;gBACjC,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;YACnE,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,yBAAyB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE;YAC3D,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,gBAAgB,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YAC1F,OAAO,IAAI,CAAC;QACd,CAAC;QACD,8BAA8B,EAAE,CAAC,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE;YAChE,MAAM,GAAG,GAAG,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC7C,IAAI,YAAY,KAAK,MAAM;gBAAE,OAAO,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC1D,IAAI,YAAY,KAAK,SAAS;gBAAE,OAAO,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChE,IAAI,YAAY,KAAK,MAAM;gBAAE,OAAO,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC1D,IAAI,YAAY,KAAK,UAAU;gBAAE,OAAO,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClE,OAAO,KAAK,CAAC;QACf,CAAC;QACD,iCAAiC,EAAE,CAAC,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE;YACnE,MAAM,GAAG,GAAG,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC7C,IAAI,YAAY,KAAK,MAAM;gBAAE,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;iBACjD,IAAI,YAAY,KAAK,SAAS;gBAAE,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;iBAC5D,IAAI,YAAY,KAAK,MAAM;gBAAE,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;iBACtD,IAAI,YAAY,KAAK,UAAU;gBAAE,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrE,CAAC;QACD,qBAAqB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE;YACzC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,SAAS,KAAK,IAAI,EAAE,CAAC;gBAAE,OAAO,SAAS,CAAC;YACjE,OAAO,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC;iBAC9B,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,IAAI,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,EAAE,YAAY,CAAC;QAClG,CAAC;QACD,4BAA4B,EAAE,CAAC,SAAS,EAAE,EAAE;YAC1C,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YACpC,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc;gBAAE,OAAO,KAAK,CAAC;YAC9E,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACtD,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC5C,IAAI,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;oBAAE,SAAS;gBACtC,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;gBACxD,KAAK,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,UAAU,CAAC,IAAI;oBACrB,WAAW,EAAE,UAAU,CAAC,WAAW;oBACnC,UAAU,EAAE,UAAU,CAAC,UAAU;oBACjC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,qBAAqB,CAAC;wBAClD,CAAC,CAAC,EAAE,mBAAmB,EAAE,UAAU,CAAC,mBAAmB,EAAE;wBACzD,CAAC,CAAC,EAAE,CAAC;oBACP,gBAAgB,EAAE,UAAU,CAAC,gBAAgB;oBAC7C,UAAU;iBACX,CAAC,CAAC;gBACH,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC1B,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,4BAA4B,EAAE,CAAC,SAAS,EAAE,EAAE;YAC1C,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc;gBAAE,OAAO,OAAO,CAAC,WAAW,EAAE,CAAC;YAC9F,MAAM,MAAM,GAAG,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC7D,GAAG,OAAO;gBACV,QAAQ,EAAE,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;aACvD,CAAC,CAAC,CAAC;YACJ,KAAK,MAAM,OAAO,IAAI,MAAM;gBAAE,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YACjG,IAAI,CAAC;gBACH,OAAO,OAAO,CAAC,WAAW,EAAE,CAAC;YAC/B,CAAC;oBAAS,CAAC;gBACT,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;oBAC7B,IAAI,OAAO,CAAC,QAAQ;wBAAE,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;;wBAChF,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;QACH,CAAC;QACD,6BAA6B,EAAE,GAAG,EAAE;YAClC,OAAO,CAAC,YAAY,EAAE,CAAC;YACvB,IAAI,UAAU,GAAG,CAAC,IAAI,sBAAsB;gBAAE,sBAAsB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;QAClG,CAAC;QACD,iCAAiC,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,mBAAmB,EAAE,EAAE;YAClF,IAAI,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK,mBAAmB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACjG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QACD,+BAA+B,EAAE,CAAC,SAAS,EAAE,EAAE;YAC7C,MAAM,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;YACxC,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc;gBAAE,OAAO,MAAM,CAAC;YAC/E,IAAI,sBAAsB;gBAAE,OAAO,CAAC,GAAG,sBAAsB,CAAC,CAAC;YAC/D,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;YAC9B,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE;gBAAE,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACrE,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;QACpB,CAAC;QACD,+BAA+B,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE;YACxD,IAAI,UAAU,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAC;YACnC,sBAAsB,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC;YACxC,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc;gBAAE,OAAO,KAAK,CAAC;YAC9E,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1E,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACxE,OAAO,IAAI,CAAC;QACd,CAAC;QACD,YAAY;QACZ,UAAU,EAAE,CAAC,OAAO,EAAE,EAAE;YACtB,KAAK,CAAC,YAAY,KAAK,OAAO;mBACzB,6WAA6W,CAAC;QACrX,CAAC;QACD,gBAAgB,EAAE,CAAC,cAAc,EAAE,YAAY,EAAE,aAAa,GAAG,WAAW,EAAE,EAAE;YAC9E,IAAI,OAAO,cAAc,KAAK,QAAQ,EAAE,CAAC;gBACvC,OAAO,CAAC,4BAA4B,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,MAAM,EAAE,YAAmD,EAAE,aAAa,EAAE,aAAuB,EAAE,CAAC,CAAC;YAC3K,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,4BAA4B,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,aAAa,EAAE,OAAO,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,aAAuB,EAAE,CAAC,CAAC;YACpK,CAAC;QACH,CAAC;QACD,kBAAkB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC3B,OAAO,CAAC,4BAA4B,GAAG,OAAO,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAClG,UAAU,IAAI,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QACjG,CAAC;KACF,CAAC;IACF,OAAO,OAAO,CAAC;AACjB,CAAC","sourcesContent":["import type {\n  Extension,\n  ExtensionFlag,\n  ExtensionRuntime,\n  ExtensionShortcut,\n  RegisteredCommand,\n  RegisteredTool,\n} from \"./types.ts\";\n\nexport async function runResourceRegistrationBatch<T>(runtime: ExtensionRuntime, run: () => Promise<T>): Promise<T> {\n  if (!runtime.beginResourceRegistrationBatch || !runtime.endResourceRegistrationBatch) return run();\n  runtime.beginResourceRegistrationBatch();\n  try {\n    return await run();\n  } finally {\n    runtime.endResourceRegistrationBatch();\n  }\n}\n\nfunction registrationKey(extension: Extension, name: string): string {\n  return `${extension.path}\\0${name}`;\n}\n\n/** Create a runtime with throwing stubs for action methods. */\nexport function createExtensionRuntime(): ExtensionRuntime {\n  const notInitialized = () => {\n    throw new Error(\"Extension runtime not initialized. Action methods cannot be called during extension loading.\");\n  };\n  const state: { staleMessage?: string } = {};\n  let batchDepth = 0;\n  const pendingTools = new Map<string, { extension: Extension; name: string; registration: RegisteredTool }>();\n  const pendingCommands = new Map<string, { extension: Extension; name: string; registration: RegisteredCommand }>();\n  const pendingFlags = new Map<string, { extension: Extension; name: string; registration: ExtensionFlag; defaultValue?: boolean | string }>();\n  const pendingShortcuts = new Map<string, { extension: Extension; name: string; registration: ExtensionShortcut }>();\n  const shouldStage = (extension: Extension) => batchDepth > 0 && extension.sourceInfo.configurationOrigin === \"inherited-pi\";\n  let pendingActiveToolNames: string[] | undefined;\n  const assertActive = () => {\n    if (state.staleMessage) throw new Error(state.staleMessage);\n  };\n\n  const runtime: ExtensionRuntime = {\n    sendMessage: notInitialized,\n    sendMessages: notInitialized,\n    sendUserMessage: notInitialized,\n    appendEntry: notInitialized,\n    setSessionName: notInitialized,\n    getSessionName: notInitialized,\n    setLabel: notInitialized,\n    getActiveTools: notInitialized,\n    getAllTools: notInitialized,\n    setActiveTools: notInitialized,\n    refreshTools: () => {},\n    getCommands: notInitialized,\n    setModel: () => Promise.reject(new Error(\"Extension runtime not initialized\")),\n    getThinkingLevel: notInitialized,\n    setThinkingLevel: notInitialized,\n    flagValues: new Map(),\n    explicitFlagNames: new Set(),\n    flagOwners: new Map(),\n    flagOwnerOrigins: new Map(),\n    pendingProviderRegistrations: [],\n    canRegisterResource: () => true,\n    beginResourceRegistrationBatch: () => { batchDepth += 1; },\n    endResourceRegistrationBatch: () => {\n      batchDepth -= 1;\n      if (batchDepth !== 0) return;\n      const refreshTools = pendingTools.size > 0;\n      for (const pending of pendingTools.values()) pending.extension.tools.set(pending.name, pending.registration);\n      for (const pending of pendingCommands.values()) pending.extension.commands.set(pending.name, pending.registration);\n      for (const pending of pendingFlags.values()) {\n        pending.extension.flags.set(pending.name, pending.registration);\n        const ownerOrigin = runtime.flagOwnerOrigins?.get(pending.name);\n        if (ownerOrigin === pending.extension.sourceInfo.configurationOrigin\n          && pending.defaultValue !== undefined && !runtime.flagValues.has(pending.name)) {\n          runtime.flagValues.set(pending.name, pending.defaultValue);\n        }\n      }\n      for (const pending of pendingShortcuts.values()) pending.extension.shortcuts.set(pending.name as never, pending.registration);\n      pendingTools.clear();\n      pendingCommands.clear();\n      pendingFlags.clear();\n      pendingShortcuts.clear();\n      if (refreshTools) runtime.refreshTools();\n      if (pendingActiveToolNames) runtime.setActiveTools(pendingActiveToolNames);\n      pendingActiveToolNames = undefined;\n    },\n    stageToolRegistration: (extension, name, registration) => {\n      if (!shouldStage(extension)) return false;\n      pendingTools.set(registrationKey(extension, name), { extension, name, registration });\n      if (pendingActiveToolNames && !pendingActiveToolNames.includes(name)) pendingActiveToolNames.push(name);\n      return true;\n    },\n    stageCommandRegistration: (extension, name, registration) => {\n      if (!shouldStage(extension)) return false;\n      pendingCommands.set(registrationKey(extension, name), { extension, name, registration });\n      return true;\n    },\n    stageFlagRegistration: (extension, name, registration, defaultValue) => {\n      if (!shouldStage(extension)) return false;\n      const key = registrationKey(extension, name);\n      const firstDefault = pendingFlags.get(key)?.defaultValue;\n      pendingFlags.set(key, { extension, name, registration, defaultValue: firstDefault ?? defaultValue });\n      const owners = runtime.flagOwners ??= new Map();\n      const ownerOrigins = runtime.flagOwnerOrigins ??= new Map();\n      if (!owners.has(name)) {\n        owners.set(name, extension.path);\n        ownerOrigins.set(name, extension.sourceInfo.configurationOrigin);\n      }\n      return true;\n    },\n    stageShortcutRegistration: (extension, name, registration) => {\n      if (!shouldStage(extension)) return false;\n      pendingShortcuts.set(registrationKey(extension, name), { extension, name, registration });\n      return true;\n    },\n    hasPendingResourceRegistration: (extension, resourceType, name) => {\n      const key = registrationKey(extension, name);\n      if (resourceType === \"tool\") return pendingTools.has(key);\n      if (resourceType === \"command\") return pendingCommands.has(key);\n      if (resourceType === \"flag\") return pendingFlags.has(key);\n      if (resourceType === \"shortcut\") return pendingShortcuts.has(key);\n      return false;\n    },\n    deletePendingResourceRegistration: (extension, resourceType, name) => {\n      const key = registrationKey(extension, name);\n      if (resourceType === \"tool\") pendingTools.delete(key);\n      else if (resourceType === \"command\") pendingCommands.delete(key);\n      else if (resourceType === \"flag\") pendingFlags.delete(key);\n      else if (resourceType === \"shortcut\") pendingShortcuts.delete(key);\n    },\n    getPendingFlagDefault: (ownerPath, name) => {\n      if (!pendingFlags.has(`${ownerPath}\\0${name}`)) return undefined;\n      return [...pendingFlags.values()]\n        .find((pending) => pending.name === name && pending.defaultValue !== undefined)?.defaultValue;\n    },\n    getAllToolsAfterRegistration: (extension) => {\n      const tools = runtime.getAllTools();\n      if (extension.sourceInfo.configurationOrigin !== \"inherited-pi\") return tools;\n      const names = new Set(tools.map((tool) => tool.name));\n      for (const pending of pendingTools.values()) {\n        if (names.has(pending.name)) continue;\n        const { definition, sourceInfo } = pending.registration;\n        tools.push({\n          name: definition.name,\n          description: definition.description,\n          parameters: definition.parameters,\n          ...(Object.hasOwn(definition, \"constrainedSampling\")\n            ? { constrainedSampling: definition.constrainedSampling }\n            : {}),\n          promptGuidelines: definition.promptGuidelines,\n          sourceInfo,\n        });\n        names.add(pending.name);\n      }\n      return tools;\n    },\n    getCommandsAfterRegistration: (extension) => {\n      if (extension.sourceInfo.configurationOrigin !== \"inherited-pi\") return runtime.getCommands();\n      const active = [...pendingCommands.values()].map((pending) => ({\n        ...pending,\n        previous: pending.extension.commands.get(pending.name),\n      }));\n      for (const pending of active) pending.extension.commands.set(pending.name, pending.registration);\n      try {\n        return runtime.getCommands();\n      } finally {\n        for (const pending of active) {\n          if (pending.previous) pending.extension.commands.set(pending.name, pending.previous);\n          else pending.extension.commands.delete(pending.name);\n        }\n      }\n    },\n    refreshToolsAfterRegistration: () => {\n      runtime.refreshTools();\n      if (batchDepth > 0 && pendingActiveToolNames) pendingActiveToolNames = runtime.getActiveTools();\n    },\n    applyFlagDefaultAfterRegistration: (name, _ownerPath, value, configurationOrigin) => {\n      if (runtime.flagOwnerOrigins?.get(name) === configurationOrigin && !runtime.flagValues.has(name)) {\n        runtime.flagValues.set(name, value);\n      }\n    },\n    getActiveToolsAfterRegistration: (extension) => {\n      const active = runtime.getActiveTools();\n      if (extension.sourceInfo.configurationOrigin !== \"inherited-pi\") return active;\n      if (pendingActiveToolNames) return [...pendingActiveToolNames];\n      const names = new Set(active);\n      for (const pending of pendingTools.values()) names.add(pending.name);\n      return [...names];\n    },\n    setActiveToolsAfterRegistration: (extension, toolNames) => {\n      if (batchDepth === 0) return false;\n      pendingActiveToolNames = [...toolNames];\n      if (extension.sourceInfo.configurationOrigin !== \"inherited-pi\") return false;\n      const liveNames = new Set(runtime.getAllTools().map((tool) => tool.name));\n      runtime.setActiveTools(toolNames.filter((name) => liveNames.has(name)));\n      return true;\n    },\n    assertActive,\n    invalidate: (message) => {\n      state.staleMessage ??= message\n        ?? \"This extension ctx is stale after session replacement or reload. Do not use a captured pi or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload().\";\n    },\n    registerProvider: (nameOrProvider, configOrPath, extensionPath = \"<unknown>\") => {\n      if (typeof nameOrProvider === \"string\") {\n        runtime.pendingProviderRegistrations.push({ name: nameOrProvider, config: configOrPath as import(\"./types.ts\").ProviderConfig, extensionPath: extensionPath as string });\n      } else {\n        runtime.pendingProviderRegistrations.push({ provider: nameOrProvider, extensionPath: typeof configOrPath === \"string\" ? configOrPath : extensionPath as string });\n      }\n    },\n    unregisterProvider: (name) => {\n      runtime.pendingProviderRegistrations = runtime.pendingProviderRegistrations.filter((registration) =>\n        \"provider\" in registration ? registration.provider.id !== name : registration.name !== name);\n    },\n  };\n  return runtime;\n}\n"]}
+{"version":3,"file":"loader-runtime.js","sourceRoot":"","sources":["../../../src/core/extensions/loader-runtime.ts"],"names":[],"mappings":"AASA,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAI,OAAyB,EAAE,GAAqB;IACrG,IAAI,CAAC,OAAO,CAAC,8BAA8B,IAAI,CAAC,OAAO,CAAC,4BAA4B;QAAE,OAAO,GAAG,EAAE,CAAC;IACnG,OAAO,CAAC,8BAA8B,EAAE,CAAC;IACzC,IAAI,CAAC;QACJ,OAAO,MAAM,GAAG,EAAE,CAAC;IACpB,CAAC;YAAS,CAAC;QACV,OAAO,CAAC,4BAA4B,EAAE,CAAC;IACxC,CAAC;AACF,CAAC;AAED,SAAS,eAAe,CAAC,SAAoB,EAAE,IAAY;IAC1D,OAAO,GAAG,SAAS,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;AACrC,CAAC;AAED,+DAA+D;AAC/D,MAAM,UAAU,sBAAsB;IACrC,MAAM,cAAc,GAAG,GAAG,EAAE;QAC3B,MAAM,IAAI,KAAK,CAAC,8FAA8F,CAAC,CAAC;IACjH,CAAC,CAAC;IACF,MAAM,KAAK,GAA8B,EAAE,CAAC;IAC5C,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,MAAM,YAAY,GAAG,IAAI,GAAG,EAAgF,CAAC;IAC7G,MAAM,eAAe,GAAG,IAAI,GAAG,EAAmF,CAAC;IACnH,MAAM,YAAY,GAAG,IAAI,GAAG,EAGzB,CAAC;IACJ,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAmF,CAAC;IACpH,MAAM,WAAW,GAAG,CAAC,SAAoB,EAAE,EAAE,CAC5C,UAAU,GAAG,CAAC,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc,CAAC;IAC/E,IAAI,sBAA4C,CAAC;IACjD,MAAM,YAAY,GAAG,GAAG,EAAE;QACzB,IAAI,KAAK,CAAC,YAAY;YAAE,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC7D,CAAC,CAAC;IAEF,MAAM,OAAO,GAAqB;QACjC,WAAW,EAAE,cAAc;QAC3B,YAAY,EAAE,cAAc;QAC5B,eAAe,EAAE,cAAc;QAC/B,WAAW,EAAE,cAAc;QAC3B,cAAc,EAAE,cAAc;QAC9B,cAAc,EAAE,cAAc;QAC9B,QAAQ,EAAE,cAAc;QACxB,cAAc,EAAE,cAAc;QAC9B,WAAW,EAAE,cAAc;QAC3B,cAAc,EAAE,cAAc;QAC9B,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;QACtB,WAAW,EAAE,cAAc;QAC3B,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAC9E,gBAAgB,EAAE,cAAc;QAChC,gBAAgB,EAAE,cAAc;QAChC,UAAU,EAAE,IAAI,GAAG,EAAE;QACrB,iBAAiB,EAAE,IAAI,GAAG,EAAE;QAC5B,UAAU,EAAE,IAAI,GAAG,EAAE;QACrB,gBAAgB,EAAE,IAAI,GAAG,EAAE;QAC3B,4BAA4B,EAAE,EAAE;QAChC,mBAAmB,EAAE,GAAG,EAAE,CAAC,IAAI;QAC/B,8BAA8B,EAAE,GAAG,EAAE;YACpC,UAAU,IAAI,CAAC,CAAC;QACjB,CAAC;QACD,4BAA4B,EAAE,GAAG,EAAE;YAClC,UAAU,IAAI,CAAC,CAAC;YAChB,IAAI,UAAU,KAAK,CAAC;gBAAE,OAAO;YAC7B,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,GAAG,CAAC,CAAC;YAC3C,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE;gBAAE,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YAC7G,KAAK,MAAM,OAAO,IAAI,eAAe,CAAC,MAAM,EAAE;gBAC7C,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YACpE,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC7C,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;gBAChE,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChE,IACC,WAAW,KAAK,OAAO,CAAC,SAAS,CAAC,UAAU,CAAC,mBAAmB;oBAChE,OAAO,CAAC,YAAY,KAAK,SAAS;oBAClC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EACpC,CAAC;oBACF,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;gBAC5D,CAAC;YACF,CAAC;YACD,KAAK,MAAM,OAAO,IAAI,gBAAgB,CAAC,MAAM,EAAE;gBAC9C,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,IAAa,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YAC9E,YAAY,CAAC,KAAK,EAAE,CAAC;YACrB,eAAe,CAAC,KAAK,EAAE,CAAC;YACxB,YAAY,CAAC,KAAK,EAAE,CAAC;YACrB,gBAAgB,CAAC,KAAK,EAAE,CAAC;YACzB,IAAI,YAAY;gBAAE,OAAO,CAAC,YAAY,EAAE,CAAC;YACzC,IAAI,sBAAsB;gBAAE,OAAO,CAAC,cAAc,CAAC,sBAAsB,CAAC,CAAC;YAC3E,sBAAsB,GAAG,SAAS,CAAC;QACpC,CAAC;QACD,qBAAqB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE;YACxD,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YACtF,IAAI,sBAAsB,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAAE,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxG,OAAO,IAAI,CAAC;QACb,CAAC;QACD,wBAAwB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE;YAC3D,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,eAAe,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YACzF,OAAO,IAAI,CAAC;QACb,CAAC;QACD,qBAAqB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE;YACtE,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,MAAM,GAAG,GAAG,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC7C,MAAM,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,YAAY,CAAC;YACzD,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,IAAI,YAAY,EAAE,CAAC,CAAC;YACrG,OAAO,CAAC,UAAU,KAAK,IAAI,GAAG,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;YAClC,OAAO,CAAC,gBAAgB,KAAK,IAAI,GAAG,EAAE,CAAC;YACvC,MAAM,YAAY,GAAG,OAAO,CAAC,gBAAgB,CAAC;YAC9C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvB,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;gBACjC,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;YAClE,CAAC;YACD,OAAO,IAAI,CAAC;QACb,CAAC;QACD,yBAAyB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE;YAC5D,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,gBAAgB,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YAC1F,OAAO,IAAI,CAAC;QACb,CAAC;QACD,8BAA8B,EAAE,CAAC,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE;YACjE,MAAM,GAAG,GAAG,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC7C,IAAI,YAAY,KAAK,MAAM;gBAAE,OAAO,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC1D,IAAI,YAAY,KAAK,SAAS;gBAAE,OAAO,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChE,IAAI,YAAY,KAAK,MAAM;gBAAE,OAAO,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC1D,IAAI,YAAY,KAAK,UAAU;gBAAE,OAAO,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAClE,OAAO,KAAK,CAAC;QACd,CAAC;QACD,iCAAiC,EAAE,CAAC,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE;YACpE,MAAM,GAAG,GAAG,eAAe,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC7C,IAAI,YAAY,KAAK,MAAM;gBAAE,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;iBACjD,IAAI,YAAY,KAAK,SAAS;gBAAE,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;iBAC5D,IAAI,YAAY,KAAK,MAAM;gBAAE,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;iBACtD,IAAI,YAAY,KAAK,UAAU;gBAAE,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACpE,CAAC;QACD,qBAAqB,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE;YAC1C,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,SAAS,KAAK,IAAI,EAAE,CAAC;gBAAE,OAAO,SAAS,CAAC;YACjE,OAAO,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CACrC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,IAAI,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,CACxE,EAAE,YAAY,CAAC;QACjB,CAAC;QACD,4BAA4B,EAAE,CAAC,SAAS,EAAE,EAAE;YAC3C,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YACpC,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc;gBAAE,OAAO,KAAK,CAAC;YAC9E,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACtD,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC7C,IAAI,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;oBAAE,SAAS;gBACtC,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;gBACxD,KAAK,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,UAAU,CAAC,IAAI;oBACrB,WAAW,EAAE,UAAU,CAAC,WAAW;oBACnC,UAAU,EAAE,UAAU,CAAC,UAAU;oBACjC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,qBAAqB,CAAC;wBACnD,CAAC,CAAC,EAAE,mBAAmB,EAAE,UAAU,CAAC,mBAAmB,EAAE;wBACzD,CAAC,CAAC,EAAE,CAAC;oBACN,gBAAgB,EAAE,UAAU,CAAC,gBAAgB;oBAC7C,UAAU;iBACV,CAAC,CAAC;gBACH,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;YACD,OAAO,KAAK,CAAC;QACd,CAAC;QACD,4BAA4B,EAAE,CAAC,SAAS,EAAE,EAAE;YAC3C,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc;gBAAE,OAAO,OAAO,CAAC,WAAW,EAAE,CAAC;YAC9F,MAAM,MAAM,GAAG,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC9D,GAAG,OAAO;gBACV,QAAQ,EAAE,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;aACtD,CAAC,CAAC,CAAC;YACJ,KAAK,MAAM,OAAO,IAAI,MAAM;gBAAE,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YACjG,IAAI,CAAC;gBACJ,OAAO,OAAO,CAAC,WAAW,EAAE,CAAC;YAC9B,CAAC;oBAAS,CAAC;gBACV,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;oBAC9B,IAAI,OAAO,CAAC,QAAQ;wBAAE,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;;wBAChF,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACtD,CAAC;YACF,CAAC;QACF,CAAC;QACD,6BAA6B,EAAE,GAAG,EAAE;YACnC,OAAO,CAAC,YAAY,EAAE,CAAC;YACvB,IAAI,UAAU,GAAG,CAAC,IAAI,sBAAsB;gBAAE,sBAAsB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;QACjG,CAAC;QACD,iCAAiC,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,mBAAmB,EAAE,EAAE;YACnF,IAAI,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK,mBAAmB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACrC,CAAC;QACF,CAAC;QACD,+BAA+B,EAAE,CAAC,SAAS,EAAE,EAAE;YAC9C,MAAM,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;YACxC,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc;gBAAE,OAAO,MAAM,CAAC;YAC/E,IAAI,sBAAsB;gBAAE,OAAO,CAAC,GAAG,sBAAsB,CAAC,CAAC;YAC/D,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;YAC9B,KAAK,MAAM,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE;gBAAE,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACrE,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;QACnB,CAAC;QACD,+BAA+B,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE;YACzD,IAAI,UAAU,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAC;YACnC,sBAAsB,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC;YACxC,IAAI,SAAS,CAAC,UAAU,CAAC,mBAAmB,KAAK,cAAc;gBAAE,OAAO,KAAK,CAAC;YAC9E,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1E,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACxE,OAAO,IAAI,CAAC;QACb,CAAC;QACD,YAAY;QACZ,UAAU,EAAE,CAAC,OAAO,EAAE,EAAE;YACvB,KAAK,CAAC,YAAY;gBACjB,OAAO;oBACP,6WAA6W,CAAC;QAChX,CAAC;QACD,gBAAgB,EAAE,CAAC,cAAc,EAAE,YAAY,EAAE,aAAa,GAAG,WAAW,EAAE,EAAE;YAC/E,IAAI,OAAO,cAAc,KAAK,QAAQ,EAAE,CAAC;gBACxC,OAAO,CAAC,4BAA4B,CAAC,IAAI,CAAC;oBACzC,IAAI,EAAE,cAAc;oBACpB,MAAM,EAAE,YAAmD;oBAC3D,aAAa,EAAE,aAAuB;iBACtC,CAAC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACP,OAAO,CAAC,4BAA4B,CAAC,IAAI,CAAC;oBACzC,QAAQ,EAAE,cAAc;oBACxB,aAAa,EAAE,OAAO,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAE,aAAwB;iBAC1F,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,kBAAkB,EAAE,CAAC,IAAI,EAAE,EAAE;YAC5B,OAAO,CAAC,4BAA4B,GAAG,OAAO,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CACnG,UAAU,IAAI,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,KAAK,IAAI,CAC3F,CAAC;QACH,CAAC;KACD,CAAC;IACF,OAAO,OAAO,CAAC;AAChB,CAAC","sourcesContent":["import type {\n\tExtension,\n\tExtensionFlag,\n\tExtensionRuntime,\n\tExtensionShortcut,\n\tRegisteredCommand,\n\tRegisteredTool,\n} from \"./types.ts\";\n\nexport async function runResourceRegistrationBatch<T>(runtime: ExtensionRuntime, run: () => Promise<T>): Promise<T> {\n\tif (!runtime.beginResourceRegistrationBatch || !runtime.endResourceRegistrationBatch) return run();\n\truntime.beginResourceRegistrationBatch();\n\ttry {\n\t\treturn await run();\n\t} finally {\n\t\truntime.endResourceRegistrationBatch();\n\t}\n}\n\nfunction registrationKey(extension: Extension, name: string): string {\n\treturn `${extension.path}\\0${name}`;\n}\n\n/** Create a runtime with throwing stubs for action methods. */\nexport function createExtensionRuntime(): ExtensionRuntime {\n\tconst notInitialized = () => {\n\t\tthrow new Error(\"Extension runtime not initialized. Action methods cannot be called during extension loading.\");\n\t};\n\tconst state: { staleMessage?: string } = {};\n\tlet batchDepth = 0;\n\tconst pendingTools = new Map<string, { extension: Extension; name: string; registration: RegisteredTool }>();\n\tconst pendingCommands = new Map<string, { extension: Extension; name: string; registration: RegisteredCommand }>();\n\tconst pendingFlags = new Map<\n\t\tstring,\n\t\t{ extension: Extension; name: string; registration: ExtensionFlag; defaultValue?: boolean | string }\n\t>();\n\tconst pendingShortcuts = new Map<string, { extension: Extension; name: string; registration: ExtensionShortcut }>();\n\tconst shouldStage = (extension: Extension) =>\n\t\tbatchDepth > 0 && extension.sourceInfo.configurationOrigin === \"inherited-pi\";\n\tlet pendingActiveToolNames: string[] | undefined;\n\tconst assertActive = () => {\n\t\tif (state.staleMessage) throw new Error(state.staleMessage);\n\t};\n\n\tconst runtime: ExtensionRuntime = {\n\t\tsendMessage: notInitialized,\n\t\tsendMessages: notInitialized,\n\t\tsendUserMessage: notInitialized,\n\t\tappendEntry: notInitialized,\n\t\tsetSessionName: notInitialized,\n\t\tgetSessionName: notInitialized,\n\t\tsetLabel: notInitialized,\n\t\tgetActiveTools: notInitialized,\n\t\tgetAllTools: notInitialized,\n\t\tsetActiveTools: notInitialized,\n\t\trefreshTools: () => {},\n\t\tgetCommands: notInitialized,\n\t\tsetModel: () => Promise.reject(new Error(\"Extension runtime not initialized\")),\n\t\tgetThinkingLevel: notInitialized,\n\t\tsetThinkingLevel: notInitialized,\n\t\tflagValues: new Map(),\n\t\texplicitFlagNames: new Set(),\n\t\tflagOwners: new Map(),\n\t\tflagOwnerOrigins: new Map(),\n\t\tpendingProviderRegistrations: [],\n\t\tcanRegisterResource: () => true,\n\t\tbeginResourceRegistrationBatch: () => {\n\t\t\tbatchDepth += 1;\n\t\t},\n\t\tendResourceRegistrationBatch: () => {\n\t\t\tbatchDepth -= 1;\n\t\t\tif (batchDepth !== 0) return;\n\t\t\tconst refreshTools = pendingTools.size > 0;\n\t\t\tfor (const pending of pendingTools.values()) pending.extension.tools.set(pending.name, pending.registration);\n\t\t\tfor (const pending of pendingCommands.values())\n\t\t\t\tpending.extension.commands.set(pending.name, pending.registration);\n\t\t\tfor (const pending of pendingFlags.values()) {\n\t\t\t\tpending.extension.flags.set(pending.name, pending.registration);\n\t\t\t\tconst ownerOrigin = runtime.flagOwnerOrigins?.get(pending.name);\n\t\t\t\tif (\n\t\t\t\t\townerOrigin === pending.extension.sourceInfo.configurationOrigin &&\n\t\t\t\t\tpending.defaultValue !== undefined &&\n\t\t\t\t\t!runtime.flagValues.has(pending.name)\n\t\t\t\t) {\n\t\t\t\t\truntime.flagValues.set(pending.name, pending.defaultValue);\n\t\t\t\t}\n\t\t\t}\n\t\t\tfor (const pending of pendingShortcuts.values())\n\t\t\t\tpending.extension.shortcuts.set(pending.name as never, pending.registration);\n\t\t\tpendingTools.clear();\n\t\t\tpendingCommands.clear();\n\t\t\tpendingFlags.clear();\n\t\t\tpendingShortcuts.clear();\n\t\t\tif (refreshTools) runtime.refreshTools();\n\t\t\tif (pendingActiveToolNames) runtime.setActiveTools(pendingActiveToolNames);\n\t\t\tpendingActiveToolNames = undefined;\n\t\t},\n\t\tstageToolRegistration: (extension, name, registration) => {\n\t\t\tif (!shouldStage(extension)) return false;\n\t\t\tpendingTools.set(registrationKey(extension, name), { extension, name, registration });\n\t\t\tif (pendingActiveToolNames && !pendingActiveToolNames.includes(name)) pendingActiveToolNames.push(name);\n\t\t\treturn true;\n\t\t},\n\t\tstageCommandRegistration: (extension, name, registration) => {\n\t\t\tif (!shouldStage(extension)) return false;\n\t\t\tpendingCommands.set(registrationKey(extension, name), { extension, name, registration });\n\t\t\treturn true;\n\t\t},\n\t\tstageFlagRegistration: (extension, name, registration, defaultValue) => {\n\t\t\tif (!shouldStage(extension)) return false;\n\t\t\tconst key = registrationKey(extension, name);\n\t\t\tconst firstDefault = pendingFlags.get(key)?.defaultValue;\n\t\t\tpendingFlags.set(key, { extension, name, registration, defaultValue: firstDefault ?? defaultValue });\n\t\t\truntime.flagOwners ??= new Map();\n\t\t\tconst owners = runtime.flagOwners;\n\t\t\truntime.flagOwnerOrigins ??= new Map();\n\t\t\tconst ownerOrigins = runtime.flagOwnerOrigins;\n\t\t\tif (!owners.has(name)) {\n\t\t\t\towners.set(name, extension.path);\n\t\t\t\townerOrigins.set(name, extension.sourceInfo.configurationOrigin);\n\t\t\t}\n\t\t\treturn true;\n\t\t},\n\t\tstageShortcutRegistration: (extension, name, registration) => {\n\t\t\tif (!shouldStage(extension)) return false;\n\t\t\tpendingShortcuts.set(registrationKey(extension, name), { extension, name, registration });\n\t\t\treturn true;\n\t\t},\n\t\thasPendingResourceRegistration: (extension, resourceType, name) => {\n\t\t\tconst key = registrationKey(extension, name);\n\t\t\tif (resourceType === \"tool\") return pendingTools.has(key);\n\t\t\tif (resourceType === \"command\") return pendingCommands.has(key);\n\t\t\tif (resourceType === \"flag\") return pendingFlags.has(key);\n\t\t\tif (resourceType === \"shortcut\") return pendingShortcuts.has(key);\n\t\t\treturn false;\n\t\t},\n\t\tdeletePendingResourceRegistration: (extension, resourceType, name) => {\n\t\t\tconst key = registrationKey(extension, name);\n\t\t\tif (resourceType === \"tool\") pendingTools.delete(key);\n\t\t\telse if (resourceType === \"command\") pendingCommands.delete(key);\n\t\t\telse if (resourceType === \"flag\") pendingFlags.delete(key);\n\t\t\telse if (resourceType === \"shortcut\") pendingShortcuts.delete(key);\n\t\t},\n\t\tgetPendingFlagDefault: (ownerPath, name) => {\n\t\t\tif (!pendingFlags.has(`${ownerPath}\\0${name}`)) return undefined;\n\t\t\treturn [...pendingFlags.values()].find(\n\t\t\t\t(pending) => pending.name === name && pending.defaultValue !== undefined,\n\t\t\t)?.defaultValue;\n\t\t},\n\t\tgetAllToolsAfterRegistration: (extension) => {\n\t\t\tconst tools = runtime.getAllTools();\n\t\t\tif (extension.sourceInfo.configurationOrigin !== \"inherited-pi\") return tools;\n\t\t\tconst names = new Set(tools.map((tool) => tool.name));\n\t\t\tfor (const pending of pendingTools.values()) {\n\t\t\t\tif (names.has(pending.name)) continue;\n\t\t\t\tconst { definition, sourceInfo } = pending.registration;\n\t\t\t\ttools.push({\n\t\t\t\t\tname: definition.name,\n\t\t\t\t\tdescription: definition.description,\n\t\t\t\t\tparameters: definition.parameters,\n\t\t\t\t\t...(Object.hasOwn(definition, \"constrainedSampling\")\n\t\t\t\t\t\t? { constrainedSampling: definition.constrainedSampling }\n\t\t\t\t\t\t: {}),\n\t\t\t\t\tpromptGuidelines: definition.promptGuidelines,\n\t\t\t\t\tsourceInfo,\n\t\t\t\t});\n\t\t\t\tnames.add(pending.name);\n\t\t\t}\n\t\t\treturn tools;\n\t\t},\n\t\tgetCommandsAfterRegistration: (extension) => {\n\t\t\tif (extension.sourceInfo.configurationOrigin !== \"inherited-pi\") return runtime.getCommands();\n\t\t\tconst active = [...pendingCommands.values()].map((pending) => ({\n\t\t\t\t...pending,\n\t\t\t\tprevious: pending.extension.commands.get(pending.name),\n\t\t\t}));\n\t\t\tfor (const pending of active) pending.extension.commands.set(pending.name, pending.registration);\n\t\t\ttry {\n\t\t\t\treturn runtime.getCommands();\n\t\t\t} finally {\n\t\t\t\tfor (const pending of active) {\n\t\t\t\t\tif (pending.previous) pending.extension.commands.set(pending.name, pending.previous);\n\t\t\t\t\telse pending.extension.commands.delete(pending.name);\n\t\t\t\t}\n\t\t\t}\n\t\t},\n\t\trefreshToolsAfterRegistration: () => {\n\t\t\truntime.refreshTools();\n\t\t\tif (batchDepth > 0 && pendingActiveToolNames) pendingActiveToolNames = runtime.getActiveTools();\n\t\t},\n\t\tapplyFlagDefaultAfterRegistration: (name, _ownerPath, value, configurationOrigin) => {\n\t\t\tif (runtime.flagOwnerOrigins?.get(name) === configurationOrigin && !runtime.flagValues.has(name)) {\n\t\t\t\truntime.flagValues.set(name, value);\n\t\t\t}\n\t\t},\n\t\tgetActiveToolsAfterRegistration: (extension) => {\n\t\t\tconst active = runtime.getActiveTools();\n\t\t\tif (extension.sourceInfo.configurationOrigin !== \"inherited-pi\") return active;\n\t\t\tif (pendingActiveToolNames) return [...pendingActiveToolNames];\n\t\t\tconst names = new Set(active);\n\t\t\tfor (const pending of pendingTools.values()) names.add(pending.name);\n\t\t\treturn [...names];\n\t\t},\n\t\tsetActiveToolsAfterRegistration: (extension, toolNames) => {\n\t\t\tif (batchDepth === 0) return false;\n\t\t\tpendingActiveToolNames = [...toolNames];\n\t\t\tif (extension.sourceInfo.configurationOrigin !== \"inherited-pi\") return false;\n\t\t\tconst liveNames = new Set(runtime.getAllTools().map((tool) => tool.name));\n\t\t\truntime.setActiveTools(toolNames.filter((name) => liveNames.has(name)));\n\t\t\treturn true;\n\t\t},\n\t\tassertActive,\n\t\tinvalidate: (message) => {\n\t\t\tstate.staleMessage ??=\n\t\t\t\tmessage ??\n\t\t\t\t\"This extension ctx is stale after session replacement or reload. Do not use a captured pi or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload().\";\n\t\t},\n\t\tregisterProvider: (nameOrProvider, configOrPath, extensionPath = \"<unknown>\") => {\n\t\t\tif (typeof nameOrProvider === \"string\") {\n\t\t\t\truntime.pendingProviderRegistrations.push({\n\t\t\t\t\tname: nameOrProvider,\n\t\t\t\t\tconfig: configOrPath as import(\"./types.ts\").ProviderConfig,\n\t\t\t\t\textensionPath: extensionPath as string,\n\t\t\t\t});\n\t\t\t} else {\n\t\t\t\truntime.pendingProviderRegistrations.push({\n\t\t\t\t\tprovider: nameOrProvider,\n\t\t\t\t\textensionPath: typeof configOrPath === \"string\" ? configOrPath : (extensionPath as string),\n\t\t\t\t});\n\t\t\t}\n\t\t},\n\t\tunregisterProvider: (name) => {\n\t\t\truntime.pendingProviderRegistrations = runtime.pendingProviderRegistrations.filter((registration) =>\n\t\t\t\t\"provider\" in registration ? registration.provider.id !== name : registration.name !== name,\n\t\t\t);\n\t\t},\n\t};\n\treturn runtime;\n}\n"]}

package/dist/core/extensions/loader-virtual-modules.d.ts

@@ -7,6 +7,19 @@ declare function loadVirtualModules(): Promise<Record<string, object>>;
  * in the background.
  */
 declare function getTranspileCacheDir(): string;
+/**
+ * Per-extension-path record of the complete file graph a transformed (jiti)
+ * evaluation read, mapping each absolute file path to the SHA256 hex digest of
+ * its content at evaluation time. Persisted under the versioned transpile
+ * cache directory so pruneStaleTranspileCaches handles version cleanup.
+ */
+export interface ExtensionGraphManifest {
+    version: 1;
+    extensionPath: string;
+    files: Record<string, string>;
+}
+declare function graphManifestPath(extensionPath: string): string;
+export declare function readExtensionGraphManifest(extensionPath: string): ExtensionGraphManifest | undefined;
 export interface ExtensionCacheToken {
     cwd: string;
     generation: number;
@@ -33,6 +46,10 @@ export declare const extensionLoaderTestHooks: {
     getAliases: typeof getAliases;
     findPackageRoot: typeof findPackageRoot;
     getTranspileCacheDir: typeof getTranspileCacheDir;
+    graphManifestPath: typeof graphManifestPath;
+    readExtensionGraphManifest: typeof readExtensionGraphManifest;
+    loadExtensionModuleTransformed: (extensionPath: string, cacheToken?: ExtensionCacheToken) => Promise<ExtensionFactory | undefined>;
+    loadTransformedExtensionModule: (extensionPath: string, cacheToken?: ExtensionCacheToken) => Promise<ExtensionFactory | undefined>;
 };
 export declare function loadExtensionModule(extensionPath: string, cacheToken?: ExtensionCacheToken): Promise<ExtensionFactory | undefined>;
 export {};

package/dist/core/extensions/loader-virtual-modules.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"loader-virtual-modules.d.ts","sourceRoot":"","sources":["../../../src/core/extensions/loader-virtual-modules.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAMnD,iBAAe,kBAAkB,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAqCnE;AAoBD;;;;;GAKG;AACH,iBAAS,oBAAoB,IAAI,MAAM,CAMtC;AAqBD,MAAM,WAAW,mBAAmB;IAClC,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,wBAAgB,mBAAmB,IAAI,IAAI,CAI1C;AAED,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,MAAM,GAAG,mBAAmB,CAOrE;AAiBD;;;;;;;;GAQG;AACH,iBAAS,eAAe,CAAC,WAAW,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAQ5E;AAMD;;;GAGG;AACH,iBAAS,UAAU,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAuD5C;AAED,eAAO,MAAM,wBAAwB;;;;;CAKpC,CAAC;AAUF,wBAAsB,mBAAmB,CACvC,aAAa,EAAE,MAAM,EACrB,UAAU,CAAC,EAAE,mBAAmB,GAC/B,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC,CAmCvC"}
+{"version":3,"file":"loader-virtual-modules.d.ts","sourceRoot":"","sources":["../../../src/core/extensions/loader-virtual-modules.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAMnD,iBAAe,kBAAkB,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAsCnE;AAoBD;;;;;GAKG;AACH,iBAAS,oBAAoB,IAAI,MAAM,CAMtC;AAiBD;;;;;GAKG;AACH,MAAM,WAAW,sBAAsB;IACtC,OAAO,EAAE,CAAC,CAAC;IACX,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC9B;AAWD,iBAAS,iBAAiB,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM,CAGxD;AAgFD,wBAAgB,0BAA0B,CAAC,aAAa,EAAE,MAAM,GAAG,sBAAsB,GAAG,SAAS,CAYpG;AA8GD,MAAM,WAAW,mBAAmB;IACnC,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,wBAAgB,mBAAmB,IAAI,IAAI,CAI1C;AAED,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,MAAM,GAAG,mBAAmB,CAOrE;AAiBD;;;;;;;;GAQG;AACH,iBAAS,eAAe,CAAC,WAAW,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAQ5E;AAKD;;;GAGG;AACH,iBAAS,UAAU,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAuD5C;AAED,eAAO,MAAM,wBAAwB;;;;;;;oDAOY,MAAM,eAAe,mBAAmB;oDAExC,MAAM,eAAe,mBAAmB;CAExF,CAAC;AAgFF,wBAAsB,mBAAmB,CACxC,aAAa,EAAE,MAAM,EACrB,UAAU,CAAC,EAAE,mBAAmB,GAC9B,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC,CA4BvC"}

package/dist/core/extensions/loader-virtual-modules.js

@@ -1,12 +1,13 @@
+import * as crypto from "node:crypto";
 import * as fs from "node:fs";
 import { createRequire } from "node:module";
 import * as path from "node:path";
 import { pathToFileURL } from "node:url";
 import { createJiti } from "jiti/static";
 import { getExtensionTranspileCacheDir, isBunBinary, isBundledBuild } from "../../config.js";
-import { moduleDirFromMetaUrl } from "../../utils/split-launcher.js";
 import { resolutionBaseUrl } from "../../utils/module-require.js";
 import { resolvePath } from "../../utils/paths.js";
+import { moduleDirFromMetaUrl } from "../../utils/split-launcher.js";
 const require = createRequire(import.meta.url);
 let _virtualModules = null;
 let _virtualModulesPromise = null;
@@ -22,7 +23,7 @@ async function loadVirtualModules() {
         // "@earendil-works/pi-ai"`, so we load the compat module here and key it
         // under the root specifier below to keep every extension working unchanged.
         import("@earendil-works/pi-ai/compat"),
-        import("../oauth-compat.js"),
+        import("@earendil-works/pi-ai/oauth"),
         // NOTE: This import works because loader.ts exports are NOT re-exported from index.ts,
         // avoiding a circular dependency while preserving the package-name extension import path.
         import("../../index.js"),
@@ -86,6 +87,215 @@ function pruneStaleTranspileCaches(currentDir) {
         .map((entry) => fs.promises.rm(path.join(parent, entry), { recursive: true, force: true }))))
         .catch(() => { });
 }
+const GRAPH_MANIFEST_VERSION = 1;
+const GRAPH_MANIFEST_DIR_NAME = "graph-manifests";
+const graphManifestFileExtensions = new Set([".ts", ".tsx", ".mts", ".cts", ".js", ".mjs", ".cjs", ".jsx", ".json"]);
+function graphManifestDir() {
+    return path.join(getTranspileCacheDir(), GRAPH_MANIFEST_DIR_NAME);
+}
+function graphManifestPath(extensionPath) {
+    const key = crypto.createHash("sha256").update(extensionPath).digest("hex").slice(0, 32);
+    return path.join(graphManifestDir(), `${key}.json`);
+}
+function sha256Hex(content) {
+    return crypto.createHash("sha256").update(content).digest("hex");
+}
+// jiti's CJS bundle reads every module source in the require graph through
+// fs.readFileSync (even on fsCache hits, since the cache key is a hash of the
+// source), so observing readFileSync while a transformed import is in flight
+// sees the complete transitive graph. The observer records into every active
+// recorder; unrelated reads that interleave only over-record, which at worst
+// forces a redundant re-evaluation later — never a stale one.
+const fsExports = require("node:fs");
+const activeGraphRecorders = new Set();
+let observedReadFileSync = null;
+function recordObservedRead(target, content) {
+    if (typeof target !== "string" || !path.isAbsolute(target))
+        return;
+    if (!graphManifestFileExtensions.has(path.extname(target).toLowerCase()))
+        return;
+    const normalizedTarget = path.normalize(target);
+    const cachePrefix = path.normalize(getTranspileCacheDir()) + path.sep;
+    if (normalizedTarget.toLowerCase().startsWith(cachePrefix.toLowerCase()))
+        return;
+    const hash = sha256Hex(content);
+    for (const recorder of activeGraphRecorders) {
+        recorder.set(normalizedTarget, hash);
+    }
+}
+function installReadFileSyncObserver() {
+    if (observedReadFileSync)
+        return;
+    const original = fsExports.readFileSync;
+    observedReadFileSync = original;
+    const wrapped = ((...args) => {
+        const content = original(...args);
+        recordObservedRead(args[0], content);
+        return content;
+    });
+    fsExports.readFileSync = wrapped;
+}
+function uninstallReadFileSyncObserver() {
+    if (!observedReadFileSync || activeGraphRecorders.size > 0)
+        return;
+    fsExports.readFileSync = observedReadFileSync;
+    observedReadFileSync = null;
+}
+function writeGraphManifest(manifest) {
+    const manifestFile = graphManifestPath(manifest.extensionPath);
+    let tempFile;
+    try {
+        fs.mkdirSync(path.dirname(manifestFile), { recursive: true });
+        tempFile = `${manifestFile}.${process.pid}.${Math.random().toString(36).slice(2)}.tmp`;
+        fs.writeFileSync(tempFile, JSON.stringify(manifest));
+        fs.renameSync(tempFile, manifestFile);
+    }
+    catch {
+        // Manifest persistence is best-effort: a missing manifest only means the
+        // next reload cannot prove the graph unchanged and re-evaluates. A failed
+        // replacement must not leave an older manifest behind describing a graph
+        // that no longer matches this evaluation, so invalidate the destination
+        // first, independently of removing the temporary file.
+        try {
+            fs.rmSync(manifestFile, { force: true });
+        }
+        catch {
+            // A leftover stale manifest cannot be reused without also matching
+            // the in-memory manifest of the same evaluation, and readers
+            // tolerate missing or corrupt files.
+        }
+        try {
+            if (tempFile)
+                fs.rmSync(tempFile, { force: true });
+        }
+        catch {
+            // Temporary files carry a unique suffix and are never read back.
+        }
+    }
+}
+function isStringRecord(value) {
+    if (typeof value !== "object" || value === null || Array.isArray(value))
+        return false;
+    return Object.values(value).every((entry) => typeof entry === "string");
+}
+export function readExtensionGraphManifest(extensionPath) {
+    try {
+        const parsed = JSON.parse(fs.readFileSync(graphManifestPath(extensionPath), "utf-8"));
+        if (parsed.version !== GRAPH_MANIFEST_VERSION)
+            return undefined;
+        if (parsed.extensionPath !== extensionPath)
+            return undefined;
+        if (!isStringRecord(parsed.files))
+            return undefined;
+        return { version: GRAPH_MANIFEST_VERSION, extensionPath, files: parsed.files };
+    }
+    catch {
+        return undefined;
+    }
+}
+async function recordExtensionGraph(extensionPath, load) {
+    const files = new Map();
+    activeGraphRecorders.add(files);
+    installReadFileSyncObserver();
+    let result;
+    try {
+        result = await load();
+    }
+    finally {
+        activeGraphRecorders.delete(files);
+        uninstallReadFileSyncObserver();
+    }
+    const manifest = {
+        version: GRAPH_MANIFEST_VERSION,
+        extensionPath,
+        files: Object.fromEntries([...files.entries()].sort(([a], [b]) => a.localeCompare(b))),
+    };
+    writeGraphManifest(manifest);
+    return { result, manifest };
+}
+/**
+ * Factories from transformed evaluations, retained across extension cache
+ * generation bumps and paired with the graph manifest recorded by the
+ * evaluation that produced them. A retained factory is reused on a reload only
+ * when the persisted manifest still equals the paired one and every file in it
+ * rehashes to its recorded digest; any doubt (missing or corrupt manifest,
+ * missing file, hash mismatch, changed graph shape) falls back to a fresh
+ * transformed re-evaluation, which re-records the manifest.
+ */
+const retainedTransformedLoads = new Map();
+function manifestFilesEqual(a, b) {
+    const aEntries = Object.entries(a);
+    if (aEntries.length !== Object.keys(b).length)
+        return false;
+    return aEntries.every(([filePath, hash]) => b[filePath] === hash);
+}
+function graphUnchangedSince(manifest) {
+    const stored = readExtensionGraphManifest(manifest.extensionPath);
+    if (!stored || !manifestFilesEqual(stored.files, manifest.files))
+        return false;
+    const entries = Object.entries(manifest.files);
+    if (entries.length === 0)
+        return false;
+    for (const [filePath, hash] of entries) {
+        try {
+            if (sha256Hex(fs.readFileSync(filePath)) !== hash)
+                return false;
+        }
+        catch {
+            return false;
+        }
+    }
+    return true;
+}
+/**
+ * A factory can defer loading local code until it is invoked (e.g. a
+ * `require()` inside the factory body). Those reads are invisible to the
+ * import-time recording, so retained factories are wrapped to record reads
+ * that happen while the factory executes and merge them into the retained and
+ * persisted manifest — editing a deferred dependency then invalidates reuse
+ * like any other graph file.
+ */
+function withDeferredGraphRecording(extensionPath, factory) {
+    return (pi) => {
+        const files = new Map();
+        activeGraphRecorders.add(files);
+        installReadFileSyncObserver();
+        const finish = () => {
+            activeGraphRecorders.delete(files);
+            uninstallReadFileSyncObserver();
+            if (files.size > 0)
+                mergeDeferredGraphReads(extensionPath, files);
+        };
+        try {
+            const result = factory(pi);
+            if (result !== null &&
+                (typeof result === "object" || typeof result === "function") &&
+                typeof result.then === "function") {
+                return Promise.resolve(result).finally(finish);
+            }
+            finish();
+            return result;
+        }
+        catch (error) {
+            finish();
+            throw error;
+        }
+    };
+}
+function mergeDeferredGraphReads(extensionPath, files) {
+    const retained = retainedTransformedLoads.get(extensionPath);
+    if (!retained)
+        return;
+    let changed = false;
+    for (const [filePath, hash] of files) {
+        if (retained.manifest.files[filePath] !== hash) {
+            retained.manifest.files[filePath] = hash;
+            changed = true;
+        }
+    }
+    if (changed)
+        writeGraphManifest(retained.manifest);
+}
 let extensionCacheCwd;
 let extensionCacheGeneration = 0;
 const extensionCache = new Map();
@@ -163,8 +373,8 @@ function getAliases() {
     // upstream layout change moves these files, this join needs updating to
     // match the package's real dist paths.
     const piAiEntry = resolveWorkspaceOrImport("ai/dist/compat.js", "@earendil-works/pi-ai");
+    const piAiOauthEntry = resolveWorkspaceOrImport("ai/dist/oauth.js", "@earendil-works/pi-ai");
     const piAiProvidersEntry = resolveWorkspaceOrImport("ai/dist/providers/all.js", "@earendil-works/pi-ai");
-    const piAiOauthEntry = path.resolve(__dirname, "../oauth-compat.js");
     _aliases = {
         "@bastani/atomic": piCodingAgentEntry,
         "@earendil-works/pi-coding-agent": piCodingAgentEntry,
@@ -194,6 +404,10 @@ export const extensionLoaderTestHooks = {
     getAliases,
     findPackageRoot,
     getTranspileCacheDir,
+    graphManifestPath,
+    readExtensionGraphManifest,
+    loadExtensionModuleTransformed: (extensionPath, cacheToken) => importExtensionModule(extensionPath, cacheToken, true),
+    loadTransformedExtensionModule: (extensionPath, cacheToken) => loadTransformedExtensionModule(extensionPath, cacheToken),
 };
 /**
  * Extension paths already evaluated via native import() in this process. Bun on
@@ -202,22 +416,9 @@ export const extensionLoaderTestHooks = {
  * fresh module evaluation.
  */
 const nativelyImportedPaths = new Set();
-export async function loadExtensionModule(extensionPath, cacheToken) {
-    if (isCurrentCacheToken(cacheToken)) {
-        const cachedFactory = extensionCache.get(extensionPath);
-        if (cachedFactory)
-            return cachedFactory;
-    }
+async function importExtensionModule(extensionPath, cacheToken, forceTransformedImports) {
     const isWindows = process.platform === "win32";
-    // Single-file builds (compiled binary or dev bundle) cannot alias host
-    // package specifiers to files on disk: extensions must share the live
-    // module instances baked into the build, so virtualModules is used instead
-    // (which requires jiti's transformed-import path).
     const isSingleFileBuild = isBunBinary || isBundledBuild;
-    // Windows first-load fast path: native import() (jiti's default tryNative)
-    // skips per-launch transpilation of the extension module graph. Re-loads of
-    // the same path fall back to transformed imports for fresh evaluation.
-    const forceTransformedImports = isSingleFileBuild || (isWindows && nativelyImportedPaths.has(extensionPath));
     const jiti = createJiti(resolutionBaseUrl(import.meta.url), {
         moduleCache: false,
         ...(forceTransformedImports
@@ -227,16 +428,84 @@ export async function loadExtensionModule(extensionPath, cacheToken) {
                 : {}),
         ...(isSingleFileBuild ? { virtualModules: await getVirtualModules() } : { alias: getAliases() }),
     });
-    const module = await jiti.import(extensionImportSpecifier(extensionPath, cacheToken), { default: true });
+    const specifier = extensionImportSpecifier(extensionPath, cacheToken);
+    // Transformed evaluations are the loads whose repeat cost is the Windows
+    // /reload regression, so they are also the loads whose file graph is
+    // recorded into a content-hash manifest.
+    let module;
+    let recordedManifest;
+    if (forceTransformedImports) {
+        const recorded = await recordExtensionGraph(extensionPath, () => jiti.import(specifier, { default: true }));
+        module = recorded.result;
+        recordedManifest = recorded.manifest;
+    }
+    else {
+        module = await jiti.import(specifier, { default: true });
+    }
     if (isWindows && !forceTransformedImports) {
         nativelyImportedPaths.add(extensionPath);
     }
-    const factory = module;
+    let factory = module;
     if (typeof factory !== "function")
         return undefined;
+    if (recordedManifest) {
+        factory = withDeferredGraphRecording(extensionPath, factory);
+        retainedTransformedLoads.set(extensionPath, { factory, manifest: recordedManifest });
+    }
     if (isCurrentCacheToken(cacheToken)) {
         extensionCache.set(extensionPath, factory);
     }
     return factory;
 }
+/**
+ * Load an extension through jiti's transformed-import path, reusing the
+ * previously evaluated factory when the recorded graph manifest proves the
+ * entire transitive file graph is byte-identical. This lets the
+ * unchanged-graph case survive the /reload generation bump instead of
+ * re-evaluating hundreds of files; anything else re-evaluates as before.
+ *
+ * Note: when the graph is byte-identical, the previously evaluated module
+ * instance (including any mutable module-scoped state it closed over) is
+ * reused. An extension that requires clean module state on every reload must
+ * change a file in its graph to force re-evaluation.
+ */
+async function loadTransformedExtensionModule(extensionPath, cacheToken) {
+    const retained = retainedTransformedLoads.get(extensionPath);
+    if (retained && graphUnchangedSince(retained.manifest)) {
+        if (isCurrentCacheToken(cacheToken)) {
+            extensionCache.set(extensionPath, retained.factory);
+        }
+        return retained.factory;
+    }
+    return importExtensionModule(extensionPath, cacheToken, true);
+}
+export async function loadExtensionModule(extensionPath, cacheToken) {
+    if (isCurrentCacheToken(cacheToken)) {
+        const cachedFactory = extensionCache.get(extensionPath);
+        if (cachedFactory)
+            return cachedFactory;
+    }
+    const isWindows = process.platform === "win32";
+    // Single-file builds (compiled binary or dev bundle) cannot alias host
+    // package specifiers to files on disk: extensions must share the live
+    // module instances baked into the build, so virtualModules is used instead
+    // (which requires jiti's transformed-import path).
+    const isSingleFileBuild = isBunBinary || isBundledBuild;
+    // Windows first-load fast path: native import() (jiti's default tryNative)
+    // skips per-launch transpilation of the extension module graph. Re-loads of
+    // the same path fall back to transformed imports for fresh evaluation.
+    //
+    // That fallback is not free. Measured on Windows CI, a transformed re-import
+    // costs ~15 ms per module file in the extension's transitive graph, so a
+    // reload of all five builtin packages (621 files) takes ~10.7 s cold and
+    // ~2.5 s against a warm jiti fsCache, against ~40 ms on Linux and macOS. The
+    // cost is correctness-driven and is paid interactively on every Windows
+    // `/reload`; removing it needs a content-hash-keyed evaluation cache or a
+    // narrower trigger, not a larger timeout.
+    const forceTransformedImports = isSingleFileBuild || (isWindows && nativelyImportedPaths.has(extensionPath));
+    if (forceTransformedImports) {
+        return loadTransformedExtensionModule(extensionPath, cacheToken);
+    }
+    return importExtensionModule(extensionPath, cacheToken, false);
+}
 //# sourceMappingURL=loader-virtual-modules.js.map