agent-orchestrator/src-tauri/src/commands/btmsg.rs

use crate::btmsg;
use crate::groups;

#[tauri::command]
pub fn btmsg_get_agents(group_id: String) -> Result<Vec<btmsg::BtmsgAgent>, String> {
    btmsg::get_agents(&group_id)
}

#[tauri::command]
pub fn btmsg_unread_count(agent_id: String) -> Result<i32, String> {
    btmsg::unread_count(&agent_id)
}

#[tauri::command]
pub fn btmsg_unread_messages(agent_id: String) -> Result<Vec<btmsg::BtmsgMessage>, String> {
    btmsg::unread_messages(&agent_id)
}

#[tauri::command]
pub fn btmsg_history(agent_id: String, other_id: String, limit: i32) -> Result<Vec<btmsg::BtmsgMessage>, String> {
    btmsg::history(&agent_id, &other_id, limit)
}

#[tauri::command]
pub fn btmsg_send(from_agent: String, to_agent: String, content: String) -> Result<String, String> {
    btmsg::send_message(&from_agent, &to_agent, &content)
}

#[tauri::command]
pub fn btmsg_set_status(agent_id: String, status: String) -> Result<(), String> {
    btmsg::set_status(&agent_id, &status)
}

#[tauri::command]
pub fn btmsg_ensure_admin(group_id: String) -> Result<(), String> {
    btmsg::ensure_admin(&group_id)
}

#[tauri::command]
pub fn btmsg_all_feed(group_id: String, limit: i32) -> Result<Vec<btmsg::BtmsgFeedMessage>, String> {
    btmsg::all_feed(&group_id, limit)
}

#[tauri::command]
pub fn btmsg_mark_read(reader_id: String, sender_id: String) -> Result<(), String> {
    btmsg::mark_read_conversation(&reader_id, &sender_id)
}

#[tauri::command]
pub fn btmsg_get_channels(group_id: String) -> Result<Vec<btmsg::BtmsgChannel>, String> {
    btmsg::get_channels(&group_id)
}

#[tauri::command]
pub fn btmsg_channel_messages(channel_id: String, limit: i32) -> Result<Vec<btmsg::BtmsgChannelMessage>, String> {
    btmsg::get_channel_messages(&channel_id, limit)
}

#[tauri::command]
pub fn btmsg_channel_send(channel_id: String, from_agent: String, content: String) -> Result<String, String> {
    btmsg::send_channel_message(&channel_id, &from_agent, &content)
}

#[tauri::command]
pub fn btmsg_create_channel(name: String, group_id: String, created_by: String) -> Result<String, String> {
    btmsg::create_channel(&name, &group_id, &created_by)
}

#[tauri::command]
pub fn btmsg_add_channel_member(channel_id: String, agent_id: String) -> Result<(), String> {
    btmsg::add_channel_member(&channel_id, &agent_id)
}

/// Register all agents from a GroupsFile into the btmsg database.
/// Creates/updates agent records, sets up contact permissions, ensures review channels.
#[tauri::command]
pub fn btmsg_register_agents(config: groups::GroupsFile) -> Result<(), String> {
    btmsg::register_agents_from_groups(&config)
}

// ---- Per-message acknowledgment (seen_messages) ----

#[tauri::command]
pub fn btmsg_unseen_messages(agent_id: String, session_id: String) -> Result<Vec<btmsg::BtmsgMessage>, String> {
    btmsg::unseen_messages(&agent_id, &session_id)
}

#[tauri::command]
pub fn btmsg_mark_seen(session_id: String, message_ids: Vec<String>) -> Result<(), String> {
    btmsg::mark_messages_seen(&session_id, &message_ids)
}

#[tauri::command]
pub fn btmsg_prune_seen() -> Result<u64, String> {
    btmsg::prune_seen_messages(7 * 24 * 3600, 200_000)
}

// ---- Heartbeat monitoring ----

#[tauri::command]
pub fn btmsg_record_heartbeat(agent_id: String) -> Result<(), String> {
    btmsg::record_heartbeat(&agent_id)
}

#[tauri::command]
pub fn btmsg_get_stale_agents(group_id: String, threshold_secs: i64) -> Result<Vec<String>, String> {
    btmsg::get_stale_agents(&group_id, threshold_secs)
}

#[tauri::command]
pub fn btmsg_get_agent_heartbeats(group_id: String) -> Result<Vec<btmsg::AgentHeartbeat>, String> {
    btmsg::get_agent_heartbeats(&group_id)
}

// ---- Dead letter queue ----

#[tauri::command]
pub fn btmsg_get_dead_letters(group_id: String, limit: i32) -> Result<Vec<btmsg::DeadLetter>, String> {
    btmsg::get_dead_letters(&group_id, limit)
}

#[tauri::command]
pub fn btmsg_clear_dead_letters(group_id: String) -> Result<(), String> {
    btmsg::clear_dead_letters(&group_id)
}

#[tauri::command]
pub fn btmsg_clear_all_comms(group_id: String) -> Result<(), String> {
    btmsg::clear_all_communications(&group_id)
}

#[tauri::command]
pub fn btmsg_queue_dead_letter(
    from_agent: String,
    to_agent: String,
    content: String,
    error: String,
) -> Result<(), String> {
    btmsg::queue_dead_letter(&from_agent, &to_agent, &content, &error)
}

// ---- Audit log ----

#[tauri::command]
pub fn audit_log_event(agent_id: String, event_type: String, detail: String) -> Result<(), String> {
    btmsg::log_audit_event(&agent_id, &event_type, &detail)
}

#[tauri::command]
pub fn audit_log_list(group_id: String, limit: i32, offset: i32) -> Result<Vec<btmsg::AuditEntry>, String> {
    btmsg::get_audit_log(&group_id, limit, offset)
}

#[tauri::command]
pub fn audit_log_for_agent(agent_id: String, limit: i32) -> Result<Vec<btmsg::AuditEntry>, String> {
    btmsg::get_audit_log_for_agent(&agent_id, limit)
}