buildkit-llb 0.2.0

use std::path::{Path, PathBuf};

use crate::utils::{OperationOutput, OutputIdx};

/// Operand of *command execution operation* that specifies how are input sources mounted.
#[derive(Debug, Clone)]
pub enum Mount<'a, P: AsRef<Path>> {
    /// Read-only output of another operation.
    ReadOnlyLayer(OperationOutput<'a>, P),

    /// Read-only output of another operation with a selector.
    ReadOnlySelector(OperationOutput<'a>, P, P),

    /// Empty layer that produces an output.
    Scratch(OutputIdx, P),

    /// Writable output of another operation.
    Layer(OutputIdx, OperationOutput<'a>, P),

    /// Writable persistent cache.
    SharedCache(P),

    /// Optional SSH agent socket at the specified path.
    OptionalSshAgent(P),
}

impl<'a, P: AsRef<Path>> Mount<'a, P> {
    /// Transform the mount into owned variant (basically, with `PathBuf` as the path).
    pub fn into_owned(self) -> Mount<'a, PathBuf> {
        use Mount::*;

        match self {
            ReadOnlySelector(op, path, selector) => {
                ReadOnlySelector(op, path.as_ref().into(), selector.as_ref().into())
            }

            ReadOnlyLayer(op, path) => ReadOnlyLayer(op, path.as_ref().into()),
            Scratch(output, path) => Scratch(output, path.as_ref().into()),
            Layer(output, input, path) => Layer(output, input, path.as_ref().into()),
            SharedCache(path) => SharedCache(path.as_ref().into()),
            OptionalSshAgent(path) => OptionalSshAgent(path.as_ref().into()),
        }
    }

    pub fn is_root(&self) -> bool {
        use Mount::*;

        let path = match self {
            ReadOnlySelector(_, path, ..) => path,
            ReadOnlyLayer(_, path) => path,
            Scratch(_, path) => path,
            Layer(_, _, path) => path,
            SharedCache(path) => path,
            OptionalSshAgent(_) => return false,
        };

        path.as_ref() == Path::new("/")
    }
}