ChatProvider.swift

import Foundation
import Preferences
import SwiftUI

public final class ChatProvider: ObservableObject {
    public typealias MessageID = String
    let id = UUID()
    @Published public var history: [ChatMessage] = []
    @Published public var isReceivingMessage = false
    public var systemPrompt = ""
    public var extraSystemPrompt = ""
    public var onMessageSend: (String) -> Void
    public var onStop: () -> Void
    public var onClear: () -> Void
    public var onClose: () -> Void
    public var onSwitchContext: () -> Void
    public var onDeleteMessage: (MessageID) -> Void
    public var onResendMessage: (MessageID) -> Void
    public var onResetPrompt: () -> Void
    public var onRunCustomCommand: (CustomCommand) -> Void = { _ in }
    public var onSetAsExtraPrompt: (MessageID) -> 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 (MessageID) -> Void = { _ in },
        onResendMessage: @escaping (MessageID) -> Void = { _ in },
        onResetPrompt: @escaping () -> Void = {},
        onRunCustomCommand: @escaping (CustomCommand) -> Void = { _ in },
        onSetAsExtraPrompt: @escaping (MessageID) -> 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
        self.onResetPrompt = onResetPrompt
        self.onRunCustomCommand = onRunCustomCommand
        self.onSetAsExtraPrompt = onSetAsExtraPrompt
    }

    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: MessageID) { onDeleteMessage(id) }
    public func resendMessage(id: MessageID) { onResendMessage(id) }
    public func resetPrompt() { onResetPrompt() }
    public func triggerCustomCommand(_ command: CustomCommand) {
        onRunCustomCommand(command)
    }
    public func setAsExtraPrompt(id: MessageID) { onSetAsExtraPrompt(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
    }
}