typescript 7.1.0-dev.20260924.1 → 7.1.0-dev.20260925.1

package/dist/api/sync/api.js

@@ -21,7 +21,7 @@ import { TypePredicateKind } from "#enums/typePredicateKind";
 import { ModifierFlags, unescapeLeadingUnderscores, } from "../../ast/index.js";
 import { assertNever } from "../../internal/utils.js";
 import { encodeNode, uint8ArrayToBase64, } from "../node/encoder.js";
-import { decodeNode, getNodeId, parseNodeHandle, readParseOptionsKey, readSourceFileHash, RemoteSourceFile, } from "../node/node.js";
+import { decodeNode, getNodeId, parseNodeHandle, readParseOptionsKey, readSourceFileHash, readSourceFileLease, RemoteSourceFile, } from "../node/node.js";
 import { Wtf8Decoder } from "../node/wtf8.js";
 import { createGetCanonicalFileName, toPath, } from "../path.js";
 import { resolveFileName, toCreateSnapshotRequest, } from "../proto.js";
@@ -59,6 +59,7 @@ export class API {
     initialized = false;
     initializing;
     activeSnapshots = new Map();
+    activeSourceFileLeases = new Map();
     printer;
     internal;
     constructor(options = {}) {
@@ -185,6 +186,10 @@ export class API {
             return yield* apiRequest("parseJsonConfigFileContent", { json, ...options });
         });
     }
+    /**
+     * Create and retain a source file independently of a program.
+     * Dispose the returned lease when the source file no longer needs to remain available remotely.
+     */
     get createSourceFile() {
         const owner = this;
         return cacheGeneratorMethod(owner, "createSourceFile", function (fileName, sourceText, options = {}) {
@@ -193,16 +198,20 @@ export class API {
             if (!data) {
                 throw new Error("createSourceFile returned no source file");
             }
-            return new RemoteSourceFile(data, owner.decoder, owner.client.getTimingCollector());
+            return owner.retainSourceFileResponse(data);
         }, function* (fileName, sourceText, options = {}) {
             yield* owner.ensureInitialized.gen();
             const data = sourceFileResponseToUint8Array(yield* apiRequest("createSourceFile", { fileName, sourceText, options }));
             if (!data) {
                 throw new Error("createSourceFile returned no source file");
             }
-            return new RemoteSourceFile(data, owner.decoder, owner.client.getTimingCollector());
+            return owner.retainSourceFileResponse(data);
         });
     }
+    /**
+     * Read, create, and retain a source file independently of a program.
+     * Dispose the returned lease when the source file no longer needs to remain available remotely.
+     */
     get createSourceFileFromFile() {
         const owner = this;
         return cacheGeneratorMethod(owner, "createSourceFileFromFile", function (file, options = {}) {
@@ -211,16 +220,37 @@ export class API {
             if (!data) {
                 throw new Error("createSourceFileFromFile returned no source file");
             }
-            return new RemoteSourceFile(data, owner.decoder, owner.client.getTimingCollector());
+            return owner.retainSourceFileResponse(data);
         }, function* (file, options = {}) {
             yield* owner.ensureInitialized.gen();
             const data = sourceFileResponseToUint8Array(yield* apiRequest("createSourceFileFromFile", { fileName: resolveFileName(file), options }));
             if (!data) {
                 throw new Error("createSourceFileFromFile returned no source file");
             }
-            return new RemoteSourceFile(data, owner.decoder, owner.client.getTimingCollector());
+            return owner.retainSourceFileResponse(data);
         });
     }
+    retainSourceFileResponse(data) {
+        const view = new DataView(data.buffer, data.byteOffset, data.byteLength);
+        const lease = readSourceFileLease(view);
+        try {
+            const decoded = new RemoteSourceFile(data, this.decoder, this.client.getTimingCollector());
+            const sourceFile = this.sourceFileCache.setForLease(decoded.path, decoded, readParseOptionsKey(view), readSourceFileHash(view), lease);
+            const retained = new RetainedSourceFile(sourceFile, lease, this.client, () => {
+                this.activeSourceFileLeases.delete(lease);
+                this.sourceFileCache.releaseLease(lease);
+            });
+            this.activeSourceFileLeases.set(lease, retained);
+            return retained;
+        }
+        catch (error) {
+            try {
+                this.client.apiRequest("releaseSourceFile", { lease });
+            }
+            catch { }
+            throw error;
+        }
+    }
     get transpileModule() {
         const owner = this;
         return cacheGeneratorMethod(owner, "transpileModule", function (input, options = {}) {
@@ -420,26 +450,38 @@ export class API {
     get close() {
         const owner = this;
         return cacheGeneratorMethod(owner, "close", function () {
-            // Dispose all active snapshots
             try {
-                for (const snapshot of [...owner.activeSnapshots.values()]) {
-                    snapshot.dispose();
+                for (const retained of [...owner.activeSourceFileLeases.values()]) {
+                    retained.dispose();
                 }
-                owner.sourceFileCache.clear();
             }
             finally {
-                owner.client.close(); // always close the underlying connection
+                try {
+                    for (const snapshot of [...owner.activeSnapshots.values()]) {
+                        snapshot.dispose();
+                    }
+                    owner.sourceFileCache.clear();
+                }
+                finally {
+                    owner.client.close(); // always close the underlying connection
+                }
             }
         }, function* () {
-            // Dispose all active snapshots
             try {
-                for (const snapshot of [...owner.activeSnapshots.values()]) {
-                    yield* snapshot.dispose.gen();
+                for (const retained of [...owner.activeSourceFileLeases.values()]) {
+                    yield* retained.dispose.gen();
                 }
-                owner.sourceFileCache.clear();
             }
             finally {
-                owner.client.close(); // always close the underlying connection
+                try {
+                    for (const snapshot of [...owner.activeSnapshots.values()]) {
+                        yield* snapshot.dispose.gen();
+                    }
+                    owner.sourceFileCache.clear();
+                }
+                finally {
+                    owner.client.close(); // always close the underlying connection
+                }
             }
         });
     }
@@ -582,6 +624,56 @@ export class API {
         });
     }
 }
+/** An independently retained source file and its disposable remote-lifetime lease. */
+export class RetainedSourceFile {
+    sourceFile;
+    lease;
+    client;
+    onDispose;
+    disposed = false;
+    disposePromise;
+    constructor(sourceFile, lease, client, onDispose) {
+        this.sourceFile = sourceFile;
+        this.lease = lease;
+        this.client = client;
+        this.onDispose = onDispose;
+    }
+    [globalThis.Symbol.dispose]() {
+        this.dispose();
+    }
+    get dispose() {
+        const owner = this;
+        return cacheGeneratorMethod(owner, "dispose", function () {
+            return owner.disposePromise ??= owner.disposeWorker();
+        }, function* () {
+            return owner.disposePromise ??= yield* owner.disposeWorker.gen();
+        });
+    }
+    get disposeWorker() {
+        const owner = this;
+        return cacheGeneratorMethod(owner, "disposeWorker", function () {
+            if (owner.disposed)
+                return;
+            owner.disposed = true;
+            try {
+                owner.client.apiRequest("releaseSourceFile", { lease: owner.lease });
+            }
+            finally {
+                owner.onDispose();
+            }
+        }, function* () {
+            if (owner.disposed)
+                return;
+            owner.disposed = true;
+            try {
+                yield* apiRequest("releaseSourceFile", { lease: owner.lease });
+            }
+            finally {
+                owner.onDispose();
+            }
+        });
+    }
+}
 export class InternalAPI {
     client;
     ensureInitialized;