41f072a8b4
* Add init command This makes setting up the shell part of A'tuin much easier. Eval the output of "atuin init". * Update readme, add up binding
66 lines
1.5 KiB
Rust
66 lines
1.5 KiB
Rust
#![feature(str_split_once)]
|
|
#![feature(proc_macro_hygiene)]
|
|
#![feature(decl_macro)]
|
|
#![warn(clippy::pedantic, clippy::nursery)]
|
|
|
|
use std::path::PathBuf;
|
|
|
|
use directories::ProjectDirs;
|
|
use eyre::{eyre, Result};
|
|
use structopt::StructOpt;
|
|
|
|
#[macro_use]
|
|
extern crate log;
|
|
|
|
#[macro_use]
|
|
extern crate rocket;
|
|
|
|
use command::AtuinCmd;
|
|
use local::database::Sqlite;
|
|
|
|
mod command;
|
|
mod local;
|
|
mod remote;
|
|
|
|
#[derive(StructOpt)]
|
|
#[structopt(
|
|
author = "Ellie Huxtable <e@elm.sh>",
|
|
version = "0.3.2",
|
|
about = "Magical shell history"
|
|
)]
|
|
struct Atuin {
|
|
#[structopt(long, parse(from_os_str), help = "db file path")]
|
|
db: Option<PathBuf>,
|
|
|
|
#[structopt(subcommand)]
|
|
atuin: AtuinCmd,
|
|
}
|
|
|
|
impl Atuin {
|
|
fn run(self) -> Result<()> {
|
|
let db_path = if let Some(db_path) = self.db {
|
|
let path = db_path
|
|
.to_str()
|
|
.ok_or_else(|| eyre!("path {:?} was not valid UTF-8", db_path))?;
|
|
let path = shellexpand::full(path)?;
|
|
PathBuf::from(path.as_ref())
|
|
} else {
|
|
ProjectDirs::from("com", "elliehuxtable", "atuin")
|
|
.ok_or_else(|| {
|
|
eyre!("could not determine db file location\nspecify one using the --db flag")
|
|
})?
|
|
.data_dir()
|
|
.join("history.db")
|
|
};
|
|
|
|
let mut db = Sqlite::new(db_path)?;
|
|
|
|
self.atuin.run(&mut db)
|
|
}
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
pretty_env_logger::init();
|
|
|
|
Atuin::from_args().run()
|
|
}
|