feat(pro): add analytics, export, and multi-account commercial features

3 new agor-pro modules: analytics.rs (usage dashboard queries),
export.rs (session/project Markdown report generation),
profiles.rs (multi-account switching via accounts.json).
9 Tauri plugin commands. Frontend IPC bridge (pro-bridge.ts).
168 cargo tests, 14 commercial vitest tests.
This commit is contained in:
Hibryda 2026-03-17 01:52:46 +01:00
parent 6973c70c5a
commit 03fe2e2237
8 changed files with 805 additions and 3 deletions

241
agor-pro/src/export.rs Normal file
View file

@ -0,0 +1,241 @@
// SPDX-License-Identifier: LicenseRef-Commercial
// Session Export & Reporting — generate Markdown reports from agent sessions.
use serde::Serialize;
#[derive(Debug, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct SessionReport {
pub project_id: String,
pub session_id: String,
pub markdown: String,
pub cost_usd: f64,
pub turn_count: i64,
pub tool_call_count: i64,
pub duration_minutes: f64,
pub model: String,
}
#[derive(Debug, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct ProjectSummaryReport {
pub project_id: String,
pub markdown: String,
pub total_sessions: i64,
pub total_cost_usd: f64,
pub period_days: i64,
}
#[tauri::command]
pub fn pro_export_session(project_id: String, session_id: String) -> Result<SessionReport, String> {
let conn = super::open_sessions_db()?;
// Get metric for this session
let mut stmt = conn.prepare(
"SELECT start_time, end_time, peak_tokens, turn_count, tool_call_count, cost_usd, model, status, error_message
FROM session_metrics WHERE project_id = ?1 AND session_id = ?2"
).map_err(|e| format!("Query failed: {e}"))?;
let (start, end, tokens, turns, tools, cost, model, status, error): (i64, i64, i64, i64, i64, f64, Option<String>, String, Option<String>) =
stmt.query_row(rusqlite::params![project_id, session_id], |row| {
Ok((row.get(0)?, row.get(1)?, row.get(2)?, row.get(3)?,
row.get(4)?, row.get(5)?, row.get(6)?, row.get(7)?, row.get(8)?))
}).map_err(|e| format!("Session not found: {e}"))?;
let model_name = model.clone().unwrap_or_else(|| "unknown".into());
let duration_min = (end - start) as f64 / 60.0;
let start_str = epoch_to_iso(start);
let end_str = epoch_to_iso(end);
// Get messages for this session
let mut msg_stmt = conn.prepare(
"SELECT message_type, content, created_at
FROM agent_messages WHERE session_id = ?1 ORDER BY created_at ASC"
).map_err(|e| format!("Messages query failed: {e}"))?;
let messages: Vec<(String, String, i64)> = msg_stmt
.query_map(rusqlite::params![session_id], |row| {
Ok((row.get(0)?, row.get(1)?, row.get(2)?))
})
.map_err(|e| format!("Messages query failed: {e}"))?
.collect::<Result<Vec<_>, _>>()
.map_err(|e| format!("Row read failed: {e}"))?;
let mut md = String::new();
md.push_str(&format!("# Session Report: {session_id}\n\n"));
md.push_str(&format!("**Project:** {project_id} \n"));
md.push_str(&format!("**Model:** {model_name} \n"));
md.push_str(&format!("**Status:** {status} \n"));
md.push_str(&format!("**Duration:** {duration_min:.1} min \n"));
md.push_str(&format!("**Start:** {start_str} \n"));
md.push_str(&format!("**End:** {end_str} \n\n"));
md.push_str("## Metrics\n\n");
md.push_str(&format!("| Metric | Value |\n|--------|-------|\n"));
md.push_str(&format!("| Cost | ${cost:.4} |\n"));
md.push_str(&format!("| Peak Tokens | {tokens} |\n"));
md.push_str(&format!("| Turns | {turns} |\n"));
md.push_str(&format!("| Tool Calls | {tools} |\n\n"));
if let Some(err) = &error {
md.push_str(&format!("## Error\n\n```\n{err}\n```\n\n"));
}
if !messages.is_empty() {
md.push_str("## Conversation\n\n");
for (msg_type, content, ts) in &messages {
let time = epoch_to_time(*ts);
let prefix = match msg_type.as_str() {
"user" | "human" => "**User**",
"assistant" => "**Assistant**",
"tool_call" => "**Tool Call**",
"tool_result" => "**Tool Result**",
_ => msg_type.as_str(),
};
// Truncate long content
let display = if content.len() > 500 {
format!("{}... *(truncated, {} chars)*", &content[..500], content.len())
} else {
content.clone()
};
md.push_str(&format!("### [{time}] {prefix}\n\n{display}\n\n"));
}
}
Ok(SessionReport {
project_id,
session_id,
markdown: md,
cost_usd: cost,
turn_count: turns,
tool_call_count: tools,
duration_minutes: duration_min,
model: model_name,
})
}
#[tauri::command]
pub fn pro_export_project_summary(project_id: String, days: Option<i64>) -> Result<ProjectSummaryReport, String> {
let conn = super::open_sessions_db()?;
let period = days.unwrap_or(30);
let cutoff = super::analytics::now_epoch() - (period * 86400);
let mut stmt = conn.prepare(
"SELECT session_id, start_time, end_time, cost_usd, turn_count, tool_call_count, model, status
FROM session_metrics WHERE project_id = ?1 AND end_time >= ?2 ORDER BY end_time DESC"
).map_err(|e| format!("Query failed: {e}"))?;
let sessions: Vec<(String, i64, i64, f64, i64, i64, Option<String>, String)> = stmt
.query_map(rusqlite::params![project_id, cutoff], |row| {
Ok((row.get(0)?, row.get(1)?, row.get(2)?, row.get(3)?,
row.get(4)?, row.get(5)?, row.get(6)?, row.get(7)?))
})
.map_err(|e| format!("Query failed: {e}"))?
.collect::<Result<Vec<_>, _>>()
.map_err(|e| format!("Row read failed: {e}"))?;
let total_cost: f64 = sessions.iter().map(|s| s.3).sum();
let total_sessions = sessions.len() as i64;
let mut md = String::new();
md.push_str(&format!("# Project Summary: {project_id}\n\n"));
md.push_str(&format!("**Period:** last {period} days \n"));
md.push_str(&format!("**Total Sessions:** {total_sessions} \n"));
md.push_str(&format!("**Total Cost:** ${total_cost:.4} \n\n"));
if !sessions.is_empty() {
md.push_str("## Sessions\n\n");
md.push_str("| Date | Session | Model | Duration | Cost | Turns | Tools | Status |\n");
md.push_str("|------|---------|-------|----------|------|-------|-------|--------|\n");
for (sid, start, end, cost, turns, tools, model, status) in &sessions {
let date = epoch_to_date(*start);
let dur = (*end - *start) as f64 / 60.0;
let m = model.as_deref().unwrap_or("?");
let short_sid = if sid.len() > 8 { &sid[..8] } else { sid };
md.push_str(&format!("| {date} | {short_sid}.. | {m} | {dur:.0}m | ${cost:.4} | {turns} | {tools} | {status} |\n"));
}
}
Ok(ProjectSummaryReport {
project_id,
markdown: md,
total_sessions,
total_cost_usd: total_cost,
period_days: period,
})
}
fn epoch_to_iso(epoch: i64) -> String {
let secs = epoch;
let days = secs / 86400;
let time = secs % 86400;
let h = time / 3600;
let m = (time % 3600) / 60;
// Simple date from epoch (2000-01-01 = day 10957 from 1970)
format!("{}T{:02}:{:02}Z", epoch_to_date(secs), h, m)
}
fn epoch_to_time(epoch: i64) -> String {
let time = epoch % 86400;
format!("{:02}:{:02}", time / 3600, (time % 3600) / 60)
}
pub(crate) fn epoch_to_date(epoch: i64) -> String {
// Days since 1970-01-01
let mut days = epoch / 86400;
let mut year = 1970i64;
loop {
let days_in_year = if is_leap(year) { 366 } else { 365 };
if days < days_in_year { break; }
days -= days_in_year;
year += 1;
}
let months = if is_leap(year) {
[31,29,31,30,31,30,31,31,30,31,30,31]
} else {
[31,28,31,30,31,30,31,31,30,31,30,31]
};
let mut month = 0usize;
for (i, &d) in months.iter().enumerate() {
if days < d { month = i; break; }
days -= d;
}
format!("{year:04}-{:02}-{:02}", month + 1, days + 1)
}
fn is_leap(y: i64) -> bool {
(y % 4 == 0 && y % 100 != 0) || y % 400 == 0
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_epoch_to_date() {
assert_eq!(epoch_to_date(0), "1970-01-01");
assert_eq!(epoch_to_date(1710000000), "2024-03-09");
}
#[test]
fn test_epoch_to_iso() {
let iso = epoch_to_iso(1710000000);
assert!(iso.starts_with("2024-"));
assert!(iso.ends_with('Z'));
}
#[test]
fn test_session_report_struct() {
let r = SessionReport {
project_id: "test".into(),
session_id: "abc".into(),
markdown: "# Report".into(),
cost_usd: 0.5,
turn_count: 10,
tool_call_count: 5,
duration_minutes: 15.0,
model: "opus".into(),
};
let json = serde_json::to_string(&r).unwrap();
assert!(json.contains("durationMinutes"));
}
}