@goast/kotlin 0.5.1-beta.1 → 0.5.2

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

@@ -0,0 +1,73 @@
+import { type AstNodeOptions, notNullish, type Nullable, type SourceBuilder } from '@goast/core';
+
+import { KtNode } from '../node.js';
+import { writeKtNode, writeKtNodes } from '../utils/write-kt-node.js';
+import { ktParameter } from './parameter.js';
+import type { KtType } from './types.js';
+
+type Injects = 'extensionFor' | 'params' | 'returnType';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof KtNode<TBuilder, TInjects | Injects>,
+  {
+    extensionFor?: Nullable<KtType<TBuilder>>;
+    parameters?: Nullable<Nullable<KtType<TBuilder>>[]>;
+    returnType: KtType<TBuilder>;
+    suspend?: boolean;
+  }
+>;
+
+export class KtLambdaType<TBuilder extends SourceBuilder, TInjects extends string = never> extends KtNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public extensionFor: KtType<TBuilder> | null;
+  public parameters: KtType<TBuilder>[];
+  public returnType: KtType<TBuilder>;
+  public suspend: boolean;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.extensionFor = options.extensionFor ?? null;
+    this.parameters = options.parameters?.filter(notNullish) ?? [];
+    this.returnType = options.returnType;
+    this.suspend = options.suspend ?? false;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    if (this.suspend) {
+      builder.append('suspend ');
+    }
+
+    if (this.extensionFor) {
+      builder.append(this.inject.beforeExtensionFor);
+      writeKtNode(builder, this.extensionFor);
+      builder.append(this.inject.afterExtensionFor);
+      builder.append('.');
+    }
+
+    builder.append(this.inject.beforeParams);
+    if (this.parameters.length === 0) {
+      builder.append('()');
+    } else {
+      ktParameter.write(builder, this.parameters);
+    }
+    builder.append(this.inject.afterParams);
+
+    builder.append(' -> ');
+
+    builder.append(this.inject.beforeReturnType);
+    writeKtNode(builder, this.returnType);
+    builder.append(this.inject.afterReturnType);
+  }
+}
+
+const createLambdaType = <TBuilder extends SourceBuilder>(
+  parameters: Options<TBuilder>['parameters'],
+  returnType: Options<TBuilder>['returnType'],
+  options?: Omit<Options<TBuilder>, 'parameters' | 'returnType'>,
+): KtLambdaType<TBuilder> => new KtLambdaType<TBuilder>({ ...options, parameters, returnType });
+
+export const ktLambdaType: typeof createLambdaType & { write: typeof writeKtNodes } = Object.assign(createLambdaType, {
+  write: writeKtNodes,
+});

package/src/src/ast/nodes/lambda.ts

@@ -0,0 +1,74 @@
+import {
+  type AppendValue,
+  type AstNodeOptions,
+  type BasicAppendValue,
+  notNullish,
+  type Nullable,
+  type SourceBuilder,
+} from '@goast/core';
+
+import { KtNode } from '../node.js';
+import { writeKtNodes } from '../utils/write-kt-node.js';
+
+type Injects = 'arguments' | 'body';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof KtNode<TBuilder, TInjects | Injects>,
+  {
+    arguments?: Nullable<Nullable<BasicAppendValue<TBuilder>>[]>;
+    body?: Nullable<AppendValue<TBuilder>>;
+    singleline?: boolean;
+  }
+>;
+
+export class KtLambda<TBuilder extends SourceBuilder, TInjects extends string = never> extends KtNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public arguments: BasicAppendValue<TBuilder>[];
+  public body: AppendValue<TBuilder> | null;
+  public singleline: boolean;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.arguments = options.arguments?.filter(notNullish) ?? [];
+    this.body = options.body ?? null;
+    this.singleline = options.singleline ?? false;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    const singleline = this.singleline || !this.body;
+
+    builder.parenthesize('{}', (b) => {
+      if (this.arguments.length > 0) {
+        b.append(' ', this.inject.beforeArguments);
+        writeKtNodes(b, this.arguments, { separator: ', ' });
+        b.append(this.inject.afterArguments);
+
+        b.append(' ->');
+      }
+
+      b.append(singleline ? ' ' : '\n');
+
+      b.append(this.inject.beforeBody, this.body, this.inject.afterBody);
+
+      if (singleline) {
+        if (!b.isCurrentLineEmpty) {
+          b.append(' ');
+        }
+      } else {
+        b.ensureCurrentLineEmpty();
+      }
+    });
+  }
+}
+
+const createLambda = <TBuilder extends SourceBuilder>(
+  args: Options<TBuilder>['arguments'],
+  body: Options<TBuilder>['body'],
+  options?: Omit<Options<TBuilder>, 'arguments' | 'body'>,
+): KtLambda<TBuilder> => new KtLambda<TBuilder>({ ...options, arguments: args, body });
+
+export const ktLambda: typeof createLambda & { write: typeof writeKtNodes } = Object.assign(createLambda, {
+  write: writeKtNodes,
+});

package/src/src/ast/nodes/object.ts

@@ -0,0 +1,102 @@
+import { type AstNodeOptions, type BasicAppendValue, notNullish, type Nullable, type SourceBuilder } from '@goast/core';
+
+import { KtNode } from '../node.js';
+import { writeKtMembers } from '../utils/write-kt-members.js';
+import { writeKtNode, writeKtNodes } from '../utils/write-kt-node.js';
+import { type KtArgument, ktArgument } from './argument.js';
+import type { KtFunction } from './function.js';
+import type { KtInitBlock } from './init-block.js';
+import type { KtProperty } from './property.js';
+import type { KtType, KtValue } from './types.js';
+
+type Injects = 'modifiers' | 'name' | 'inheritList' | 'body' | 'members';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof KtNode<TBuilder, TInjects | Injects>,
+  {
+    data?: boolean;
+    name?: Nullable<string>;
+    class?: KtType<TBuilder>;
+    classArguments?: Nullable<Nullable<KtArgument<TBuilder> | KtValue<TBuilder>>[]>;
+    implements?: Nullable<Nullable<KtType<TBuilder>>[]>;
+    members?: Nullable<Nullable<KtObjectMember<TBuilder>>[]>;
+    companion?: Nullable<boolean>;
+  }
+>;
+
+export type KtObjectMember<TBuilder extends SourceBuilder> =
+  | KtInitBlock<TBuilder>
+  | KtProperty<TBuilder>
+  | KtFunction<TBuilder>
+  | BasicAppendValue<TBuilder>;
+
+export class KtObject<TBuilder extends SourceBuilder, TInjects extends string = never> extends KtNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public data: boolean;
+  public name: string | null;
+  public class: KtType<TBuilder> | null;
+  public classArguments: (KtArgument<TBuilder> | KtValue<TBuilder>)[];
+  public implements: KtType<TBuilder>[];
+  public members: KtObjectMember<TBuilder>[];
+  public companion: boolean;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.data = options.data ?? false;
+    this.name = options.name ?? null;
+    this.class = options.class ?? null;
+    this.classArguments = options.classArguments?.filter(notNullish) ?? [];
+    this.implements = options.implements?.filter(notNullish) ?? [];
+    this.members = options.members?.filter(notNullish) ?? [];
+    this.companion = options.companion ?? false;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append(this.inject.beforeModifiers);
+    if (this.data) builder.append('data ');
+    if (this.companion) builder.append('companion ');
+    builder.append(this.inject.afterModifiers);
+
+    builder.append('object');
+
+    if (this.name) {
+      builder.append(' ', this.inject.beforeName, this.name, this.inject.afterName);
+    }
+
+    if (this.class || this.implements.length > 0) {
+      builder.append(' : ', this.inject.beforeInheritList);
+
+      if (this.class) {
+        writeKtNode(builder, this.class);
+        ktArgument.write(builder, this.classArguments);
+        if (this.implements.length > 0) builder.append(', ');
+      }
+
+      writeKtNodes(builder, this.implements, { separator: ', ' });
+      builder.append(this.inject.afterInheritList);
+    }
+
+    builder.append(' ', this.inject.beforeBody);
+    builder.parenthesize(
+      '{}',
+      (b) => {
+        builder.append(this.inject.beforeMembers);
+        writeKtMembers(b, this.members);
+        builder.append(this.inject.afterMembers);
+      },
+      { multiline: this.members.length > 0 },
+    );
+    builder.append(this.inject.afterBody);
+
+    if (this.name) builder.appendLine();
+  }
+}
+
+const createObject = <TBuilder extends SourceBuilder>(options?: Options<TBuilder>): KtObject<TBuilder> =>
+  new KtObject<TBuilder>(options ?? {});
+
+export const ktObject: typeof createObject & { write: typeof writeKtNodes } = Object.assign(createObject, {
+  write: writeKtNodes,
+});

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

@@ -0,0 +1,118 @@
+import {
+  type AstNodeOptions,
+  type BasicAppendValue,
+  notNullish,
+  type Nullable,
+  type Prettify,
+  type SourceBuilder,
+} from '@goast/core';
+
+import type { KtAccessModifier } from '../common.js';
+import { KtNode } from '../node.js';
+import { writeKtNode } from '../utils/write-kt-node.js';
+import { writeKtParameters } from '../utils/write-kt-parameters.js';
+import { type KtAnnotation, ktAnnotation } from './annotation.js';
+import type { KtType, KtValue } from './types.js';
+
+type Injects = 'modifiers' | 'annotations' | 'name' | 'type' | 'default';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof KtNode<TBuilder, TInjects | Injects>,
+  {
+    name: string;
+    type: KtType<TBuilder>;
+    annotations?: Nullable<Nullable<KtAnnotation<TBuilder>>[]>;
+    default?: Nullable<KtValue<TBuilder>>;
+    vararg?: Nullable<boolean>;
+    description?: Nullable<BasicAppendValue<TBuilder>>;
+
+    // class parameter options
+    accessModifier?: Nullable<KtAccessModifier>;
+    property?: Nullable<'readonly' | 'mutable'>;
+    propertyDescription?: Nullable<BasicAppendValue<TBuilder>>;
+    override?: Nullable<boolean>;
+  }
+>;
+
+export class KtParameter<TBuilder extends SourceBuilder, TInjects extends string = never> extends KtNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public name: string;
+  public type: KtType<TBuilder>;
+  public annotations: KtAnnotation<TBuilder>[];
+  public default: KtValue<TBuilder> | null;
+  public vararg: boolean;
+  public description: BasicAppendValue<TBuilder> | null;
+
+  // class parameter options
+  public accessModifier: KtAccessModifier | null;
+  public property: 'readonly' | 'mutable' | null;
+  public propertyDescription: BasicAppendValue<TBuilder> | null;
+  public override: boolean;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.name = options.name;
+    this.type = options.type;
+    this.annotations = options.annotations?.filter(notNullish) ?? [];
+    this.default = options.default ?? null;
+    this.vararg = options.vararg ?? false;
+    this.description = options.description ?? null;
+
+    this.accessModifier = options.accessModifier ?? null;
+    this.property = options.property ?? null;
+    this.propertyDescription = options.propertyDescription ?? null;
+    this.override = options.override ?? false;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append(this.inject.beforeAnnotations);
+    ktAnnotation.write(builder, this.annotations, { multiline: true });
+    builder.append(this.inject.afterAnnotations);
+
+    builder.append(this.inject.beforeModifiers);
+    if (this.accessModifier && this.property) builder.append(this.accessModifier, ' ');
+    if (this.override && this.property) builder.append('override ');
+    if (this.vararg) builder.append('vararg ');
+    builder.append(this.inject.afterModifiers);
+
+    if (this.property) builder.append(this.property === 'mutable' ? 'var' : 'val', ' ');
+    builder.append(this.inject.beforeName, this.name, this.inject.afterName);
+
+    builder.append(': ');
+
+    builder.append(this.inject.beforeType);
+    writeKtNode(builder, this.type);
+    builder.append(this.inject.afterType);
+
+    if (this.default) {
+      builder.append(' = ');
+      builder.append(this.inject.beforeDefault);
+      writeKtNode(builder, this.default);
+      builder.append(this.inject.afterDefault);
+    }
+  }
+}
+
+const createParameter = <TBuilder extends SourceBuilder>(
+  name: KtParameter<TBuilder>['name'],
+  type: KtParameter<TBuilder>['type'],
+  options?: Prettify<
+    Omit<Options<TBuilder>, 'name' | 'type' | 'accessModifier' | 'property' | 'propertyDescription' | 'override'>
+  >,
+): KtParameter<TBuilder> => new KtParameter<TBuilder>({ ...options, name, type });
+
+const createClassParameter = <TBuilder extends SourceBuilder>(
+  name: KtParameter<TBuilder>['name'],
+  type: KtParameter<TBuilder>['type'],
+  options?: Prettify<Omit<Options<TBuilder>, 'name' | 'type'>>,
+): KtParameter<TBuilder> => new KtParameter<TBuilder>({ ...options, name, type });
+
+export const ktParameter: typeof createParameter & {
+  class: typeof createClassParameter;
+  write: typeof writeKtParameters;
+} = Object.assign(createParameter, {
+  class: createClassParameter,
+  write: writeKtParameters,
+});

package/src/src/ast/nodes/property.ts

@@ -0,0 +1,225 @@
+import {
+  type AppendValue,
+  type AstNodeOptions,
+  notNullish,
+  type Nullable,
+  type Prettify,
+  type SourceBuilder,
+} from '@goast/core';
+
+import type { KtAccessModifier } from '../common.js';
+import { KtNode } from '../node.js';
+import { writeKtNode, writeKtNodes } from '../utils/write-kt-node.js';
+import { type KtAnnotation, ktAnnotation } from './annotation.js';
+import { type KtArgument, ktArgument } from './argument.js';
+import type { KtDoc } from './doc.js';
+import type { KtType, KtValue } from './types.js';
+
+type AccessorInjects = 'annotations' | 'modifiers' | 'params' | 'body';
+
+type AccessorOptions<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof KtNode<TBuilder, TInjects | AccessorInjects>,
+  {
+    accessModifier?: Nullable<KtAccessModifier>;
+    annotations?: Nullable<Nullable<KtAnnotation<TBuilder>>[]>;
+    body?: Nullable<AppendValue<TBuilder>>;
+    singleExpression?: Nullable<boolean>;
+  }
+>;
+
+export abstract class KtPropertyAccessor<
+  TBuilder extends SourceBuilder,
+  TInjects extends string = never,
+> extends KtNode<TBuilder, TInjects | AccessorInjects> {
+  public abstract get kind(): 'get' | 'set';
+  public accessModifier: KtAccessModifier | null;
+  public annotations: KtAnnotation<TBuilder>[];
+  public body: AppendValue<TBuilder> | null;
+  public singleExpression: boolean;
+
+  constructor(options: AccessorOptions<TBuilder, TInjects>) {
+    super(options);
+    this.accessModifier = options.accessModifier ?? null;
+    this.annotations = options.annotations?.filter(notNullish) ?? [];
+    this.body = options.body ?? null;
+    this.singleExpression = options.singleExpression ?? false;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append(this.inject.beforeAnnotations);
+    ktAnnotation.write(builder, this.annotations, { multiline: true });
+    builder.append(this.inject.afterAnnotations);
+
+    builder.append(this.inject.beforeModifiers);
+    if (this.accessModifier) builder.append(this.accessModifier, ' ');
+    builder.append(this.inject.afterModifiers);
+
+    builder.append(this.kind);
+
+    if (this.body) {
+      builder.append(this.inject.beforeParams);
+      builder.append(this.kind === 'set' ? '(value)' : '()');
+      builder.append(this.inject.afterParams);
+
+      if (this.singleExpression) {
+        builder.append(' = ', this.inject.beforeBody);
+        builder.indent(this.body);
+      } else {
+        builder.append(' ', this.inject.beforeBody);
+        builder.parenthesize('{}', this.body, { multiline: true });
+      }
+      builder.append(this.inject.afterBody);
+    }
+
+    builder.appendLine();
+  }
+}
+export class KtPropertyGetter<TBuilder extends SourceBuilder> extends KtPropertyAccessor<TBuilder> {
+  public override readonly kind = 'get' as const;
+}
+export class KtPropertySetter<TBuilder extends SourceBuilder> extends KtPropertyAccessor<TBuilder> {
+  public override readonly kind = 'set' as const;
+}
+
+type Injects = 'doc' | 'annotations' | 'modifiers' | 'name' | 'type' | 'default' | 'delegate';
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof KtNode<TBuilder, TInjects | Injects>,
+  {
+    doc?: Nullable<KtDoc<TBuilder>>;
+    name: string;
+    type?: KtType<TBuilder>;
+    annotations?: Nullable<Nullable<KtAnnotation<TBuilder>>[]>;
+    accessModifier?: Nullable<KtAccessModifier>;
+    getter?: Nullable<KtPropertyGetter<TBuilder>>;
+    setter?: Nullable<KtPropertySetter<TBuilder>>;
+    default?: Nullable<KtValue<TBuilder>>;
+    delegate?: Nullable<KtType<TBuilder>>;
+    delegateArguments?: Nullable<Nullable<KtArgument<TBuilder> | KtValue<TBuilder>>[]>;
+    mutable?: Nullable<boolean>;
+    const?: Nullable<boolean>;
+    lateinit?: Nullable<boolean>;
+    open?: Nullable<boolean>;
+    override?: Nullable<boolean>;
+    abstract?: Nullable<boolean>;
+  }
+>;
+
+export class KtProperty<TBuilder extends SourceBuilder, TInjects extends string = never> extends KtNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public doc: KtDoc<TBuilder> | null;
+  public name: string;
+  public type: KtType<TBuilder> | null;
+  public annotations: KtAnnotation<TBuilder>[];
+  public accessModifier: KtAccessModifier | null;
+  public getter: KtPropertyGetter<TBuilder> | null;
+  public setter: KtPropertySetter<TBuilder> | null;
+  public default: KtValue<TBuilder> | null;
+  public delegate: KtType<TBuilder> | null;
+  public delegateArguments: (KtArgument<TBuilder> | KtValue<TBuilder>)[] | null;
+  public mutable: boolean;
+  public const: boolean;
+  public lateinit: boolean;
+  public open: boolean;
+  public override: boolean;
+  public abstract: boolean;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.doc = options.doc ?? null;
+    this.name = options.name;
+    this.type = options.type ?? null;
+    this.annotations = options.annotations?.filter(notNullish) ?? [];
+    this.accessModifier = options.accessModifier ?? null;
+    this.getter = options.getter ?? null;
+    this.setter = options.setter ?? null;
+    this.default = options.default ?? null;
+    this.delegate = options.delegate ?? null;
+    this.delegateArguments = options.delegateArguments?.filter(notNullish) ?? null;
+    this.mutable = options.mutable ?? false;
+    this.const = options.const ?? false;
+    this.lateinit = options.lateinit ?? false;
+    this.open = options.open ?? false;
+    this.override = options.override ?? false;
+    this.abstract = options.abstract ?? false;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append(this.inject.beforeDoc);
+    this.doc?.write(builder);
+    builder.append(this.inject.afterDoc);
+
+    builder.append(this.inject.beforeAnnotations);
+    ktAnnotation.write(builder, this.annotations, { multiline: true });
+    builder.append(this.inject.afterAnnotations);
+
+    builder.append(this.inject.beforeModifiers);
+    if (this.accessModifier) builder.append(this.accessModifier, ' ');
+    if (this.const) builder.append('const ');
+    if (this.lateinit) builder.append('lateinit ');
+    if (this.abstract) builder.append('abstract ');
+    if (this.override) builder.append('override ');
+    if (this.open) builder.append('open ');
+    builder.append(this.inject.afterModifiers);
+
+    builder.append(this.mutable || !!this.setter ? 'var ' : 'val ');
+    builder.append(this.inject.beforeName, this.name, this.inject.afterName);
+
+    if (this.type || (!this.default && !this.getter?.body)) {
+      builder.append(': ', this.inject.beforeType);
+      writeKtNode(builder, this.type ?? 'Any?');
+      builder.append(this.inject.afterType);
+    }
+
+    if (this.default) {
+      builder.append(' = ', this.inject.beforeDefault);
+      writeKtNode(builder, this.default);
+      builder.append(this.inject.afterDefault);
+    }
+
+    if (this.delegate) {
+      builder.append(' by ', this.inject.beforeDelegate);
+      writeKtNode(builder, this.delegate);
+      if (this.delegateArguments) {
+        ktArgument.write(builder, this.delegateArguments);
+      }
+      builder.append(this.inject.afterDelegate);
+    }
+
+    builder.appendLine();
+
+    if (this.getter || this.setter) {
+      builder.indent((b) => {
+        this.getter?.write(b);
+        this.setter?.write(b);
+      });
+    }
+
+    builder.ensureCurrentLineEmpty();
+  }
+}
+
+const createPropertyGetter = <TBuilder extends SourceBuilder>(
+  options?: AccessorOptions<TBuilder>,
+): KtPropertyGetter<TBuilder> => new KtPropertyGetter<TBuilder>(options ?? {});
+
+const createPropertySetter = <TBuilder extends SourceBuilder>(
+  options?: AccessorOptions<TBuilder>,
+): KtPropertySetter<TBuilder> => new KtPropertySetter<TBuilder>(options ?? {});
+
+const createProperty = <TBuilder extends SourceBuilder>(
+  name: Options<TBuilder>['name'],
+  options?: Prettify<Omit<Options<TBuilder>, 'name'>>,
+): KtProperty<TBuilder> => new KtProperty<TBuilder>({ ...options, name });
+
+export const ktProperty: typeof createProperty & {
+  getter: typeof createPropertyGetter;
+  setter: typeof createPropertySetter;
+  write: typeof writeKtNodes;
+} = Object.assign(createProperty, {
+  getter: createPropertyGetter,
+  setter: createPropertySetter,
+  write: writeKtNodes,
+});