codeberg_cli/actions/repo/
fork.rs

1use crate::actions::GlobalArgs;
2use crate::render::spinner::{spin_and_try_every_second_for, spin_until_ready};
3use crate::render::ui::confirm_with_prompt;
4use crate::types::context::BergContext;
5use crate::types::git::{Git, OwnerRepo};
6use forgejo_api::structs::CreateForkOption;
7use miette::{Context, IntoDiagnostic};
8use url::Url;
9
10use clap::Parser;
11
12/// Fork a repository
13#[derive(Parser, Debug)]
14pub struct RepoForkArgs {
15    /// Repository to be forked
16    #[arg(value_name = "OWNER/REPO")]
17    pub owner_and_repo: OwnerRepo,
18}
19
20impl RepoForkArgs {
21    pub async fn run(self, global_args: GlobalArgs) -> miette::Result<()> {
22        let _ = global_args;
23        let ctx = BergContext::new(self, global_args).await?;
24
25        let OwnerRepo { owner, repo } = ctx.args.owner_and_repo.clone();
26        let url = spin_until_ready(start_fork_repo(&ctx, owner.as_str(), repo.as_str())).await?;
27        ask_confirm_clone(repo.as_str())?;
28        Git::clone(url, repo)?;
29        Ok(())
30    }
31}
32
33async fn start_fork_repo(
34    ctx: &BergContext<RepoForkArgs>,
35    owner: &str,
36    repo: &str,
37) -> miette::Result<Url> {
38    // just to check if the repo exists
39    let _ssh_url_original = get_ssh_url(ctx, owner, repo).await?;
40
41    ctx.client
42        .create_fork(
43            owner,
44            repo,
45            CreateForkOption {
46                name: None,
47                organization: None,
48            },
49        )
50        .await
51        .into_diagnostic()?;
52    let user = ctx.client.user_get_current().await.into_diagnostic()?;
53    let username = user
54        .login
55        .as_ref()
56        .cloned()
57        .context("Current user has no username")?;
58    let new_url = spin_and_try_every_second_for(|| get_ssh_url(ctx, &username, repo), 10).await?;
59    tracing::debug!("Forked Repo SSH URL: {new_url:?}");
60    Ok(new_url)
61}
62
63async fn get_ssh_url(
64    ctx: &BergContext<RepoForkArgs>,
65    owner: &str,
66    repo: &str,
67) -> miette::Result<Url> {
68    ctx.client
69        .repo_get(owner, repo)
70        .await
71        .into_diagnostic()
72        .and_then(|repo| repo.ssh_url.context("No SSH url on repo"))
73        .context("User doesn't own the repo that was specified.")
74}
75
76fn ask_confirm_clone(repo: &str) -> miette::Result<()> {
77    let current_path = std::env::current_dir().into_diagnostic()?;
78    if !confirm_with_prompt(
79        format!("Do you really to fork {repo} into the directory {current_path:?}").as_str(),
80    )? {
81        miette::bail!("Abort cloning the repository.")
82    }
83    Ok(())
84}