aft/db/
bash_tasks.rs

1use rusqlite::{params, Connection, OptionalExtension, Row};
2
3#[derive(Debug, Clone)]
4pub struct BashTaskRow {
5    pub harness: String,
6    pub session_id: String,
7    pub task_id: String,
8    pub project_key: String,
9    pub command: String,
10    pub cwd: String,
11    pub status: String,
12    pub exit_code: Option<i32>,
13    pub pid: Option<i64>,
14    pub pgid: Option<i64>,
15    pub started_at: i64,
16    pub completed_at: Option<i64>,
17    pub stdout_path: Option<String>,
18    pub stderr_path: Option<String>,
19    pub compressed: bool,
20    pub timeout_ms: Option<i64>,
21    pub completion_delivered: bool,
22    pub output_bytes: Option<i64>,
23    pub metadata: String,
24}
25
26pub fn upsert_bash_task(conn: &Connection, row: &BashTaskRow) -> rusqlite::Result<()> {
27    conn.execute(
28        "INSERT INTO bash_tasks (
29            harness, session_id, task_id, project_key, command, cwd, status,
30            exit_code, pid, pgid, started_at, completed_at, stdout_path, stderr_path,
31            compressed, timeout_ms, completion_delivered, output_bytes, metadata
32         ) VALUES (
33            ?1, ?2, ?3, ?4, ?5, ?6, ?7,
34            ?8, ?9, ?10, ?11, ?12, ?13, ?14,
35            ?15, ?16, ?17, ?18, ?19
36         )
37         ON CONFLICT(harness, session_id, task_id) DO UPDATE SET
38            project_key = excluded.project_key,
39            command = excluded.command,
40            cwd = excluded.cwd,
41            status = excluded.status,
42            exit_code = excluded.exit_code,
43            pid = excluded.pid,
44            pgid = excluded.pgid,
45            started_at = excluded.started_at,
46            completed_at = excluded.completed_at,
47            stdout_path = excluded.stdout_path,
48            stderr_path = excluded.stderr_path,
49            compressed = excluded.compressed,
50            timeout_ms = excluded.timeout_ms,
51            completion_delivered = excluded.completion_delivered,
52            output_bytes = excluded.output_bytes,
53            metadata = excluded.metadata",
54        params![
55            row.harness,
56            row.session_id,
57            row.task_id,
58            row.project_key,
59            row.command,
60            row.cwd,
61            row.status,
62            row.exit_code,
63            row.pid,
64            row.pgid,
65            row.started_at,
66            row.completed_at,
67            row.stdout_path,
68            row.stderr_path,
69            row.compressed,
70            row.timeout_ms,
71            row.completion_delivered,
72            row.output_bytes,
73            row.metadata,
74        ],
75    )?;
76    Ok(())
77}
78
79pub fn get_bash_task(
80    conn: &Connection,
81    harness: &str,
82    session_id: &str,
83    task_id: &str,
84) -> rusqlite::Result<Option<BashTaskRow>> {
85    conn.query_row(
86        "SELECT harness, session_id, task_id, project_key, command, cwd, status,
87                exit_code, pid, pgid, started_at, completed_at, stdout_path, stderr_path,
88                compressed, timeout_ms, completion_delivered, output_bytes, metadata
89         FROM bash_tasks
90         WHERE harness = ?1 AND session_id = ?2 AND task_id = ?3",
91        params![harness, session_id, task_id],
92        map_bash_task_row,
93    )
94    .optional()
95}
96
97pub fn list_bash_tasks_for_session(
98    conn: &Connection,
99    harness: &str,
100    session_id: &str,
101) -> rusqlite::Result<Vec<BashTaskRow>> {
102    let mut stmt = conn.prepare(
103        "SELECT harness, session_id, task_id, project_key, command, cwd, status,
104                exit_code, pid, pgid, started_at, completed_at, stdout_path, stderr_path,
105                compressed, timeout_ms, completion_delivered, output_bytes, metadata
106         FROM bash_tasks
107         WHERE harness = ?1 AND session_id = ?2
108         ORDER BY started_at ASC, task_id ASC",
109    )?;
110
111    let rows = stmt
112        .query_map(params![harness, session_id], map_bash_task_row)?
113        .collect();
114    rows
115}
116
117pub fn find_bash_task_for_project(
118    conn: &Connection,
119    harness: &str,
120    project_key: &str,
121    task_id: &str,
122) -> rusqlite::Result<Option<BashTaskRow>> {
123    conn.query_row(
124        "SELECT harness, session_id, task_id, project_key, command, cwd, status,
125                exit_code, pid, pgid, started_at, completed_at, stdout_path, stderr_path,
126                compressed, timeout_ms, completion_delivered, output_bytes, metadata
127         FROM bash_tasks
128         WHERE harness = ?1 AND project_key = ?2 AND task_id = ?3
129         ORDER BY started_at DESC
130         LIMIT 1",
131        params![harness, project_key, task_id],
132        map_bash_task_row,
133    )
134    .optional()
135}
136
137fn map_bash_task_row(row: &Row<'_>) -> rusqlite::Result<BashTaskRow> {
138    Ok(BashTaskRow {
139        harness: row.get(0)?,
140        session_id: row.get(1)?,
141        task_id: row.get(2)?,
142        project_key: row.get(3)?,
143        command: row.get(4)?,
144        cwd: row.get(5)?,
145        status: row.get(6)?,
146        exit_code: row.get(7)?,
147        pid: row.get(8)?,
148        pgid: row.get(9)?,
149        started_at: row.get(10)?,
150        completed_at: row.get(11)?,
151        stdout_path: row.get(12)?,
152        stderr_path: row.get(13)?,
153        compressed: row.get::<_, i64>(14)? != 0,
154        timeout_ms: row.get(15)?,
155        completion_delivered: row.get::<_, i64>(16)? != 0,
156        output_bytes: row.get(17)?,
157        metadata: row.get::<_, Option<String>>(18)?.unwrap_or_default(),
158    })
159}
aft/db/bash_tasks.rs

aft/db/
bash_tasks.rs
