mirror of https://tildegit.org/ben/dotfiles
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
|
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()
|