Initial commit

1 files changed, 30 insertions, 0 deletions

diff --git a/rofi.py b/rofi.py
new file mode 100644
index 0000000..101c1c6
--- /dev/null
+++ b/rofi.py
@@ -0,0 +1,30 @@
+import subprocess
+
+def rofi(prompt, options, rofi_args=[], fuzzy=True):
+    optionstr = '\n'.join(option.replace('\n', ' ') for option in options)
+    args = ['rofi', '-sort', '-no-levenshtein-sort']
+    if fuzzy:
+        args += ['-matching', 'fuzzy']
+    args += ['-dmenu', '-p', prompt, '-format', 's', '-i']
+    args += rofi_args
+    args = [str(arg) for arg in args]
+
+
+    result = subprocess.run(args, input=optionstr, stdout=subprocess.PIPE, universal_newlines=True)
+    returncode = result.returncode
+    stdout = result.stdout.strip()
+
+    selected = stdout.strip()
+    try:
+        index = [opt.strip() for opt in options].index(selected)
+    except ValueError:
+        index = -1
+
+    if returncode == 0:
+        key = 0
+    elif returncode == 1:
+        key = -1
+    elif returncode > 9:
+        key = returncode - 9
+
+    return key, index, selected