180 lines · 5.3 KB · Rust
RawDownload
// Copyright 2022 The Jujutsu Authors//// Licensed under the Apache License, Version 2.0 (the "License");// you may not use this file except in compliance with the License.// You may obtain a copy of the License at//// https://www.apache.org/licenses/LICENSE-2.0//// Unless required by applicable law or agreed to in writing, software// distributed under the License is distributed on an "AS IS" BASIS,// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.// See the License for the specific language governing permissions and// limitations under the License. use std::any::Any;use std::io::Read;use std::path::Path;use std::time::SystemTime; use async_trait::async_trait;use jj_cli::cli_util::{CliRunner, CommandHelper};use jj_cli::command_error::CommandError;use jj_cli::ui::Ui;use jj_lib::backend::{    Backend, BackendInitError, BackendLoadError, BackendResult, ChangeId, Commit, CommitId,    Conflict, ConflictId, FileId, SigningFn, SymlinkId, Tree, TreeId,};use jj_lib::git_backend::GitBackend;use jj_lib::index::Index;use jj_lib::repo::StoreFactories;use jj_lib::repo_path::RepoPath;use jj_lib::settings::UserSettings;use jj_lib::signing::Signer;use jj_lib::workspace::{Workspace, WorkspaceInitError}; #[derive(clap::Parser, Clone, Debug)]enum CustomCommand {    /// Initialize a workspace using the Jit backend    InitJit,} fn create_store_factories() -> StoreFactories {    let mut store_factories = StoreFactories::default();    // Register the backend so it can be loaded when the repo is loaded. The name    // must match `Backend::name()`.    store_factories.add_backend(        "jit",        Box::new(|settings, store_path| Ok(Box::new(JitBackend::load(settings, store_path)?))),    );    store_factories} fn run_custom_command(    _ui: &mut Ui,    command_helper: &CommandHelper,    command: CustomCommand,) -> Result<(), CommandError> {    match command {        CustomCommand::InitJit => {            let wc_path = command_helper.cwd();            // Initialize a workspace with the custom backend            Workspace::init_with_backend(                command_helper.settings(),                wc_path,                &|settings, store_path| Ok(Box::new(JitBackend::init(settings, store_path)?)),                Signer::from_settings(command_helper.settings())                    .map_err(WorkspaceInitError::SignInit)?,            )?;            Ok(())        }    }} fn main() -> std::process::ExitCode {    CliRunner::init()        .set_store_factories(create_store_factories())        .add_subcommand(run_custom_command)        .run()} /// A commit backend that's extremely similar to the Git backend#[derive(Debug)]struct JitBackend {    inner: GitBackend,} impl JitBackend {    fn init(settings: &UserSettings, store_path: &Path) -> Result<Self, BackendInitError> {        let inner = GitBackend::init_internal(settings, store_path)?;        Ok(JitBackend { inner })    }     fn load(settings: &UserSettings, store_path: &Path) -> Result<Self, BackendLoadError> {        let inner = GitBackend::load(settings, store_path)?;        Ok(JitBackend { inner })    }} #[async_trait]impl Backend for JitBackend {    fn as_any(&self) -> &dyn Any {        self    }     fn name(&self) -> &str {        "jit"    }     fn commit_id_length(&self) -> usize {        self.inner.commit_id_length()    }     fn change_id_length(&self) -> usize {        self.inner.change_id_length()    }     fn root_commit_id(&self) -> &CommitId {        self.inner.root_commit_id()    }     fn root_change_id(&self) -> &ChangeId {        self.inner.root_change_id()    }     fn empty_tree_id(&self) -> &TreeId {        self.inner.empty_tree_id()    }     fn concurrency(&self) -> usize {        1    }     async fn read_file(&self, path: &RepoPath, id: &FileId) -> BackendResult<Box<dyn Read>> {        self.inner.read_file(path, id).await    }     fn write_file(&self, path: &RepoPath, contents: &mut dyn Read) -> BackendResult<FileId> {        self.inner.write_file(path, contents)    }     async fn read_symlink(&self, path: &RepoPath, id: &SymlinkId) -> BackendResult<String> {        self.inner.read_symlink(path, id).await    }     fn write_symlink(&self, path: &RepoPath, target: &str) -> BackendResult<SymlinkId> {        self.inner.write_symlink(path, target)    }     async fn read_tree(&self, path: &RepoPath, id: &TreeId) -> BackendResult<Tree> {        self.inner.read_tree(path, id).await    }     fn write_tree(&self, path: &RepoPath, contents: &Tree) -> BackendResult<TreeId> {        self.inner.write_tree(path, contents)    }     fn read_conflict(&self, path: &RepoPath, id: &ConflictId) -> BackendResult<Conflict> {        self.inner.read_conflict(path, id)    }     fn write_conflict(&self, path: &RepoPath, contents: &Conflict) -> BackendResult<ConflictId> {        self.inner.write_conflict(path, contents)    }     async fn read_commit(&self, id: &CommitId) -> BackendResult<Commit> {        self.inner.read_commit(id).await    }     fn write_commit(        &self,        contents: Commit,        sign_with: Option<&mut SigningFn>,    ) -> BackendResult<(CommitId, Commit)> {        self.inner.write_commit(contents, sign_with)    }     fn gc(&self, index: &dyn Index, keep_newer: SystemTime) -> BackendResult<()> {        self.inner.gc(index, keep_newer)    }}