1
0
mirror of https://github.com/dcarrillo/dotfiles.git synced 2024-11-20 15:31:12 +00:00
dotfiles/.config/kitty/secrets.py

61 lines
1.9 KiB
Python
Raw Permalink Normal View History

2021-12-05 18:59:14 +00:00
from contextlib import closing
2021-12-06 11:21:18 +00:00
from typing import Dict, List, Optional
2021-12-05 18:59:14 +00:00
2021-12-06 11:21:18 +00:00
from prompt_toolkit.application.current import get_app
from prompt_toolkit import PromptSession
from prompt_toolkit.completion import WordCompleter
2021-12-05 18:59:14 +00:00
import secretstorage
from kitty.boss import Boss
2023-09-30 15:35:44 +00:00
#
# To add passwords to the Secret Service that can be retrieved for this kitten:
# secret-tool store --label user1 description "Kitty password" id $(uuidgen)
# secret-tool store --label user2 description "Kitty password" id $(uuidgen)
# Add map to kitty.conf:
# map ctrl+alt+p kitten secrets.py "description" "Kitty password"
2021-12-05 18:59:14 +00:00
2021-12-06 11:21:18 +00:00
def main(args: List[str]) -> Optional[str]:
secrets = get_secret_names(args[1], args[2])
entries = WordCompleter(secrets)
session = PromptSession(completer=entries)
try:
entry = session.prompt('> ', pre_run=expand_prompt)
except (KeyboardInterrupt, EOFError):
pass
else:
return secrets[entry]
2021-12-05 18:59:14 +00:00
2021-12-06 11:21:18 +00:00
return None
2021-12-05 18:59:14 +00:00
2021-12-06 11:21:18 +00:00
def expand_prompt() -> None:
app = get_app()
buffer = app.current_buffer
if buffer.complete_state:
buffer.complete_next()
else:
buffer.start_completion(select_first=True)
2021-12-05 18:59:14 +00:00
def get_secret_names(attribute: str, value: str) -> Dict[str, str]:
secrets = {}
with closing(secretstorage.dbus_init()) as bus:
for keyring in secretstorage.get_all_collections(bus):
for item in keyring.get_all_items():
if item.is_locked():
item.unlock()
attr = item.get_attributes()
if attr.get(attribute) == value:
secrets[item.get_label()] = item.get_secret().decode('utf-8')
return secrets
def handle_result(args: List[str], answer: str, target_window_id: int, boss: Boss) -> None:
window = boss.window_id_map.get(target_window_id)
if window is not None:
window.paste(answer)