@goast/kotlin 0.5.9-beta.2 → 0.5.10

package/src/src/ast/nodes/reference.ts

@@ -0,0 +1,178 @@
+import {
+  type AstNodeOptions,
+  notNullish,
+  type Nullable,
+  type Prettify,
+  type SourceBuilder,
+  type TupleWithCount,
+} from '@goast/core';
+
+import { KotlinFileBuilder } from '../../file-builder.js';
+import { KtNode } from '../node.js';
+import { writeKtNode, writeKtNodes } from '../utils/write-kt-node.js';
+import { ktGenericParameter } from './generic-parameter.js';
+import type { KtType } from './types.js';
+
+import type { KtCall } from './call.js';
+
+type Injects = never;
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof KtNode<TBuilder, TInjects | Injects>,
+  {
+    name: string;
+    packageName?: Nullable<string>;
+    generics?: Nullable<Nullable<KtType<TBuilder>>[]>;
+    nullable?: Nullable<boolean>;
+    classReference?: Nullable<boolean>;
+    subReference?: Nullable<KtReference<TBuilder> | KtCall<TBuilder>>;
+  }
+>;
+
+type _AddImportHandler<T> = {
+  // deno-lint-ignore no-explicit-any
+  builderClass: abstract new (...args: any) => T;
+  // deno-lint-ignore no-explicit-any
+  handler: (builder: T, reference: KtReference<any>) => void;
+};
+
+export class KtReference<TBuilder extends SourceBuilder, TInjects extends string = never> extends KtNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  // deno-lint-ignore no-explicit-any
+  private static readonly addImportHandlers: _AddImportHandler<any>[] = [
+    {
+      builderClass: KotlinFileBuilder,
+      handler: (builder, reference) => builder.addImport(reference.name, reference.packageName),
+    },
+  ];
+
+  public name: string;
+  public packageName: string | null;
+  public generics: KtType<TBuilder>[];
+  public nullable: boolean;
+  public classReference: boolean;
+  public subReference: KtReference<TBuilder> | KtCall<TBuilder> | null;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.name = options.name;
+    this.packageName = options.packageName ?? null;
+    this.generics = options.generics?.filter(notNullish) ?? [];
+    this.nullable = options.nullable ?? false;
+    this.classReference = options.classReference ?? false;
+    this.subReference = options.subReference ?? null;
+  }
+
+  public addImport(builder: TBuilder): void {
+    if (this.packageName) {
+      const handler = KtReference.addImportHandlers.find((h) => builder instanceof h.builderClass);
+      if (handler) {
+        handler.handler(builder, this);
+      }
+    }
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    builder.append(this.name);
+    ktGenericParameter.write(builder, this.generics);
+    if (this.classReference) {
+      builder.append('::class');
+    } else if (this.subReference) {
+      builder.append('::');
+      writeKtNode(builder, this.subReference);
+    } else if (this.nullable) {
+      builder.append('?');
+    }
+
+    this.addImport(builder);
+  }
+
+  public static registerAddImportHandler<TBuilder extends SourceBuilder>(
+    builderClass: _AddImportHandler<TBuilder>['builderClass'],
+    handler: _AddImportHandler<TBuilder>['handler'],
+  ): void {
+    KtReference.addImportHandlers.push({ builderClass, handler });
+  }
+}
+
+const createReference = <TBuilder extends SourceBuilder>(
+  name: Options<TBuilder>['name'],
+  packageName?: Options<TBuilder>['packageName'],
+  options?: Prettify<Omit<Options<TBuilder>, 'name' | 'packageName'>>,
+): KtReference<TBuilder> => new KtReference<TBuilder>({ ...options, name, packageName });
+
+export type KtReferenceFactory =
+  & (<TBuilder extends SourceBuilder>(
+    options?: Prettify<Omit<Options<TBuilder>, 'name' | 'packageName'>>,
+  ) => KtReference<TBuilder>)
+  & {
+    refName: string;
+    packageName: Nullable<string>;
+    matches: (value: unknown) => value is KtReference<never>;
+  };
+
+export type KtGenericReferenceFactory<TGenericCount extends number | number[]> =
+  & (<TBuilder extends SourceBuilder>(
+    generics: TupleWithCount<KtType<TBuilder>, TGenericCount>,
+    options?: Prettify<Omit<Options<TBuilder>, 'name' | 'packageName' | 'generics'>>,
+  ) => KtReference<TBuilder>)
+  & {
+    refName: string;
+    packageName: Nullable<string>;
+    infer: <TBuilder extends SourceBuilder>(
+      options?: Prettify<Omit<Options<TBuilder>, 'name' | 'packageName' | 'generics'>>,
+    ) => KtReference<TBuilder>;
+    matches: (value: unknown) => value is KtReference<never>;
+  };
+
+const createFactory = (name: string, packageName?: Nullable<string>): KtReferenceFactory => {
+  return Object.assign(
+    <TBuilder extends SourceBuilder>(options?: Prettify<Omit<Options<TBuilder>, 'name' | 'packageName'>>) =>
+      createReference<TBuilder>(name, packageName, options),
+    {
+      refName: name,
+      packageName,
+      matches: (value: unknown): value is KtReference<never> =>
+        value instanceof KtReference && value.name === name && value.packageName === (packageName ?? null),
+    },
+  );
+};
+
+const createGenericFactory = <TGenericCount extends number | number[]>(
+  name: string,
+  packageName?: Nullable<string>,
+): KtGenericReferenceFactory<TGenericCount> => {
+  return Object.assign(
+    <TBuilder extends SourceBuilder>(
+      generics: TupleWithCount<KtType<TBuilder>, TGenericCount>,
+      options?: Prettify<Omit<Options<TBuilder>, 'name' | 'packageName' | 'generics'>>,
+    ) => createReference<TBuilder>(name, packageName, { ...options, generics }),
+    {
+      refName: name,
+      packageName,
+      infer: <TBuilder extends SourceBuilder>(
+        options?: Prettify<Omit<Options<TBuilder>, 'name' | 'packageName' | 'generics'>>,
+      ) => createReference<TBuilder>(name, packageName, options),
+      matches: (value: unknown): value is KtReference<never> =>
+        value instanceof KtReference && value.name === name && value.packageName === (packageName ?? null),
+    },
+  );
+};
+
+const importRefs = <TBuilder extends SourceBuilder>(builder: TBuilder, references: KtReference<TBuilder>[]): void => {
+  references.forEach((ref) => ref.addImport(builder));
+};
+
+export const ktReference: typeof createReference & {
+  factory: typeof createFactory;
+  genericFactory: typeof createGenericFactory;
+  write: typeof writeKtNodes;
+  import: typeof importRefs;
+} = Object.assign(createReference, {
+  factory: createFactory,
+  genericFactory: createGenericFactory,
+  write: writeKtNodes,
+  import: importRefs,
+});

package/src/src/ast/nodes/string.ts

@@ -0,0 +1,114 @@
+import { type AstNodeOptions, type Nullable, type Prettify, type SourceBuilder, spliceString } from '@goast/core';
+
+import { KtNode } from '../node.js';
+import { writeKtNodes } from '../utils/write-kt-node.js';
+import { ktArgument } from './argument.js';
+
+type Injects = never;
+
+type Options<TBuilder extends SourceBuilder, TInjects extends string = never> = AstNodeOptions<
+  typeof KtNode<TBuilder, TInjects | Injects>,
+  {
+    value?: Nullable<string>;
+    template?: Nullable<boolean>;
+    multiline?: Nullable<boolean>;
+    trimMargin?: Nullable<boolean>;
+    marginPrefix?: Nullable<string>;
+    autoAddMarginPrefix?: Nullable<boolean>;
+  }
+>;
+
+export class KtString<TBuilder extends SourceBuilder, TInjects extends string = never> extends KtNode<
+  TBuilder,
+  TInjects | Injects
+> {
+  public value: string | null;
+  public template: boolean;
+  public multiline: boolean;
+  public trimMargin: boolean;
+  public marginPrefix: string | null;
+  public autoAddMarginPrefix: boolean;
+
+  constructor(options: Options<TBuilder, TInjects>) {
+    super(options);
+    this.value = options?.value ?? null;
+    this.template = options?.template ?? false;
+    this.multiline = options?.multiline ?? false;
+    this.trimMargin = options?.trimMargin ?? true;
+    this.marginPrefix = options?.marginPrefix ?? null;
+    this.autoAddMarginPrefix = options?.autoAddMarginPrefix ?? true;
+  }
+
+  protected override onWrite(builder: TBuilder): void {
+    if (this.value === null) {
+      builder.append('null');
+      return;
+    }
+
+    let value = JSON.stringify(this.value).slice(1, -1);
+    if (this.multiline) {
+      value = value.replace(/\\n/g, '\n').replace(/\\r/g, '\r');
+    }
+
+    if (!this.template) {
+      value = value.replace(/\$/g, '\\$');
+    } else {
+      const originalParts = this.findTemplateParts(this.value);
+      const escapedParts = this.findTemplateParts(value);
+      if (originalParts.length !== escapedParts.length) {
+        throw new Error('Template parts count mismatch');
+      }
+      for (let i = originalParts.length - 1; i >= 0; i--) {
+        const original = originalParts[i];
+        const escaped = escapedParts[i];
+        value = spliceString(value, escaped.index, escaped.value.length, original.value);
+      }
+    }
+
+    builder.append(this.multiline ? '"""' + (this.trimMargin ? '\n' : '') : '"');
+
+    if (this.multiline && this.trimMargin) {
+      builder.indent((b) => {
+        if (this.autoAddMarginPrefix) {
+          b.appendWithLinePrefix(this.marginPrefix ?? '|', value);
+        } else {
+          b.append(value);
+        }
+
+        b.append('\n""".trimMargin');
+        ktArgument.write(b, [this.marginPrefix ? new KtString<TBuilder>({ value: this.marginPrefix }) : null]);
+      });
+    } else {
+      builder.append(value, this.multiline ? '"""' : '"');
+    }
+  }
+
+  protected findTemplateParts(value: string): { value: string; index: number }[] {
+    const parts: { value: string; index: number }[] = [];
+    let start: number | undefined = undefined;
+    let bracketCount = 0;
+    for (let i = 0; i < value.length; i++) {
+      if (value[i] === '$' && value[i + 1] === '{') {
+        start = i;
+      } else if (value[i] === '{') {
+        bracketCount++;
+      } else if (value[i] === '}') {
+        bracketCount--;
+        if (bracketCount === 0 && start !== undefined) {
+          parts.push({ value: value.slice(start, i + 1), index: start });
+          start = undefined;
+        }
+      }
+    }
+    return parts;
+  }
+}
+
+const createString = <TBuilder extends SourceBuilder>(
+  value: Options<TBuilder>['value'],
+  options?: Prettify<Omit<Options<TBuilder>, 'value'>>,
+): KtString<TBuilder> => new KtString<TBuilder>({ ...options, value });
+
+export const ktString: typeof createString & { write: typeof writeKtNodes } = Object.assign(createString, {
+  write: writeKtNodes,
+});

package/src/src/ast/nodes/types.ts

@@ -0,0 +1,23 @@
+import type { BasicAppendValue, SourceBuilder } from '@goast/core';
+
+import type { KtCall } from './call.js';
+import type { KtCollectionLiteral } from './collection-literal.js';
+import type { KtLambdaType } from './lambda-type.js';
+import type { KtLambda } from './lambda.js';
+import type { KtObject } from './object.js';
+import type { KtReference } from './reference.js';
+import type { KtString } from './string.js';
+
+export type KtType<TBuilder extends SourceBuilder> =
+  | KtReference<TBuilder>
+  | KtLambdaType<TBuilder>
+  | BasicAppendValue<TBuilder>;
+
+export type KtValue<TBuilder extends SourceBuilder> =
+  | KtReference<TBuilder>
+  | KtString<TBuilder>
+  | KtCall<TBuilder>
+  | KtObject<TBuilder>
+  | KtLambda<TBuilder>
+  | KtCollectionLiteral<TBuilder>
+  | BasicAppendValue<TBuilder>;

package/src/src/ast/references/index.ts

@@ -0,0 +1,10 @@
+export * as jackson from './jackson.js';
+export * as jakarta from './jakarta.js';
+export * as java from './java.js';
+export * from './kotlin.js';
+export * as kotlinx from './kotlinx.js';
+export * as okhttp3 from './okhttp3.js';
+export * as reactor from './reactor.js';
+export * as springReactive from './spring-reactive.js';
+export * as spring from './spring.js';
+export * as swagger from './swagger.js';

package/src/src/ast/references/jackson.ts

@@ -0,0 +1,59 @@
+import type { SourceBuilder } from '@goast/core';
+
+import type { SpringBootVersion } from '../../config.js';
+import { type KtReference, ktReference, type KtReferenceFactory } from '../nodes/reference.js';
+
+/**
+ * Resolves a Jackson package name for the targeted Spring Boot version.
+ *
+ * Jackson 3 (the default in Spring Boot 4) moved every package **except annotations** from
+ * `com.fasterxml.jackson.*` to `tools.jackson.*`. Annotations stay on `com.fasterxml.jackson.annotation`.
+ */
+const jacksonPackage = (subPackage: string, springBootVersion: SpringBootVersion): string =>
+  `${springBootVersion === 4 ? 'tools.jackson' : 'com.fasterxml.jackson'}.${subPackage}`;
+
+// com.fasterxml.jackson.annotation
+export const jsonTypeInfo: KtReferenceFactory = ktReference.factory('JsonTypeInfo', 'com.fasterxml.jackson.annotation');
+export const jsonSubTypes: KtReferenceFactory = ktReference.factory('JsonSubTypes', 'com.fasterxml.jackson.annotation');
+export const jsonClassDescription: KtReferenceFactory = ktReference.factory(
+  'JsonClassDescription',
+  'com.fasterxml.jackson.annotation',
+);
+export const jsonProperty: KtReferenceFactory = ktReference.factory('JsonProperty', 'com.fasterxml.jackson.annotation');
+export const jsonPropertyDescription: KtReferenceFactory = ktReference.factory(
+  'JsonPropertyDescription',
+  'com.fasterxml.jackson.annotation',
+);
+export const jsonInclude: KtReferenceFactory = ktReference.factory('JsonInclude', 'com.fasterxml.jackson.annotation');
+export const jsonIgnore: KtReferenceFactory = ktReference.factory('JsonIgnore', 'com.fasterxml.jackson.annotation');
+export const jsonAnySetter: KtReferenceFactory = ktReference.factory(
+  'JsonAnySetter',
+  'com.fasterxml.jackson.annotation',
+);
+export const jsonAnyGetter: KtReferenceFactory = ktReference.factory(
+  'JsonAnyGetter',
+  'com.fasterxml.jackson.annotation',
+);
+
+// com.fasterxml.jackson.databind (Jackson 2) / tools.jackson.databind (Jackson 3, Spring Boot 4)
+export const objectMapper = <TBuilder extends SourceBuilder>(
+  springBootVersion: SpringBootVersion = 3,
+): KtReference<TBuilder> => ktReference<TBuilder>('ObjectMapper', jacksonPackage('databind', springBootVersion));
+export const serializationFeature = <TBuilder extends SourceBuilder>(
+  springBootVersion: SpringBootVersion = 3,
+): KtReference<TBuilder> =>
+  ktReference<TBuilder>('SerializationFeature', jacksonPackage('databind', springBootVersion));
+export const deserializationFeature = <TBuilder extends SourceBuilder>(
+  springBootVersion: SpringBootVersion = 3,
+): KtReference<TBuilder> =>
+  ktReference<TBuilder>('DeserializationFeature', jacksonPackage('databind', springBootVersion));
+
+// com.fasterxml.jackson.module.kotlin (Jackson 2) / tools.jackson.module.kotlin (Jackson 3, Spring Boot 4)
+export const jacksonObjectMapper = <TBuilder extends SourceBuilder>(
+  springBootVersion: SpringBootVersion = 3,
+): KtReference<TBuilder> =>
+  ktReference<TBuilder>('jacksonObjectMapper', jacksonPackage('module.kotlin', springBootVersion));
+export const jacksonMapperBuilder = <TBuilder extends SourceBuilder>(
+  springBootVersion: SpringBootVersion = 3,
+): KtReference<TBuilder> =>
+  ktReference<TBuilder>('jacksonMapperBuilder', jacksonPackage('module.kotlin', springBootVersion));

package/src/src/ast/references/jakarta.ts

@@ -0,0 +1,14 @@
+import { ktReference, type KtReferenceFactory } from '../nodes/reference.js';
+
+// jakarta.annotation
+export const generated: KtReferenceFactory = ktReference.factory('Generated', 'jakarta.annotation');
+
+// jakarta.validation
+export const valid: KtReferenceFactory = ktReference.factory('Valid', 'jakarta.validation');
+
+// jakarta.validation.constraints
+export const pattern: KtReferenceFactory = ktReference.factory('Pattern', 'jakarta.validation.constraints');
+export const min: KtReferenceFactory = ktReference.factory('Min', 'jakarta.validation.constraints');
+export const max: KtReferenceFactory = ktReference.factory('Max', 'jakarta.validation.constraints');
+export const notEmpty: KtReferenceFactory = ktReference.factory('NotEmpty', 'jakarta.validation.constraints');
+export const size: KtReferenceFactory = ktReference.factory('Size', 'jakarta.validation.constraints');

package/src/src/ast/references/java.ts

@@ -0,0 +1,27 @@
+import { type KtGenericReferenceFactory, ktReference, type KtReferenceFactory } from '../nodes/reference.js';
+
+// java.io
+export const file: KtReferenceFactory = ktReference.factory('File', 'java.io');
+export const ioException: KtReferenceFactory = ktReference.factory('IOException', 'java.io');
+
+// java.lang
+export const illegalStateException: KtReferenceFactory = ktReference.factory('IllegalStateException', 'java.lang');
+export const system: KtReferenceFactory = ktReference.factory('System', 'java.lang');
+export const unsupportedOperationException: KtReferenceFactory = ktReference.factory(
+  'UnsupportedOperationException',
+  'java.lang',
+);
+export const void_: KtReferenceFactory = ktReference.factory('Void', 'java.lang');
+
+// java.net
+export const uri: KtReferenceFactory = ktReference.factory('URI', 'java.net');
+
+// java.time
+export const offsetDateTime: KtReferenceFactory = ktReference.factory('OffsetDateTime', 'java.time');
+export const offsetTime: KtReferenceFactory = ktReference.factory('OffsetTime', 'java.time');
+export const localDateTime: KtReferenceFactory = ktReference.factory('LocalDateTime', 'java.time');
+export const localDate: KtReferenceFactory = ktReference.factory('LocalDate', 'java.time');
+export const localTime: KtReferenceFactory = ktReference.factory('LocalTime', 'java.time');
+
+// java.util
+export const optional: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>('Optional', 'java.util');

package/src/src/ast/references/kotlin.ts

@@ -0,0 +1,41 @@
+import { type KtGenericReferenceFactory, ktReference, type KtReferenceFactory } from '../nodes/reference.js';
+
+// kotlin
+export const any: KtReferenceFactory = ktReference.factory('Any', 'kotlin');
+export const nothing: KtReferenceFactory = ktReference.factory('Nothing', 'kotlin');
+export const unit: KtReferenceFactory = ktReference.factory('Unit', 'kotlin');
+export const string: KtReferenceFactory = ktReference.factory('String', 'kotlin');
+export const int: KtReferenceFactory = ktReference.factory('Int', 'kotlin');
+export const long: KtReferenceFactory = ktReference.factory('Long', 'kotlin');
+export const float: KtReferenceFactory = ktReference.factory('Float', 'kotlin');
+export const double: KtReferenceFactory = ktReference.factory('Double', 'kotlin');
+export const boolean: KtReferenceFactory = ktReference.factory('Boolean', 'kotlin');
+export const lazyFun: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>('lazy', 'kotlin');
+export const throws: KtReferenceFactory = ktReference.factory('Throws', 'kotlin');
+export const deprecated: KtReferenceFactory = ktReference.factory('Deprecated', 'kotlin');
+export const throwable: KtReferenceFactory = ktReference.factory('Throwable', 'kotlin');
+
+// kotlin.collections
+export const list: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>('List', 'kotlin.collections');
+export const mutableList: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>(
+  'MutableList',
+  'kotlin.collections',
+);
+export const map: KtGenericReferenceFactory<2> = ktReference.genericFactory<2>('Map', 'kotlin.collections');
+export const mutableMap: KtGenericReferenceFactory<2> = ktReference.genericFactory<2>(
+  'MutableMap',
+  'kotlin.collections',
+);
+export const listOf: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>('listOf', 'kotlin.collections');
+export const mutableListOf: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>(
+  'mutableListOf',
+  'kotlin.collections',
+);
+export const mapOf: KtGenericReferenceFactory<2> = ktReference.genericFactory<2>('mapOf', 'kotlin.collections');
+export const mutableMapOf: KtGenericReferenceFactory<2> = ktReference.genericFactory<2>(
+  'mutableMapOf',
+  'kotlin.collections',
+);
+
+// kotlin.jvm
+export const jvmStatic: KtReferenceFactory = ktReference.factory('JvmStatic', 'kotlin.jvm');

package/src/src/ast/references/kotlinx.ts

@@ -0,0 +1,14 @@
+import { type KtGenericReferenceFactory, ktReference } from '../nodes/reference.js';
+
+// kotlinx.coroutines.reactive
+export const awaitFirst: KtGenericReferenceFactory<1> = ktReference.genericFactory(
+  'awaitFirst',
+  'kotlinx.coroutines.reactive',
+);
+export const awaitFirstOrNull: KtGenericReferenceFactory<1> = ktReference.genericFactory(
+  'awaitFirstOrNull',
+  'kotlinx.coroutines.reactive',
+);
+
+// kotlinx.coroutines.reactor
+export const mono: KtGenericReferenceFactory<1> = ktReference.genericFactory('mono', 'kotlinx.coroutines.reactor');

package/src/src/ast/references/okhttp3.ts

@@ -0,0 +1,5 @@
+import { ktReference, type KtReferenceFactory } from '../nodes/reference.js';
+
+// okhttp3
+export const okHttpClient: KtReferenceFactory = ktReference.factory('Factory', 'okhttp3.Call');
+export const httpUrl: KtReferenceFactory = ktReference.factory('HttpUrl', 'okhttp3');

package/src/src/ast/references/reactor.ts

@@ -0,0 +1,5 @@
+import { type KtGenericReferenceFactory, ktReference } from '../nodes/reference.js';
+
+// reactor.core.publisher
+export const flux: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>('Flux', 'reactor.core.publisher');
+export const mono: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>('Mono', 'reactor.core.publisher');

package/src/src/ast/references/spring-reactive.ts

@@ -0,0 +1,33 @@
+import { type KtGenericReferenceFactory, ktReference, type KtReferenceFactory } from '../nodes/reference.js';
+
+// org.springframework.web.reactive.function
+export const bodyInserters: KtReferenceFactory = ktReference.factory(
+  'BodyInserters',
+  'org.springframework.web.reactive.function',
+);
+
+// org.springframework.web.reactive.function.client
+export const webClient: KtReferenceFactory = ktReference.factory(
+  'WebClient',
+  'org.springframework.web.reactive.function.client',
+);
+export const clientResponse: KtReferenceFactory = ktReference.factory(
+  'ClientResponse',
+  'org.springframework.web.reactive.function.client',
+);
+export const requestHeadersSpec: KtGenericReferenceFactory<1> = ktReference.genericFactory(
+  'RequestHeadersSpec',
+  'org.springframework.web.reactive.function.client.WebClient',
+);
+export const toEntity: KtGenericReferenceFactory<1> = ktReference.genericFactory(
+  'toEntity',
+  'org.springframework.web.reactive.function.client',
+);
+export const awaitBody: KtGenericReferenceFactory<1> = ktReference.genericFactory(
+  'awaitBody',
+  'org.springframework.web.reactive.function.client',
+);
+export const awaitBodilessEntity: KtReferenceFactory = ktReference.factory(
+  'awaitBodilessEntity',
+  'org.springframework.web.reactive.function.client',
+);

package/src/src/ast/references/spring.ts

@@ -0,0 +1,86 @@
+import { type KtGenericReferenceFactory, ktReference, type KtReferenceFactory } from '../nodes/reference.js';
+
+// org.springframework.beans.factory.annotation
+export const autowired: KtReferenceFactory = ktReference.factory(
+  'Autowired',
+  'org.springframework.beans.factory.annotation',
+);
+
+// org.springframework.http
+export const httpStatus: KtReferenceFactory = ktReference.factory('HttpStatus', 'org.springframework.http');
+export const httpMethod: KtReferenceFactory = ktReference.factory('HttpMethod', 'org.springframework.http');
+export const responseEntity: KtGenericReferenceFactory<1> = ktReference.genericFactory<1>(
+  'ResponseEntity',
+  'org.springframework.http',
+);
+export const mediaType: KtReferenceFactory = ktReference.factory('MediaType', 'org.springframework.http');
+
+// org.springframework.http.client
+export const multipartBodyBuilder: KtReferenceFactory = ktReference.factory(
+  'MultipartBodyBuilder',
+  'org.springframework.http.client',
+);
+
+// org.springframework.http.codec.multipart
+export const filePart: KtReferenceFactory = ktReference.factory('FilePart', 'org.springframework.http.codec.multipart');
+
+// org.springframework.stereotype
+export const controller: KtReferenceFactory = ktReference.factory('Controller', 'org.springframework.stereotype');
+
+// org.springframework.util
+export const multiValueMap: KtGenericReferenceFactory<2> = ktReference.genericFactory<2>(
+  'MultiValueMap',
+  'org.springframework.util',
+);
+export const linkedMultiValueMap: KtGenericReferenceFactory<2> = ktReference.genericFactory<2>(
+  'LinkedMultiValueMap',
+  'org.springframework.util',
+);
+
+// org.springframework.validation.annotation
+export const validated: KtReferenceFactory = ktReference.factory(
+  'Validated',
+  'org.springframework.validation.annotation',
+);
+
+// org.springframework.web.bind.annotation
+export const pathVariable: KtReferenceFactory = ktReference.factory(
+  'PathVariable',
+  'org.springframework.web.bind.annotation',
+);
+export const requestBody: KtReferenceFactory = ktReference.factory(
+  'RequestBody',
+  'org.springframework.web.bind.annotation',
+);
+export const requestMapping: KtReferenceFactory = ktReference.factory(
+  'RequestMapping',
+  'org.springframework.web.bind.annotation',
+);
+export const requestMethod: KtReferenceFactory = ktReference.factory(
+  'RequestMethod',
+  'org.springframework.web.bind.annotation',
+);
+export const requestParam: KtReferenceFactory = ktReference.factory(
+  'RequestParam',
+  'org.springframework.web.bind.annotation',
+);
+export const requestPart: KtReferenceFactory = ktReference.factory(
+  'RequestPart',
+  'org.springframework.web.bind.annotation',
+);
+export const requestHeader: KtReferenceFactory = ktReference.factory(
+  'RequestHeader',
+  'org.springframework.web.bind.annotation',
+);
+
+// org.springframework.web.context.request
+export const nativeWebRequest: KtReferenceFactory = ktReference.factory(
+  'NativeWebRequest',
+  'org.springframework.web.context.request',
+);
+
+// org.springframework.web.util
+export const uriComponentsBuilder: KtReferenceFactory = ktReference.factory(
+  'UriComponentsBuilder',
+  'org.springframework.web.util',
+);

package/src/src/ast/references/swagger.ts

@@ -0,0 +1,23 @@
+import { ktReference, type KtReferenceFactory } from '../nodes/reference.js';
+
+// io.swagger.v3.oas.annotations
+export const parameter: KtReferenceFactory = ktReference.factory('Parameter', 'io.swagger.v3.oas.annotations');
+export const operation: KtReferenceFactory = ktReference.factory('Operation', 'io.swagger.v3.oas.annotations');
+
+// io.swagger.v3.oas.annotations.media
+export const schema: KtReferenceFactory = ktReference.factory('Schema', 'io.swagger.v3.oas.annotations.media');
+export const arraySchema: KtReferenceFactory = ktReference.factory(
+  'ArraySchema',
+  'io.swagger.v3.oas.annotations.media',
+);
+export const content: KtReferenceFactory = ktReference.factory('Content', 'io.swagger.v3.oas.annotations.media');
+
+// io.swagger.v3.oas.annotations.responses
+export const apiResponse: KtReferenceFactory = ktReference.factory(
+  'ApiResponse',
+  'io.swagger.v3.oas.annotations.responses',
+);
+export const apiResponses: KtReferenceFactory = ktReference.factory(
+  'ApiResponses',
+  'io.swagger.v3.oas.annotations.responses',
+);

package/src/src/ast/utils/get-kotlin-builder-options.ts

@@ -0,0 +1,19 @@
+import { defaultOpenApiGeneratorConfig, type SourceBuilder } from '@goast/core';
+
+import { defaultKotlinGeneratorConfig, type KotlinGeneratorConfig } from '../../config.js';
+import { KotlinFileBuilder } from '../../file-builder.js';
+
+const ktConfigSymbol = Symbol();
+type _BuilderWithConfig = SourceBuilder & { [ktConfigSymbol]?: KotlinGeneratorConfig };
+
+export function getKotlinBuilderOptions(builder: SourceBuilder): KotlinGeneratorConfig {
+  if (builder instanceof KotlinFileBuilder) {
+    return builder.options;
+  }
+  if (ktConfigSymbol in builder.options) {
+    return builder.options[ktConfigSymbol] as KotlinGeneratorConfig;
+  }
+  const options = { ...defaultOpenApiGeneratorConfig, ...defaultKotlinGeneratorConfig, ...builder.options };
+  (builder as _BuilderWithConfig)[ktConfigSymbol] = options;
+  return options;
+}