pctx 0.1.3

Generate LLM-ready context from your codebase
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

//! File output writing.

use std::fs::OpenOptions;
use std::io::Write;
use std::path::Path;

use crate::error::PctxError;

/// Write content to a file (atomic check-and-create to avoid TOCTOU race)
pub fn write(path: &Path, content: &str, force: bool) -> Result<(), PctxError> {
    let mut options = OpenOptions::new();
    options.write(true);

    if force {
        options.create(true).truncate(true);
    } else {
        options.create_new(true);
    }

    let mut file = options.open(path).map_err(|e| {
        if e.kind() == std::io::ErrorKind::AlreadyExists {
            PctxError::OutputExists(path.to_path_buf())
        } else if e.kind() == std::io::ErrorKind::PermissionDenied {
            PctxError::PermissionDenied(path.to_path_buf())
        } else {
            PctxError::Io(e)
        }
    })?;

    file.write_all(content.as_bytes())?;
    file.flush()?;
    Ok(())
}