ascii-agents-hook 0.1.1

use std::io::{Read, Write};
use std::os::unix::net::UnixStream;
use std::time::{Duration, SystemTime, UNIX_EPOCH};

use anyhow::{Context, Result};
use serde_json::Value;

const WRITE_TIMEOUT: Duration = Duration::from_millis(200);

fn default_socket_path() -> String {
    if let Ok(p) = std::env::var("ASCII_AGENTS_SOCKET") {
        return p;
    }
    if let Ok(dir) = std::env::var("XDG_RUNTIME_DIR") {
        return format!("{dir}/ascii-agents.sock");
    }
    // Safety: getuid is always safe on Unix.
    let uid = unsafe { libc::getuid() };
    format!("/tmp/ascii-agents-{uid}.sock")
}

fn main() -> Result<()> {
    let socket = default_socket_path();

    let mut buf = String::new();
    std::io::stdin()
        .read_to_string(&mut buf)
        .context("read stdin")?;
    let mut payload: Value = match serde_json::from_str(&buf) {
        Ok(v) => v,
        // If we can't parse, exit 0 silently so CC isn't blocked.
        Err(_) => return Ok(()),
    };

    if let Value::Object(map) = &mut payload {
        let ts = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .map(|d| d.as_millis())
            .unwrap_or(0);
        map.insert("_shim_ts_ms".into(), Value::from(ts as u64));
    }

    // Best-effort send with a hard write timeout so a stuck daemon can never
    // block CC's subprocess wait. If the daemon isn't running or is slow,
    // we drop the event and exit 0.
    if let Ok(s) = UnixStream::connect(&socket) {
        let _ = s.set_write_timeout(Some(WRITE_TIMEOUT));
        let mut s = s;
        let mut line = serde_json::to_vec(&payload).unwrap_or_default();
        line.push(b'\n');
        let _ = s.write_all(&line);
    }
    Ok(())
}