1 files changed, 247 insertions, 0 deletions

diff --git a/src/sub_commands/init.rs b/src/sub_commands/init.rs
new file mode 100644
index 0000000..e5fa52c
--- /dev/null
+++ b/src/sub_commands/init.rs
@@ -0,0 +1,247 @@
+use std::{str::FromStr, fs::{OpenOptions, File}, io::Write};
+use dialoguer::{theme::ColorfulTheme, Confirm, Input};
+use clap::{Args};
+use indicatif::ProgressBar;
+use nostr_sdk::prelude::*;
+
+use crate::{config::{load_config, save_conifg}, groups::{init::{InitializeGroup}, group::{Group}}, repos::{init::InitializeRepo, repo::Repo}, utils::{save_event, create_client, get_or_generate_keys}, cli_helpers::select_relays, repo_config::RepoConfig};
+
+#[derive(Args)]
+struct InitRepo {
+    // Repo Name
+    #[arg(short, long)]
+    name: String,
+    /// Admin Group ID
+    #[arg(long)]
+    admin_group_id: Option<String>,
+    /// Relays
+    #[arg(short, long)]
+    relays: Option<String>,
+}
+
+#[derive(Args)]
+pub struct InitSubCommand {
+    /// Repo Name
+    #[arg(short, long)]
+    name: Option<String>,
+}
+
+pub fn create_and_broadcast_init(
+    relays: Vec<String>,
+    sub_command_args: &InitSubCommand,
+) -> Result<()> {
+    
+    let mut cfg = load_config();
+
+    let repo_dir_path = std::env::current_dir().unwrap();
+    
+    // check for potential problems
+    let ngit_path = repo_dir_path.clone().join(".ngit");
+    if ngit_path.is_dir() && (
+        !Confirm::with_theme(&ColorfulTheme::default())
+            .with_prompt("ngit already initialized! Do you want overwrite it with a fresh repoisotry?")
+            .default(false)
+            .interact()
+            .unwrap()
+        || !Confirm::with_theme(&ColorfulTheme::default())
+            .with_prompt("Are you sure?")
+            .default(false)
+            .interact()
+            .unwrap()
+    ) { panic!("aborted as ngit repository already exists."); };
+
+    let git_path = repo_dir_path.clone().join(".git");
+    if git_path.is_dir() && (
+        !Confirm::with_theme(&ColorfulTheme::default())
+            .with_prompt("git has already been initialized here. For this alpha ngit prototype its best to start with a fresh repository. Continue anyway?")
+            .default(false)
+            .interact()
+            .unwrap()
+    ) { panic!("aborted as git repository already initialized."); };
+
+    // collect information from user
+
+    let dir_name: String =String::from(repo_dir_path.file_name().unwrap().to_string_lossy());
+    let repo_name: String = match &sub_command_args.name {
+        Some(name) => name.clone(),
+        None => {
+            Input::with_theme(&ColorfulTheme::default())
+                .with_prompt("Repo Name")
+                .default(dir_name)
+                .interact_text()
+                .unwrap()
+        },
+    };    
+    let repo_relays = select_relays(&mut cfg, &relays)?;
+
+    let mut repo_group_members: Vec<String> = vec![];
+    loop {
+        if Confirm::with_theme(&ColorfulTheme::default())
+            .with_prompt("Would you like add other maintainers now?")
+            .default(false)
+            .interact()
+            .unwrap()
+        {
+            let member_key_input: String = Input::with_theme(&ColorfulTheme::default())
+                .with_prompt("npub or hex (n to abort)")
+                .interact_text()
+                .unwrap()
+            ;
+            if member_key_input.starts_with("npub") {
+                match XOnlyPublicKey::from_bech32(member_key_input) {
+                    Ok(k) => { repo_group_members.push(k.to_string()); },
+                    Err(e) => { println!("{}",e) },
+                }}
+            else {
+                match XOnlyPublicKey::from_str(member_key_input.as_str()) {
+                    Ok(k) => { repo_group_members.push(k.to_string()); },
+                    Err(e) => { println!("{}",e) },
+                }
+            }
+        }
+        else { break; }
+    }
+
+    let keys = get_or_generate_keys(&mut cfg);
+    let mut events_to_broadcast: Vec<Event> = vec![];
+
+    // delay adding user as group member so keys are the last thing asked for
+    repo_group_members.push(keys.public_key().to_string());
+
+    let admin_group_event = match cfg.default_admin_group_event_serialized {
+        None => {
+            let new_admin_group = Group::new(
+                &InitializeGroup::new()
+                    .members(
+                        vec![
+                            keys.public_key().to_string(),
+                        ],
+                        vec![],
+                    )
+                    .relays(&repo_relays),
+                &keys,
+            ).unwrap();
+            cfg.default_admin_group_event_serialized = Some(new_admin_group.events[0].as_json());
+            save_conifg(&cfg);
+            events_to_broadcast.push(new_admin_group.events[0].clone());
+            new_admin_group.events[0].clone()
+        },
+        Some(admin) => Group::new_from_json_event(admin.clone())
+            .expect("default_admin_group_event_serialized in MyConfig to load into Group")
+            .events[0].clone(),
+    };
+
+    let new_repo_group = Group::new(
+        &InitializeGroup::new()
+            .name(format!("{repo_name} maintainers (ngit)"))
+            .members(
+                repo_group_members,
+                vec![],
+            )
+            .relays(&repo_relays)
+        ,
+        &keys,
+    ).unwrap();
+    events_to_broadcast.push(new_repo_group.events[0].clone());   
+
+    let new_repo = Repo::new(
+        &InitializeRepo::new()
+            .name(&repo_name)
+            .relays(&repo_relays)
+            .maintainers_group(new_repo_group.get_ref())
+        ,
+        &keys,
+    ).unwrap();    
+    events_to_broadcast.push(new_repo.events[0].clone());   
+
+    // crate .ngit folder and store the repo and group reference and associated events (?)
+    for p in [
+        "groups",
+        "branches",
+        "patches",
+        "merges",
+        "prs",
+        "issues",
+        "comments",
+    ] { std::fs::create_dir_all(ngit_path.join(p)).unwrap(); }
+
+    // save repo event locally
+    save_event(
+        ngit_path.join(format!("groups/{}.json",admin_group_event.id.to_string())),
+        &admin_group_event,
+    ).unwrap();
+    
+    save_event(
+        ngit_path.join(format!("groups/{}.json",new_repo_group.id.to_string())),
+        &new_repo_group.events[0],
+    ).unwrap();
+    save_event(
+        ngit_path.join("repo.json"),
+        &new_repo.events[0],
+    ).unwrap();
+
+    // set repo config
+    let mut repo_config = RepoConfig::open(&repo_dir_path);
+    for b in ["main", "master"] {
+        repo_config.set_mapping(&b.to_string(), &new_repo.events[0].id.to_string());
+
+    }
+    repo_config.set_last_branch_ref_update_time(new_repo.events[0].created_at.clone());
+
+    // initialise git
+    git2::Repository::init(repo_dir_path.clone()).unwrap();
+
+    // add .gitignore
+    let gitignore_path = repo_dir_path.join(".gitignore");
+    let mut gitignore_file = if gitignore_path.is_file() {
+        OpenOptions::new()
+            .write(true)
+            .append(true)
+            .open(&gitignore_path)
+            .expect(".gitignore to open")
+    } else {
+        File::create(gitignore_path)
+            .expect("create and open .gitignore file")
+    };
+    writeln!(gitignore_file, ".ngit")
+        .expect(".ngit added to gitignore");
+
+    let spinner = ProgressBar::new_spinner();
+    spinner.set_message("Broadcasting... if this takes 20s+, there was a problem broadcasting to one or more relays even if it says 'Repository Initialised'.");
+
+    let client = create_client(&keys, repo_relays.clone())?;
+    for e in &events_to_broadcast { 
+        match client.send_event(e.clone()) {
+            Ok(_) => (),
+            // TODO: this isn't working - if a relay is specified with a type it will wait 30ish secs and then return successful
+            Err(e) => { println!("error broadcasting repo event: {}",e); },
+        }
+        // TODO: better error handling here / reporting. potentially warn if taking a while and report on troublesome relays
+    }
+    spinner.finish_with_message(format!(
+        "Repository Initialised! id: {}",
+        Nip19Event::new(
+            new_repo.id.clone(),
+            vec![&repo_relays[0]],
+        )
+            .to_bech32()
+            .expect("Nip19Event to convert to to_bech32")
+
+    ));
+    // println!("Hint: only maintainers can push to master branch and merge PRs but anyone can clone and push a forked branch.");
+
+    // Instructions:
+    // 1. make some commits on 'master' or 'main' branch.
+    // 2. 'ngit push' will push them to the nostr repo (via a patch).
+    // 3. make a branch called 'feature-1'
+    // 3. 'ngit push' will create a PR
+
+    // repo
+        // repo 
+    // PRs
+
+    // patch
+
+
+    Ok(())
+}