zesdex/src/app/state/runtime.rs

use std::path::PathBuf;
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Copy, Serialize, Deserialize, Default)]
pub struct UsageStats {
    pub tokens_in: u64,
    pub tokens_out: u64,
    pub api_calls: u64,
    pub review_tokens: u64,
    pub total_ms: u64,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SessionRuntime {
    pub messages: Vec<crate::dto::chat::message::ChatMessage>,
    pub tool_call_results: Vec<ToolCallResult>,
    pub pending_tool_queue: Vec<PendingTool>,
    pub bash_jobs: Vec<BashJobRef>,
    pub subagent_queue: usize,
    pub edit_count: u32,
    pub consecutive_empty_reviews: u32,
    pub session_start: i64,
    pub lesson_count: u32,
    pub lessons_user: u32,
    pub lessons_feedback: u32,
    pub lessons_project: u32,
    pub lessons_reference: u32,
    pub lessons_active: u32,
    pub lessons_stale: u32,
    pub lessons_contradicted: u32,
    pub lessons_human: u32,
    pub lessons_verified: u32,
    pub lessons_unverified: u32,
    pub review_count: u32,
    pub session_dir: PathBuf,
    pub usage: UsageStats,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ToolCallResult {
    pub tool_call_id: String,
    pub tool_name: String,
    pub output: String,
    pub is_error: bool,
    pub duration_ms: u64,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PendingTool {
    pub tool_name: String,
    pub args: serde_json::Value,
    pub execution_model: crate::app::state::types::ExecutionModel,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct BashJobRef {
    pub id: String,
    pub command: String,
    pub started_at: i64,
    pub running: bool,
}

#[derive(Debug, Clone)]
pub enum TurnEvent {
    AssistantMessage(crate::dto::chat::message::ChatMessage),
    ToolResult {
        tool_call_id: String,
        tool_name: String,
        output: String,
        is_error: bool,
    },
    SystemNote {
        kind: String,
        message: String,
    },
    Error(String),
    Done,
}

impl SessionRuntime {
    pub fn new(session_dir: PathBuf) -> Self {
        SessionRuntime {
            messages: Vec::new(),
            tool_call_results: Vec::new(),
            pending_tool_queue: Vec::new(),
            bash_jobs: Vec::new(),
            subagent_queue: 0,
            edit_count: 0,
            consecutive_empty_reviews: 0,
            session_start: chrono::Utc::now().timestamp_millis(),
            lesson_count: 0,
            lessons_user: 0,
            lessons_feedback: 0,
            lessons_project: 0,
            lessons_reference: 0,
            lessons_active: 0,
            lessons_stale: 0,
            lessons_contradicted: 0,
            lessons_human: 0,
            lessons_verified: 0,
            lessons_unverified: 0,
            review_count: 0,
            session_dir,
            usage: UsageStats::default(),
        }
    }

    pub fn push_message(&mut self, msg: crate::dto::chat::message::ChatMessage) {
        self.messages.push(msg);
    }

    pub fn record_api_call(&mut self, tokens_in: u64, tokens_out: u64, duration_ms: u64) {
        self.usage.tokens_in += tokens_in;
        self.usage.tokens_out += tokens_out;
        self.usage.api_calls += 1;
        self.usage.total_ms += duration_ms;
    }

    pub fn record_review_tokens(&mut self, tokens: u64) {
        self.usage.review_tokens += tokens;
    }
}