feat: extensions base

Cargo.toml

@@ -37,6 +37,7 @@ rusqlite = { version = "0.40.1", features = ["bundled"] }
 serde = { version = "1.0.228", features = ["derive"] }
 serde_json = "1.0.149"
 sha2 = "0.11.0"
+steel-core = { git = "https://github.com/mattwparas/steel.git" }
 tempfile = "3.27.0"
 tokio = { version = "1.48.0", features = ["full"] }
 toml = "0.9.8"

src/app.rs

@@ -1,5 +1,6 @@
 //! Main logic for the app
 use std::collections::HashMap;
+use std::path::PathBuf;
 
 use crate::app::apps::{App, AppCommand, AppIcon, ICNS_ICON};
 use crate::commands::Function;
@@ -139,6 +140,11 @@ pub enum Message {
     KeyPressed(Shortcut),
     FocusTextInput(Move),
     HideWindow(Id),
+    AddExtensionApp {
+        display: String,
+        search: String,
+        icon_path: PathBuf,
+    },
     RunFunction(Function),
     OpenFocused,
     SetConfig(SetConfigFields),

src/app/tile.rs

@@ -8,6 +8,7 @@ use crate::autoupdate::new_version_available;
 use crate::clipboard::ClipBoardContentType;
 use crate::config::{Config, Shelly};
 use crate::debounce::Debouncer;
+use crate::extensions::ExtensionEngine;
 use crate::platform::default_app_paths;
 use crate::platform::macos::events::Event;
 use crate::platform::macos::launching::{EventTapHandle, Shortcut};
@@ -188,6 +189,7 @@ pub struct Tile {
     query_lc: String,
     results: Vec<App>,
     options: AppIndex,
+    extension_apps: AppIndex,
     emoji_apps: AppIndex,
     visible: bool,
     focused: bool,
@@ -206,6 +208,7 @@ pub struct Tile {
     debouncer: Debouncer,
     pub settings_window: Option<window::Id>,
     previous_input_source: Option<String>,
+    extension_engine: Option<ExtensionEngine>,
     conn: Connection,
 }
 

src/app/tile/elm.rs

@@ -21,8 +21,8 @@ use crate::app::pages::settings::settings_page;
 use crate::app::tile::{AppIndex, Hotkeys};
 use crate::app::{DEFAULT_WINDOW_HEIGHT, SettingsTab, ToApp, ToApps};
 use crate::config::Theme;
-use crate::database::load_clipboard;
 use crate::debounce::Debouncer;
+use crate::extensions::ExtensionEngine;
 use crate::platform::macos::events::Event;
 use crate::styles::{
     contents_style, glass_border, glass_surface, results_scrollbar_style, rustcast_text_input_style,
@@ -92,6 +92,7 @@ pub fn new(hotkeys: Hotkeys, config: &Config) -> (Tile, Task<Message>) {
             options,
             hotkeys,
             events,
+            extension_apps: AppIndex::empty(),
             emoji_apps: AppIndex::from_apps(App::emoji_apps()),
             visible: true,
             frontmost: None,
@@ -102,6 +103,7 @@ pub fn new(hotkeys: Hotkeys, config: &Config) -> (Tile, Task<Message>) {
             clipboard_content: vec![],
             tray_icon: None,
             sender: None,
+            extension_engine: None,
             page: Page::Main,
             height: DEFAULT_WINDOW_HEIGHT,
             file_search_sender: None,

src/app/tile/update.rs

@@ -48,6 +48,7 @@ use crate::config::ThemeMode;
 use crate::database::load_clipboard;
 use crate::database::store_clipboard_content;
 use crate::debounce::DebouncePolicy;
+use crate::extensions::ExtensionEngine;
 use crate::platform::macos::events::Event;
 use crate::platform::macos::launching::Shortcut;
 use crate::platform::macos::launching::global_handler;
@@ -208,7 +209,20 @@ pub fn handle_update(tile: &mut Tile, message: Message) -> Task<Message> {
             thread::spawn(move || fs::write(home + "/.config/rustcast/config.toml", confg_str));
             Task::none()
         }
+        Message::AddExtensionApp {
+            display,
+            search,
+            icon_path,
+        } => {
+            dbg!("Adding new extension app");
+            Task::none()
+        }
         Message::SetSender(sender) => {
+            if tile.config.extensions {
+                let eng = ExtensionEngine::new(sender.clone());
+                tile.extension_engine = Some(eng)
+            }
+
             tile.sender = Some(sender.clone());
             match global_handler(sender.clone(), tile.hotkeys.all_hotkeys()) {
                 Ok(a) => tile.hotkeys.handle = Some(a),

src/config.rs

@@ -98,6 +98,7 @@ pub struct Config {
     pub buffer_rules: Buffer,
     pub event_duration: u32,
     pub main_page: MainPage,
+    pub extensions: bool,
     pub start_at_login: bool,
     pub theme: Theme,
     pub window_location: Position,
@@ -128,6 +129,7 @@ impl Default for Config {
             theme: Theme::default(),
             start_at_login: true,
             event_duration: 60,
+            extensions: true,
             placeholder: String::from("Time to be productive!"),
             search_url: "https://duckduckgo.com/search?q=%s".to_string(),
             cbhist: true,

src/extensions.rs

@@ -0,0 +1,120 @@
+use std::{
+    collections::HashMap,
+    fs,
+    path::PathBuf,
+    sync::{Arc, Mutex},
+    time::Duration,
+};
+
+use serde::{Deserialize, Serialize};
+use steel::steel_vm::{builtin::BuiltInModule, engine::Engine, register_fn::RegisterFn};
+
+use crate::app::tile::ExtSender;
+
+#[derive(Deserialize, Serialize)]
+#[serde(rename_all = "lowercase")]
+pub struct ExtensionItem {
+    src_file: PathBuf,
+    name: String,
+    icon_source_file: PathBuf,
+    ext_type: ExtensionType,
+    ext_config: HashMap<String, String>,
+}
+
+#[derive(Deserialize, Serialize)]
+#[serde(rename_all = "lowercase")]
+pub enum ExtensionType {
+    Constant,
+    Polled(Duration),
+    Dynamic,
+}
+
+pub struct Action {
+    run_file: PathBuf,
+}
+
+pub struct ExtensionEngine {
+    engine: Arc<Mutex<Engine>>,
+    extensions: Vec<ExtensionItem>,
+}
+
+impl ExtensionEngine {
+    pub fn new(sender: ExtSender) -> Self {
+        let mut engine = Engine::new();
+        let mut module = BuiltInModule::new("rustcast/core");
+        let sender = sender.clone();
+
+        module.register_fn("cli", extension_prelude::call_command);
+        module.register_fn("fswrite", extension_prelude::create_file);
+        module.register_fn("fsread", extension_prelude::read_file);
+        module.register_fn(
+            "populate",
+            move |name: String, search: String, icon_path: String| {
+                extension_prelude::populate(sender.clone(), name, search, icon_path)
+            },
+        );
+
+        engine.register_module(module);
+
+        ExtensionEngine {
+            engine: Arc::new(Mutex::new(engine)),
+            extensions: vec![],
+        }
+    }
+
+    pub fn load_extension(&mut self, extension_config_path: PathBuf) {
+        let Ok(config_str) = fs::read_to_string(extension_config_path) else {
+            return;
+        };
+
+        if let Some(conf) = toml::from_str(&config_str).ok() {
+            self.extensions.push(conf);
+        }
+    }
+}
+
+mod extension_prelude {
+    use std::path::Path;
+
+    use iced::futures::SinkExt;
+    use log::info;
+
+    use crate::app::tile::ExtSender;
+    pub fn call_command(command: String) -> String {
+        info!("Extension is calling CLI: {command}");
+        std::process::Command::new("sh")
+            .arg("-c")
+            .arg(command)
+            .output()
+            .ok()
+            .and_then(|x| String::from_utf8(x.stdout).ok())
+            .unwrap_or_default()
+    }
+
+    pub fn create_file(path: String, contents: String) {
+        info!("Extension called file write function");
+        std::fs::write(path, contents).ok();
+    }
+
+    pub fn read_file(path: String) -> String {
+        info!("Extension called file write function");
+        std::fs::read_to_string(path).unwrap_or_default()
+    }
+
+    pub fn populate(sender: ExtSender, name: String, search_name: String, icon_path: String) {
+        let sender = sender.clone();
+        tokio::task::spawn_blocking(async move || {
+            let icon_path = icon_path.clone();
+            let mut sender = sender.clone();
+            sender
+                .0
+                .send(crate::app::Message::AddExtensionApp {
+                    display: name,
+                    search: search_name,
+                    icon_path: Path::new(&(icon_path.clone())).to_path_buf(),
+                })
+                .await
+                .unwrap();
+        });
+    }
+}

src/main.rs

@@ -1,12 +1,15 @@
 #![deny(clippy::dbg_macro)]
 
+pub use steel::steel_vm;
+
 mod app;
 mod autoupdate;
 mod clipboard;
 mod commands;
 mod config;
 mod database;
 mod debounce;
+mod extensions;
 mod platform;
 mod quit;
 mod styles;