import Foundation

/// 文本后处理：保留系统标点，不清洗，只在缺少句末标点时补句号。
enum TextPostProcessor {
    private static let terminalPunctuation: Set<Character> = [
        "。", "！", "？", "…", "；", "：", ".", "!", "?", ";", ":", "\n"
    ]

    private static let closingMarks: Set<Character> = ["”", "\"", "」", "』", "）", ")", "》", "】", "’", "'"]

    static func finalize(_ raw: String, addPeriod: Bool) -> String {
        var text = raw.trimmingCharacters(in: .whitespacesAndNewlines)
        guard !text.isEmpty else { return "" }

        guard addPeriod else { return text }

        var probe = text
        while let last = probe.last, closingMarks.contains(last) {
            probe.removeLast()
        }
        guard let last = probe.last else { return text }
        if !terminalPunctuation.contains(last) {
            text += "。"
        }
        return text
    }

}