@goast/typescript 0.8.0-beta.1 → 0.8.0

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,
+  },
+);

package/src/src/ast/nodes/call.ts

@@ -0,0 +1,55 @@
+import {
+  type ArrayItem,
+  type AstNodeOptions,
+  notNullish,
+  type Nullable,
+  type Prettify,
+  type SourceBuilder,
+} from '@goast/core';
+
+import { TsNode } from '../node.js';
+import { writeTsNodes } from '../utils/write-ts-nodes.js';
+import { type TsArgument, tsArgument } from './argument.js';
+import type { TsType, TsValue } from './types.js';
+
+type Injects = never;
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof TsNode<TBuilder, TInjects | Injects>,
+  {
+    path: Nullable<TsType<TBuilder> | TsValue<TBuilder>>[];
+    arguments?: Nullable<Nullable<TsArgument<TBuilder> | TsValue<TBuilder>>[]>;
+  }
+>;
+
+export class TsCall<TBuilder extends SourceBuilder, TInjects extends string = never> extends TsNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public path: (TsType<TBuilder> | TsValue<TBuilder>)[];
+  public arguments: (TsArgument<TBuilder> | TsValue<TBuilder>)[] | null;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.path = options.path.filter(notNullish);
+    this.arguments = options.arguments?.filter(notNullish) ?? null;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    writeTsNodes(builder, this.path, { separator: '.' });
+
+    if (this.arguments) {
+      tsArgument.write(builder, this.arguments);
+    }
+  }
+}
+
+const createCall = <TBuilder extends SourceBuilder>(
+  path: Options<TBuilder>['path'] | ArrayItem<Options<TBuilder>['path']>,
+  args?: Options<TBuilder>['arguments'],
+  options?: Prettify<Omit<Options<TBuilder>, 'path' | 'arguments'>>,
+): TsCall<TBuilder> => new TsCall<TBuilder>({ ...options, path: Array.isArray(path) ? path : [path], arguments: args });
+
+export const tsCall: typeof createCall & { write: typeof writeTsNodes } = Object.assign(createCall, {
+  write: writeTsNodes,
+});

package/src/src/ast/nodes/class.ts

@@ -0,0 +1,133 @@
+import {
+  type AstNodeOptions,
+  type BasicAppendValue,
+  notNullish,
+  type Nullable,
+  type Prettify,
+  type SourceBuilder,
+} from '@goast/core';
+
+import { TsNode } from '../node.js';
+import { writeTsMembers } from '../utils/write-ts-members.js';
+import { writeTsNode, writeTsNodes } from '../utils/write-ts-nodes.js';
+import type { TsConstructor } from './constructor.js';
+import { type TsDecorator, tsDecorator } from './decorator.js';
+import { type TsDoc, tsDoc } from './doc.js';
+import { type TsGenericParameter, tsGenericParameter } from './generic-parameter.js';
+import type { TsIndexer } from './indexer.js';
+import type { TsMethod } from './method.js';
+import type { TsProperty } from './property.js';
+import type { TsReference } from './reference.js';
+
+type Injects = 'doc' | 'decorators' | 'modifiers' | 'name' | 'generics' | 'extends' | 'implements' | 'body' | 'members';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof TsNode<TBuilder, TInjects | Injects>,
+  {
+    doc?: Nullable<TsDoc<TBuilder>>;
+    decorators?: Nullable<Nullable<TsDecorator<TBuilder>>[]>;
+    export?: Nullable<boolean>;
+    abstract?: Nullable<boolean>;
+    name: string;
+    generics?: Nullable<Nullable<TsGenericParameter<TBuilder> | BasicAppendValue<TBuilder>>[]>;
+    extends?: Nullable<TsReference<TBuilder> | BasicAppendValue<TBuilder>>;
+    implements?: Nullable<Nullable<TsReference<TBuilder> | BasicAppendValue<TBuilder>>[]>;
+    members?: Nullable<Nullable<Member<TBuilder>>[]>;
+  }
+>;
+
+type Member<TBuilder extends SourceBuilder> =
+  | TsProperty<TBuilder>
+  | TsMethod<TBuilder>
+  | TsIndexer<TBuilder>
+  | TsConstructor<TBuilder>
+  | BasicAppendValue<TBuilder>;
+
+export class TsClass<TBuilder extends SourceBuilder, TInjects extends string = never> extends TsNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public doc: TsDoc<TBuilder> | null;
+  public decorators: TsDecorator<TBuilder>[];
+  public export: boolean;
+  public abstract: boolean;
+  public name: string;
+  public generics: (TsGenericParameter<TBuilder> | BasicAppendValue<TBuilder>)[];
+  public extends: TsReference<TBuilder> | BasicAppendValue<TBuilder> | null;
+  public implements: (TsReference<TBuilder> | BasicAppendValue<TBuilder>)[];
+  public members: Member<TBuilder>[];
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.doc = options.doc ?? null;
+    this.decorators = options.decorators?.filter(notNullish) ?? [];
+    this.export = options.export ?? false;
+    this.abstract = options.abstract ?? false;
+    this.name = options.name;
+    this.generics = options.generics?.filter(notNullish) ?? [];
+    this.extends = options.extends ?? null;
+    this.implements = options.implements?.filter(notNullish) ?? [];
+    this.members = options.members?.filter(notNullish) ?? [];
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append(this.inject.beforeDoc);
+    tsDoc.write(builder, tsDoc.get(this.doc, { generics: this.generics }));
+    builder.append(this.inject.afterDoc);
+
+    builder.append(this.inject.beforeDecorators);
+    tsDecorator.write(builder, this.decorators, { multiline: true });
+    builder.append(this.inject.afterDecorators);
+
+    builder.append(this.inject.beforeModifiers);
+    if (this.export) builder.append('export ');
+    if (this.abstract) builder.append('abstract ');
+    builder.append(this.inject.afterModifiers);
+
+    builder.append('class ');
+    builder.append(this.inject.beforeName, this.name, this.inject.afterName);
+
+    builder.append(this.inject.beforeGenerics);
+    tsGenericParameter.write(builder, this.generics);
+    builder.append(this.inject.afterGenerics);
+
+    if (this.extends) {
+      builder.append(' extends ');
+      builder.append(this.inject.beforeExtends);
+      writeTsNode(builder, this.extends);
+      builder.append(this.inject.afterExtends);
+    }
+
+    if (this.implements.length > 0) {
+      builder.append(' implements ');
+      builder.append(this.inject.beforeImplements);
+      writeTsNodes(builder, this.implements, { separator: ', ' });
+      builder.append(this.inject.afterImplements);
+    }
+
+    builder.append(' ');
+
+    builder.append(this.inject.beforeBody);
+    builder.parenthesize(
+      '{}',
+      (b) => {
+        b.append(this.inject.beforeMembers);
+        writeTsMembers(b, this.members);
+        b.append(this.inject.afterMembers);
+      },
+      { multiline: this.members.length > 0 },
+    );
+    builder.append(this.inject.afterBody);
+
+    builder.appendLine();
+  }
+}
+
+const createClass = <TBuilder extends SourceBuilder>(
+  name: Options<TBuilder>['name'],
+  options?: Prettify<Omit<Options<TBuilder>, 'name'>>,
+): TsClass<TBuilder> => new TsClass<TBuilder>({ ...options, name });
+
+export const tsClass: typeof createClass & { write: typeof writeTsNodes } = Object.assign(createClass, {
+  write: writeTsNodes,
+});

package/src/src/ast/nodes/constructor-parameter.ts

@@ -0,0 +1,70 @@
+import type { AstNodeOptions, Nullable, Prettify, SourceBuilder } from '@goast/core';
+
+import type { TsAccessModifier } from '../common.js';
+import { writeTsNode } from '../utils/write-ts-nodes.js';
+import { writeTsParameters } from '../utils/write-ts-parameters.js';
+import { tsDecorator } from './decorator.js';
+import { TsParameter } from './parameter.js';
+
+type Injects = 'modifiers';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof TsParameter<TBuilder, TInjects | Injects>,
+  {
+    accessModifier?: Nullable<TsAccessModifier>;
+    readonly?: Nullable<boolean>;
+  }
+>;
+
+export class TsConstructorParameter<TBuilder extends SourceBuilder, TInjects extends string = never>
+  extends TsParameter<
+    TBuilder,
+    TInjects | Injects
+  > {
+  public accessModifier: TsAccessModifier | null;
+  public readonly: boolean;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.accessModifier = options.accessModifier ?? null;
+    this.readonly = options.readonly ?? false;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append(this.inject.beforeDecorators);
+    tsDecorator.write(builder, this.decorators);
+    builder.append(this.inject.afterDecorators);
+
+    builder.append(this.inject.beforeModifiers);
+    if (this.accessModifier) builder.append(this.accessModifier, ' ');
+    if (this.readonly) builder.append('readonly ');
+    builder.append(this.inject.afterModifiers);
+
+    builder.append(this.inject.beforeName, this.name, this.inject.afterName);
+    if (this.optional) builder.append('?');
+
+    if (this.type) {
+      builder.append(': ');
+      builder.append(this.inject.beforeType);
+      writeTsNode(builder, this.type);
+      builder.append(this.inject.afterType);
+    }
+
+    if (this.default) {
+      builder.append(' = ');
+      builder.append(this.inject.beforeDefault);
+      writeTsNode(builder, this.default);
+      builder.append(this.inject.afterDefault);
+    }
+  }
+}
+
+const createConstructorParameter = <TBuilder extends SourceBuilder>(
+  name: Options<TBuilder>['name'],
+  options?: Prettify<Omit<Options<TBuilder>, 'name'>>,
+): TsConstructorParameter<TBuilder> => new TsConstructorParameter<TBuilder>({ ...options, name });
+
+export const tsConstructorParameter: typeof createConstructorParameter & { write: typeof writeTsParameters } = Object
+  .assign(createConstructorParameter, {
+    write: writeTsParameters,
+  });

package/src/src/ast/nodes/constructor.ts

@@ -0,0 +1,77 @@
+import {
+  type AppendValue,
+  type AstNodeOptions,
+  type BasicAppendValue,
+  notNullish,
+  type Nullable,
+  type SourceBuilder,
+} from '@goast/core';
+
+import { TsNode } from '../node.js';
+import { writeTsNodes } from '../utils/write-ts-nodes.js';
+import type { TsConstructorParameter } from './constructor-parameter.js';
+import { type TsDecorator, tsDecorator } from './decorator.js';
+import { type TsDoc, tsDoc } from './doc.js';
+import { tsParameter } from './parameter.js';
+
+type Injects = 'params' | 'doc' | 'decorators' | 'body';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof TsNode<TBuilder, TInjects | Injects>,
+  {
+    doc?: Nullable<TsDoc<TBuilder>>;
+    decorators?: Nullable<Nullable<TsDecorator<TBuilder>>[]>;
+    parameters?: Nullable<Nullable<TsConstructorParameter<TBuilder> | BasicAppendValue<TBuilder>>[]>;
+    body?: Nullable<AppendValue<TBuilder>>;
+  }
+>;
+
+export class TsConstructor<TBuilder extends SourceBuilder, TInjects extends string = never> extends TsNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public doc: TsDoc<TBuilder> | null;
+  public decorators: TsDecorator<TBuilder>[];
+  public parameters: (TsConstructorParameter<TBuilder> | BasicAppendValue<TBuilder>)[];
+  public body: AppendValue<TBuilder> | null;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.doc = options.doc ?? null;
+    this.decorators = options.decorators?.filter(notNullish) ?? [];
+    this.parameters = options.parameters?.filter(notNullish) ?? [];
+    this.body = options.body ?? null;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append(this.inject.beforeDoc);
+    tsDoc.get(this.doc, { parameters: this.parameters })?.write(builder);
+    builder.append(this.inject.afterDoc);
+
+    builder.append(this.inject.beforeDecorators);
+    tsDecorator.write(builder, this.decorators, { multiline: true });
+    builder.append(this.inject.afterDecorators);
+
+    builder.append('constructor');
+    builder.append(this.inject.beforeParams);
+    tsParameter.write(builder, this.parameters);
+    builder.append(this.inject.afterParams);
+    builder.append(' ');
+
+    builder.append(this.inject.beforeBody);
+    builder.parenthesize('{}', this.body, { multiline: !!this.body });
+    builder.append(this.inject.afterBody);
+
+    builder.appendLine();
+  }
+}
+
+const createConstructor = <TBuilder extends SourceBuilder>(options?: Options<TBuilder>): TsConstructor<TBuilder> =>
+  new TsConstructor<TBuilder>(options ?? {});
+
+export const tsConstructor: typeof createConstructor & { write: typeof writeTsNodes } = Object.assign(
+  createConstructor,
+  {
+    write: writeTsNodes,
+  },
+);

package/src/src/ast/nodes/decorator.ts

@@ -0,0 +1,76 @@
+import {
+  type AppendValue,
+  type AstNodeOptions,
+  type BasicAppendValue,
+  notNullish,
+  type Nullable,
+  type Prettify,
+  type SingleOrMultiple,
+  type SourceBuilder,
+  toArray,
+} from '@goast/core';
+
+import { TsNode } from '../node.js';
+import { writeTsNode } from '../utils/write-ts-nodes.js';
+import { type TsArgument, tsArgument } from './argument.js';
+import type { TsReference } from './reference.js';
+import type { TsValue } from './types.js';
+
+type Injects = 'function' | 'arguments';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof TsNode<TBuilder, TInjects | Injects>,
+  {
+    function: TsReference<TBuilder> | BasicAppendValue<TBuilder>;
+    arguments?: Nullable<Nullable<TsArgument<TBuilder> | TsValue<TBuilder>>[]>;
+  }
+>;
+
+export class TsDecorator<TBuilder extends SourceBuilder, TInjects extends string = never> extends TsNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public function: TsReference<TBuilder> | BasicAppendValue<TBuilder>;
+  public arguments: (TsArgument<TBuilder> | TsValue<TBuilder>)[] | null;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.function = options.function;
+    this.arguments = options.arguments?.filter(notNullish) ?? null;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append('@');
+    builder.append(this.inject.beforeFunction);
+    writeTsNode(builder, this.function);
+    builder.append(this.inject.afterFunction);
+
+    if (this.arguments) {
+      builder.append(this.inject.beforeArguments);
+      tsArgument.write(builder, this.arguments);
+      builder.append(this.inject.afterArguments);
+    }
+  }
+}
+
+const createDecorator = <TBuilder extends SourceBuilder>(
+  fn: Options<TBuilder>['function'],
+  args?: Options<TBuilder>['arguments'],
+  options?: Prettify<Omit<Options<TBuilder>, 'function' | 'arguments'>>,
+): TsDecorator<TBuilder> => new TsDecorator<TBuilder>({ ...options, function: fn, arguments: args });
+
+const writeDecorators = <TBuilder extends SourceBuilder>(
+  builder: TBuilder,
+  nodes: SingleOrMultiple<Nullable<TsDecorator<TBuilder> | AppendValue<TBuilder>>>,
+  options?: { multiline?: boolean },
+): void => {
+  const filteredNodes = toArray(nodes).filter(notNullish);
+  builder.forEach(filteredNodes, (b, d) => {
+    writeTsNode(b, d);
+    b.append(options?.multiline ? '\n' : ' ');
+  });
+};
+
+export const tsDecorator: typeof createDecorator & { write: typeof writeDecorators } = Object.assign(createDecorator, {
+  write: writeDecorators,
+});