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:
parent
6973c70c5a
commit
03fe2e2237
8 changed files with 805 additions and 3 deletions
241
agor-pro/src/export.rs
Normal file
241
agor-pro/src/export.rs
Normal 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"));
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue