add new search engine commands

2026-04-09 08:44:56 +02:00
parent ab4ba97886
commit 50dc2bb5e5
2 changed files with 107 additions and 14 deletions
@@ -0,0 +1,90 @@
+import click
+from urllib.parse import quote_plus
+from browser_cli.client import send_command, BrowserNotConnected
+from rich.console import Console
+
+console = Console()
+
+ENGINES = {
+    "google":        "https://www.google.com/search?q={query}",
+    "brave":         "https://search.brave.com/search?q={query}",
+    "duckduckgo":    "https://duckduckgo.com/?q={query}",
+    "ddg":           "https://duckduckgo.com/?q={query}",
+    "youtube":       "https://www.youtube.com/results?search_query={query}",
+    "yt":            "https://www.youtube.com/results?search_query={query}",
+    "spotify":       "https://open.spotify.com/search/{query}",
+    "amazon":        "https://www.amazon.com/s?k={query}",
+    "ecosia":        "https://www.ecosia.org/search?q={query}",
+    "furaffinity":   "https://www.furaffinity.net/search/?q={query}",
+    "fa":            "https://www.furaffinity.net/search/?q={query}",
+    "bing":          "https://www.bing.com/search?q={query}",
+    "github":        "https://github.com/search?q={query}",
+    "wikipedia":     "https://en.wikipedia.org/wiki/Special:Search?search={query}",
+    "wiki":          "https://en.wikipedia.org/wiki/Special:Search?search={query}",
+    "reddit":        "https://www.reddit.com/search/?q={query}",
+    "stackoverflow": "https://stackoverflow.com/search?q={query}",
+    "so":            "https://stackoverflow.com/search?q={query}",
+}
+
+_DISPLAY_NAMES = {
+    "google": "Google", "brave": "Brave Search", "duckduckgo": "DuckDuckGo",
+    "ddg": "DuckDuckGo", "youtube": "YouTube", "yt": "YouTube",
+    "spotify": "Spotify", "amazon": "Amazon", "ecosia": "Ecosia",
+    "furaffinity": "FurAffinity", "fa": "FurAffinity", "bing": "Bing",
+    "github": "GitHub", "wikipedia": "Wikipedia", "wiki": "Wikipedia",
+    "reddit": "Reddit", "stackoverflow": "Stack Overflow", "so": "Stack Overflow",
+}
+
+_SUBCOMMANDS = [
+    ("google",        "Search with Google."),
+    ("brave",         "Search with Brave Search."),
+    ("duckduckgo",    "Search with DuckDuckGo."),
+    ("ddg",           "Search with DuckDuckGo (alias for duckduckgo)."),
+    ("youtube",       "Search YouTube videos."),
+    ("yt",            "Search YouTube (alias for youtube)."),
+    ("spotify",       "Search Spotify."),
+    ("amazon",        "Search Amazon."),
+    ("ecosia",        "Search with Ecosia."),
+    ("furaffinity",   "Search FurAffinity."),
+    ("fa",            "Search FurAffinity (alias for furaffinity)."),
+    ("bing",          "Search with Bing."),
+    ("github",        "Search GitHub."),
+    ("wikipedia",     "Search Wikipedia."),
+    ("wiki",          "Search Wikipedia (alias for wikipedia)."),
+    ("reddit",        "Search Reddit."),
+    ("stackoverflow", "Search Stack Overflow."),
+    ("so",            "Search Stack Overflow (alias for stackoverflow)."),
+]
+
+
+@click.group("search")
+def search_group():
+    """Search the web — open a query in a search engine."""
+
+
+def _build_command(engine_key: str, help_text: str) -> click.Command:
+    @click.command(engine_key, help=help_text)
+    @click.argument("query", nargs=-1, required=True)
+    @click.option("--bg", is_flag=True, help="Open in background (no focus)")
+    @click.option("--window", "window", default=None, help="Open in named window")
+    @click.option("--group", "group", default=None, help="Open in tab group (name or ID)")
+    def _cmd(query, bg, window, group):
+        terms = " ".join(query)
+        url = ENGINES[engine_key].format(query=quote_plus(terms))
+        try:
+            send_command("navigate.open", {"url": url, "background": bg, "window": window, "group": group})
+        except BrowserNotConnected as e:
+            console.print(f"[red]Error:[/red] {e}")
+            raise SystemExit(1)
+        except RuntimeError as e:
+            console.print(f"[red]Browser error:[/red] {e}")
+            raise SystemExit(1)
+        suffix = f" in group '{group}'" if group else (f" in window '{window}'" if window else "")
+        display = _DISPLAY_NAMES.get(engine_key, engine_key.capitalize())
+        console.print(f"[green]Searching[/green] [cyan]{display}[/cyan]: {terms}{suffix}")
+
+    return _cmd
+
+
+for _name, _help in _SUBCOMMANDS:
+    search_group.add_command(_build_command(_name, _help))