import Foundation
import SwiftUI

public final class ChatProvider: ObservableObject {
    let id = UUID()
    @Published public var history: [ChatMessage] = []
    @Published public var isReceivingMessage = false
    public var onMessageSend: (String) -> Void
    public var onStop: () -> Void
    public var onClear: () -> Void
    public var onClose: () -> Void
    public var onSwitchContext: () -> Void
    public var onDeleteMessage: (String) -> Void
    public var onResendMessage: (String) -> Void

    public init(
        history: [ChatMessage] = [],
        isReceivingMessage: Bool = false,
        onMessageSend: @escaping (String) -> Void = { _ in },
        onStop: @escaping () -> Void = {},
        onClear: @escaping () -> Void = {},
        onClose: @escaping () -> Void = {},
        onSwitchContext: @escaping () -> Void = {},
        onDeleteMessage: @escaping (String) -> Void = { _ in },
        onResendMessage: @escaping (String) -> Void = { _ in }
    ) {
        self.history = history
        self.isReceivingMessage = isReceivingMessage
        self.onMessageSend = onMessageSend
        self.onStop = onStop
        self.onClear = onClear
        self.onClose = onClose
        self.onSwitchContext = onSwitchContext
        self.onDeleteMessage = onDeleteMessage
        self.onResendMessage = onResendMessage
    }

    public func send(_ message: String) { onMessageSend(message) }
    public func stop() { onStop() }
    public func clear() { onClear() }
    public func close() { onClose() }
    public func switchContext() { onSwitchContext() }
    public func deleteMessage(id: String) { onDeleteMessage(id) }
    public func resendMessage(id: String) { onResendMessage(id) }
}

public struct ChatMessage: Equatable {
    public var id: String
    public var isUser: Bool
    public var text: String

    public init(id: String, isUser: Bool, text: String) {
        self.id = id
        self.isUser = isUser
        self.text = text
    }
}