composiohq__composio/ts/packages/ts-builders/src/Parameter.ts

import type { BasicBuilder } from './BasicBuilder';
import { TypeContext } from './OperatorPrecedence';
import { TypeBuilder } from './TypeBuilder';
import { Writer } from './Writer';

export class Parameter implements BasicBuilder {
  private isOptional = false;
  constructor(
    private name: string,
    private type: TypeBuilder
  ) {}

  optional(): this {
    this.isOptional = true;
    return this;
  }

  write(writer: Writer): void {
    writer.write(this.name);
    if (this.isOptional) {
      writer.write('?');
    }
    writer.write(': ');
    this.type.writeInContext(writer, TypeContext.FunctionParameter);
  }
}

export function parameter(name: string, type: TypeBuilder) {
  return new Parameter(name, type);
}