//===----------------------------------------------------------------------===//
// Copyright © 2026 Apple Inc. and the Containerization project authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//   https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//===----------------------------------------------------------------------===//

import ContainerizationOS
import Foundation

enum EscapeSequence {
    static let hideCursor = "\u{001B}[?25l"
    static let showCursor = "\u{001B}[?25h"
    static let moveUp = "\u{001B}[1A"
    static let clearToEndOfLine = "\u{001B}[K"
}

extension ProgressBar {
    var termWidth: Int {
        guard
            let terminalHandle = term,
            let terminal = try? Terminal(descriptor: terminalHandle.fileDescriptor)
        else {
            return 0
        }

        return (try? Int(terminal.size.width)) ?? 0
    }

    /// Clears the progress bar and resets the cursor.
    public func clearAndResetCursor() {
        state.withLock { s in
            clear(state: &s)
            resetCursor()
        }
    }

    /// Clears the progress bar.
    public func clear() {
        state.withLock { s in
            clear(state: &s)
        }
    }

    /// Clears the progress bar (caller must hold state lock).
    func clear(state: inout State) {
        displayText("", state: &state)
    }

    /// Resets the cursor.
    public func resetCursor() {
        display(EscapeSequence.showCursor)
    }

    func display(_ text: String) {
        guard let term else {
            return
        }
        termQueue.sync {
            try? term.write(contentsOf: Data(text.utf8))
            try? term.synchronize()
        }
    }

    func displayText(_ text: String, terminating: String = "\r") {
        state.withLock { s in
            displayText(text, state: &s, terminating: terminating)
        }
    }

    func displayText(_ text: String, state: inout State, terminating: String = "\r") {
        state.output = text

        // Clears previously printed lines.
        var lines = ""
        if terminating.hasSuffix("\r") && termWidth > 0 {
            let lineCount = (text.count - 1) / termWidth
            for _ in 0..<lineCount {
                lines += EscapeSequence.moveUp
            }
        }

        let output = "\(text)\(EscapeSequence.clearToEndOfLine)\(terminating)\(lines)"
        display(output)
    }
}
