Skip to content

Instantly share code, notes, and snippets.

@an-empty-string
Created July 27, 2023 04:00
Show Gist options
  • Save an-empty-string/dc9ab3604396e375cd82bab06b64f5dc to your computer and use it in GitHub Desktop.
Save an-empty-string/dc9ab3604396e375cd82bab06b64f5dc to your computer and use it in GitHub Desktop.
#!/usr/bin/env python3
import functools
import json
import os
import shlex
import subprocess
import sys
from typing import List
PATH = os.path.expanduser("~/.clipboard")
def run(s: str, inp: str = None) -> str:
return subprocess.check_output(shlex.split(s), text=True, input=inp)
def notify(subj, message):
subprocess.run(["notify-send", subj, message])
def read() -> List[str]:
with open(PATH) as f:
return json.load(f)
def write(xs: List[str]):
with open(PATH, "w") as f:
json.dump(xs, f)
def copy():
xs = read()
xs.insert(0, run("wl-paste").rstrip("\n"))
write(xs)
notify(f"added to buffer, {len(xs)} items now", menuval(xs[0]))
def pop(n: int):
xs = read()
x = xs.pop(n)
run("wl-copy", inp=x)
write(xs)
notify(f"popped from buffer, {len(xs)} items now", menuval(x))
def clear():
write([])
notify("cleared", "all data cleared from buffer")
def pop0():
pop(0)
def paste(n: int):
xs = read()
run("wl-copy", inp=xs[n])
notify(f"pulled from buffer", menuval(xs[n]))
def rotate():
xs = read()
write(xs[1:] + xs[0])
notify(f"rotated buffer, top is now", menuval(read()[0]))
def menuval(x):
x = x.replace("\n", " ")
return x
def dmenu(ispop=False):
entries = read()
ilist = [f"{idx:3} {menuval(val)}" for idx, val in enumerate(entries)]
ilist.extend(["!clear", "!rotate"])
ilist.append("!paste" if ispop else "!pop")
inp = "\n".join(ilist)
prompt = "pop" if ispop else "paste"
result = run(f"rofi -dmenu -p {prompt}", inp=inp)
result = result.split()
if result[0].startswith("!"):
{
"!clear": clear,
"!rotate": rotate,
"!pop": functools.partial(dmenu, ispop=True),
"!paste": dmenu,
}[result[0]]()
else:
[paste, pop][ispop](int(result[0]))
def main():
{"copy": copy, "pop": pop, "dmenu": dmenu}[sys.argv[1]]()
if __name__ == "__main__":
main()
# vim: ft=python
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment