dotfiles/weechat/.weechat/python/fzf.py

75 lines
2.6 KiB
Python
Raw Normal View History

from typing import Iterator, Tuple
import weechat
SCRIPT_NAME = "fzf"
SCRIPT_AUTHOR = "Trygve Aaberge <trygveaa@gmail.com>"
SCRIPT_VERSION = "0.1.0"
SCRIPT_LICENSE = "MIT"
SCRIPT_DESC = "Switch buffer using fzf (currently only works inside tmux)"
REPO_URL = "https://github.com/trygveaa/weechat-fzf"
def print_error(message: str) -> None:
weechat.prnt("", weechat.prefix("error") + message)
def fzf_process_cb(
data: str, command: str, return_code: int, out: str, err: str
) -> int:
if return_code == weechat.WEECHAT_HOOK_PROCESS_ERROR or return_code == 2 or err:
print_error("Error running fzf (code {}): {}".format(return_code, err))
return weechat.WEECHAT_RC_OK
if out != "":
pointer, _ = out.split("\t", 1)
weechat.buffer_set(pointer, "display", "1")
return weechat.WEECHAT_RC_OK
def fzf_command_cb(data: str, buffer: str, args: str) -> int:
cmd = (
"fzf-tmux -- --delimiter='\t' --with-nth=3.. "
"--preview='tail -$LINES {2} 2>/dev/null'"
)
hook = weechat.hook_process_hashtable(cmd, {"stdin": "1"}, 0, "fzf_process_cb", "")
for buffer_info in buffers():
weechat.hook_set(hook, "stdin", "\t".join(buffer_info) + "\n")
weechat.hook_set(hook, "stdin_close", "")
return weechat.WEECHAT_RC_OK
def buffers() -> Iterator[Tuple[str, str, str, str]]:
logger_filenames = {}
logger_infolist = weechat.infolist_get("logger_buffer", "", "")
while weechat.infolist_next(logger_infolist):
buffer = weechat.infolist_pointer(logger_infolist, "buffer")
filename = weechat.infolist_string(logger_infolist, "log_filename")
logger_filenames[buffer] = filename
weechat.infolist_free(logger_infolist)
buffer_infolist = weechat.infolist_get("buffer", "", "")
while weechat.infolist_next(buffer_infolist):
pointer = weechat.infolist_pointer(buffer_infolist, "pointer")
number = weechat.infolist_integer(buffer_infolist, "number")
name = weechat.infolist_string(buffer_infolist, "name")
yield (pointer, logger_filenames.get(pointer, ""), str(number), name)
weechat.infolist_free(buffer_infolist)
def main() -> None:
if not weechat.register(
SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, SCRIPT_DESC, "", ""
):
return
tmux = weechat.string_eval_expression("${env:TMUX}", {}, {}, {})
if not tmux:
print_error("Error: fzf.py currently only supports being run inside tmux")
return
weechat.hook_command(SCRIPT_NAME, SCRIPT_DESC, "", "", "", "fzf_command_cb", "")
if __name__ == "__main__":
main()