plugins/ssh.lua

   1 --[[
   2 =pod
   3
   4 =head1 NAME
   5
   6 ssh.lua - SSH menu for known hosts.
   7
   8 =head1 SYNOPSIS
   9
  10 Add something like:
  11
  12   wmii.load_plugin ("ssh")
  13   wmii.add_key_handler ("Mod1-z", ssh.show_menu)
  14
  15 into your wmiirc.lua.
  16
  17 =head1 DESCRIPTION
  18
  19 This reads ~/.ssh/known_hosts in order to display a menu of hosts (and IP
  20 addresses) found in the file.  It assumes 'HashKnownHosts no' is set in
  21 ~/.ssh/config (otherwise it displays the hashed hosts).
  22
  23 =head1 SEE ALSO
  24
  25 L<wmii(1)>, L<lua(1)>
  26
  27 =head1 AUTHOR
  28
  29 David Leadbeater <dgl@dgl.cx>
  30
  31 =head1 LICENCE AND COPYRIGHT
  32
  33 Copyright (c) 2008, David Leadbeater <dgl@dgl.cx>
  34
  35 This is free software.  You may redistribute copies of it under the terms of
  36 the GNU General Public License L<http://www.gnu.org/licenses/gpl.html>.  There
  37 is NO WARRANTY, to the extent permitted by law.
  38
  39 =cut
  40 --]]
  41
  42 local wmii = require("wmii")
  43 local os = require("os")
  44 local io = require("io")
  45 local type = type
  46
  47 module ("ssh")
  48 api_version=0.1
  49
  50 wmii.set_conf ("ssh.askforuser", true);
  51
  52 local hosts
  53 local users
  54
  55 function load_hosts()
  56   hosts = {}
  57
  58   local file = io.open(os.getenv("HOME") .. "/.ssh/known_hosts", "r")
  59   if file then
  60     local line = file:read("*line")
  61
  62     while line do
  63       local host = line:match("([^ ,]+)")
  64       hosts[host] = 1
  65       line = file:read("*line")
  66     end
  67     file:close()
  68   end
  69 end
  70
  71 function load_users()
  72   users = {}
  73
  74   local file = io.open("/etc/passwd", "r")
  75   if file then
  76     local line = file:read("*line")
  77
  78         users[""] = 1
  79     while line do
  80       local user = line:match("([^:]+)")
  81       users[user] = 1
  82       line = file:read("*line")
  83     end
  84     file:close()
  85   end
  86 end
  87
  88 function show_menu()
  89   local str = wmii.menu(hosts, "ssh:")
  90   if type(str) == "string" then
  91     local cmd = wmii.get_conf("xterm") .. " -e /bin/sh -c \"exec ssh "
  92         if wmii.get_conf("ssh.askforuser") then
  93                 local user = wmii.menu(users, "username:")
  94                 if type(user) == "string" and user ~= "" then
  95                         cmd = cmd .. "-l " .. user .. " "
  96                 end
  97         end
  98         cmd = cmd .. str .. "\" &"
  99     os.execute(cmd)
 100   end
 101 end
 102
 103 load_hosts()
 104 load_users()