agent-first-http 0.12.0

Give your AI agent its own private browser — so it reads the real page, past logins and bot walls, without ever touching yours.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

//! Atomic artifact writes into `--out/<request_id>/`.
//!
//! `write_bytes` creates the destination directory lazily and writes to a
//! `.tmp` sibling first, then renames into place so a crashed fetch never
//! leaves a half-written artifact that a later run would mistake for
//! complete.

use std::path::{Path, PathBuf};

use tokio::fs;
use tokio::io::AsyncWriteExt;

use crate::shared::error::{Error, ErrorCode};

pub async fn ensure_dir(dir: &Path) -> Result<(), Error> {
    fs::create_dir_all(dir).await.map_err(|e| {
        Error::new(
            ErrorCode::IoError,
            format!("create_dir_all({}): {e}", dir.display()),
        )
    })
}

pub async fn write_bytes(target: &Path, bytes: &[u8]) -> Result<(), Error> {
    if let Some(parent) = target.parent() {
        ensure_dir(parent).await?;
    }
    let tmp = sibling_tmp(target);
    let mut file = fs::File::create(&tmp).await.map_err(|e| {
        Error::new(
            ErrorCode::IoError,
            format!("create({}): {e}", tmp.display()),
        )
    })?;
    file.write_all(bytes)
        .await
        .map_err(|e| Error::new(ErrorCode::IoError, format!("write({}): {e}", tmp.display())))?;
    file.flush()
        .await
        .map_err(|e| Error::new(ErrorCode::IoError, format!("flush({}): {e}", tmp.display())))?;
    drop(file);
    fs::rename(&tmp, target).await.map_err(|e| {
        Error::new(
            ErrorCode::IoError,
            format!("rename({} -> {}): {e}", tmp.display(), target.display()),
        )
    })
}

fn sibling_tmp(target: &Path) -> PathBuf {
    let mut name = target
        .file_name()
        .map(|s| s.to_os_string())
        .unwrap_or_default();
    name.push(".tmp");
    target.with_file_name(name)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[tokio::test]
    async fn writes_and_renames_atomically() {
        let dir = tempfile::tempdir().unwrap();
        let target = dir.path().join("nested/body.txt");
        write_bytes(&target, b"hello").await.unwrap();
        let read = fs::read(&target).await.unwrap();
        assert_eq!(read, b"hello");
        // No leftover .tmp
        let tmp = sibling_tmp(&target);
        assert!(!tmp.exists());
    }
}