From 308f3a6ece0353980ab8061b410e0d6b8b17adb9 Mon Sep 17 00:00:00 2001
From: Adrien Abraham <aabraham@freebox.fr>
Date: Thu, 18 Aug 2022 11:10:44 +0200
Subject: [PATCH] rofibangs: add option to specify bang

---
 rofibangs.py | 35 +++++++++++++++++------------------
 1 file changed, 17 insertions(+), 18 deletions(-)

diff --git a/rofibangs.py b/rofibangs.py
index 4bd6eee..eb0110e 100755
--- a/rofibangs.py
+++ b/rofibangs.py
@@ -28,9 +28,8 @@ 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")
+        config_path = os.environ.get("ROFIBANGS_CONFIG_PATH") or os.path.expanduser(
+            "~/.config/rofibangs.json"
         )
     try:
         with open(config_path, "rt") as bangs_file:
@@ -52,7 +51,7 @@ def run_rofi(config, input_text="", title="bang"):
         [rofi_path, "-dmenu", "-p", title],
         text=True,
         capture_output=True,
-        input=input_text
+        input=input_text,
     )
     output = completed_process.stdout
     if not output:
@@ -73,12 +72,9 @@ def open_bang(config, handle, query):
 
 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"
-    )
+    ap.add_argument("-c", "--config", help="path to JSON config file")
+    ap.add_argument("-l", "--list", action="store_true", help="show available bangs")
+    ap.add_argument("-b", "--bang", help="launch with this bang already set")
     args = ap.parse_args()
 
     config = load_config()
@@ -89,16 +85,19 @@ def main():
         list_bangs(config)
         return
 
-    process_input = "\n".join(i["handle"] for i in config["bangs"]) + "\n"
-    output = run_rofi(config, input_text=process_input)
-    parts = output.split(maxsplit=1)
-    if len(parts) < 1:
-        exit("Bad Rofi output.")
-    if len(parts) == 1:
-        handle = parts[0]
+    if handle := args.bang:
         query = run_rofi(config, title=handle)
     else:
-        handle, query = parts
+        process_input = "\n".join(i["handle"] for i in config["bangs"]) + "\n"
+        output = run_rofi(config, input_text=process_input)
+        parts = output.split(maxsplit=1)
+        if len(parts) < 1:
+            exit("Bad Rofi output.")
+        if len(parts) == 1:
+            handle = parts[0]
+            query = run_rofi(config, title=handle)
+        else:
+            handle, query = parts
     open_bang(config, handle, query)