Class ConstDeclaration

packages/ts-builders/src/ConstDeclaration.ts:7–38

import { Writer } from './Writer'

export class ConstDeclaration implements BasicBuilder {
  private docComment?: DocComment
  private value?: ValueBuilder

  constructor(
    readonly name: string,
    readonly type?: TypeBuilder,
  ) {}

  setDocComment(docComment: DocComment): this {
    this.docComment = docComment
    return this
  }

  setValue(value: ValueBuilder): this {
    this.value = value
    return this
  }

  write(writer: Writer): void {
    if (this.docComment) {
      writer.write(this.docComment)
    }
    writer.write('const ').write(this.name)
    if (this.type) {
      writer.write(': ').write(this.type)
    }
    if (this.value) {
      writer.write(' = ').write(this.value)
    }
  }
}

export function constDeclaration(name: string, type?: TypeBuilder) {
  return new ConstDeclaration(name, type)