otkeep 0.1.0

use std::ffi::OsStr;

use anyhow::{bail, Context};
use otkeep::{database::NoSuchScriptForCurrentTree, AppContext};

fn main() {
    match try_main() {
        Ok(code) => std::process::exit(code),
        Err(e) => {
            eprintln!("Error: {:?}", e);
            std::process::exit(1);
        }
    }
}

fn try_main() -> anyhow::Result<i32> {
    let mut args = std::env::args_os().skip(1);
    let db = otkeep::load_db()?;
    let root_id = match otkeep::find_root(&db)? {
        Some((id, _)) => id,
        None => {
            otkeep::print_established_trees(&db)?;
            bail!("No OtKeep tree root was found. To establish one, use okeep establish");
        }
    };

    let mut app = AppContext { db, root_id };
    let cmd_name = match args.next() {
        Some(arg) => arg,
        None => {
            otkeep::list_scripts(&app)?;
            eprintln!("\nFor more options, try okeep",);
            return Ok(1);
        }
    };
    run(
        cmd_name.to_str().context("Command name not utf-8")?,
        &mut app,
        args,
    )
    .context("Failed to run script")
}

fn run(
    name: &str,
    ctx: &mut AppContext,
    args: impl Iterator<Item = impl AsRef<OsStr>>,
) -> anyhow::Result<i32> {
    match ctx.db.run_script(ctx.root_id, name, args) {
        Ok(status) => Ok(status.code().unwrap_or(1)),
        Err(e) => match e.downcast_ref::<NoSuchScriptForCurrentTree>() {
            Some(_) => {
                eprintln!("No script named '{}' for the current tree.\n", name);
                otkeep::list_scripts(ctx)?;
                eprintln!("\nFor more options, try okeep");
                Ok(1)
            }
            None => Err(e),
        },
    }
}