gix-config 0.61.0

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

use bstr::{BStr, BString, ByteSlice, ByteVec};
use gix_error::{ErrorExt, ExnResult, OptionExt, ResultExt, message, not_found, validation};
use gix_features::threading::OwnShared;
use gix_ref::Category;

use crate::{
    File, file,
    file::{Metadata, SectionId, includes, init},
};

impl File {
    /// Traverse all `include` and `includeIf` directives found in this instance and follow them, loading the
    /// referenced files from their location and adding their content right past the value that included them.
    ///
    /// # Limitations
    ///
    /// - Note that this method is _not idempotent_ and calling it multiple times will resolve includes multiple
    ///   times. It's recommended use is as part of a multi-step bootstrapping which needs fine-grained control,
    ///   and unless that's given one should prefer one of the other ways of initialization that resolve includes
    ///   at the right time.
    ///
    /// # Deviation
    ///
    /// - included values are added after the _section_ that included them, not directly after the value. This is
    ///   a deviation from how git does it, as it technically adds new value right after the include path itself,
    ///   technically 'splitting' the section. This can only make a difference if the `include` section also has values
    ///   which later overwrite portions of the included file, which seems unusual as these would be related to `includes`.
    ///   We can fix this by 'splitting' the include section if needed so the included sections are put into the right place.
    /// - `hasconfig:remote.*.url` will not prevent itself to include files with `[remote "name"]\nurl = x` values, but it also
    ///   won't match them, i.e. one cannot include something that will cause the condition to match or to always be true.
    pub fn resolve_includes(&mut self, options: init::Options<'_>) -> ExnResult {
        if options.includes.max_depth == 0 {
            return Ok(());
        }
        let mut buf = Vec::new();
        resolve(self, &mut buf, options)
    }
}

pub(crate) fn resolve(config: &mut File, buf: &mut Vec<u8>, options: init::Options<'_>) -> ExnResult {
    resolve_includes_recursive(None, config, 0, buf, options)
}

fn resolve_includes_recursive(
    search_config: Option<&File>,
    target_config: &mut File,
    depth: u8,
    buf: &mut Vec<u8>,
    options: init::Options<'_>,
) -> ExnResult {
    if depth == options.includes.max_depth {
        return if options.includes.err_on_max_depth_exceeded {
            Err(validation(format!(
                "The maximum allowed length {} of the file include chain built by following nested resolve_includes is exceeded",
                options.includes.max_depth
            ))
            .raise_erased())
        } else {
            Ok(())
        };
    }

    for id in target_config.section_order.clone().into_iter() {
        let section = &target_config.sections[&id];
        let header = &section.header;
        let backing = &target_config.backing;
        let header_name = header.name.as_bstr_in(backing);
        let mut paths = None;
        if header_name == "include" && header.subsection_name.is_none() {
            paths = Some(gather_paths(section, id, backing));
        } else if header_name == "includeIf"
            && let Some(condition) = &header.subsection_name
        {
            let target_config_path = section.meta.path.as_deref();
            if include_condition_match(
                condition.value_in(backing),
                target_config_path,
                search_config.unwrap_or(target_config),
                options.includes,
            )? {
                paths = Some(gather_paths(section, id, backing));
            }
        }
        if let Some(paths) = paths {
            insert_includes_recursively(paths, target_config, depth, options, buf)?;
        }
    }
    Ok(())
}

fn insert_includes_recursively(
    section_ids_and_include_paths: Vec<(SectionId, crate::Path)>,
    target_config: &mut File,
    depth: u8,
    options: init::Options<'_>,
    buf: &mut Vec<u8>,
) -> ExnResult {
    for (section_id, config_path) in section_ids_and_include_paths {
        let meta = OwnShared::clone(&target_config.sections[&section_id].meta);
        let target_config_path = meta.path.as_deref();
        let config_path = match resolve_path(config_path, target_config_path, options.includes)? {
            Some(p) => p,
            None => continue,
        };
        if !config_path.is_file() {
            continue;
        }

        buf.clear();
        std::io::copy(
            &mut std::fs::File::open(&config_path).or_raise_erased(|| {
                message!(
                    "Could not read included configuration file at '{}'",
                    config_path.display()
                )
            })?,
            buf,
        )
        .or_raise_erased(|| message("Failed to copy configuration file into buffer"))?;
        let config_meta = Metadata {
            path: Some(config_path),
            trust: meta.trust,
            level: meta.level + 1,
            source: meta.source,
        };
        let no_follow_options = init::Options {
            includes: includes::Options::no_follow(),
            ..options
        };

        let mut include_config = File::from_bytes_owned(buf, config_meta, no_follow_options)
            .or_raise_erased(|| message("Could not parse included configuration file"))?;
        resolve_includes_recursive(Some(target_config), &mut include_config, depth + 1, buf, options)?;

        target_config
            .append_or_insert(include_config, Some(section_id))
            .or_raise_erased(|| message("Could not append included configuration"))?;
    }
    Ok(())
}

fn gather_paths(section: &file::SectionData, id: SectionId, backing: &[u8]) -> Vec<(SectionId, crate::Path)> {
    section
        .body
        .values_in(backing, "path")
        .into_iter()
        .map(|path| (id, crate::Path::from(path)))
        .collect()
}

fn include_condition_match(
    condition: &BStr,
    target_config_path: Option<&Path>,
    search_config: &File,
    options: Options<'_>,
) -> ExnResult<bool> {
    let mut tokens = condition.splitn(2, |b| *b == b':');
    let (prefix, condition) = match (tokens.next(), tokens.next()) {
        (Some(a), Some(b)) => (a, b),
        _ => return Ok(false),
    };
    let condition = condition.as_bstr();
    match prefix {
        b"gitdir" => gitdir_matches(
            condition,
            target_config_path,
            options,
            gix_glob::wildmatch::Mode::empty(),
        ),
        b"gitdir/i" => gitdir_matches(
            condition,
            target_config_path,
            options,
            gix_glob::wildmatch::Mode::IGNORE_CASE,
        ),
        b"onbranch" => Ok(onbranch_matches(condition, options.conditional).is_some()),
        b"hasconfig" => {
            let mut tokens = condition.splitn(2, |b| *b == b':');
            let (key_glob, value_glob) = match (tokens.next(), tokens.next()) {
                (Some(a), Some(b)) => (a, b),
                _ => return Ok(false),
            };
            if key_glob.as_bstr() != "remote.*.url" {
                return Ok(false);
            }
            let Some(sections) = search_config.sections_by_name("remote") else {
                return Ok(false);
            };
            for remote in sections {
                for url in remote.values("url") {
                    let glob_matches = gix_glob::wildmatch(
                        value_glob.as_bstr(),
                        url.as_ref(),
                        gix_glob::wildmatch::Mode::NO_MATCH_SLASH_LITERAL,
                    );
                    if glob_matches {
                        return Ok(true);
                    }
                }
            }
            Ok(false)
        }
        _ => Ok(false),
    }
}

fn onbranch_matches(
    condition: &BStr,
    conditional::Context { branch_name, .. }: conditional::Context<'_>,
) -> Option<()> {
    let branch_name = branch_name?;
    let (_, branch_name) = branch_name
        .category_and_short_name()
        .filter(|(cat, _)| *cat == Category::LocalBranch)?;

    let condition: BString = if condition.ends_with(b"/") {
        let mut condition: BString = condition.into();
        condition.push_str("**");
        condition
    } else {
        condition.into()
    };

    gix_glob::wildmatch(
        condition.as_bstr(),
        branch_name,
        gix_glob::wildmatch::Mode::NO_MATCH_SLASH_LITERAL,
    )
    .then_some(())
}

fn gitdir_matches(
    condition_path: &BStr,
    target_config_path: Option<&Path>,
    Options {
        conditional: conditional::Context { git_dir, .. },
        interpolate: context,
        err_on_interpolation_failure,
        err_on_missing_config_path,
        ..
    }: Options<'_>,
    wildmatch_mode: gix_glob::wildmatch::Mode,
) -> ExnResult<bool> {
    if !err_on_interpolation_failure && git_dir.is_none() {
        return Ok(false);
    }
    let git_dir = gix_path::to_unix_separators_on_windows(gix_path::into_bstr(git_dir.ok_or_raise_erased(|| {
        not_found("The git directory must be provided to support `gitdir:` conditional includes")
    })?));

    let mut pattern_path = match check_interpolation_result(
        err_on_interpolation_failure,
        crate::Path::from(condition_path.to_owned()).interpolate(context),
    )
    .or_raise_erased(|| message("Could not interpolate conditional include path"))?
    {
        Some(path) => gix_path::into_bstr(path).into_owned(),
        // Git keeps the original condition pattern when interpolation fails.
        None => condition_path.to_owned(),
    };
    // NOTE: yes, only if we do path interpolation will the slashes be forced to unix separators on windows
    if pattern_path != condition_path {
        pattern_path = gix_path::to_unix_separators_on_windows(pattern_path).into_owned();
    }

    if let Some(relative_pattern_path) = pattern_path.strip_prefix(b"./") {
        if !err_on_missing_config_path && target_config_path.is_none() {
            return Ok(false);
        }
        let parent_dir = target_config_path
            .ok_or_raise_erased(|| {
                not_found(
                    "Include paths from environment variables must not be relative as no config file path exists as root",
                )
            })?
            .parent()
            .expect("config path can never be /");
        let mut joined_path = gix_path::to_unix_separators_on_windows(gix_path::into_bstr(parent_dir)).into_owned();
        joined_path.push(b'/');
        joined_path.extend_from_slice(relative_pattern_path);
        pattern_path = joined_path;
    }

    // NOTE: this special handling of leading backslash is needed to do it like git does
    if pattern_path.iter().next() != Some(&(std::path::MAIN_SEPARATOR as u8))
        && !gix_path::from_bstr(pattern_path.clone()).is_absolute()
    {
        pattern_path.insert_str(0, "**/");
    }
    if pattern_path.ends_with(b"/") {
        pattern_path.push_str("**");
    }

    let match_mode = gix_glob::wildmatch::Mode::NO_MATCH_SLASH_LITERAL | wildmatch_mode;
    let is_match = gix_glob::wildmatch(pattern_path.as_bstr(), git_dir.as_bstr(), match_mode);
    if is_match {
        return Ok(true);
    }

    let expanded_git_dir = gix_path::to_unix_separators_on_windows(gix_path::into_bstr(
        gix_path::realpath(gix_path::from_byte_slice(&git_dir))
            .or_raise_erased(|| message("Could not resolve the git directory to its real path"))?,
    ));
    Ok(gix_glob::wildmatch(
        pattern_path.as_bstr(),
        expanded_git_dir.as_ref(),
        match_mode,
    ))
}

fn check_interpolation_result(disable: bool, res: ExnResult<impl Into<PathBuf>>) -> ExnResult<Option<PathBuf>> {
    if disable {
        return res.map(|path| Some(path.into()));
    }
    match res {
        Ok(good) => Ok(Some(good.into())),
        Err(err) if err.is_validation() => Err(err),
        Err(_) => Ok(None),
    }
}

fn resolve_path(
    path: crate::Path,
    target_config_path: Option<&Path>,
    includes::Options {
        interpolate: context,
        err_on_interpolation_failure,
        err_on_missing_config_path,
        ..
    }: includes::Options<'_>,
) -> ExnResult<Option<PathBuf>> {
    let path = match check_interpolation_result(err_on_interpolation_failure, path.interpolate(context))
        .or_raise_erased(|| message("Could not interpolate include path"))?
    {
        Some(p) => p,
        None => return Ok(None),
    };
    let path: PathBuf = if path.is_relative() {
        if !err_on_missing_config_path && target_config_path.is_none() {
            return Ok(None);
        }
        target_config_path
            .ok_or_raise_erased(|| {
                not_found(
                    "Include paths from environment variables must not be relative as no config file path exists as root",
                )
            })?
            .parent()
            .expect("path is a config file which naturally lives in a directory")
            .join(path)
    } else {
        path
    };
    Ok(Some(path))
}

mod types;
pub use types::{Options, conditional};