pub mod git;
mod parser;
pub mod user_info;

use comfy_table::Table;
use parser::*;
use rustyline::error::ReadlineError;
use std::{io, ops::ControlFlow};
use thiserror::Error;

pub fn run_command(user_input: &str) -> Result<ControlFlow<(), ()>, ShackleError> {
    match user_input.parse::<ShackleCommand>() {
        Err(parse_error) => {
            println!("{}", parse_error);
        }
        Ok(ShackleCommand::Exit) => {
            return Ok(ControlFlow::Break(()));
        }
        Ok(ShackleCommand::List) => {
            let mut table = Table::new();
            table.set_header(vec!["path", "description"]);
            let listing = git::list()?;
            for meta in listing {
                table.add_row(vec![meta.path.display().to_string(), meta.description]);
            }

            println!("{table}");
        }
        Ok(ShackleCommand::SetDescription(SetDescriptionArgs {
            directory,
            description,
        })) => {
            git::set_description(&directory, &description)?;
            println!("Successfully updated description");
        }
        Ok(ShackleCommand::SetBranch(SetBranchArgs { directory, branch })) => {
            git::set_branch(&directory, &branch)?;
            println!("Successfully updated branch");
        }
        Ok(ShackleCommand::Init(InitArgs {
            repo_name,
            group,
            description,
            branch,
        })) => {
            let init_result = git::init(&repo_name, &group, &description, &branch)?;
            println!("Successfully created \"{}\"", init_result.path.display());
        }
        Ok(ShackleCommand::GitUploadPack(upload_pack_args)) => {
            git::upload_pack(&upload_pack_args)?;
        }
        Ok(ShackleCommand::GitReceivePack(receive_pack_args)) => {
            git::receive_pack(&receive_pack_args)?;
        }
    }
    Ok(ControlFlow::Continue(()))
}

#[derive(Error, Debug)]
pub enum ShackleError {
    #[error(transparent)]
    IoError(#[from] io::Error),
    #[error(transparent)]
    GitError(#[from] git2::Error),
    #[error(transparent)]
    ReadlineError(#[from] ReadlineError),
    #[error("Could not get the current user name")]
    UserReadError,
    #[error("Path is not accessible")]
    InvalidDirectory,
    #[error("Unknown group")]
    InvalidGroup,
}