55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
from contextlib import closing
|
|
|
|
from typing import Dict, List, Optional
|
|
|
|
from prompt_toolkit.application.current import get_app
|
|
from prompt_toolkit import PromptSession
|
|
from prompt_toolkit.completion import WordCompleter
|
|
|
|
import secretstorage
|
|
|
|
from kitty.boss import Boss
|
|
|
|
|
|
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]
|
|
|
|
return None
|
|
|
|
|
|
def expand_prompt() -> None:
|
|
app = get_app()
|
|
buffer = app.current_buffer
|
|
if buffer.complete_state:
|
|
buffer.complete_next()
|
|
else:
|
|
buffer.start_completion(select_first=False)
|
|
|
|
|
|
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)
|