112 lines · 4.3 KB · Rust
RawDownload
#![allow(clippy::result_large_err)]use std::path::Path; use gix_ref::{    Category, FullName, Target,    store::WriteReflog,    transaction::{PreviousValue, RefEdit},}; use crate::{    ThreadSafeRepository,    bstr::{BString, ByteSlice},    config::tree::Init,}; /// The name of the branch to use if non is configured via git configuration.////// # Deviation////// We use `main` instead of `master`.pub const DEFAULT_BRANCH_NAME: &str = "main"; /// The error returned by [`crate::init()`].#[derive(Debug, thiserror::Error)]#[expect(missing_docs)]pub enum Error {    #[error("Could not obtain the current directory")]    CurrentDir(#[from] std::io::Error),    #[error(transparent)]    Init(#[from] crate::create::Error),    #[error(transparent)]    Open(#[from] crate::open::Error),    #[error("Invalid default branch name: {name:?}")]    InvalidBranchName {        name: BString,        source: gix_validate::reference::name::Error,    },    #[error("Could not edit HEAD reference with new default name")]    EditHeadForDefaultBranch(#[from] crate::reference::edit::Error),} impl ThreadSafeRepository {    /// Create a repository with work-tree within `directory`, creating intermediate directories as needed.    ///    /// Fails without action if the destination directory isn't empty unless    /// [`create::Options::destination_must_be_empty`][crate::create::Options::destination_must_be_empty] is `None`    /// or `Some(false)`. Note that initialization still fails if a `.git` directory already exists in    /// the destination.    pub fn init(        directory: impl AsRef<Path>,        kind: crate::create::Kind,        options: crate::create::Options,    ) -> Result<Self, Error> {        use gix_sec::trust::DefaultForLevel;        let open_options = crate::open::Options::default_for_level(gix_sec::Trust::Full);        Self::init_opts(directory, kind, options, open_options)    }     /// Similar to [`init`][Self::init()], but allows to determine how exactly to open the newly created repository.    ///    /// # Deviation    ///    /// Instead of naming the default branch `master`, we name it `main` unless configured explicitly using the `init.defaultBranch`    /// configuration key.    pub fn init_opts(        directory: impl AsRef<Path>,        kind: crate::create::Kind,        create_options: crate::create::Options,        mut open_options: crate::open::Options,    ) -> Result<Self, Error> {        let path = crate::create::into(directory.as_ref(), kind, create_options)?;        let (git_dir, worktree_dir) = path.into_repository_and_work_tree_directories();        open_options.git_dir_trust = Some(gix_sec::Trust::Full);        // The repo will use `core.precomposeUnicode` to adjust the value as needed.        open_options.current_dir = gix_fs::current_dir(false)?.into();        let repo = ThreadSafeRepository::open_from_paths(git_dir, worktree_dir, open_options)?;         let branch_name = repo            .config            .resolved            .string(Init::DEFAULT_BRANCH)            .unwrap_or_else(|| DEFAULT_BRANCH_NAME.into());        if branch_name.as_bstr() != DEFAULT_BRANCH_NAME {            let configured_branch_name = branch_name;            let sym_ref: FullName = Category::LocalBranch                .to_full_name(configured_branch_name.as_bstr())                .map_err(|err| Error::InvalidBranchName {                    name: configured_branch_name.clone(),                    source: err,                })?;            gix_validate::reference::branch_name(sym_ref.as_bstr()).map_err(|err| Error::InvalidBranchName {                name: configured_branch_name,                source: err,            })?;            let mut repo = repo.to_thread_local();            let prev_write_reflog = repo.refs.write_reflog;            repo.refs.write_reflog = WriteReflog::Disable;            repo.edit_reference(RefEdit {                change: gix_ref::transaction::Change::Update {                    log: Default::default(),                    expected: PreviousValue::Any,                    new: Target::Symbolic(sym_ref),                },                name: "HEAD".try_into().expect("valid"),                deref: false,            })?;            repo.refs.write_reflog = prev_write_reflog;        }         Ok(repo)    }}