heddle-cli 0.3.1

An AI-native version control system
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

// SPDX-License-Identifier: Apache-2.0
//! Stash worktree helper operations.

use std::fs;

use anyhow::{Result, anyhow};
use objects::{
    fs_ops::remove_path_recursively,
    object::{Blob, ContentHash, Tree, TreeEntry},
    store::ObjectStore,
    worktree::WorktreeStatus,
};
use repo::{Repository, StashEntry};

pub(super) fn build_worktree_tree(repo: &Repository, status: &WorktreeStatus) -> Result<Tree> {
    let mut tree = Tree::new();

    for path in &status.modified {
        let full_path = repo.root().join(path);
        if full_path.exists() {
            let content = fs::read(&full_path)?;
            let blob = Blob::new(content);
            let hash = repo.store().put_blob(&blob)?;
            let entry = TreeEntry::file(path.to_string_lossy().to_string(), hash, false)?;
            tree.insert(entry);
        }
    }

    for path in &status.added {
        let full_path = repo.root().join(path);
        if full_path.exists() {
            let content = fs::read(&full_path)?;
            let blob = Blob::new(content);
            let hash = repo.store().put_blob(&blob)?;
            let entry = TreeEntry::file(path.to_string_lossy().to_string(), hash, false)?;
            tree.insert(entry);
        }
    }

    Ok(tree)
}

pub(super) fn restore_worktree(
    repo: &Repository,
    tree: &Tree,
    status: &WorktreeStatus,
) -> Result<()> {
    for path in &status.modified {
        let full_path = repo.root().join(path);
        if let Some(entry) = tree.get(&path.to_string_lossy()) {
            let blob = repo.require_blob(&entry.hash)?;
            fs::write(&full_path, blob.content())?;
        }
    }

    for path in &status.added {
        let full_path = repo.root().join(path);
        if full_path.is_symlink() {
            fs::remove_file(&full_path)?;
        } else if full_path.is_dir() {
            remove_path_recursively(&full_path)?;
        } else if full_path.exists() {
            fs::remove_file(&full_path)?;
        }
    }

    Ok(())
}

pub(super) fn apply_stash(repo: &Repository, stash: &StashEntry) -> Result<()> {
    let stash_tree_hash = ContentHash::from_hex(&stash.tree_hash)
        .map_err(|e| anyhow!("Invalid stash tree hash: {}", e))?;
    let stash_tree = repo.require_tree(&stash_tree_hash)?;

    for entry in stash_tree.entries() {
        let full_path = repo.root().join(&entry.name);
        let blob = repo.require_blob(&entry.hash)?;
        if let Some(parent) = full_path.parent()
            && !parent.exists()
        {
            fs::create_dir_all(parent)?;
        }
        fs::write(&full_path, blob.content())?;
    }

    Ok(())
}