keyble 0.1.0

data/.document

@@ -0,0 +1,5 @@
+lib/**/*.rb
+bin/*
+- 
+features/**/*.feature
+LICENSE.txt

data/Gemfile

@@ -0,0 +1,8 @@
+source :rubygems
+
+gem "net-scp"
+
+group :development do
+  gem "bundler"
+  gem "jeweler"
+end

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2012 Scott Tadman
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,19 @@
+# keyble
+
+SSH keychain management for servers.
+
+## Usage
+
+```
+keyble --import keyfile
+keyble --import keyfile server [server [...]]
+keyble --add user@host server [server [...]]
+keyble --remove user@host server [server [...]]
+keyble --list
+keyble --list server [server [...]]
+```
+
+## Copyright
+
+Copyright (c) 2012 Scott Tadman, The Working Group Inc.
+See LICENSE.txt for further details.

data/Rakefile

@@ -0,0 +1,37 @@
+# encoding: utf-8
+
+require 'rubygems'
+require 'bundler'
+
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+
+require 'rake'
+require 'jeweler'
+
+Jeweler::Tasks.new do |gem|
+  gem.name = "keyble"
+  gem.homepage = "http://github.com/twg/keyble"
+  gem.license = "MIT"
+  gem.summary = %Q{SSH Key Management}
+  gem.description = %Q{Unubtrusive SSH key management}
+  gem.email = "scott@twg.ca"
+  gem.authors = [ "Scott Tadman" ]
+  # Dependencies defined in Gemfile
+end
+Jeweler::RubygemsDotOrgTasks.new
+
+require 'rake/testtask'
+
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+task :default => :test

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/keyble

@@ -0,0 +1,118 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH << File.expand_path('../lib', File.dirname(__FILE__))
+
+require 'keyble'
+require 'optparse'
+require 'net/scp'
+
+method = :list
+key = nil
+
+parser = OptionParser.new do |parser|
+  parser.on("-c", "--cache=s") do |s|
+    Keyble.cache_file_path = s
+  end
+  parser.on("-a", "--add=s") do |s|
+    method = :add
+    key = s
+  end
+  parser.on("-r", "--remove=s") do |s|
+    method = :remove
+    key = s
+  end
+  parser.on("-i", "--import") do |s|
+    method = :import
+  end
+  parser.on("-l", "--list") do
+    method = :list
+  end
+end
+
+args = parser.parse(*ARGV)
+
+case (method)
+when :add
+  if (File.exist?(key))
+    imported = Keyble.keys_import(key)
+
+    if (args.any?)
+      Keyble.keys_add(imported, args)
+    else
+      STDERR.puts "No servers specified."
+      exit(-11)
+    end
+  elsif (args.any?)
+    if (entry = Keyble.cache[key])
+      Keyble.keys_add({ key => entry }, args)
+    else
+      STDERR.puts "Could not locate key for #{key}"
+    end
+  else
+    puts parser
+    exit(-1)
+  end
+when :remove
+  if (File.exist?(key))
+    imported = Keyble.keys_import(key)
+
+    if (args.any?)
+      Keyble.keys_remove(imported, args)
+    else
+      STDERR.puts "No servers specified."
+      exit(-11)
+    end
+  elsif (args.any?)
+    Keyble.keys_remove([ key ], args)
+  else
+    puts parser
+    exit(-1)
+  end
+when :import
+  if (args.any?)
+    imported = { }
+
+    servers = args.reject do |path|
+      if (File.exist?(path))
+        imported.merge!(Keyble.read(path))
+
+        true
+      else
+        false
+      end
+    end
+
+    if (servers.any?)
+      server_keys = Keyble.keys_get(servers)
+      
+      server_keys.each do |server, keys|
+        imported.merge!(keys)
+      end
+    end
+
+    Keyble.cache_merge!(imported)
+    Keyble.cache_save!
+
+    Keyble.keys_display(imported)
+  else
+    STDERR.puts "No servers specified"
+    exit(-11)
+  end
+when :list
+  if (args.any?)
+    Keyble.keys_get(args).each do |server, keys|
+      if (args.length > 1)
+        puts server
+        puts '-' * 78
+      end
+
+      Keyble.keys_display(keys)
+    end
+  else
+    if (Keyble.cache.any?)
+      Keyble.keys_display(Keyble.cache)
+    else
+      puts "No keys in #{Keyble.cache_file_path}"
+    end
+  end
+end

data/keyble.gemspec

@@ -0,0 +1,57 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "keyble"
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Scott Tadman"]
+  s.date = "2013-02-27"
+  s.description = "Unubtrusive SSH key management"
+  s.email = "scott@twg.ca"
+  s.executables = ["keyble"]
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.md"
+  ]
+  s.files = [
+    ".document",
+    "Gemfile",
+    "LICENSE.txt",
+    "README.md",
+    "Rakefile",
+    "VERSION",
+    "bin/keyble",
+    "keyble.gemspec",
+    "lib/keyble.rb",
+    "test/helper.rb",
+    "test/test_keyble.rb"
+  ]
+  s.homepage = "http://github.com/twg/keyble"
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.23"
+  s.summary = "SSH Key Management"
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<net-scp>, [">= 0"])
+      s.add_development_dependency(%q<bundler>, [">= 0"])
+      s.add_development_dependency(%q<jeweler>, [">= 0"])
+    else
+      s.add_dependency(%q<net-scp>, [">= 0"])
+      s.add_dependency(%q<bundler>, [">= 0"])
+      s.add_dependency(%q<jeweler>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<net-scp>, [">= 0"])
+    s.add_dependency(%q<bundler>, [">= 0"])
+    s.add_dependency(%q<jeweler>, [">= 0"])
+  end
+end
+

data/lib/keyble.rb

@@ -0,0 +1,133 @@
+module Keyble
+  def parse(data)
+    result = { }
+
+    data.split(/\r?\n/).collect do |line|
+      type, key, comment = line.split(/\s+/)
+
+      if (comment and !comment.empty? and comment.match(/\S+\@\S+/))
+        result[comment] = line
+      end
+    end
+
+    result
+  end
+
+  def read(path)
+    result = { }
+
+    File.open(path) do |f|
+      result = parse(f.read)
+    end
+
+    result
+  end
+
+  def write(path, data)
+    File.open(path, 'w') do |f|
+      data.each do |key, line|
+        f.puts(line)
+      end
+    end
+  end
+
+  def ssh_authorized_keys_path
+    ".ssh/authorized_keys"
+  end
+
+  def cache_file_path=(value)
+    @cache_file_path = value
+  end
+
+  def cache_file_path
+    @cache_file_path or File.expand_path(".keyble", ENV['HOME'])
+  end
+
+  def cache
+    @cache ||=
+      if (File.exist?(cache_file_path))
+        read(cache_file_path)
+      else
+        { }
+      end
+  end
+
+  def cache_merge!(data)
+    cache.merge!(data)
+
+    cache
+  end
+
+  def cache_save!
+    return unless (@cache)
+
+    write(cache_file_path, @cache)
+  end
+
+  def keys_get(servers)
+    result = { }
+
+    servers.each do |server|
+      Net::SCP.start(server, nil) do |scp|
+        result[server] = parse(scp.download!(ssh_authorized_keys_path))
+      end
+    end
+
+    result
+  end
+
+  def keys_reassign(servers)
+    servers.each do |server|
+      Net::SCP.start(server, nil) do |scp|
+        existing = parse(scp.download!(ssh_authorized_keys_path))
+
+        merged = yield(server, existing.dup)
+
+        if (merged != existing)
+          authorized_keys = StringIO.new(merged.values.join("\n"))
+
+          scp.upload!(authorized_keys, ssh_authorized_keys_path)
+        end
+      end
+    end
+  end
+
+  def keys_add(keys, servers)
+    keys_reassign(servers) do |server, existing_keys|
+      existing_keys.merge(keys)
+    end
+  end
+
+  def keys_remove(keys, servers)
+    if (keys.respond_to?(:keys))
+      keys = keys.keys
+    end
+
+    keys_reassign(servers) do |server, existing_keys|
+      keys.each do |key|
+        existing_keys.delete(key)
+      end
+
+      existing_keys
+    end
+  end
+
+  def keys_import(file)
+    imported = read(file)
+
+    cache_merge!(imported)
+    cache_save!
+
+    imported
+  end
+
+  def keys_display(keys)
+    longest_key = keys.keys.collect { |k| k.to_s.length }.sort[-1]
+
+    keys.sort.each do |key, line|
+      puts "%-#{longest_key}s  %s" % [ key, line.split(/\s+/)[1][-20, 20] ]
+    end
+  end
+
+  extend self
+end

data/test/helper.rb

@@ -0,0 +1,18 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'keyble'
+
+class Test::Unit::TestCase
+end

data/test/test_keyble.rb

@@ -0,0 +1,4 @@
+require File.expand_path('helper', File.dirname(__FILE__))
+
+class TestKeyble < Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification
+name: keyble
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Scott Tadman
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: net-scp
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Unubtrusive SSH key management
+email: scott@twg.ca
+executables:
+- keyble
+extensions: []
+extra_rdoc_files:
+- LICENSE.txt
+- README.md
+files:
+- .document
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- VERSION
+- bin/keyble
+- keyble.gemspec
+- lib/keyble.rb
+- test/helper.rb
+- test/test_keyble.rb
+homepage: http://github.com/twg/keyble
+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.23
+signing_key: 
+specification_version: 3
+summary: SSH Key Management
+test_files: []