about summary refs log tree commit diff
path: root/tvix/store/src/fuse/inodes.rs
blob: a52ba7989e80f7570792c2b3040488039c0d6d76 (plain) (blame)
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
//! This module contains all the data structures used to track information
//! about inodes, which present tvix-store nodes in a filesystem.
use crate::{proto, B3Digest};

#[derive(Clone, Debug)]
pub enum InodeData {
    Regular(B3Digest, u32, bool),  // digest, size, executable
    Symlink(Vec<u8>),              // target
    Directory(DirectoryInodeData), // either [DirectoryInodeData:Sparse] or [DirectoryInodeData:Populated]
}

/// This encodes the two different states of [InodeData::Directory].
/// Either the data still is sparse (we only saw a [proto::DirectoryNode], but
/// didn't fetch the [proto::Directory] struct yet,
/// or we processed a lookup and did fetch the data.
#[derive(Clone, Debug)]
pub enum DirectoryInodeData {
    Sparse(B3Digest, u32),                              // digest, size
    Populated(B3Digest, Vec<(u64, proto::node::Node)>), // [(child_inode, node)]
}

impl From<&proto::node::Node> for InodeData {
    fn from(value: &proto::node::Node) -> Self {
        match value {
            proto::node::Node::Directory(directory_node) => directory_node.into(),
            proto::node::Node::File(file_node) => file_node.into(),
            proto::node::Node::Symlink(symlink_node) => symlink_node.into(),
        }
    }
}

impl From<&proto::SymlinkNode> for InodeData {
    fn from(value: &proto::SymlinkNode) -> Self {
        InodeData::Symlink(value.target.clone())
    }
}

impl From<&proto::FileNode> for InodeData {
    fn from(value: &proto::FileNode) -> Self {
        InodeData::Regular(
            value.digest.clone().try_into().unwrap(),
            value.size,
            value.executable,
        )
    }
}

/// Converts a DirectoryNode to a sparsely populated InodeData::Directory.
impl From<&proto::DirectoryNode> for InodeData {
    fn from(value: &proto::DirectoryNode) -> Self {
        InodeData::Directory(DirectoryInodeData::Sparse(
            value.digest.clone().try_into().unwrap(),
            value.size,
        ))
    }
}

/// converts a proto::Directory to a InodeData::Directory(DirectoryInodeData::Populated(..)).
/// The inodes for each child are 0, because it's up to the InodeTracker to allocate them.
impl From<proto::Directory> for InodeData {
    fn from(value: proto::Directory) -> Self {
        let digest = value.digest();

        let children: Vec<(u64, proto::node::Node)> = value.nodes().map(|node| (0, node)).collect();

        InodeData::Directory(DirectoryInodeData::Populated(digest, children))
    }
}

impl From<&InodeData> for fuser::FileType {
    fn from(val: &InodeData) -> Self {
        match val {
            InodeData::Regular(..) => fuser::FileType::RegularFile,
            InodeData::Symlink(_) => fuser::FileType::Symlink,
            InodeData::Directory(..) => fuser::FileType::Directory,
        }
    }
}