import ranger.api
import ranger.core.fm
import ranger.ext.signals
import subprocess

hook_init_old = ranger.api.hook_init


def hook_init(fm: ranger.core.fm.FM):
    def zoxide_add(signal: ranger.ext.signals.Signal):
        path = signal.new.path
        process = subprocess.Popen(["zoxide", "add", path])
        process.wait()

    fm.signal_bind("cd", zoxide_add)
    return hook_init_old(fm)


ranger.api.hook_init = hook_init


class z(ranger.api.commands.Command):
    def execute(self):
        try:
            output = subprocess.check_output(["zoxide", "query"] + self.args[1:])
            output = output.decode("utf-8")

            query_prefix = "query: "

            if output.startswith(query_prefix):
                directory = output[len(query_prefix) :].strip()
                self.fm.cd(directory)
                self.fm.notify(directory)
            else:
                self.fm.notify("no match found", bad=True)

        except Exception as e:
            self.fm.notify(e, bad=True)