rofibangs

rofibangs.py

@@ -0,0 +1,98 @@
+#!/usr/bin/env python3
+"""Bangs in DDG style without using DDG but just Rofi.
+
+Load a JSON config file to have the bangs available. Rofi should return, using
+its dmenu option, the handle followed by your query, e.g. "w burger" if you
+want a Wikipedia article for "burger", or at least a much needed
+disambiguation…
+
+Config file example:
+    { "bangs": [ {
+        "handle": "w",
+        "name": "Wikipedia",
+        "url": "https://en.wikipedia.org/wiki/Special:Search?search={}&go=Go"
+    } ] }
+
+By default this scripts attempts to load your config file from
+`~/.config/rofibangs.json`, but you can specify the ROFIBANGS_CONFIG_PATH
+environment variable or pass the path through the -c command-line option.
+"""
+
+import argparse
+import json
+import os
+import subprocess
+import urllib.parse
+import webbrowser
+
+
+def load_config(config_path=None):
+    if config_path is None:
+        config_path = os.environ.get(
+            "ROFIBANGS_CONFIG_PATH",
+            os.path.expanduser("~/.config/rofibangs.json")
+        )
+    try:
+        with open(config_path, "rt") as bangs_file:
+            return json.load(bangs_file)
+    except OSError:
+        return None
+
+
+def list_bangs(config):
+    for item in config["bangs"]:
+        name = item["name"]
+        handle = item["handle"]
+        print(f"- {handle} {name}")
+
+
+def open_bang(config, handle, query):
+    for bang in config["bangs"]:
+        if handle == bang["handle"]:
+            break
+    else:
+        print("Unknown handle.")
+        return
+    url = bang["url"].format(urllib.parse.quote(query))
+    webbrowser.open_new_tab(url)
+
+
+def main():
+    ap = argparse.ArgumentParser()
+    ap.add_argument(
+            "-c", "--config", help="path to JSON config file"
+    )
+    ap.add_argument(
+            "-l", "--list", action="store_true", help="show available bangs"
+    )
+    args = ap.parse_args()
+
+    config = load_config()
+    if config is None:
+        exit("Can't load config file.")
+
+    if args.list:
+        list_bangs(config)
+        return
+
+    rofi_path = config.get("rofi_path", "rofi")
+    process_input = "\n".join(i["handle"] for i in config["bangs"]) + "\n"
+    completed_process = subprocess.run(
+        [rofi_path, "-dmenu", "-p", "bang"],
+        text=True,
+        capture_output=True,
+        input=process_input
+    )
+    output = completed_process.stdout
+    if not output:
+        exit("Empty Rofi output.")
+
+    parts = output.split(maxsplit=1)
+    if len(parts) != 2:
+        exit("Bad Rofi output.")
+
+    open_bang(config, *parts)
+
+
+if __name__ == "__main__":
+    main()