feat: websearch plugin (#74)

plugins/websearch/Cargo.toml

@@ -0,0 +1,16 @@
+[package]
+name = "websearch"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+crate-type = ["cdylib"]
+
+[dependencies]
+anyrun-plugin = { git = "https://github.com/Kirottu/anyrun" }
+abi_stable = "0.11.1"
+urlencoding = "2.1.3"
+ron = "0.8.0"
+serde = { version = "1.0.152", features = ["derive"] }
+strum = "0.25.0"
+strum_macros = "0.25.1"

plugins/websearch/src/lib.rs

@@ -0,0 +1,115 @@
+use abi_stable::std_types::{ROption, RString, RVec};
+use anyrun_plugin::*;
+use serde::Deserialize;
+use std::{fmt, fs, process::Command};
+use strum::IntoEnumIterator;
+use strum_macros::EnumIter;
+use urlencoding::encode;
+
+#[derive(Debug, Clone, Copy, Deserialize, EnumIter)]
+enum Engine {
+    Google,
+    Ecosia,
+    Bing,
+    DuckDuckGo,
+}
+
+impl Engine {
+    fn value(&self) -> &str {
+        match *self {
+            Engine::Google => "google.com/search?q=",
+            Engine::Ecosia => "www.ecosia.org/search?q=",
+            Engine::Bing => "www.bing.com/search?q=",
+            Engine::DuckDuckGo => "duckduckgo.com/?q=",
+        }
+    }
+}
+
+impl fmt::Display for Engine {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match self {
+            Engine::Google => write!(f, "Google"),
+            Engine::Ecosia => write!(f, "Ecosia"),
+            Engine::Bing => write!(f, "Bing"),
+            Engine::DuckDuckGo => write!(f, "DuckDuckGo"),
+        }
+    }
+}
+
+#[derive(Deserialize)]
+struct Config {
+    prefix: String,
+    engines: Vec<Engine>,
+}
+
+impl Default for Config {
+    fn default() -> Self {
+        Config {
+            prefix: "?".to_string(),
+            engines: vec![Engine::Google],
+        }
+    }
+}
+
+#[init]
+fn init(config_dir: RString) -> Config {
+    match fs::read_to_string(format!("{}/websearch.ron", config_dir)) {
+        Ok(content) => ron::from_str(&content).unwrap_or_default(),
+        Err(_) => Config::default(),
+    }
+}
+
+#[info]
+fn info() -> PluginInfo {
+    PluginInfo {
+        name: "Websearch".into(),
+        icon: "help-about".into(),
+    }
+}
+
+#[get_matches]
+fn get_matches(input: RString, config: &Config) -> RVec<Match> {
+    if !input.starts_with(&config.prefix) {
+        RVec::new()
+    } else {
+        config
+            .engines
+            .clone()
+            .into_iter()
+            .map(|engine| Match {
+                title: input.trim_start_matches(&config.prefix).into(),
+                description: ROption::RSome(format!("Search with {}", engine.to_string()).into()),
+                use_pango: false,
+                icon: ROption::RNone,
+                id: ROption::RNone,
+            })
+            .collect()
+    }
+}
+
+#[handler]
+fn handler(selection: Match) -> HandleResult {
+    for engine in Engine::iter() {
+        if selection
+            .description
+            .clone()
+            .unwrap()
+            .to_string()
+            .contains(&engine.to_string())
+        {
+            if let Err(why) = Command::new("sh")
+                .arg("-c")
+                .arg(format!(
+                    "xdg-open https://{}{}",
+                    engine.value(),
+                    encode(&selection.title.to_string())
+                ))
+                .spawn()
+            {
+                println!("Failed to perform websearch: {}", why);
+            }
+        }
+    }
+
+    HandleResult::Close
+}