export class PrintContext {
  private output: string = ""
  private indentLevel: number = 0
  private currentColumn: number = 0
  private preserveStack: string[] = []

  /**
   * Write text to the output
   */
  write(text: string): void {
    this.output += text
    this.currentColumn += text.length
  }

  /**
   * Write text and update column tracking for newlines
   */
  writeWithColumnTracking(text: string): void {
    this.output += text

    const lines = text.split('\n')

    if (lines.length > 1) {
      this.currentColumn = lines[lines.length - 1].length
    } else {
      this.currentColumn += text.length
    }
  }

  /**
   * Increase indentation level
   */
  indent(): void {
    this.indentLevel++
  }

  /**
   * Decrease indentation level
   */
  dedent(): void {
    if (this.indentLevel > 0) {
      this.indentLevel--
    }
  }

  /**
   * Enter a tag that may preserve whitespace
   */
  enterTag(tagName: string): void {
    this.preserveStack.push(tagName.toLowerCase())
  }

  /**
   * Exit the current tag
   */
  exitTag(): void {
    this.preserveStack.pop()
  }

  /**
   * Check if the output written so far ends with a whitespace character
   */
  endsWithWhitespace(): boolean {
    return /\s$/.test(this.output)
  }

  /**
   * Check if we're at the start of a line
   */
  isAtStartOfLine(): boolean {
    return this.currentColumn === 0
  }

  /**
   * Get current indentation level
   */
  getCurrentIndentLevel(): number {
    return this.indentLevel
  }

  /**
   * Get current column position
   */
  getCurrentColumn(): number {
    return this.currentColumn
  }

  /**
   * Get the current tag stack (for debugging)
   */
  getTagStack(): string[] {
    return [...this.preserveStack]
  }

  /**
   * Get the complete output string
   */
  getOutput(): string {
    return this.output
  }

  /**
   * Reset the context for reuse
   */
  reset(): void {
    this.output = ""
    this.indentLevel = 0
    this.currentColumn = 0
    this.preserveStack = []
  }
}