toru/src/main.rs

248 lines
7.5 KiB
Rust
Raw Normal View History

2022-08-20 07:06:55 +00:00
mod git;
2022-08-21 00:59:09 +00:00
mod edit;
2022-08-20 04:01:43 +00:00
mod vault;
mod error;
mod tasks;
mod state;
mod config;
mod colour;
2022-08-21 00:59:09 +00:00
use tasks::Id;
2022-08-20 04:01:43 +00:00
use std::path;
#[derive(clap::Parser, Debug)]
struct Args {
#[clap(subcommand)]
command : Command,
}
#[derive(clap::Subcommand, Debug)]
2022-08-20 07:06:55 +00:00
#[clap(version, about, author, global_setting=clap::AppSettings::DisableHelpSubcommand)]
2022-08-20 04:01:43 +00:00
enum Command {
/// Create a new task.
New {
#[clap(short, long)]
name : String,
#[clap(short, long)]
info : Option<String>,
#[clap(short, long)]
tags : Vec<String>,
#[clap(short, long)]
2022-08-21 00:59:09 +00:00
dependencies : Vec<Id>,
2022-08-20 04:01:43 +00:00
#[clap(short, long, value_enum)]
priority : Option<tasks::Priority>,
},
/// Displays the specified task in detail.
View {
id_or_name : String,
2022-08-21 00:59:09 +00:00
},
2022-08-21 02:50:03 +00:00
/// Edit a note directly.
2022-08-21 00:59:09 +00:00
Edit {
id_or_name : String,
2022-08-21 02:50:03 +00:00
/// Edit the info specifically in its own file.
#[clap(short, long)]
info : bool,
},
2022-08-20 04:01:43 +00:00
/// Delete a task completely.
Delete {
id_or_name : String,
2022-08-20 04:01:43 +00:00
},
/// Discard a task without deleting the underlying file.
Discard {
id_or_name : String,
2022-08-20 04:01:43 +00:00
},
/// Mark a task as complete.
Complete {
id_or_name : String,
2022-08-20 04:01:43 +00:00
},
2022-08-20 07:06:55 +00:00
/// Run Git commands at the root of the vault.
#[clap(trailing_var_arg=true)]
Git {
args : Vec<String>,
},
/// Lists tasks according to the specified ordering and filters.
List {
// Need to have options for:
// - column to order by
// - ascending or descending
// - which columns to include
// - filters which exclude values
},
2022-08-21 07:34:44 +00:00
/// For making changes to global configuration.
#[clap(subcommand)]
Config(ConfigCommand),
2022-08-20 04:01:43 +00:00
/// Commands for interacting with vaults.
#[clap(subcommand)]
Vault(VaultCommand),
/// Switches to the specified vault.
Switch {
name : String,
},
2022-08-20 04:01:43 +00:00
}
2022-08-21 07:34:44 +00:00
#[derive(clap::Subcommand, Debug)]
enum ConfigCommand {
/// For checking or changing default text editor command.
Editor {
/// Command to launch editor. Omit to view current editor.
editor : Option<String>,
}
}
2022-08-20 04:01:43 +00:00
#[derive(clap::Subcommand, Debug)]
enum VaultCommand {
/// Creates a new vault at the specified location of the given name.
New {
name : String,
path : path::PathBuf,
},
/// Disconnects the specified vault from toru, without altering the files.
Disconnect {
name : String,
},
/// Connects an existing fault to toru.
Connect {
name : String,
path : path::PathBuf,
},
/// Deletes the specified vault along with all of its data.
Delete {
name : String,
},
/// Lists all configured vaults.
List,
}
fn main() {
let result = program();
match result {
Ok(()) => (),
Err(error::Error::Generic(message)) => {
println!("{} {}", colour::error("Error:"), message);
}
result => println!("{:?}", result),
}
}
fn program() -> Result<(), error::Error> {
let command = {
use clap::Parser;
Args::parse().command
};
let mut config = config::Config::load()?;
use Command::*;
if let Vault(command) = command {
use VaultCommand::*;
match command {
New { name, path } => {
vault::new(name.clone(), path, &mut config)?;
println!("Created vault {}", colour::vault(&name));
},
Disconnect { name } => {
vault::disconnect(&name, &mut config)?;
println!("Disconnected vault {}", colour::vault(&name));
},
Connect { name , path } => {
vault::connect(name.clone(), path, &mut config)?;
println!("Connected vault {}", colour::vault(&name));
},
Delete { name } => {
vault::delete(&name, &mut config)?;
println!("Deleted vault {}", colour::vault(&name));
},
List => {
config.list_vaults()?;
},
}
}
else if let 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);
2022-08-21 07:34:44 +00:00
}
}
}
}
}
else if let Switch { name } = command {
config.switch(&name)?;
println!("Switched to vault {}", colour::vault(&name));
}
else if let Git { args } = command {
let vault_folder = &config.current_vault()?.1;
git::run_command(args, 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 {
New { name, info, tags, dependencies, priority } => {
let task = tasks::Task::new(name, info, tags, dependencies, priority, vault_folder, &mut state)?;
println!("Created task {} (ID: {})", colour::task_name(&task.data.name), colour::id(&task.data.id.to_string()));
},
Delete { id_or_name } => {
let id = state.name_or_id_to_id(&id_or_name)?;
let task = tasks::Task::load(id, vault_folder, false)?;
let name = task.data.name.clone();
state.index_remove(task.data.name.clone(), task.data.id);
task.delete()?;
println!("Deleted task {} (ID: {})", colour::task_name(&name), colour::id(&id.to_string()));
},
View { id_or_name } => {
let id = state.name_or_id_to_id(&id_or_name)?;
let task = tasks::Task::load(id, vault_folder, true)?;
task.display()?;
},
Edit { id_or_name, info } => {
let id = state.name_or_id_to_id(&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 {}", colour::id(&id.to_string()));
},
Discard { id_or_name } => {
let id = state.name_or_id_to_id(&id_or_name)?;
let mut task = tasks::Task::load(id, vault_folder, false)?;
task.data.discarded = true;
task.save()?;
println!("Discarded task {}", colour::id(&id.to_string()));
},
Complete { id_or_name } => {
let id = state.name_or_id_to_id(&id_or_name)?;
let mut task = tasks::Task::load(id, vault_folder, false)?;
task.data.complete = true;
task.save()?;
println!("Marked task {} as complete", colour::id(&id.to_string()));
},
List {} => {
tasks::list(vault_folder)?;
},
// All commands which are dealt with in if let chain at start.
Vault(_) | Config(_) | Git { args : _ } | Switch { name : _ } => unreachable!(),
2022-08-20 04:01:43 +00:00
}
state.save()?;
2022-08-20 04:01:43 +00:00
}
config.save()?;
Ok(())
}