sshwitch 0.0.2

data/bin/sshwitch

@@ -0,0 +1,174 @@
+#!/usr/bin/env ruby
+require 'fileutils'
+require 'optparse'
+
+include FileUtils
+
+HOME_PATH = File.expand_path('~')+'/'
+SSH_PATH = HOME_PATH + '.ssh/'
+SWITCH_FILE = SSH_PATH+'.sshwitch'
+
+def get_current
+  begin
+    File.open(SWITCH_FILE).first
+  rescue
+    File.new(SWITCH_FILE, 'w')
+    nil
+  end
+end
+
+def list
+  begin
+    Dir[SSH_PATH+'*/'].map { |a| File.basename(a) }
+  end
+end
+
+def switch(name)
+  if File.readable?(SSH_PATH+name+"/id_rsa") && File.readable?(SSH_PATH+name+"/id_rsa.pub")
+    begin
+      cp(SSH_PATH+name+"/id_rsa", SSH_PATH+"id_rsa", :verbose => true)
+      cp(SSH_PATH+name+"/id_rsa.pub", SSH_PATH+"id_rsa.pub", :verbose => true)
+      File.open(SWITCH_FILE, 'w') {|f| f.write(name) }
+      puts "Changed key pair to: " + name
+    rescue => e
+      puts "Could not copy, check if you have permission to write on #{SSH_PATH}"
+      puts e.message
+    end
+  else
+    puts "Could not read key pair in #{SSH_PATH+name}"
+    puts "Check if key pair exists in #{SSH_PATH+name}\n\n"
+    puts "If not you can create a new one with: \"sshwitch -n #{name}\""
+    puts "Or backup the current key pair in #{SSH_PATH} with: \"sshwitch -b #{name}\""
+  end
+end
+
+def new(name)
+  if File.directory? SSH_PATH+name
+    puts "#{SSH_PATH+name} already exists, skipping"
+  else
+    puts "Creating key pair in #{SSH_PATH+name}"
+    begin
+      mkdir SSH_PATH+name
+      `ssh-keygen -t rsa -f #{SSH_PATH+name}/id_rsa`
+      rmdir SSH_PATH+name
+    rescue => e
+      puts "Could not create key pair, check if you have permission to write on #{SSH_PATH+name}"
+      puts e.message
+    end
+  end
+end
+
+def backup(name)
+  if File.directory? SSH_PATH+name
+    puts "#{SSH_PATH+name} already exists, skipping"
+  else
+   puts "Copying current key pair in #{SSH_PATH} to #{SSH_PATH+name}"
+   begin
+     mkdir SSH_PATH+name
+     cp(SSH_PATH+"/id_rsa", SSH_PATH+name+"id_rsa", :verbose => true)
+     cp(SSH_PATH+"/id_rsa.pub", SSH_PATH+name+"id_rsa.pub", :verbose => true)
+    rescue => e
+      puts "Could not backup key pair, check if you have permission to write on #{SSH_PATH+name}"
+      puts e.message
+   end
+  end
+end
+
+def rename(oldname, newname)
+  if File.directory? SSH_PATH+newname
+    puts "#{SSH_PATH+newname} already exists, cannot rename"
+  else
+    begin
+      mv SSH_PATH+oldname, SSH_PATH+newname, :verbose => true
+      if get_current == oldname
+        File.open(SWITCH_FILE, 'w') {|f| f.write(newname) }
+        puts "Renamed the current key"
+      end
+    rescue => e
+      puts e.message
+      puts "Could not rename"
+    end
+  end
+end
+
+def delete(name)
+  begin
+    Dir.glob(File.join(SSH_PATH+name, "*")).each {|file| rm file, :verbose => true}
+    rmdir SSH_PATH+name, :verbose => true
+  rescue => e
+    puts e.message
+  end
+end
+
+options = {}
+
+option_parser = OptionParser.new do |opts|
+  exec_name = File.split($0)[1]
+  opts.banner = "Switch and manage key pairs in #{SSH_PATH} \n\n"
+  opts.banner += "Default usage: #{exec_name} key_pair_name \n\n"
+  opts.banner += "Advanced usage: #{exec_name} [option]\n"
+
+  opts.on("-c", "--current", "Get name of current key pair") do
+    options[:current] = true
+  end
+
+  opts.on("-n NEW", "--new", "New key pair name") do |new|
+    options[:new] = new
+  end
+
+  opts.on("-b BACKUP", "--backup", "Backup key pair in #{SSH_PATH}") do |backup|
+    options[:backup] = backup
+  end
+
+  opts.on("-r OLDNAME,NEWNAME", "--rename", Array, "Rename a key pair") do |rename_names|
+    if rename_names.count == 2
+      puts "Rename #{rename_names[0]} to #{rename_names[1]}"
+      options[:oldname] = rename_names[0]
+      options[:newname] = rename_names[1]
+    else
+      puts "Wrong number of arguments for Rename (#{rename_names.count} for 2)"
+    end
+  end
+
+  opts.on("-l", "--list", "Get list of key pairs in #{SSH_PATH} \n\n") do
+    options[:list] = true
+  end
+
+  opts.on("-d DELETE", "--delete", "Delete key pair \n\n") do |delete|
+    options[:delete] = delete
+  end
+end.parse!
+
+#If no options and no name param print current key
+if options.empty? && ARGV.empty?
+  puts get_current
+#If no options and name param use default behavior switch()
+elsif options.empty? && ARGV.count > 0
+  name = ARGV.shift
+  if get_current.nil?
+    puts "WARNING! This will overwrite the key pair in #{SSH_PATH}, press Y to confirm"
+    if (gets.chr.upcase == 'Y')
+      switch(name)
+    end
+  elsif get_current == name
+    puts "#{name} is the current key pair"
+  else
+    switch(name)
+  end
+#If other options...
+elsif options[:current]
+  puts get_current
+elsif options[:list]
+  puts list
+elsif options[:new]
+  new(options[:new])
+elsif options[:backup]
+  backup(options[:backup])
+elsif options[:delete]
+  puts "WARNING! This will delete the directory and all files in #{SSH_PATH+options[:delete]}, press Y to confirm"
+  if (gets.chr.upcase == 'Y')
+    delete(options[:delete])
+  end
+elsif options[:oldname] && options[:newname]
+  rename(options[:oldname], options[:newname])
+end

metadata

@@ -0,0 +1,47 @@
+--- !ruby/object:Gem::Specification
+name: sshwitch
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Agustín Leñero
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-29 00:00:00.000000000 Z
+dependencies: []
+description: Manage different sets of keys in your home folder
+email: agush@roca9.com
+executables:
+- sshwitch
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/sshwitch
+homepage: http://rubygems.org/gems/sshwitch
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Manage your ssh keys
+test_files: []