ELY-Browser/crates/ely_app/src/shell/command_actions.rs

use ely_domain::CommandIntent;
use gpui::{Context, Window};

use super::ElyShell;
use ely_browser_core::SpaceImportProfileMapping;

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
enum SpaceFileCommand {
    ExportActiveSpace,
    ImportToActiveProfile,
    ImportPreservingProfiles,
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
enum ShortcutFileCommand {
    Export,
    Import,
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
enum BookmarkFileCommand {
    Export,
    Import,
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
enum LocalDataFileCommand {
    Export,
}

impl ElyShell {
    pub(super) fn handle_shell_command_intent(
        &mut self,
        intent: Option<&CommandIntent>,
        window: &mut Window,
        cx: &mut Context<Self>,
    ) {
        let Some(CommandIntent::Command(command)) = intent else {
            return;
        };

        if install_plugin_from_file_command(command) {
            self.choose_plugin_package(window, cx);
        }
        if download_current_page_command(command) {
            self.download_active_tab(window, cx);
        }

        match space_file_command(command) {
            Some(SpaceFileCommand::ExportActiveSpace) => self.export_active_space(window, cx),
            Some(SpaceFileCommand::ImportToActiveProfile) => {
                self.choose_space_import(SpaceImportProfileMapping::UseActiveProfile, window, cx);
            }
            Some(SpaceFileCommand::ImportPreservingProfiles) => {
                self.choose_space_import(SpaceImportProfileMapping::PreserveExisting, window, cx);
            }
            None => {}
        }

        match shortcut_file_command(command) {
            Some(ShortcutFileCommand::Export) => self.export_shortcuts(window, cx),
            Some(ShortcutFileCommand::Import) => self.choose_shortcut_import(window, cx),
            None => {}
        }

        match bookmark_file_command(command) {
            Some(BookmarkFileCommand::Export) => self.export_bookmarks(window, cx),
            Some(BookmarkFileCommand::Import) => self.choose_bookmark_import(window, cx),
            None => {}
        }

        match local_data_file_command(command) {
            Some(LocalDataFileCommand::Export) => self.export_local_data(window, cx),
            None => {}
        }
    }
}

fn install_plugin_from_file_command(command: &str) -> bool {
    matches!(
        command.trim().to_ascii_lowercase().as_str(),
        "install-plugin-from-file"
            | "install plugin from file"
            | "install-plugin"
            | "install plugin"
    )
}

fn download_current_page_command(command: &str) -> bool {
    matches!(
        command.trim().to_ascii_lowercase().as_str(),
        "download-current-page"
            | "download current page"
            | "save-page"
            | "save page"
            | "save-current-page"
            | "save current page"
    )
}

fn space_file_command(command: &str) -> Option<SpaceFileCommand> {
    match command.trim().to_ascii_lowercase().as_str() {
        "export-space" | "export space" | "export-active-space" | "export active space" => {
            Some(SpaceFileCommand::ExportActiveSpace)
        }
        "import-space"
        | "import space"
        | "import-space-active-profile"
        | "import space active profile" => Some(SpaceFileCommand::ImportToActiveProfile),
        "import-space-with-profiles"
        | "import space with profiles"
        | "import-space-preserve-profiles"
        | "import space preserve profiles" => Some(SpaceFileCommand::ImportPreservingProfiles),
        _ => None,
    }
}

fn shortcut_file_command(command: &str) -> Option<ShortcutFileCommand> {
    match command.trim().to_ascii_lowercase().as_str() {
        "export-shortcuts" | "export shortcuts" | "export-keybindings" | "export keybindings" => {
            Some(ShortcutFileCommand::Export)
        }
        "import-shortcuts" | "import shortcuts" | "import-keybindings" | "import keybindings" => {
            Some(ShortcutFileCommand::Import)
        }
        _ => None,
    }
}

fn bookmark_file_command(command: &str) -> Option<BookmarkFileCommand> {
    match command.trim().to_ascii_lowercase().as_str() {
        "export-bookmarks" | "export bookmarks" => Some(BookmarkFileCommand::Export),
        "import-bookmarks" | "import bookmarks" => Some(BookmarkFileCommand::Import),
        _ => None,
    }
}

fn local_data_file_command(command: &str) -> Option<LocalDataFileCommand> {
    match command.trim().to_ascii_lowercase().as_str() {
        "export-local-data"
        | "export local data"
        | "export-privacy-data"
        | "export privacy data" => Some(LocalDataFileCommand::Export),
        _ => None,
    }
}

#[cfg(test)]
mod tests {
    use super::{
        BookmarkFileCommand, LocalDataFileCommand, ShortcutFileCommand, SpaceFileCommand,
        bookmark_file_command, download_current_page_command, install_plugin_from_file_command,
        local_data_file_command, shortcut_file_command, space_file_command,
    };

    #[test]
    fn install_plugin_from_file_command_matches_prd_aliases() {
        assert!(install_plugin_from_file_command("install-plugin-from-file"));
        assert!(install_plugin_from_file_command("Install Plugin from File"));
        assert!(install_plugin_from_file_command("install plugin"));
    }

    #[test]
    fn install_plugin_from_file_command_rejects_other_plugin_commands() {
        assert!(!install_plugin_from_file_command("plugins"));
        assert!(!install_plugin_from_file_command("open plugins"));
    }

    #[test]
    fn download_current_page_command_matches_save_aliases() {
        assert!(download_current_page_command("download-current-page"));
        assert!(download_current_page_command("save current page"));
        assert!(!download_current_page_command("open downloads"));
    }

    #[test]
    fn space_file_command_matches_export_and_import_aliases() {
        assert_eq!(space_file_command("export-space"), Some(SpaceFileCommand::ExportActiveSpace));
        assert_eq!(
            space_file_command("import space"),
            Some(SpaceFileCommand::ImportToActiveProfile)
        );
        assert_eq!(
            space_file_command("import-space-with-profiles"),
            Some(SpaceFileCommand::ImportPreservingProfiles)
        );
    }

    #[test]
    fn space_file_command_rejects_other_space_commands() {
        assert_eq!(space_file_command("new-space Research"), None);
        assert_eq!(space_file_command("spaces"), None);
    }

    #[test]
    fn shortcut_file_command_matches_export_and_import_aliases() {
        assert_eq!(shortcut_file_command("export-shortcuts"), Some(ShortcutFileCommand::Export));
        assert_eq!(shortcut_file_command("import keybindings"), Some(ShortcutFileCommand::Import));
    }

    #[test]
    fn bookmark_file_command_matches_export_and_import_aliases() {
        assert_eq!(bookmark_file_command("export-bookmarks"), Some(BookmarkFileCommand::Export));
        assert_eq!(bookmark_file_command("import bookmarks"), Some(BookmarkFileCommand::Import));
    }

    #[test]
    fn local_data_file_command_matches_export_aliases() {
        assert_eq!(
            local_data_file_command("export-local-data"),
            Some(LocalDataFileCommand::Export)
        );
        assert_eq!(
            local_data_file_command("export privacy data"),
            Some(LocalDataFileCommand::Export)
        );
    }
}