@goast/typescript 0.6.3-beta.1 → 0.7.0

package/LICENSE

@@ -1,21 +1,21 @@
-MIT License
-
-Copyright (c) 2021 Marc Schmidt
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+MIT License
+
+Copyright (c) 2021 Marc Schmidt
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/assets/stubs/easy-network-stub/easy-network-stub.utils.ts

@@ -1,4 +1,4 @@
-import type { EasyNetworkStub, ErrorResponse, HttpMethod, RouteResponseCallback } from 'easy-network-stub';
+import type { EasyNetworkStub, ErrorResponse, HttpMethod, RouteResponseCallback, StubHandle } from 'easy-network-stub';
 
 /**
  * Options for the `EasyNetworkStubWrapper`.
@@ -128,8 +128,8 @@ export class EasyNetworkStubWrapper {
     method: HttpMethod,
     route: Route,
     response: RouteResponseCallback<Route, unknown>,
-  ): void {
-    this.wrappedStub.stub<Route>(method, route, async (request) => {
+  ): StubHandle {
+    return this.wrappedStub.stub<Route>(method, route, async (request) => {
       if (this.options.delay > 0) await sleep(this.options.delay);
       return await this.runRequest(method, route, response, request);
     });
@@ -152,9 +152,9 @@ export class EasyNetworkStubWrapper {
     method: HttpMethod,
     route: Route,
     response: RouteResponseCallback<Route, T>,
-  ) => void {
+  ) => StubHandle {
     return <Route extends string>(method: HttpMethod, route: Route, response: RouteResponseCallback<Route, T>) => {
-      this.wrappedStub.stub2<T>()<Route>(method, route, async (request) => {
+      return this.wrappedStub.stub2<T>()<Route>(method, route, async (request) => {
         if (this.options.delay > 0) await sleep(this.options.delay);
         return await this.runRequest(method, route, response, request);
       });
@@ -166,7 +166,7 @@ export class EasyNetworkStubWrapper {
    */
   public reset(): void {
     this._requests = [];
-    this.wrappedStub['_config'].stubs = [];
+    this.wrappedStub.reset();
   }
 
   private async runRequest<Route extends string, T>(
@@ -219,7 +219,9 @@ export class EasyNetworkStubBase {
   }
 
   public reset() {
-    this.stubWrapper.reset();
+    if (this.ownsStubWrapper) {
+      this.stubWrapper.reset();
+    }
   }
 }
 

package/esm/src/generators/services/easy-network-stub/easy-network-stub-generator.d.ts

@@ -23,8 +23,11 @@ export declare class DefaultTypeScriptEasyNetworkStubGenerator extends TypeScrip
     protected getStubClassName(ctx: Context): string;
     protected getEndpointPathPropertyName(ctx: Context, endpoint: ApiEndpoint): string;
     protected getEndpointRequestsFieldName(ctx: Context, endpoint: ApiEndpoint): string;
+    protected getEndpointStubHandleFieldName(ctx: Context, endpoint: ApiEndpoint): string;
     protected getEndpointRequestsPropertyName(ctx: Context, endpoint: ApiEndpoint): string;
     protected getEndpointStubMethodName(ctx: Context, endpoint: ApiEndpoint): string;
+    protected getEndpointUnregisterMethodName(ctx: Context, endpoint: ApiEndpoint): string;
+    protected getEndpointUnregisterMethod(ctx: Context, endpoint: ApiEndpoint): ts.Method<Builder>;
     protected getStubRoute(ctx: Context, endpoint: ApiEndpoint): string;
     protected getEndpointStatusCodes(ctx: Context, endpoint: ApiEndpoint): Record<number, ts.Type<Builder>>;
     protected getSchemaType(ctx: Context, schema: ApiSchema | undefined, fallback?: ts.Type<Builder>): ts.Type<Builder>;

package/esm/src/generators/services/easy-network-stub/easy-network-stub-generator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"easy-network-stub-generator.d.ts","sourceRoot":"","sources":["../../../../../src/src/generators/services/easy-network-stub/easy-network-stub-generator.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,gBAAgB,EAIrB,KAAK,YAAY,EAElB,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,EAAE,EAAE,MAAM,uBAAuB,CAAC;AAC3C,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAClE,OAAO,KAAK,EAAE,yCAAyC,EAAE,wCAAwC,EAAE,MAAM,aAAa,CAAC;AAEvH,KAAK,OAAO,GAAG,yCAAyC,CAAC;AACzD,KAAK,MAAM,GAAG,wCAAwC,CAAC;AACvD,KAAK,OAAO,GAAG,qBAAqB,CAAC;AAErC,MAAM,WAAW,kCAAkC,CAAC,OAAO,SAAS,MAAM,GAAG,MAAM;IACjF,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;CAC/C;AAED,qBAAa,yCAA0C,SAAQ,uBAAuB,CAAC,OAAO,EAAE,MAAM,CACpG,YAAW,kCAAkC;IACtC,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC;IAwBnD,SAAS,CAAC,kBAAkB,CAAC,GAAG,EAAE,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC;IAOrE,SAAS,CAAC,oBAAoB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;IAgBzF,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;IAoDvD,SAAS,CAAC,uBAAuB,CAC/B,GAAG,EAAE,OAAO,EACZ,QAAQ,EAAE,WAAW,EACrB,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GAC/B,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;IAUnB,SAAS,CAAC,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;IAiCxF,SAAS,CAAC,eAAe,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM;IAQ/C,SAAS,CAAC,wBAAwB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAI/E,SAAS,CAAC,gBAAgB,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM;IAIhD,SAAS,CAAC,2BAA2B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIlF,SAAS,CAAC,4BAA4B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAInF,SAAS,CAAC,+BAA+B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAItF,SAAS,CAAC,yBAAyB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIhF,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAyBnE,SAAS,CAAC,sBAAsB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAYvG,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,GAAG,SAAS,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;CAKpH"}
+{"version":3,"file":"easy-network-stub-generator.d.ts","sourceRoot":"","sources":["../../../../../src/src/generators/services/easy-network-stub/easy-network-stub-generator.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,gBAAgB,EAIrB,KAAK,YAAY,EAElB,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,EAAE,EAAE,MAAM,uBAAuB,CAAC;AAC3C,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAClE,OAAO,KAAK,EAAE,yCAAyC,EAAE,wCAAwC,EAAE,MAAM,aAAa,CAAC;AAEvH,KAAK,OAAO,GAAG,yCAAyC,CAAC;AACzD,KAAK,MAAM,GAAG,wCAAwC,CAAC;AACvD,KAAK,OAAO,GAAG,qBAAqB,CAAC;AAErC,MAAM,WAAW,kCAAkC,CAAC,OAAO,SAAS,MAAM,GAAG,MAAM;IACjF,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;CAC/C;AAED,qBAAa,yCAA0C,SAAQ,uBAAuB,CAAC,OAAO,EAAE,MAAM,CACpG,YAAW,kCAAkC;IACtC,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC;IAwBnD,SAAS,CAAC,kBAAkB,CAAC,GAAG,EAAE,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC;IAOrE,SAAS,CAAC,oBAAoB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;IAgBzF,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;IA+DvD,SAAS,CAAC,uBAAuB,CAC/B,GAAG,EAAE,OAAO,EACZ,QAAQ,EAAE,WAAW,EACrB,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GAC/B,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;IAUnB,SAAS,CAAC,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;IAuCxF,SAAS,CAAC,eAAe,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM;IAQ/C,SAAS,CAAC,wBAAwB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAI/E,SAAS,CAAC,gBAAgB,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM;IAIhD,SAAS,CAAC,2BAA2B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIlF,SAAS,CAAC,4BAA4B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAInF,SAAS,CAAC,8BAA8B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIrF,SAAS,CAAC,+BAA+B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAItF,SAAS,CAAC,yBAAyB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIhF,SAAS,CAAC,+BAA+B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAItF,SAAS,CAAC,2BAA2B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;IAc9F,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAyBnE,SAAS,CAAC,sBAAsB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAYvG,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,GAAG,SAAS,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;CAKpH"}

package/esm/src/generators/services/easy-network-stub/easy-network-stub-generator.js

@@ -55,6 +55,11 @@ export class DefaultTypeScriptEasyNetworkStubGenerator extends TypeScriptFileGen
                     value: ts.tuple([]),
                 })),
                 '\n',
+                ...ctx.service.endpoints.map((endpoint) => ts.property(this.getEndpointStubHandleFieldName(ctx, endpoint), {
+                    accessModifier: 'private',
+                    type: s `${ctx.refs.stubHandle()} | undefined`,
+                })),
+                '\n',
                 ...ctx.service.endpoints.map((endpoint) => ts.property(this.getEndpointRequestsPropertyName(ctx, endpoint), {
                     readonly: true,
                     accessModifier: 'public',
@@ -64,11 +69,13 @@ export class DefaultTypeScriptEasyNetworkStubGenerator extends TypeScriptFileGen
                     }),
                 })),
                 ...ctx.service.endpoints.map((endpoint) => this.getEndpointStubMethod(ctx, endpoint)),
+                ...ctx.service.endpoints.map((endpoint) => this.getEndpointUnregisterMethod(ctx, endpoint)),
                 ts.method('reset', {
                     accessModifier: 'public',
                     override: true,
                     returnType: ts.refs.void_(),
                     body: appendValueGroup([
+                        ...ctx.service.endpoints.map((endpoint) => `this.${this.getEndpointStubHandleFieldName(ctx, endpoint)} = undefined;`),
                         ...ctx.service.endpoints.map((endpoint) => `this.${this.getEndpointRequestsFieldName(ctx, endpoint)}.length = 0;`),
                         'super.reset();',
                     ], '\n'),
@@ -99,7 +106,11 @@ export class DefaultTypeScriptEasyNetworkStubGenerator extends TypeScriptFileGen
             ],
             returnType: 'this',
             body: appendValueGroup([
-                s `this.stubWrapper.stub2<${requestType}>()(${s.indent `
+                s `if (this.${this.getEndpointStubHandleFieldName(ctx, endpoint)}) {${s.indent `
+              throw new Error('${this.getEndpointStubMethodName(ctx, endpoint)} has already been called. Call ${this.getEndpointUnregisterMethodName(ctx, endpoint)}() or reset() before stubbing again.');`}
+            }`,
+                s `this.${this.getEndpointStubHandleFieldName(ctx, endpoint)} = this.stubWrapper.stub2<${requestType}>()(${s
+                    .indent `
               ${ts.string(endpoint.method.toUpperCase())},
               ${this.getStubClassName(ctx)}.${this.getEndpointPathPropertyName(ctx, endpoint)},
               async (request) => {${s.indent `
@@ -129,12 +140,28 @@ export class DefaultTypeScriptEasyNetworkStubGenerator extends TypeScriptFileGen
     getEndpointRequestsFieldName(ctx, endpoint) {
         return toCasing(endpoint.name + '_Requests', adjustCasing(ctx.config.propertyNameCasing, { prefix: '_' }));
     }
+    getEndpointStubHandleFieldName(ctx, endpoint) {
+        return toCasing('stub_' + endpoint.name + '_Handle', adjustCasing(ctx.config.propertyNameCasing, { prefix: '_' }));
+    }
     getEndpointRequestsPropertyName(ctx, endpoint) {
         return toCasing(endpoint.name + '_Requests', ctx.config.propertyNameCasing);
     }
     getEndpointStubMethodName(ctx, endpoint) {
         return toCasing('stub_' + endpoint.name, ctx.config.functionNameCasing);
     }
+    getEndpointUnregisterMethodName(ctx, endpoint) {
+        return toCasing('unregister_stub_' + endpoint.name, ctx.config.functionNameCasing);
+    }
+    getEndpointUnregisterMethod(ctx, endpoint) {
+        return ts.method(this.getEndpointUnregisterMethodName(ctx, endpoint), {
+            accessModifier: 'public',
+            returnType: ts.refs.void_(),
+            body: appendValueGroup([
+                `this.${this.getEndpointStubHandleFieldName(ctx, endpoint)}?.unregister();`,
+                `this.${this.getEndpointStubHandleFieldName(ctx, endpoint)} = undefined;`,
+            ], '\n'),
+        });
+    }
     getStubRoute(ctx, endpoint) {
         var _a, _b;
         let path = endpoint.path;

package/esm/src/generators/services/easy-network-stub/refs.d.ts

@@ -10,5 +10,6 @@ export declare function getReferenceFactories(options: TypeScriptEasyNetworkStub
     easyNetworkStubBase: ts.ModuleReferenceFactory;
     easyNetworkStubGroup: ts.GenericModuleReferenceFactory<2>;
     createEasyNetworkStubGroup: ts.GenericModuleReferenceFactory<2>;
+    stubHandle: ts.ModuleReferenceFactory;
 };
 //# sourceMappingURL=refs.d.ts.map

package/esm/src/generators/services/easy-network-stub/refs.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"refs.d.ts","sourceRoot":"","sources":["../../../../../src/src/generators/services/easy-network-stub/refs.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,EAAE,EAAE,MAAM,uBAAuB,CAAC;AAC3C,OAAO,KAAK,EAAE,yCAAyC,EAAE,MAAM,aAAa,CAAC;AAG7E,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,yCAAyC,GAAG;IACzF,6BAA6B,EAAE,EAAE,CAAC,sBAAsB,CAAC;IACzD,eAAe,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAC3C,eAAe,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACrD,2BAA2B,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACjE,gBAAgB,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACtD,sBAAsB,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAClD,mBAAmB,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAC/C,oBAAoB,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IAC1D,0BAA0B,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;CACjE,CA4BA"}
+{"version":3,"file":"refs.d.ts","sourceRoot":"","sources":["../../../../../src/src/generators/services/easy-network-stub/refs.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,EAAE,EAAE,MAAM,uBAAuB,CAAC;AAC3C,OAAO,KAAK,EAAE,yCAAyC,EAAE,MAAM,aAAa,CAAC;AAG7E,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,yCAAyC,GAAG;IACzF,6BAA6B,EAAE,EAAE,CAAC,sBAAsB,CAAC;IACzD,eAAe,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAC3C,eAAe,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACrD,2BAA2B,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACjE,gBAAgB,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACtD,sBAAsB,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAClD,mBAAmB,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAC/C,oBAAoB,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IAC1D,0BAA0B,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IAChE,UAAU,EAAE,EAAE,CAAC,sBAAsB,CAAC;CACvC,CA6BA"}

package/esm/src/generators/services/easy-network-stub/refs.js

@@ -23,5 +23,6 @@ export function getReferenceFactories(options) {
             importType: 'type-import',
         }),
         createEasyNetworkStubGroup: ts.reference.genericFactory('createEasyNetworkStubGroup', easyNetworkStubUtilsPath),
+        stubHandle: ts.reference.factory('StubHandle', 'easy-network-stub', { importType: 'type-import' }),
     };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goast/typescript",
-  "version": "0.6.3-beta.1",
+  "version": "0.7.0",
   "description": "Provides gOAst generators for generating TypeScript code from OpenAPI specifications.",
   "author": {
     "name": "Marc Schmidt (MaSch0212)",
@@ -25,7 +25,7 @@
   "dependencies": {
     "tslib": "^2.6.2",
     "fs-extra": "^11.3.3",
-    "@goast/core": "0.5.0-beta.5"
+    "@goast/core": "0.5.0"
   },
   "devDependencies": {
     "@types/node": "^20.9.0",

package/script/src/generators/services/easy-network-stub/easy-network-stub-generator.d.ts

@@ -23,8 +23,11 @@ export declare class DefaultTypeScriptEasyNetworkStubGenerator extends TypeScrip
     protected getStubClassName(ctx: Context): string;
     protected getEndpointPathPropertyName(ctx: Context, endpoint: ApiEndpoint): string;
     protected getEndpointRequestsFieldName(ctx: Context, endpoint: ApiEndpoint): string;
+    protected getEndpointStubHandleFieldName(ctx: Context, endpoint: ApiEndpoint): string;
     protected getEndpointRequestsPropertyName(ctx: Context, endpoint: ApiEndpoint): string;
     protected getEndpointStubMethodName(ctx: Context, endpoint: ApiEndpoint): string;
+    protected getEndpointUnregisterMethodName(ctx: Context, endpoint: ApiEndpoint): string;
+    protected getEndpointUnregisterMethod(ctx: Context, endpoint: ApiEndpoint): ts.Method<Builder>;
     protected getStubRoute(ctx: Context, endpoint: ApiEndpoint): string;
     protected getEndpointStatusCodes(ctx: Context, endpoint: ApiEndpoint): Record<number, ts.Type<Builder>>;
     protected getSchemaType(ctx: Context, schema: ApiSchema | undefined, fallback?: ts.Type<Builder>): ts.Type<Builder>;

package/script/src/generators/services/easy-network-stub/easy-network-stub-generator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"easy-network-stub-generator.d.ts","sourceRoot":"","sources":["../../../../../src/src/generators/services/easy-network-stub/easy-network-stub-generator.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,gBAAgB,EAIrB,KAAK,YAAY,EAElB,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,EAAE,EAAE,MAAM,uBAAuB,CAAC;AAC3C,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAClE,OAAO,KAAK,EAAE,yCAAyC,EAAE,wCAAwC,EAAE,MAAM,aAAa,CAAC;AAEvH,KAAK,OAAO,GAAG,yCAAyC,CAAC;AACzD,KAAK,MAAM,GAAG,wCAAwC,CAAC;AACvD,KAAK,OAAO,GAAG,qBAAqB,CAAC;AAErC,MAAM,WAAW,kCAAkC,CAAC,OAAO,SAAS,MAAM,GAAG,MAAM;IACjF,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;CAC/C;AAED,qBAAa,yCAA0C,SAAQ,uBAAuB,CAAC,OAAO,EAAE,MAAM,CACpG,YAAW,kCAAkC;IACtC,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC;IAwBnD,SAAS,CAAC,kBAAkB,CAAC,GAAG,EAAE,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC;IAOrE,SAAS,CAAC,oBAAoB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;IAgBzF,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;IAoDvD,SAAS,CAAC,uBAAuB,CAC/B,GAAG,EAAE,OAAO,EACZ,QAAQ,EAAE,WAAW,EACrB,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GAC/B,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;IAUnB,SAAS,CAAC,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;IAiCxF,SAAS,CAAC,eAAe,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM;IAQ/C,SAAS,CAAC,wBAAwB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAI/E,SAAS,CAAC,gBAAgB,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM;IAIhD,SAAS,CAAC,2BAA2B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIlF,SAAS,CAAC,4BAA4B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAInF,SAAS,CAAC,+BAA+B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAItF,SAAS,CAAC,yBAAyB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIhF,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAyBnE,SAAS,CAAC,sBAAsB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAYvG,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,GAAG,SAAS,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;CAKpH"}
+{"version":3,"file":"easy-network-stub-generator.d.ts","sourceRoot":"","sources":["../../../../../src/src/generators/services/easy-network-stub/easy-network-stub-generator.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,gBAAgB,EAIrB,KAAK,YAAY,EAElB,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,EAAE,EAAE,MAAM,uBAAuB,CAAC;AAC3C,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAClE,OAAO,KAAK,EAAE,yCAAyC,EAAE,wCAAwC,EAAE,MAAM,aAAa,CAAC;AAEvH,KAAK,OAAO,GAAG,yCAAyC,CAAC;AACzD,KAAK,MAAM,GAAG,wCAAwC,CAAC;AACvD,KAAK,OAAO,GAAG,qBAAqB,CAAC;AAErC,MAAM,WAAW,kCAAkC,CAAC,OAAO,SAAS,MAAM,GAAG,MAAM;IACjF,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;CAC/C;AAED,qBAAa,yCAA0C,SAAQ,uBAAuB,CAAC,OAAO,EAAE,MAAM,CACpG,YAAW,kCAAkC;IACtC,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC;IAwBnD,SAAS,CAAC,kBAAkB,CAAC,GAAG,EAAE,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC;IAOrE,SAAS,CAAC,oBAAoB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;IAgBzF,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;IA+DvD,SAAS,CAAC,uBAAuB,CAC/B,GAAG,EAAE,OAAO,EACZ,QAAQ,EAAE,WAAW,EACrB,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GAC/B,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;IAUnB,SAAS,CAAC,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;IAuCxF,SAAS,CAAC,eAAe,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM;IAQ/C,SAAS,CAAC,wBAAwB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAI/E,SAAS,CAAC,gBAAgB,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM;IAIhD,SAAS,CAAC,2BAA2B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIlF,SAAS,CAAC,4BAA4B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAInF,SAAS,CAAC,8BAA8B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIrF,SAAS,CAAC,+BAA+B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAItF,SAAS,CAAC,yBAAyB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAIhF,SAAS,CAAC,+BAA+B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAItF,SAAS,CAAC,2BAA2B,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;IAc9F,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM;IAyBnE,SAAS,CAAC,sBAAsB,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAYvG,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,GAAG,SAAS,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;CAKpH"}

package/script/src/generators/services/easy-network-stub/easy-network-stub-generator.js

@@ -58,6 +58,11 @@ class DefaultTypeScriptEasyNetworkStubGenerator extends file_generator_js_1.Type
                     value: index_js_1.ts.tuple([]),
                 })),
                 '\n',
+                ...ctx.service.endpoints.map((endpoint) => index_js_1.ts.property(this.getEndpointStubHandleFieldName(ctx, endpoint), {
+                    accessModifier: 'private',
+                    type: (0, core_1.builderTemplate) `${ctx.refs.stubHandle()} | undefined`,
+                })),
+                '\n',
                 ...ctx.service.endpoints.map((endpoint) => index_js_1.ts.property(this.getEndpointRequestsPropertyName(ctx, endpoint), {
                     readonly: true,
                     accessModifier: 'public',
@@ -67,11 +72,13 @@ class DefaultTypeScriptEasyNetworkStubGenerator extends file_generator_js_1.Type
                     }),
                 })),
                 ...ctx.service.endpoints.map((endpoint) => this.getEndpointStubMethod(ctx, endpoint)),
+                ...ctx.service.endpoints.map((endpoint) => this.getEndpointUnregisterMethod(ctx, endpoint)),
                 index_js_1.ts.method('reset', {
                     accessModifier: 'public',
                     override: true,
                     returnType: index_js_1.ts.refs.void_(),
                     body: (0, core_1.appendValueGroup)([
+                        ...ctx.service.endpoints.map((endpoint) => `this.${this.getEndpointStubHandleFieldName(ctx, endpoint)} = undefined;`),
                         ...ctx.service.endpoints.map((endpoint) => `this.${this.getEndpointRequestsFieldName(ctx, endpoint)}.length = 0;`),
                         'super.reset();',
                     ], '\n'),
@@ -102,7 +109,11 @@ class DefaultTypeScriptEasyNetworkStubGenerator extends file_generator_js_1.Type
             ],
             returnType: 'this',
             body: (0, core_1.appendValueGroup)([
-                (0, core_1.builderTemplate) `this.stubWrapper.stub2<${requestType}>()(${core_1.builderTemplate.indent `
+                (0, core_1.builderTemplate) `if (this.${this.getEndpointStubHandleFieldName(ctx, endpoint)}) {${core_1.builderTemplate.indent `
+              throw new Error('${this.getEndpointStubMethodName(ctx, endpoint)} has already been called. Call ${this.getEndpointUnregisterMethodName(ctx, endpoint)}() or reset() before stubbing again.');`}
+            }`,
+                (0, core_1.builderTemplate) `this.${this.getEndpointStubHandleFieldName(ctx, endpoint)} = this.stubWrapper.stub2<${requestType}>()(${core_1.builderTemplate
+                    .indent `
               ${index_js_1.ts.string(endpoint.method.toUpperCase())},
               ${this.getStubClassName(ctx)}.${this.getEndpointPathPropertyName(ctx, endpoint)},
               async (request) => {${core_1.builderTemplate.indent `
@@ -132,12 +143,28 @@ class DefaultTypeScriptEasyNetworkStubGenerator extends file_generator_js_1.Type
     getEndpointRequestsFieldName(ctx, endpoint) {
         return (0, core_1.toCasing)(endpoint.name + '_Requests', (0, core_1.adjustCasing)(ctx.config.propertyNameCasing, { prefix: '_' }));
     }
+    getEndpointStubHandleFieldName(ctx, endpoint) {
+        return (0, core_1.toCasing)('stub_' + endpoint.name + '_Handle', (0, core_1.adjustCasing)(ctx.config.propertyNameCasing, { prefix: '_' }));
+    }
     getEndpointRequestsPropertyName(ctx, endpoint) {
         return (0, core_1.toCasing)(endpoint.name + '_Requests', ctx.config.propertyNameCasing);
     }
     getEndpointStubMethodName(ctx, endpoint) {
         return (0, core_1.toCasing)('stub_' + endpoint.name, ctx.config.functionNameCasing);
     }
+    getEndpointUnregisterMethodName(ctx, endpoint) {
+        return (0, core_1.toCasing)('unregister_stub_' + endpoint.name, ctx.config.functionNameCasing);
+    }
+    getEndpointUnregisterMethod(ctx, endpoint) {
+        return index_js_1.ts.method(this.getEndpointUnregisterMethodName(ctx, endpoint), {
+            accessModifier: 'public',
+            returnType: index_js_1.ts.refs.void_(),
+            body: (0, core_1.appendValueGroup)([
+                `this.${this.getEndpointStubHandleFieldName(ctx, endpoint)}?.unregister();`,
+                `this.${this.getEndpointStubHandleFieldName(ctx, endpoint)} = undefined;`,
+            ], '\n'),
+        });
+    }
     getStubRoute(ctx, endpoint) {
         var _a, _b;
         let path = endpoint.path;

package/script/src/generators/services/easy-network-stub/refs.d.ts

@@ -10,5 +10,6 @@ export declare function getReferenceFactories(options: TypeScriptEasyNetworkStub
     easyNetworkStubBase: ts.ModuleReferenceFactory;
     easyNetworkStubGroup: ts.GenericModuleReferenceFactory<2>;
     createEasyNetworkStubGroup: ts.GenericModuleReferenceFactory<2>;
+    stubHandle: ts.ModuleReferenceFactory;
 };
 //# sourceMappingURL=refs.d.ts.map

package/script/src/generators/services/easy-network-stub/refs.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"refs.d.ts","sourceRoot":"","sources":["../../../../../src/src/generators/services/easy-network-stub/refs.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,EAAE,EAAE,MAAM,uBAAuB,CAAC;AAC3C,OAAO,KAAK,EAAE,yCAAyC,EAAE,MAAM,aAAa,CAAC;AAG7E,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,yCAAyC,GAAG;IACzF,6BAA6B,EAAE,EAAE,CAAC,sBAAsB,CAAC;IACzD,eAAe,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAC3C,eAAe,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACrD,2BAA2B,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACjE,gBAAgB,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACtD,sBAAsB,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAClD,mBAAmB,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAC/C,oBAAoB,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IAC1D,0BAA0B,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;CACjE,CA4BA"}
+{"version":3,"file":"refs.d.ts","sourceRoot":"","sources":["../../../../../src/src/generators/services/easy-network-stub/refs.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,EAAE,EAAE,MAAM,uBAAuB,CAAC;AAC3C,OAAO,KAAK,EAAE,yCAAyC,EAAE,MAAM,aAAa,CAAC;AAG7E,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,yCAAyC,GAAG;IACzF,6BAA6B,EAAE,EAAE,CAAC,sBAAsB,CAAC;IACzD,eAAe,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAC3C,eAAe,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACrD,2BAA2B,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACjE,gBAAgB,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IACtD,sBAAsB,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAClD,mBAAmB,EAAE,EAAE,CAAC,sBAAsB,CAAC;IAC/C,oBAAoB,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IAC1D,0BAA0B,EAAE,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC;IAChE,UAAU,EAAE,EAAE,CAAC,sBAAsB,CAAC;CACvC,CA6BA"}

package/script/src/generators/services/easy-network-stub/refs.js

@@ -26,5 +26,6 @@ function getReferenceFactories(options) {
             importType: 'type-import',
         }),
         createEasyNetworkStubGroup: index_js_1.ts.reference.genericFactory('createEasyNetworkStubGroup', easyNetworkStubUtilsPath),
+        stubHandle: index_js_1.ts.reference.factory('StubHandle', 'easy-network-stub', { importType: 'type-import' }),
     };
 }

package/src/mod.ts

@@ -0,0 +1,7 @@
+export * from './src/ast/index.js';
+export * from './src/common-results.js';
+export * from './src/config.js';
+export * from './src/file-builder.js';
+export * from './src/generators/index.js';
+export * from './src/import-collection.js';
+export * from './src/utils.js';

package/src/src/assets.ts

@@ -0,0 +1,9 @@
+import { AssetManager } from '@goast/core';
+import { pathToFileURL, URL } from 'node:url';
+
+declare const __filename: string | undefined;
+const scriptUrl = typeof __filename === 'undefined' ? import.meta.url : pathToFileURL(__filename);
+
+const manager = new AssetManager(new URL('../assets/', scriptUrl));
+export const getAssetFileContent = manager.getAssetFileContent.bind(manager);
+export const copyAssetFile = manager.copyAssetFile.bind(manager);

package/src/src/ast/_index.ts

@@ -0,0 +1,54 @@
+export { type TsAccessModifier as AccessModifier } from './common.js';
+export { TsNode as Node } from './node.js';
+export { TsArgument as Argument, tsArgument as argument } from './nodes/argument.js';
+export { TsArrayType as ArrayType, tsArrayType as arrayType } from './nodes/array-type.js';
+export { TsArrowFunction as ArrowFunction, tsArrowFunction as arrowFunction } from './nodes/arrow-function.js';
+export { TsCall as Call, tsCall as call } from './nodes/call.js';
+export { TsClass as Class, tsClass as class } from './nodes/class.js';
+export {
+  TsConstructorParameter as ConstructorParameter,
+  tsConstructorParameter as constructorParameter,
+} from './nodes/constructor-parameter.js';
+export { TsConstructor as Constructor, tsConstructor as constructor } from './nodes/constructor.js';
+export { TsDecorator as Decorator, tsDecorator as decorator } from './nodes/decorator.js';
+export { TsDocTag as DocTag, tsDocTag as docTag } from './nodes/doc-tag.js';
+export { TsDoc as Doc, tsDoc as doc } from './nodes/doc.js';
+export { TsEnumValue as EnumValue, tsEnumValue as enumValue } from './nodes/enum-value.js';
+export { TsEnum as Enum, tsEnum as enum } from './nodes/enum.js';
+export { TsExport as Export, tsExport as export } from './nodes/export.js';
+export { TsFunctionType as FunctionType, tsFunctionType as functionType } from './nodes/function-type.js';
+export { TsFunction as Function, tsFunction as function } from './nodes/function.js';
+export {
+  TsGenericParameter as GenericParameter,
+  tsGenericParameter as genericParameter,
+} from './nodes/generic-parameter.js';
+export { TsIndexer as Indexer, tsIndexer as indexer } from './nodes/indexer.js';
+export { TsInterface as Interface, tsInterface as interface } from './nodes/interface.js';
+export {
+  TsIntersectionType as IntersectionType,
+  tsIntersectionType as intersectionType,
+} from './nodes/intersection-type.js';
+export { TsLookupType as LookupType, tsLookupType as lookupType } from './nodes/lookup-type.js';
+export { TsMethod as Method, tsMethod as method } from './nodes/method.js';
+export { TsObjectType as ObjectType, tsObjectType as objectType } from './nodes/object-type.js';
+export { TsObject as Object, tsObject as object } from './nodes/object.js';
+export { TsParameter as Parameter, tsParameter as parameter } from './nodes/parameter.js';
+export { TsProperty as Property, tsProperty as property } from './nodes/property.js';
+export {
+  type TsGenericModuleReferenceFactory as GenericModuleReferenceFactory,
+  type TsGenericReferenceFactory as GenericReferenceFactory,
+  type TsModuleReferenceFactory as ModuleReferenceFactory,
+  TsReference as Reference,
+  tsReference as reference,
+  type TsReferenceFactory as ReferenceFactory,
+} from './nodes/reference.js';
+export { TsString as String, tsString as string } from './nodes/string.js';
+export { TsTuple as Tuple, tsTuple as tuple } from './nodes/tuple.js';
+export { TsTypeAlias as TypeAlias, tsTypeAlias as typeAlias } from './nodes/type-alias.js';
+export { TsTypeof as Typeof, tsTypeof as typeof } from './nodes/typeof.js';
+export { type TsType as Type, type TsValue as Value } from './nodes/types.js';
+export { TsUnionType as UnionType, tsUnionType as unionType } from './nodes/union-type.js';
+export { TsVariable as Variable, tsVariable as variable } from './nodes/variable.js';
+export * as refs from './references/index.js';
+export { getTypeScriptBuilderOptions as getBuilderOptions } from './utils/get-type-script-builder-options.js';
+export { toTsNode as toNode } from './utils/to-ts-node.js';

package/src/src/ast/common.ts

@@ -0,0 +1 @@
+export type TsAccessModifier = 'public' | 'protected' | 'private';

package/src/src/ast/index.ts

@@ -0,0 +1 @@
+export * as ts from './_index.js';

package/src/src/ast/node.ts

@@ -0,0 +1,10 @@
+import { AstNode, type AstNodeOptions, type SourceBuilder } from '@goast/core';
+
+export abstract class TsNode<TBuilder extends SourceBuilder, TInjects extends string = never> extends AstNode<
+  TBuilder,
+  TInjects
+> {
+  constructor(options: AstNodeOptions<typeof AstNode<TBuilder, TInjects>>) {
+    super(options);
+  }
+}

package/src/src/ast/nodes/argument.ts

@@ -0,0 +1,40 @@
+import type { AstNodeOptions, Prettify, SourceBuilder } from '@goast/core';
+
+import { TsNode } from '../node.js';
+import { writeTsNode } from '../utils/write-ts-nodes.js';
+import { writeTsParameters } from '../utils/write-ts-parameters.js';
+import type { TsValue } from './types.js';
+
+type Injects = never;
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof TsNode<TBuilder, TInjects | Injects>,
+  {
+    value: TsValue<TBuilder>;
+  }
+>;
+
+export class TsArgument<TBuilder extends SourceBuilder, TInjects extends string = never> extends TsNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public value: TsValue<TBuilder>;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.value = options.value;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    writeTsNode(builder, this.value);
+  }
+}
+
+const createArgument = <TBuilder extends SourceBuilder>(
+  value: Options<TBuilder>['value'],
+  options?: Prettify<Omit<Options<TBuilder>, 'value'>>,
+): TsArgument<TBuilder> => new TsArgument<TBuilder>({ ...options, value });
+
+export const tsArgument: typeof createArgument & { write: typeof writeTsParameters } = Object.assign(createArgument, {
+  write: writeTsParameters,
+});

package/src/src/ast/nodes/array-type.ts

@@ -0,0 +1,49 @@
+import type { AstNodeOptions, Nullable, Prettify, SourceBuilder } from '@goast/core';
+
+import { TsNode } from '../node.js';
+import { writeTsNode, writeTsNodes } from '../utils/write-ts-nodes.js';
+import type { TsType } from './types.js';
+
+type Injects = 'type';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof TsNode<TBuilder, TInjects | Injects>,
+  {
+    readonly?: Nullable<boolean>;
+    type: TsType<TBuilder>;
+  }
+>;
+
+export class TsArrayType<TBuilder extends SourceBuilder, TInjects extends string = never> extends TsNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public readonly: boolean;
+  public type: TsType<TBuilder>;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.readonly = options.readonly ?? false;
+    this.type = options.type;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    if (this.readonly) builder.append('readonly ');
+
+    builder
+      .append(this.inject.beforeType)
+      .parenthesize('()', (b) => writeTsNode(b, this.type))
+      .append(this.inject.afterType);
+
+    builder.append('[]');
+  }
+}
+
+const createArrayType = <TBuilder extends SourceBuilder>(
+  type: Options<TBuilder>['type'],
+  options?: Prettify<Omit<Options<TBuilder>, 'type'>>,
+): TsArrayType<TBuilder> => new TsArrayType<TBuilder>({ ...options, type });
+
+export const tsArrayType: typeof createArrayType & { write: typeof writeTsNodes } = Object.assign(createArrayType, {
+  write: writeTsNodes,
+});

package/src/src/ast/nodes/arrow-function.ts

@@ -0,0 +1,91 @@
+import {
+  type AppendValue,
+  type AstNodeOptions,
+  type BasicAppendValue,
+  notNullish,
+  type Nullable,
+  type SourceBuilder,
+} from '@goast/core';
+
+import { TsNode } from '../node.js';
+import { writeTsNode, writeTsNodes } from '../utils/write-ts-nodes.js';
+import { type TsGenericParameter, tsGenericParameter } from './generic-parameter.js';
+import { type TsParameter, tsParameter } from './parameter.js';
+import type { TsType } from './types.js';
+
+type Injects = 'generics' | 'params' | 'returnType' | 'body';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof TsNode<TBuilder, TInjects | Injects>,
+  {
+    async?: Nullable<boolean>;
+    generics?: Nullable<Nullable<TsGenericParameter<TBuilder> | BasicAppendValue<TBuilder>>[]>;
+    parameters?: Nullable<Nullable<TsParameter<TBuilder> | BasicAppendValue<TBuilder>>[]>;
+    returnType?: Nullable<TsType<TBuilder>>;
+    singleExpression?: Nullable<boolean>;
+    body?: Nullable<AppendValue<TBuilder>>;
+  }
+>;
+
+export class TsArrowFunction<TBuilder extends SourceBuilder, TInjects extends string = never> extends TsNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public async: boolean;
+  public generics: (TsGenericParameter<TBuilder> | BasicAppendValue<TBuilder>)[];
+  public parameters: (TsParameter<TBuilder> | BasicAppendValue<TBuilder>)[];
+  public returnType: TsType<TBuilder> | null;
+  public singleExpression: boolean;
+  public body: AppendValue<TBuilder> | null;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.async = options.async ?? false;
+    this.generics = options.generics?.filter(notNullish) ?? [];
+    this.parameters = options.parameters?.filter(notNullish) ?? [];
+    this.returnType = options.returnType ?? null;
+    this.singleExpression = options.singleExpression ?? false;
+    this.body = options.body ?? null;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    if (this.async) {
+      builder.append('async ');
+    }
+
+    builder.append(this.inject.beforeGenerics);
+    tsGenericParameter.write(builder, this.generics);
+    builder.append(this.inject.afterGenerics);
+
+    builder.append(this.inject.beforeParams);
+    tsParameter.write(builder, this.parameters);
+    builder.append(this.inject.afterParams);
+
+    if (this.returnType) {
+      builder.append(': ');
+      builder.append(this.inject.beforeReturnType);
+      writeTsNode(builder, this.returnType);
+      builder.append(this.inject.afterReturnType);
+    }
+
+    builder.append(' => ');
+
+    builder.append(this.inject.beforeBody);
+    if (this.singleExpression && this.body) {
+      builder.append(this.body);
+    } else {
+      builder.parenthesize('{}', this.body, { multiline: !!this.body });
+    }
+    builder.append(this.inject.afterBody);
+  }
+}
+
+const createArrowFunction = <TBuilder extends SourceBuilder>(options?: Options<TBuilder>): TsArrowFunction<TBuilder> =>
+  new TsArrowFunction<TBuilder>(options ?? {});
+
+export const tsArrowFunction: typeof createArrowFunction & { write: typeof writeTsNodes } = Object.assign(
+  createArrowFunction,
+  {
+    write: writeTsNodes,
+  },
+);