full rewrite #1

Merged
valerie merged 2 commits from rewrite into main 2023-06-26 18:49:37 -04:00
6 changed files with 202 additions and 222 deletions
Showing only changes of commit 57ebb91cbd - Show all commits

View file

@ -1,13 +1,14 @@
[package]
name = "open"
version = "0.6.0"
version = "1.0.0"
authors = ["Valerie Wolfe <sleeplessval@gmail.com>"]
edition = "2018"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
configparser = "3.0.0"
pico-args = "0.5.0"
toml = "0.7.5"
[profile.release]
strip = "debuginfo"

View file

@ -1,27 +1,5 @@
# open
A Rust reimplementation of `xdg-open` command. It's written to be quickly and easily customizable, features separate local and global configs, and a zero-operand command allowing the user to specify how files should be opened differently, and for opening a project, etc.
An opinionated, friendly alternative to `xdg-open` focused on opening files from a
terminal. Easily understandable and configurable with a simple toml file.
For example, for
```ini
[open]
# zero-operand command
command = atom .
# use $EDITOR to edit files without specified commands?
use_editor = true
[.md]
command = typora
[.rs]
command = atom
[filename:.gitignore]
command = vim
shell = true
```
I can use `open` to open the directory in Atom, or I could use `open src/main.rs` to open `main.rs` in Atom, and I can specify these on a per-project basis.
For directories with a local config, any missing values will be filled in by the global config (`~/.config/open.conf`), which means local configs can be shorter.

View file

@ -4,11 +4,14 @@ use std::{
path::Path
};
use configparser::ini::Ini;
use toml::{
Value,
map::Map
};
pub struct Config {
pub local: Option<Ini>,
pub global: Option<Ini>,
pub local: Option<Map<String, Value>>,
pub global: Option<Map<String, Value>>,
pub local_path: Option<String>,
pub global_path: Option<String>
}
@ -18,14 +21,14 @@ impl Config {
// Instantiate global config
let i_dir_global = String::from(var("HOME").ok().unwrap());
let dir_global = Path::new(i_dir_global.as_str());
let i_path_global = dir_global.join(".config/open.conf");
let i_path_global = dir_global.join(".config/open.toml");
let path_global = i_path_global.as_path();
let mut global: Option<Ini> = None;
let mut global = None;
if path_global.exists() {
let i_global = read_to_string(path_global).unwrap();
let mut tmp = Ini::new();
tmp.read(i_global).ok();
global = Some(tmp);
let raw_conf = read_to_string(path_global).unwrap();
let toml_conf: Value = toml::from_str(raw_conf.as_str()).unwrap();
let toml = toml_conf.as_table().unwrap();
global = Some(toml.to_owned());
}
// Instantiate local config, if it exists.
@ -34,16 +37,16 @@ impl Config {
let mut i_path_local = dir_local.join(".open");
let mut path_local = i_path_local.as_path();
let root = Path::new("/");
let mut local: Option<Ini> = None;
let mut local = None;
loop {
if dir_local == root {
break;
}
if path_local.exists() {
let i_local = read_to_string(path_local).unwrap();
let mut tmp = Ini::new();
tmp.read(i_local).ok();
local = Some(tmp);
let raw_conf = read_to_string(path_local).unwrap();
let toml_conf: Value = toml::from_str(raw_conf.as_str()).unwrap();
let toml = toml_conf.as_table().unwrap();
local = Some(toml.to_owned());
break;
}
dir_local = dir_local.parent().unwrap();
@ -76,55 +79,20 @@ impl Config {
};
return output;
}
pub fn get(&self, section: &str, key: &str) -> Option<String> {
let mut output: Option<String> = None;
pub fn get(&self, key: &str) -> Option<Value> {
let mut output: Option<Value> = None;
if self.local.is_some() {
output = self.local.as_ref().unwrap().get(section, key);
let result = self.local.as_ref().unwrap().get(key);
if result.is_some() {
output = Some(result.unwrap().to_owned());
}
}
if output.is_none() && self.global.is_some() {
output = self.global.as_ref().unwrap().get(section, key);
let result = self.global.as_ref().unwrap().get(key);
if result.is_some() {
output = Some(result.unwrap().to_owned());
}
}
return output;
}
pub fn getbool(&self, section: &str, key: &str) -> Option<bool> {
let mut output = None;
if self.local.is_some() {
let i_out = self.local.as_ref().unwrap().getbool(section, key);
output = i_out.unwrap_or(None);
}
if output.is_none() && self.global.is_some() {
let i_out = self.global.as_ref().unwrap().getbool(section, key);
output = i_out.unwrap_or(None);
}
return output;
}
pub fn add(&mut self, section: &str, key: &str, value: String) {
let mut ini: Ini;
let local = self.local.is_some();
if local {
ini = self.local.clone().unwrap();
} else {
ini = self.global.clone().unwrap();
}
ini.set(section, key, Some(value));
if local {
self.local = Some(ini);
} else {
self.global = Some(ini);
}
}
pub fn add_global(&mut self, section: &str, key: &str, value: String) {
let mut global = self.global.clone().unwrap();
global.set(section, key, Some(value));
self.global = Some(global);
}
pub fn write(&self) -> std::io::Result<()> {
let mut path = self.local_path.as_ref();
if path.is_some() {
let result = self.local.as_ref().unwrap().write(path.unwrap().as_str());
return result;
}
path = self.global_path.as_ref();
self.global.as_ref().unwrap().write(path.unwrap().as_str())
}
}

30
src/error.rs Normal file
View file

@ -0,0 +1,30 @@
use std::{
path::Path,
process::exit
};
pub fn no_configs() {
println!("open: no configurations found");
exit(1);
}
pub fn many_args() {
println!("open: too many arguments supplied");
exit(2);
}
pub fn editor_unset() {
println!("open: $EDITOR is not set");
exit(3);
}
pub fn not_found(path: &Path) {
println!("open: {path:?} does not exist");
exit(4);
}
pub fn no_section(path: &Path) {
println!("open: no appropriate sections for {path:?}");
exit(5);
}

View file

@ -1,157 +1,155 @@
use std::{
env::{args, current_dir, var},
env::current_dir,
path::Path,
process::{Command, exit, Stdio}
process::{ Command, Stdio }
};
use pico_args::Arguments;
use toml::value::{ Array, Value };
mod config;
mod error;
use config::Config;
fn main() {
// Prepare config file
let mut args = Arguments::from_env();
// help flag (-h / --help)
if args.contains(["-h", "--help"]) {
help_text();
return;
}
// version flag (-v / --version)
if args.contains(["-v", "--version"]) {
println!("{}", env!("CARGO_PKG_VERSION"));
return;
}
// prepare configs
let i_dir = current_dir().unwrap();
let dir = i_dir.as_path();
let mut config = Config::new();
let config = Config::new();
// Parse arguments and handle them.
let args: Vec<String> = args().collect();
// path flag (-p / --path)
if args.contains(["-p", "--path"]) {
let local = config.local_path;
let global = config.global_path;
if local.is_some() {
println!("{}", local.unwrap());
return;
}
if global.is_some() {
println!("{}", global.unwrap());
return;
}
error::no_configs();
return;
}
let mut error: Option<String> = None;
let mut file_operand = false;
for arg in &args[1..] {
match arg.as_str() {
"-h" |
"--help" => {
println!("open v0.6.0
Valerie Wolfe <sleeplessval@gmail.com>
A Rust reimplementation of \"xdg-open\" configurable with an ini file.
// get target
let arg_target = args.subcommand().unwrap();
let i_target = arg_target.unwrap_or(String::from("."));
let target = Path::new(&i_target);
if !target.exists() { error::not_found(&target); }
USAGE:
open [FLAGS] [OPERAND]
// get section
// ordering: filename -> type (ext/dir)
let mut section = None;
FLAGS:
-h, --help Prints this help text
-a, --add Add a handler for a operand type
-p, --path Prints the config path used
-v, --version Prints the version number
");
return;
},
"-a" |
"--add" => {
if args.len() < 4 {
println!("open: too few arguments.");
exit(1);
}
let ext = args.get(2).unwrap();
let exe = args.get(3).unwrap();
let tmp = args.get(4);
let shell = tmp.is_some() && tmp.unwrap() == "shell";
println!("{} {} {}", ext, exe, shell);
config.add(ext, "command", exe.to_string());
if shell {
config.add(ext, "shell", "true".to_string());
}
config.write().ok();
return;
},
"-p" |
"--path" => {
let local = config.local_path;
if local.is_some() {
println!("{}", local.unwrap());
} else {
println!("{}", config.global_path.unwrap());
}
return;
},
_ => {
if file_operand {
error = Some("open: too many file operands.".to_string());
} else {
file_operand = true;
}
// by exact filename
let filename = target.file_name();
if filename.is_some() {
let filename_section = config.get(filename.unwrap().to_str().unwrap());
if filename_section.is_some() {
section = filename_section;
}
}
// handle types; dir first
if section.is_none() && target.is_dir() {
let dir_section = config.get("dir");
if dir_section.is_some() {
section = dir_section;
}
}
// handle types; extensions second
if section.is_none() {
let extension = target.extension();
if extension.is_some() {
let extension = extension.unwrap().to_str();
// pull extension array and filter matches
let i_macrosection: Option<Value> = config.get("extension");
let macrosection: Array = i_macrosection.unwrap().as_array().unwrap().to_owned();
let matches = macrosection.iter().filter(|value| {
let table = value.as_table().unwrap();
let i_target = table.get("match").unwrap();
let target = i_target.as_str();
target == extension
}).map(|value| value.to_owned() );
let sections: Vec<Value> = matches.collect();
if sections.len() > 0 {
section = sections.get(0).cloned();
}
}
}
if error.is_some() {
println!("{}", error.unwrap());
exit(1);
}
let default = ".".to_string();
let arg_target = args.get(1);
let i_target = arg_target.unwrap_or(&default);
let target = Path::new(i_target);
if !target.exists() {
println!("open: \"{}\" does not exist.", i_target);
exit(1);
}
let i_ext = target.extension();
let i_filename: String;
let ext: &str;
if target.is_dir() {
if arg_target.is_none() {
ext = "open";
} else {
ext = "dir";
}
} else {
if i_ext.is_none() {
i_filename = ["filename", target.file_name().unwrap().to_str().unwrap()].join(":");
} else {
i_filename = [".", i_ext.unwrap().to_str().unwrap()].join("");
}
ext = i_filename.as_str();
}
let i_exe = config.get(ext, "command");
if i_exe.is_none() {
let use_editor = config.getbool("open", "use_editor");
if use_editor.unwrap_or(false) {
let i_editor = var("EDITOR");
if i_editor.is_err() {
println!("open: encountered an error trying to access $EDITOR");
exit(1);
}
let editor = i_editor.ok().unwrap();
if editor.is_empty() {
println!("open: $EDITOR is not defined.");
exit(1);
}
let mut command = Command::new(editor);
command.args(vec![i_target]);
command.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.stdin(Stdio::inherit());
command.output().ok();
exit(0);
} else {
match ext {
"dir" => println!("open: no command specified for directories."),
_ => println!("open: no command specified for \"{}\" files.", ext)
}
exit(1);
}
}
let exe = i_exe.unwrap();
let mut parts = exe.split(" ");
let mut command = Command::new(parts.next().unwrap());
let mut param: Vec<&str> = vec![];
for part in parts {
param.push(part);
}
param.push(i_target);
command.args(param)
.current_dir(dir);
let is_sh = config.getbool(ext, "shell").unwrap_or(false);
if is_sh {
command.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.stdin(Stdio::inherit());
command.output().ok();
} else {
command.stdout(Stdio::null())
.stderr(Stdio::null());
command.spawn().ok();
// default or fail on missing session
if section.is_none() {
let default = config.get("default");
if default.is_some() { section = default; }
else { error::no_section(target); }
}
// unwrap our section
let properties = section.unwrap();
// collect properties
let command = properties.get("command").unwrap().as_str().unwrap().to_string();
let shell = properties.get("shell").unwrap_or(&Value::Boolean(false)).as_bool().unwrap();
// build child
let mut parts = command.split(" ");
let mut child = Command::new(parts.next().unwrap());
let mut param: Vec<&str> = parts.collect();
param.push(&i_target);
child
.args(param)
.current_dir(dir);
// shell processes inherit stdio (and use `output()`)
if shell {
child
.stdin(Stdio::inherit())
.stdout(Stdio::inherit())
.stderr(Stdio::inherit());
child.output().ok();
}
// non-shell processes should redirect to dev/null (and use `spawn()`)
else {
child
.stdin(Stdio::null())
.stdout(Stdio::null())
.stderr(Stdio::null());
child.spawn().ok();
}
}
pub fn help_text() {
println!("open v{}
Valerie Wolfe <sleeplessval@gmail.com>
A Rust reimagining of \"xdg-open\" configurable with an toml file.
usage: open [flags] <target>
flags:
-h, --help Prints this help text
-p, --path Prints the path to the config file
-v, --version Prints the version number
",
env!("CARGO_PKG_VERSION")
);
}

5
src/util.rs Normal file
View file

@ -0,0 +1,5 @@
use std::{
process::{ Command, Stdio }
};