This commit is contained in:
Kirottu
2023-05-27 22:12:50 +03:00
6 changed files with 84 additions and 22 deletions

View File

@@ -94,6 +94,8 @@ Anyrun requires plugins to function, as they provide the results for input. The
- Calculator & unit conversion. - Calculator & unit conversion.
- [Shell](plugins/shell/README.md) - [Shell](plugins/shell/README.md)
- Run shell commands. - Run shell commands.
- [Translate](plugins/translate/README.md)
- Quickly translate text.
- [Kidex](plugins/kidex/README.md) - [Kidex](plugins/kidex/README.md)
- File search provided by [Kidex](https://github.com/Kirottu/kidex). - File search provided by [Kidex](https://github.com/Kirottu/kidex).
- [Randr](plugins/randr/README.md) - [Randr](plugins/randr/README.md)

View File

@@ -29,9 +29,6 @@ in
cargoLock = { cargoLock = {
lockFile = ./Cargo.lock; lockFile = ./Cargo.lock;
outputHashes = {
"kidex-common-0.1.0" = "sha256-sPzCTK0gdIYkKWxrtoPJ/F2zrG2ZKHOSmANW2g00fSQ=";
};
}; };
checkInputs = [cargo rustc]; checkInputs = [cargo rustc];

View File

@@ -6,6 +6,8 @@ Launch applications.
Simply search for the application you wish to launch. Simply search for the application you wish to launch.
*NOTE: The applications plugin does not look for executables in your $PATH, it looks for [desktop entries](https://specifications.freedesktop.org/desktop-entry-spec/desktop-entry-spec-latest.html) in standard locations (`XDG_DATA_DIRS`).*
## Configuration ## Configuration
```ron ```ron

View File

@@ -179,7 +179,11 @@ pub fn scrubber(config: &Config) -> Result<Vec<(DesktopEntry, u64)>, Box<dyn std
Err(_why) => return None, Err(_why) => return None,
}; };
let entries = DesktopEntry::from_dir_entry(&entry, config); let entries = DesktopEntry::from_dir_entry(&entry, config);
Some(entries.into_iter().map(|entry| (entry.name.clone(), entry))) Some(
entries
.into_iter()
.map(|entry| (format!("{}{}", entry.name, entry.icon), entry)),
)
}) })
.flatten() .flatten()
.collect(); .collect();
@@ -195,7 +199,11 @@ pub fn scrubber(config: &Config) -> Result<Vec<(DesktopEntry, u64)>, Box<dyn std
Err(_why) => return None, Err(_why) => return None,
}; };
let entries = DesktopEntry::from_dir_entry(&entry, config); let entries = DesktopEntry::from_dir_entry(&entry, config);
Some(entries.into_iter().map(|entry| (entry.name.clone(), entry))) Some(
entries
.into_iter()
.map(|entry| (format!("{}{}", entry.name, entry.icon), entry)),
)
}) })
.flatten(), .flatten(),
), ),

View File

@@ -4,7 +4,8 @@ Quickly translate text using the Google Translate API.
## Usage ## Usage
Type in `<prefix><target lang> <text to translate>`, where prefix is the configured prefix (default is in [Configuration](#Configuration)) and the rest are pretty obvious. Type in `<prefix><target lang> <text to translate>` or `<prefix><src lang><language_delimiter><target lang> <text to translate>`,
where the `prefix` and `language_delimiter` are config options (defaults are in [Configuration](#Configuration)) and the rest are pretty obvious.
## Configuration ## Configuration
@@ -12,6 +13,7 @@ Type in `<prefix><target lang> <text to translate>`, where prefix is the configu
// <Anyrun config dir>/translate.ron // <Anyrun config dir>/translate.ron
Config( Config(
prefix: ":", prefix: ":",
language_delimiter: ">",
max_entries: 3, max_entries: 3,
) )
``` ```

View File

@@ -3,11 +3,14 @@ use std::fs;
use abi_stable::std_types::{ROption, RString, RVec}; use abi_stable::std_types::{ROption, RString, RVec};
use anyrun_plugin::*; use anyrun_plugin::*;
use fuzzy_matcher::FuzzyMatcher; use fuzzy_matcher::FuzzyMatcher;
use reqwest::Client;
use serde::Deserialize; use serde::Deserialize;
use tokio::runtime::Runtime;
#[derive(Deserialize)] #[derive(Deserialize)]
struct Config { struct Config {
prefix: String, prefix: String,
language_delimiter: String,
max_entries: usize, max_entries: usize,
} }
@@ -15,6 +18,7 @@ impl Default for Config {
fn default() -> Self { fn default() -> Self {
Self { Self {
prefix: ":".to_string(), prefix: ":".to_string(),
language_delimiter: ">".to_string(),
max_entries: 3, max_entries: 3,
} }
} }
@@ -22,6 +26,8 @@ impl Default for Config {
struct State { struct State {
config: Config, config: Config,
client: Client,
runtime: Runtime,
langs: Vec<(&'static str, &'static str)>, langs: Vec<(&'static str, &'static str)>,
} }
@@ -32,6 +38,8 @@ fn init(config_dir: RString) -> State {
Ok(content) => ron::from_str(&content).unwrap_or_default(), Ok(content) => ron::from_str(&content).unwrap_or_default(),
Err(_) => Config::default(), Err(_) => Config::default(),
}, },
client: Client::new(),
runtime: Runtime::new().expect("Failed to create tokio runtime"),
langs: vec![ langs: vec![
("af", "Afrikaans"), ("af", "Afrikaans"),
("sq", "Albanian"), ("sq", "Albanian"),
@@ -151,51 +159,94 @@ fn info() -> PluginInfo {
} }
#[get_matches] #[get_matches]
fn get_matches(input: RString, data: &State) -> RVec<Match> { fn get_matches(input: RString, state: &State) -> RVec<Match> {
if !input.starts_with(&data.config.prefix) { if !input.starts_with(&state.config.prefix) {
return RVec::new(); return RVec::new();
} }
// Ignore the prefix // Ignore the prefix
let input = &input[data.config.prefix.len()..]; let input = &input[state.config.prefix.len()..];
let (lang, text) = match input.split_once(' ') { let (lang_split, text) = match input.split_once(' ') {
Some(split) => split, Some(split) => split,
None => return RVec::new(), None => return RVec::new(),
}; };
let (src, dest) = match lang_split.split_once(&state.config.language_delimiter) {
Some(split) => (Some(split.0), split.1),
None => (None, lang_split),
};
if text.is_empty() { if text.is_empty() {
return RVec::new(); return RVec::new();
} }
let matcher = fuzzy_matcher::skim::SkimMatcherV2::default().ignore_case(); let matcher = fuzzy_matcher::skim::SkimMatcherV2::default().ignore_case();
// Fuzzy match the input language with the languages in the Vec let dest_matches = state
let mut matches = data
.langs .langs
.clone() .clone()
.into_iter() .into_iter()
.filter_map(|(code, name)| { .filter_map(|(code, name)| {
matcher matcher
.fuzzy_match(code, lang) .fuzzy_match(code, dest)
.max(matcher.fuzzy_match(name, lang)) .max(matcher.fuzzy_match(name, dest))
.map(|score| (code, name, score)) .map(|score| (code, name, score))
}) })
.collect::<Vec<_>>(); .collect::<Vec<_>>();
matches.sort_by(|a, b| b.2.cmp(&a.2)); // Fuzzy match the input language with the languages in the Vec
let mut matches = match src {
Some(src) => {
let src_matches = state
.langs
.clone()
.into_iter()
.filter_map(|(code, name)| {
matcher
.fuzzy_match(code, src)
.max(matcher.fuzzy_match(name, src))
.map(|score| (code, name, score))
})
.collect::<Vec<_>>();
let mut matches = src_matches
.into_iter()
.flat_map(|src| dest_matches.clone().into_iter().map(move |dest| (Some(src), dest)))
.collect::<Vec<_>>();
matches.sort_by(|a, b| (b.1 .2 + b.0.unwrap().2).cmp(&(a.1 .2 + a.0.unwrap().2)));
matches
}
None => {
let mut matches = dest_matches
.into_iter()
.map(|dest| (None, dest))
.collect::<Vec<_>>();
matches.sort_by(|a, b| b.1 .2.cmp(&a.1 .2));
matches
}
};
// We only want 3 matches // We only want 3 matches
matches.truncate(data.config.max_entries); matches.truncate(state.config.max_entries);
tokio::runtime::Runtime::new().expect("Failed to spawn tokio runtime!").block_on(async move { state.runtime.block_on(async move {
// Create the futures for fetching the translation results // Create the futures for fetching the translation results
let futures = matches let futures = matches
.into_iter() .into_iter()
.map(|(code, name, _)| async move { .map(|(src, dest)| async move {
(name, reqwest::get(format!("https://translate.googleapis.com/translate_a/single?client=gtx&sl=auto&tl={}&dt=t&q={}", code, text)).await) match src {
Some(src) =>
(dest.1, state.client.get(format!("https://translate.googleapis.com/translate_a/single?client=gtx&sl={}&tl={}&dt=t&q={}", src.0, dest.0, text)).send().await),
None => (dest.1, state.client.get(format!("https://translate.googleapis.com/translate_a/single?client=gtx&sl=auto&tl={}&dt=t&q={}", dest.0, text)).send().await)
}
}); });
futures::future::join_all(futures) // Wait for all futures to complete
.await let res = futures::future::join_all(futures) // Wait for all futures to complete
.await;
res
.into_iter() .into_iter()
.filter_map(|(name, res)| res .filter_map(|(name, res)| res
.ok() .ok()
@@ -215,7 +266,7 @@ fn get_matches(input: RString, data: &State) -> RVec<Match> {
description: ROption::RSome( description: ROption::RSome(
format!( format!(
"{} -> {}", "{} -> {}",
data.langs.iter() state.langs.iter()
.find_map(|(code, name)| if *code == json[2].as_str().expect("Malformed JSON!") { .find_map(|(code, name)| if *code == json[2].as_str().expect("Malformed JSON!") {
Some(*name) Some(*name)
} else { } else {