phalus 0.6.0

Private Headless Automated License Uncoupling System — AI-powered clean room software reimplementation
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

use std::path::PathBuf;

use thiserror::Error;

use crate::CspSpec;

#[derive(Debug, Error)]
pub enum CacheError {
    #[error("I/O error: {0}")]
    Io(#[from] std::io::Error),
    #[error("JSON error: {0}")]
    Json(#[from] serde_json::Error),
}

pub struct CspCache {
    dir: PathBuf,
}

impl CspCache {
    pub fn new(dir: PathBuf) -> Self {
        Self { dir }
    }

    pub fn default_cache() -> Self {
        let dir = dirs::home_dir()
            .unwrap_or_else(|| PathBuf::from("."))
            .join(".phalus")
            .join("cache")
            .join("csp");
        Self { dir }
    }

    fn cache_path(&self, name: &str, version: &str, content_hash: &str) -> PathBuf {
        // Sanitize components to prevent path traversal. Replace any character
        // that could escape the cache directory (path separators, `..`) with `_`.
        let safe = |s: &str| s.replace(['/', '\\'], "_").replace("..", "_");
        self.dir.join(format!(
            "{}-{}-{}.json",
            safe(name),
            safe(version),
            safe(content_hash)
        ))
    }

    pub fn get(&self, name: &str, version: &str, content_hash: &str) -> Option<CspSpec> {
        let path = self.cache_path(name, version, content_hash);
        let data = std::fs::read_to_string(path).ok()?;
        serde_json::from_str(&data).ok()
    }

    pub fn put(
        &self,
        name: &str,
        version: &str,
        content_hash: &str,
        csp: &CspSpec,
    ) -> Result<(), CacheError> {
        std::fs::create_dir_all(&self.dir)?;
        let path = self.cache_path(name, version, content_hash);
        let json = serde_json::to_string_pretty(csp)?;
        // Write atomically: write to a temp file then rename into place so
        // concurrent readers never see a partial write.
        let tmp_path = path.with_extension("tmp");
        std::fs::write(&tmp_path, json)?;
        std::fs::rename(&tmp_path, &path)?;
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{CspDocument, CspSpec};
    use chrono::Utc;
    use tempfile::TempDir;

    fn sample_csp() -> CspSpec {
        CspSpec {
            package_name: "test-pkg".into(),
            package_version: "1.0.0".into(),
            documents: vec![CspDocument {
                filename: "01-overview.md".into(),
                content: "test".into(),
                content_hash: "abc".into(),
            }],
            generated_at: Utc::now(),
        }
    }

    #[test]
    fn test_cache_miss() {
        let dir = TempDir::new().unwrap();
        let cache = CspCache::new(dir.path().to_path_buf());
        assert!(cache.get("test-pkg", "1.0.0", "hash123").is_none());
    }

    #[test]
    fn test_cache_hit() {
        let dir = TempDir::new().unwrap();
        let cache = CspCache::new(dir.path().to_path_buf());
        let csp = sample_csp();
        cache.put("test-pkg", "1.0.0", "hash123", &csp).unwrap();
        let cached = cache.get("test-pkg", "1.0.0", "hash123");
        assert!(cached.is_some());
        assert_eq!(cached.unwrap().package_name, "test-pkg");
    }

    #[test]
    fn test_cache_different_hash_misses() {
        let dir = TempDir::new().unwrap();
        let cache = CspCache::new(dir.path().to_path_buf());
        let csp = sample_csp();
        cache.put("test-pkg", "1.0.0", "hash123", &csp).unwrap();
        assert!(cache.get("test-pkg", "1.0.0", "different-hash").is_none());
    }
}