mirror of
https://github.com/atuinsh/atuin.git
synced 2025-03-13 14:39:26 +01:00
95cc472037
I'd like to tidy up the root a little, and it's nice to have all the rust crates in one place
101 lines
2.8 KiB
Rust
101 lines
2.8 KiB
Rust
use eyre::Result;
|
|
use serde::Serialize;
|
|
|
|
use atuin_common::shell::{Shell, ShellError};
|
|
|
|
use crate::store::AliasStore;
|
|
|
|
pub mod bash;
|
|
pub mod fish;
|
|
pub mod xonsh;
|
|
pub mod zsh;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Serialize)]
|
|
pub struct Alias {
|
|
pub name: String,
|
|
pub value: String,
|
|
}
|
|
|
|
pub fn parse_alias(line: &str) -> Alias {
|
|
let mut parts = line.split('=');
|
|
|
|
let name = parts.next().unwrap().to_string();
|
|
let remaining = parts.collect::<Vec<&str>>().join("=").to_string();
|
|
|
|
Alias {
|
|
name,
|
|
value: remaining,
|
|
}
|
|
}
|
|
|
|
pub fn existing_aliases(shell: Option<Shell>) -> Result<Vec<Alias>, ShellError> {
|
|
let shell = if let Some(shell) = shell {
|
|
shell
|
|
} else {
|
|
Shell::current()
|
|
};
|
|
|
|
// this only supports posix-y shells atm
|
|
if !shell.is_posixish() {
|
|
return Err(ShellError::NotSupported);
|
|
}
|
|
|
|
// This will return a list of aliases, each on its own line
|
|
// They will be in the form foo=bar
|
|
let aliases = shell.run_interactive(["alias"])?;
|
|
let aliases: Vec<Alias> = aliases.lines().map(parse_alias).collect();
|
|
|
|
Ok(aliases)
|
|
}
|
|
|
|
/// Import aliases from the current shell
|
|
/// This will not import aliases already in the store
|
|
/// Returns aliases that were set
|
|
pub async fn import_aliases(store: AliasStore) -> Result<Vec<Alias>> {
|
|
let shell_aliases = existing_aliases(None)?;
|
|
let store_aliases = store.aliases().await?;
|
|
|
|
let mut res = Vec::new();
|
|
|
|
for alias in shell_aliases {
|
|
// O(n), but n is small, and imports infrequent
|
|
// can always make a map
|
|
if store_aliases.contains(&alias) {
|
|
continue;
|
|
}
|
|
|
|
res.push(alias.clone());
|
|
store.set(&alias.name, &alias.value).await?;
|
|
}
|
|
|
|
Ok(res)
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
#[test]
|
|
fn test_parse_simple_alias() {
|
|
let alias = super::parse_alias("foo=bar");
|
|
assert_eq!(alias.name, "foo");
|
|
assert_eq!(alias.value, "bar");
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_quoted_alias() {
|
|
let alias = super::parse_alias("emacs='TERM=xterm-24bits emacs -nw'");
|
|
assert_eq!(alias.name, "emacs");
|
|
assert_eq!(alias.value, "'TERM=xterm-24bits emacs -nw'");
|
|
|
|
let git_alias = super::parse_alias("gwip='git add -A; git rm $(git ls-files --deleted) 2> /dev/null; git commit --no-verify --no-gpg-sign --message \"--wip-- [skip ci]\"'");
|
|
assert_eq!(git_alias.name, "gwip");
|
|
assert_eq!(git_alias.value, "'git add -A; git rm $(git ls-files --deleted) 2> /dev/null; git commit --no-verify --no-gpg-sign --message \"--wip-- [skip ci]\"'");
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_quoted_alias_equals() {
|
|
let alias = super::parse_alias("emacs='TERM=xterm-24bits emacs -nw --foo=bar'");
|
|
assert_eq!(alias.name, "emacs");
|
|
assert_eq!(alias.value, "'TERM=xterm-24bits emacs -nw --foo=bar'");
|
|
}
|
|
}
|