@goast/typescript 0.4.3 → 0.4.4

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.test.ts

@@ -0,0 +1,122 @@
+import "../../../_dnt.test_polyfills.js";
+import { EOL } from 'node:os';
+
+import { expect } from '../../../deps/jsr.io/@std/expect/1.0.12/expect.js';
+import { beforeEach, describe, it } from '../../../deps/jsr.io/@std/testing/1.0.9/bdd.js';
+
+import { TypeScriptFileBuilder } from '../../file-builder.js';
+import { tsConstructorParameter } from './constructor-parameter.js';
+import { tsDecorator } from './decorator.js';
+
+let builder: TypeScriptFileBuilder;
+
+beforeEach(() => {
+  builder = new TypeScriptFileBuilder();
+});
+
+describe('tsConstructorParameter', () => {
+  it('should write the name of the parameter', () => {
+    builder.append(tsConstructorParameter('x'));
+    expect(builder.toString(false)).toBe('x');
+  });
+
+  it('should write decorators if they exist', () => {
+    builder.append(tsConstructorParameter('x', { decorators: [tsDecorator('decorator')] }));
+    expect(builder.toString(false)).toBe('@decorator x');
+  });
+
+  it('should write the type if it exists', () => {
+    builder.append(tsConstructorParameter('x', { type: 'number' }));
+    expect(builder.toString(false)).toBe('x: number');
+  });
+
+  it('should write the default if it exists', () => {
+    builder.append(tsConstructorParameter('x', { default: '42' }));
+    expect(builder.toString(false)).toBe('x = 42');
+  });
+
+  it('should write the accessibility if it exists', () => {
+    builder.append(tsConstructorParameter('x', { accessModifier: 'public' }));
+    expect(builder.toString(false)).toBe('public x');
+  });
+
+  it('should write the readonly keyword if the parameter is readonly', () => {
+    builder.append(tsConstructorParameter('x', { readonly: true }));
+    expect(builder.toString(false)).toBe('readonly x');
+  });
+
+  it('should write the optional symbol if the parameter is optional', () => {
+    builder.append(tsConstructorParameter('x', { optional: true }));
+    expect(builder.toString(false)).toBe('x?');
+  });
+
+  it('should write all the parts of the parameter', () => {
+    builder.append(
+      tsConstructorParameter('x', {
+        decorators: [tsDecorator('decorator')],
+        type: 'number',
+        default: 42,
+        accessModifier: 'public',
+        readonly: true,
+        optional: true,
+      }),
+    );
+    expect(builder.toString(false)).toBe('@decorator public readonly x?: number = 42');
+  });
+
+  it('should render injections', () => {
+    builder.append(
+      tsConstructorParameter('x', {
+        decorators: [tsDecorator('decorator')],
+        type: 'number',
+        default: 42,
+        accessModifier: 'public',
+        readonly: true,
+        optional: true,
+        inject: {
+          before: '║b║',
+          after: '║a║',
+          beforeDecorators: '║bd║',
+          afterDecorators: '║ad║',
+          beforeModifiers: '║bm║',
+          afterModifiers: '║am║',
+          beforeName: '║bn║',
+          afterName: '║an║',
+          beforeType: '║bt║',
+          afterType: '║at║',
+          beforeDefault: '║bdf║',
+          afterDefault: '║adf║',
+        },
+      }),
+    );
+    expect(builder.toString(false)).toBe(
+      '║b║║bd║@decorator ║ad║║bm║public readonly ║am║║bn║x║an║?: ║bt║number║at║ = ║bdf║42║adf║║a║',
+    );
+  });
+
+  describe('write', () => {
+    it('should write parenthesis if the node is empty', () => {
+      tsConstructorParameter.write(builder, []);
+      expect(builder.toString(false)).toBe('()');
+    });
+
+    it('should write a single parameter', () => {
+      tsConstructorParameter.write(builder, [tsConstructorParameter('x')]);
+      expect(builder.toString(false)).toBe('(x)');
+    });
+
+    it('should write all the parameters', () => {
+      tsConstructorParameter.write(builder, [tsConstructorParameter('x'), tsConstructorParameter('y')]);
+      expect(builder.toString(false)).toBe('(x, y)');
+    });
+
+    it('should write multiline if more than 2 parameters', () => {
+      tsConstructorParameter.write(builder, [
+        tsConstructorParameter('x'),
+        tsConstructorParameter('y'),
+        tsConstructorParameter('z'),
+      ]);
+      expect(builder.toString(false)).toBe(`(${EOL}  x,${EOL}  y,${EOL}  z${EOL})`);
+    });
+  });
+});

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.test.ts

@@ -0,0 +1,131 @@
+import "../../../_dnt.test_polyfills.js";
+import { EOL } from 'node:os';
+
+import { expect } from '../../../deps/jsr.io/@std/expect/1.0.12/mod.js';
+import { beforeEach, it } from '../../../deps/jsr.io/@std/testing/1.0.9/bdd.js';
+
+import { normalizeEOL } from '@goast/test-utils';
+
+import { TypeScriptFileBuilder } from '../../file-builder.js';
+import { tsConstructorParameter } from './constructor-parameter.js';
+import { tsConstructor } from './constructor.js';
+import { tsDecorator } from './decorator.js';
+import { tsDocTag } from './doc-tag.js';
+import { tsDoc } from './doc.js';
+
+let builder: TypeScriptFileBuilder;
+
+beforeEach(() => {
+  builder = new TypeScriptFileBuilder();
+});
+
+it('should write empty constructor', () => {
+  builder.append(tsConstructor());
+  expect(builder.toString(false)).toBe(`constructor() {}${EOL}`);
+});
+
+it('should write documentation if it exists', () => {
+  builder.append(tsConstructor({ doc: tsDoc({ description: 'description' }) }));
+  expect(builder.toString(false)).toBe(`/**${EOL} * description${EOL} */${EOL}constructor() {}${EOL}`);
+});
+
+it('should write decorators if they exist', () => {
+  builder.append(tsConstructor({ decorators: [tsDecorator('decorator')] }));
+  expect(builder.toString(false)).toBe(`@decorator${EOL}constructor() {}${EOL}`);
+});
+
+it('should write the parameters', () => {
+  builder.append(
+    tsConstructor({
+      parameters: ['x', 'y'],
+    }),
+  );
+  expect(builder.toString(false)).toBe(`constructor(x, y) {}${EOL}`);
+});
+
+it('should write the description of the parameters when no documentation is provided', () => {
+  builder.append(
+    tsConstructor({
+      parameters: [tsConstructorParameter('x', { description: 'description' }), 'y'],
+    }),
+  );
+  expect(builder.toString(false)).toBe(`/**${EOL} * @param x description${EOL} */${EOL}constructor(x, y) {}${EOL}`);
+});
+
+it('should write the body', () => {
+  builder.append(tsConstructor({ body: 'this.x = x;' }));
+  expect(builder.toString(false)).toBe(`constructor() {${EOL}  this.x = x;${EOL}}${EOL}`);
+});
+
+it('should write the description of the parameter when documentation is provided', () => {
+  builder.append(
+    tsConstructor({
+      doc: tsDoc({
+        description: 'description',
+        tags: [tsDocTag('param', 'y', 'y-description'), tsDocTag('returns', 'r-description')],
+      }),
+      decorators: [tsDecorator('decorator')],
+      parameters: [tsConstructorParameter('x', { description: 'x-description' }), 'y'],
+      body: 'this.x = x;',
+    }),
+  );
+  expect(builder.toString(false)).toBe(
+    normalizeEOL(6)(
+      `/**
+       * description
+       *
+       * @param x x-description
+       * @param y y-description
+       * @returns r-description
+       */
+      @decorator
+      constructor(x, y) {
+        this.x = x;
+      }
+      `,
+    ),
+  );
+});
+
+it('should render injections', () => {
+  builder.append(
+    tsConstructor({
+      doc: tsDoc({
+        description: 'description',
+        tags: [tsDocTag('param', 'y', 'y-description'), tsDocTag('returns', 'r-description')],
+      }),
+      decorators: [tsDecorator('decorator')],
+      parameters: [tsConstructorParameter('x', { description: 'x-description' }), 'y'],
+      body: 'this.x = x;',
+      inject: {
+        before: '║b║',
+        after: '║a║',
+        beforeParams: '║bp║',
+        afterParams: '║ap║',
+        beforeDoc: '║bd║',
+        afterDoc: '║ad║',
+        beforeDecorators: '║bds║',
+        afterDecorators: '║ads║',
+        beforeBody: '║bb║',
+        afterBody: '║ab║',
+      },
+    }),
+  );
+  expect(builder.toString(false)).toBe(
+    normalizeEOL(6)(
+      `║b║║bd║
+      /**
+       * description
+       *
+       * @param x x-description
+       * @param y y-description
+       * @returns r-description
+       */
+      ║ad║║bds║@decorator
+      ║ads║constructor║bp║(x, y)║ap║ ║bb║{
+        this.x = x;
+      }║ab║
+      ║a║`,
+    ),
+  );
+});

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.test.ts

@@ -0,0 +1,59 @@
+import "../../../_dnt.test_polyfills.js";
+import { EOL } from 'node:os';
+
+import { expect } from '../../../deps/jsr.io/@std/expect/1.0.12/expect.js';
+import { beforeEach, describe, it } from '../../../deps/jsr.io/@std/testing/1.0.9/bdd.js';
+
+import { TypeScriptFileBuilder } from '../../file-builder.js';
+import { tsDecorator } from './decorator.js';
+
+let builder: TypeScriptFileBuilder;
+
+beforeEach(() => {
+  builder = new TypeScriptFileBuilder();
+});
+
+describe('single', () => {
+  it('should write a static decorator', () => {
+    builder.append(tsDecorator('decorator'));
+    expect(builder.toString(false)).toBe(`@decorator`);
+  });
+
+  it('should write a decorator without arguments', () => {
+    builder.append(tsDecorator('decorator', []));
+    expect(builder.toString(false)).toBe(`@decorator()`);
+  });
+
+  it('should write a decorator with arguments', () => {
+    builder.append(tsDecorator('decorator', ['arg1', 'arg2']));
+    expect(builder.toString(false)).toBe(`@decorator(arg1, arg2)`);
+  });
+
+  it('should render injections', () => {
+    builder.append(
+      tsDecorator('decorator', ['arg1', 'arg2'], {
+        inject: {
+          before: '║b║',
+          after: '║a║',
+          beforeFunction: '║bf║',
+          afterFunction: '║af║',
+          beforeArguments: '║ba║',
+          afterArguments: '║aa║',
+        },
+      }),
+    );
+    expect(builder.toString(false)).toBe(`║b║@║bf║decorator║af║║ba║(arg1, arg2)║aa║║a║`);
+  });
+});
+
+describe('multiple', () => {
+  it('should write decorators across multiple lines', () => {
+    tsDecorator.write(builder, [tsDecorator('decorator1'), tsDecorator('decorator2')], { multiline: true });
+    expect(builder.toString(false)).toBe(`@decorator1${EOL}@decorator2${EOL}`);
+  });
+
+  it('should write decorators on the same line', () => {
+    tsDecorator.write(builder, [tsDecorator('decorator1'), tsDecorator('decorator2')], { multiline: false });
+    expect(builder.toString(false)).toBe(`@decorator1 @decorator2 `);
+  });
+});

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