monochromatic-git-policy-cli 0.5.9

//! What:
//!  The files ref storage read in process: one ref by exact name, the symbolic chain behind it, and the list of every ref.
//! Why:
//!  A loose file wins over `packed-refs`; a symbolic file names another ref; both are plain text.
//!  Every spelling the reader is not sure of (a link, a name outside a plain alphabet, a damaged value, a linked
//!  worktree's split of per-worktree and shared refs) is `Unsure`, which sends the caller to Git.
//!
//! In TS you'd write (pseudocode):
//! ```ts
//! const refs = new Refs(commonDir, hashHexLength); refs.lookup('refs/heads/main'); refs.list();
//! ```

#![cfg(unix)]

use super::private_storage::read_regular_file;
use std::path::{Path, PathBuf};

/// The longest loose ref file or `packed-refs` line this reader looks at.
const LOOSE_LIMIT: u64 = 4096;

/// The largest `packed-refs` file this reader looks at.
const PACKED_LIMIT: u64 = 256 << 20;

/// How many symbolic ref hops are followed.
const SYMBOLIC_DEPTH_LIMIT: usize = 5;

/// What a ref holds.
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum RefValue {
    /// An object name as lower-case hexadecimal.
    Direct(String),
    /// The name of another ref.
    Symbolic(String),
}

/// The answer to a lookup.
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum Lookup {
    /// The ref exists with this value.
    Found(RefValue),
    /// Neither a loose file nor a packed line names it.
    Missing,
    /// Something here is outside the plain layouts.
    Unsure,
}

/// The shared refs of one repository whose worktree owns its Git directory.
pub struct Refs {
    /// The common directory, which holds `refs/` and `packed-refs`.
    directory: PathBuf,
    /// The number of hexadecimal digits of an object name.
    hex_length: usize,
}

/// Whether `name` is a ref name of a plain alphabet: `refs/` and components of ASCII letters, digits and `._-`
/// that Git always accepts.
pub fn is_plain_ref_name(name: &str) -> bool {
    let Some(rest) = name.strip_prefix("refs/") else {
        return false;
    };
    if rest.is_empty() || rest.ends_with('.') || rest.contains("..") {
        return false;
    }
    return rest.split('/').all(|component| {
        return !component.is_empty()
            && !component.starts_with('.')
            && !component.ends_with(".lock")
            && component.bytes().all(|byte| { return byte.is_ascii_alphanumeric() || matches!(byte, b'.' | b'_' | b'-'); });
    });
}

impl Refs {
    /// Readers for the refs under `directory`, with object names of `hex_length` digits.
    pub fn new(directory: &Path, hex_length: usize) -> Refs {
        return Refs { directory: directory.to_path_buf(), hex_length };
    }

    /// Whether `text` is a plain, non-null, lower-case object name.
    fn is_name(&self, text: &str) -> bool {
        return text.len() == self.hex_length
            && text.bytes().all(|byte| { return matches!(byte, b'0'..=b'9' | b'a'..=b'f'); })
            && !text.bytes().all(|byte| { return byte == b'0'; });
    }

    /// The value of a loose file's text.
    fn parse_loose(&self, text: &str) -> Option<RefValue> {
        let line: &str = text.strip_suffix('\n')?;
        if let Some(target) = line.strip_prefix("ref: ") {
            return is_plain_ref_name(target).then(|| { return RefValue::Symbolic(target.to_string()); });
        }
        return self.is_name(line).then(|| { return RefValue::Direct(line.to_string()); });
    }

    /// The packed lines as `(name, object name)`, or `None` when the file is not a plain `packed-refs`.
    fn packed(&self) -> Option<Vec<(String, String)>> {
        let bytes: Vec<u8> = match read_regular_file(self.directory.join("packed-refs").as_path(), PACKED_LIMIT) {
            Ok(bytes) => bytes,
            Err(super::private_storage::ReadRefusal::Missing) => return Some(Vec::new()),
            Err(_) => return None,
        };
        let text: &str = std::str::from_utf8(bytes.as_slice()).ok()?;
        let mut packed: Vec<(String, String)> = Vec::new();
        // A peeled line (`^<name>`) is valid only directly after a ref line.
        let mut after_ref: bool = false;
        for (position, line) in text.lines().enumerate() {
            if line.starts_with('#') {
                if position != 0 {
                    return None;
                }
                continue;
            }
            if let Some(peeled) = line.strip_prefix('^') {
                if !after_ref || !self.is_name(peeled) {
                    return None;
                }
                after_ref = false;
                continue;
            }
            let (object, name) = line.split_once(' ')?;
            if !self.is_name(object) || !is_plain_ref_name(name) {
                return None;
            }
            packed.push((name.to_string(), object.to_string()));
            after_ref = true;
        }
        return Some(packed);
    }

    /// What the `HEAD` file of `git_dir` holds, when it is a plain symbolic or direct spelling.
    pub fn lookup_head(&self, git_dir: &Path) -> Option<RefValue> {
        let bytes: Vec<u8> = read_regular_file(git_dir.join("HEAD").as_path(), LOOSE_LIMIT).ok()?;
        return self.parse_loose(std::str::from_utf8(bytes.as_slice()).ok()?);
    }

    /// What ref `name` holds, loose first.
    pub fn lookup(&self, name: &str) -> Lookup {
        if !is_plain_ref_name(name) {
            return Lookup::Unsure;
        }
        let path: PathBuf = self.directory.join(name);
        match std::fs::symlink_metadata(&path) {
            Ok(metadata) if metadata.is_file() => {
                let Ok(bytes) = read_regular_file(path.as_path(), LOOSE_LIMIT) else {
                    return Lookup::Unsure;
                };
                let Some(value) = std::str::from_utf8(bytes.as_slice()).ok().and_then(|text| { return self.parse_loose(text); }) else {
                    return Lookup::Unsure;
                };
                return Lookup::Found(value);
            }
            Ok(metadata) if metadata.is_dir() => {}
            Ok(_) => return Lookup::Unsure,
            // A file where a directory is expected (`refs/heads/a` as a file, asked for `refs/heads/a/b`) is absence.
            Err(error) if matches!(error.kind(), std::io::ErrorKind::NotFound | std::io::ErrorKind::NotADirectory) => {}
            Err(_) => return Lookup::Unsure,
        }
        let Some(packed) = self.packed() else {
            return Lookup::Unsure;
        };
        return match packed.into_iter().find(|(candidate, _)| { return candidate == name; }) {
            Some((_, object)) => Lookup::Found(RefValue::Direct(object)),
            None => Lookup::Missing,
        };
    }

    /// The object name `name` resolves to through symbolic refs; `None` when it is missing or unsure.
    pub fn resolve(&self, name: &str) -> Option<String> {
        let mut current: String = name.to_string();
        for _ in 0..SYMBOLIC_DEPTH_LIMIT {
            match self.lookup(current.as_str()) {
                Lookup::Found(RefValue::Direct(object)) => return Some(object),
                Lookup::Found(RefValue::Symbolic(target)) => current = target,
                Lookup::Missing | Lookup::Unsure => return None,
            }
        }
        return None;
    }

    /// Every ref under `refs/`, sorted by name, loose over packed; `None` when anything is not plain.
    pub fn list(&self) -> Option<Vec<(String, RefValue)>> {
        let mut found: std::collections::BTreeMap<String, RefValue> = std::collections::BTreeMap::new();
        for (name, object) in self.packed()? {
            found.insert(name, RefValue::Direct(object));
        }
        let mut pending: Vec<String> = vec![String::from("refs")];
        while let Some(directory) = pending.pop() {
            let listing = match std::fs::read_dir(self.directory.join(&directory)) {
                Ok(listing) => listing,
                Err(error) if error.kind() == std::io::ErrorKind::NotFound => continue,
                Err(_) => return None,
            };
            for candidate in listing {
                let entry = candidate.ok()?;
                let name: String = format!("{directory}/{}", entry.file_name().to_str()?);
                let kind = entry.file_type().ok()?;
                if kind.is_dir() {
                    pending.push(name);
                } else if kind.is_file() && !name.ends_with(".lock") {
                    match self.lookup(name.as_str()) {
                        Lookup::Found(value) => {
                            found.insert(name, value);
                        }
                        Lookup::Missing | Lookup::Unsure => return None,
                    }
                } else {
                    return None;
                }
            }
        }
        return Some(found.into_iter().collect());
    }
}

#[cfg(test)]
#[path = "repo_read_refs_tests.rs"]
mod tests;