91 lines
4.1 KiB
Python
91 lines
4.1 KiB
Python
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))
|