#!/usr/bin/env ruby
 
class Ssh < Thor
  require 'net/sftp'
  require 'highline/import'
  
  desc "install_key <hostname>", "installs your public key on the remote host."
  method_options :username => :string, :password => :string, :key => :string
  def install_key(host)
    username = options['username'] || ask("Enter your username: ") { |q| q.default = ENV['USER'] }
    password = options['password'] || ask("Enter your password: ") { |q| q.echo = false }
    key = options['key'] || ask("Enter your key file location: ") { |q| q.default = '~/.ssh/id_rsa.pub' }
    key = File.expand_path(key)
    
    Net::SFTP.start(host, username, :password => password) do |sftp|
      sftp.mkdir!('.ssh', :permissions => 0700) rescue true
      current_keys = sftp.download!(".ssh/authorized_keys") rescue ''
      new_key = File.read(key) rescue ''
      
      unless current_keys.index(new_key)      
        sftp.file.open(".ssh/authorized_keys", "w", 0600) do |f|
          f.puts current_keys unless current_keys.empty?
          f.puts new_key unless new_key.empty?
        end
      end
    end
  end
end