• aardA
    link
    fedilink
    arrow-up
    8
    ·
    11 months ago

    You have a list of systems you’ve connected to in known_hosts, though. And the config file is easy enough to parse - throwing away the stuff you don’t care about - to expand on that list.

      • aardA
        link
        fedilink
        arrow-up
        5
        ·
        11 months ago

        I assume you mean “lookup”, as import doesn’t really make much sense.

        I’m currently using this with wofi, though I’ll eventually rewrite it as anyrun plugin, which provides a bit more control:

        #!/usr/bin/env python3
        from argparse import ArgumentParser
        import subprocess
        import json
        import os
         
        ssh_config_file = "~/.ssh/config"
        ssh_known_hosts_file = "~/.ssh/known_hosts"
         
        # Returns a list of all hosts
        def get_hosts():
         
            hosts = []
         
            with open(os.path.expanduser(ssh_config_file)) as f:
                content = f.readlines()
         
            for line in content:
                line = line.lstrip()
                # Ignore wildcards
                if line.startswith('Host ') and not '*' in line:
                    for host in line.split()[1:]:
                        hosts.append(host)
         
            # Removes duplicate entries
            hosts = sorted(set(hosts))
         
            return hosts
         
        def get_known_hosts():
         
            hosts = []
         
            with open(os.path.expanduser(ssh_known_hosts_file)) as f:
                content = f.readlines()
         
            for line in content:
                line = line.lstrip()
                host_entry = line.partition(" ")[0]
                hosts.append(host_entry.partition(",")[0])
         
            # Removes duplicate entries
            hosts = sorted(set(hosts))
         
            return hosts
         
        # Returns a newline seperated UFT-8 encoded string of all ssh hosts
        def parse_hosts(hosts):
            return "\n".join(hosts).encode("UTF-8")
         
        # Executes wofi with the given input string
        def show_wofi(command, hosts):
         
            process = subprocess.Popen(command,shell=True,stdin=subprocess.PIPE,stdout=subprocess.PIPE)
            ret = process.communicate(input=hosts)
            host, rest = ret
            return host
         
        # Switches the focus to the given id
        def ssh_to_host(host, terminal, ssh_command):
         
            if "]:" in host:
                host, port = host[1:].split("]:")
                command = "{terminal} \'{ssh_command} {host} -p {port}\'".format(terminal=terminal, ssh_command=ssh_command, host=host, port=port)
            else:
                command = "{terminal} \'{ssh_command} {host}\'".format(terminal=terminal, ssh_command=ssh_command, host=host)
         
            process = subprocess.Popen(command,shell=True)
         
        # Entry point
        if __name__ == "__main__":
         
            parser = ArgumentParser(description="Wofi based ssh launcher")
            parser.add_argument("terminal", help='Terminal command to use')
            parser.add_argument("--ssh-command", dest='ssh_command', default='ssh', help='ssh command to use (default=ssh)')
            parser.add_argument("--mode", dest='mode', default='known_hosts', help='where to read from (default=known_hosts)')
            parser.add_argument("--command", default='wofi -p \"SSH hosts: \" -d -i --hide-scroll', help='launcher command to use')
            args = parser.parse_args()
         
            if (args.mode == "config"):
                hosts = get_hosts()
            elif (args.mode == "known_hosts"):
                hosts = get_known_hosts()
         
            parsed_hosts = parse_hosts(hosts)
         
            selected = show_wofi(args.command, parsed_hosts)
         
            selected_host = selected.decode('utf-8').rstrip()
         
            if selected_host != "":
                ssh_to_host(selected_host, args.terminal, args.ssh_command)