toru/src/main.rs

mod vcs;
mod edit;
mod args;
mod vault;
mod index;
mod error;
mod tasks;
mod state;
mod graph;
mod stats;
mod config;
mod format;

use args::*;

fn main() {
    let result = program();

    match result {
        Ok(()) => {
            std::process::exit(0);
        },
        Err(err) => {
            println!("{}", err);
            std::process::exit(1);
        }
    }
}

fn program() -> Result<(), error::Error> {
    let command = Args::accept_command();

    let mut config = config::Config::load()?;

    if let Command::Vault(command) = command {
        match command {
            VaultCommand::New { name, path } => {
                vault::new(name.clone(), path, &mut config)?;
                println!("Created vault {}", format::vault(&name));
            },
            VaultCommand::Disconnect { name } => {
                vault::disconnect(&name, &mut config)?;
                println!("Disconnected vault {}", format::vault(&name));
            },
            VaultCommand::Connect { name , path } => {
                vault::connect(name.clone(), path, &mut config)?;
                println!("Connected vault {}", format::vault(&name));
            },
            VaultCommand::Delete { name } => {
                vault::delete(&name, &mut config)?;
                println!("Deleted vault {}", format::vault(&name));
            },
            VaultCommand::List => {
                config.list_vaults()?;
            },
            VaultCommand::Rename { old_name, new_name } => {
                config.rename_vault(&old_name, new_name.clone())?;
                println!("Renamed vault {} to {}", format::vault(&old_name), format::vault(&new_name));
            }
        }
    }
    else if let Command::Config(command) = command {
        use ConfigCommand::*;
        match command {
            Editor { editor } => {
                match editor {
                    Some(editor) => {
                        config.editor = editor;
                        println!("Updated editor command to: {}", config.editor);
                    },
                    None => {
                        println!("Current editor command: {}", config.editor);
                    }
                }
            }
        }
    }
    else if let Command::Switch { name } = command {
        config.switch(&name)?;
        println!("Switched to vault {}", format::vault(&name));
    }
    else if let Command::Git { args } = command {
        let vault_folder = &config.current_vault()?.1;
        vcs::command(args, vcs::Vcs::Git, vault_folder)?;
    }
    else if command == Command::GitIgnore {
        let vault_folder = &config.current_vault()?.1;
        vcs::create_gitignore(vault_folder)?;
        println!("Default {} file created", format::file(".gitignore"));
    }
    else if let Command::Svn { args } = command {
        let vault_folder = &config.current_vault()?.1;
        vcs::command(args, vcs::Vcs::Svn, vault_folder)?;
    }
    // Commands that require loading in the state.
    else {
        let vault_folder = &config.current_vault()?.1;
        let mut state = state::State::load(vault_folder)?;

        match command {
            Command::New { name, info, tag, dependency, priority, due } => {
                let task = tasks::Task::new(name, info, tag, dependency, priority, due, vault_folder, &mut state)?;
                println!("Created task {} (ID: {})", format::task(&task.data.name), format::id(task.data.id));
            },
            Command::Delete { id_or_name } => {
                let id = state.data.index.lookup(&id_or_name)?;
                let task = tasks::Task::load(id, vault_folder, false)?;
                let name = task.data.name.clone();
                state.data.index.remove(task.data.name.clone(), task.data.id);
                state.data.deps.remove_node(task.data.id);
                task.delete()?;

                println!("Deleted task {} (ID: {})", format::task(&name), format::id(id));
            },
            Command::View { id_or_name } => {
                let id = state.data.index.lookup(&id_or_name)?;
                let task = tasks::Task::load(id, vault_folder, true)?;
                task.display(vault_folder, &state)?;
            },
            Command::Edit { id_or_name, info } => {
                let id = state.data.index.lookup(&id_or_name)?;
                if info {
                    edit::edit_info(id, vault_folder.clone(), &config.editor)?;
                }
                else {
                    edit::edit_raw(id, vault_folder.clone(), &config.editor, &mut state)?;
                }
                println!("Updated task {}", format::id(id));
            },
            Command::Track { id_or_name, hours, minutes, date, message } => {
                let id = state.data.index.lookup(&id_or_name)?;
                let mut task = tasks::Task::load(id, vault_folder, false)?;
                let entry =  tasks::TimeEntry::new(hours, minutes, date, message);
                task.data.time_entries.push(entry);
                task.save()?;
            },
            Command::Stats(command) => {
                use StatsCommand::*;
                match command {
                    Tracked { days } => {
                        stats::time_per_tag(days, vault_folder)?;
                    },
                    Completed { days } => {
                        stats::completed_tasks(days, vault_folder)?;
                    }
                }
            },
            Command::Complete { id_or_name } => {
                let id = state.data.index.lookup(&id_or_name)?;
                let mut task = tasks::Task::load(id, vault_folder, false)?;
                task.data.completed = Some(chrono::Local::now().naive_local());
                task.save()?;
                println!("Marked task {} as complete", format::id(id));
            },
            Command::List { options } => {
                tasks::list(options, vault_folder, &state)?;
            },
            // All commands which are dealt with in if let chain at start.
            Command::Vault(_) | Command::Config(_) | Command::Git { args : _ } | Command::Svn { args : _ } | Command::Switch { name : _ } | Command::GitIgnore => unreachable!(),
        }

        state.save()?;
    }

    config.save()?;

    Ok(())
}