ssh-copy-id.rb 0.1.0

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in ssh-copy-id.rb.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Junegunn Choi
+
+MIT License
+
+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,28 @@
+# ssh-copy-id.rb
+
+A Ruby implementation of `ssh-copy-id` script. (Surprise!)
+Supports copying to/removing from multiple servers.
+
+## Installation
+
+```bash
+gem install ssh-copy-id.rb
+```
+
+## Usage
+
+```
+usage: ssh-copy-id.rb [-i IDENDITY] [-u USER] [-p PASSWORD] [USER[:PASSWORD]@]HOSTNAME[:PORT] ...
+
+    -i, --identity=IDENTITY          Identity file. Default: ~/.ssh/id_rsa.pub
+    -u, --username=USERNAME          Username. Default: current user.
+    -p, --password[=PASSWORD]        Password. Ask password if not given.
+```
+
+## Examples
+
+```ruby
+ssh-copy-id.rb   server1 server2:9022 user@server{2..5} user:password@server{6..10}
+ssh-remove-id.rb server1 server2:9022 user@server{2..5} user:password@server{6..10}
+```
+

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/ssh-copy-id.rb

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'ssh-copy-id'
+require 'optparse'
+
+options = SSHCopyID.parse_options(__FILE__, ARGV)
+SSHCopyID.grant({:hosts => ARGV.uniq, :output => STDOUT}.merge(options))

data/bin/ssh-remove-id.rb

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'ssh-copy-id'
+require 'optparse'
+
+options = SSHCopyID.parse_options(__FILE__, ARGV)
+SSHCopyID.revoke({:hosts => ARGV.uniq, :output => STDOUT}.merge(options))

data/lib/ssh-copy-id.rb

@@ -0,0 +1,111 @@
+require 'rubygems'
+require 'ssh-copy-id/version'
+require 'net/ssh'
+require 'highline'
+
+module SSHCopyID
+  DEFAULT_IDENTITY_FILE = '~/.ssh/id_rsa.pub'
+
+  def self.grant options
+    execute options, true do |identity|
+      %[
+      /bin/bash -cl '
+      umask 077;
+      test -d ~/.ssh || mkdir ~/.ssh;
+      if [ ! -f ~/.ssh/authorized_keys -o `grep "#{identity}" ~/.ssh/authorized_keys 2> /dev/null | wc -l` -eq 0 ]; then
+        echo "#{identity}" >> ~/.ssh/authorized_keys
+        echo "Key added."
+      else
+        echo "Key already exists."
+      fi
+      '
+      ]
+    end
+  end
+
+  def self.revoke options
+    execute options, false do |identity|
+      %[
+      /bin/bash -cl '
+      umask 077;
+      if [ -f ~/.ssh/authorized_keys -a `grep "#{identity}" ~/.ssh/authorized_keys 2> /dev/null | wc -l` -ge 1 ]; then
+        grep -v "#{identity}" ~/.ssh/authorized_keys | cat > ~/.ssh/authorized_keys.ruby-ssh-copy-id
+        mv -f ~/.ssh/authorized_keys.ruby-ssh-copy-id ~/.ssh/authorized_keys
+        echo "Key removed."
+      else
+        echo "Key not found."
+      fi
+      '
+      ]
+    end
+  end
+
+  def self.execute options, ask_password
+    identity_file = options[:identity] || DEFAULT_IDENTITY_FILE
+    identity = File.readlines(File.expand_path identity_file).first.chomp
+    password = options[:password]
+    output = options[:output] || Class.new { def method_missing *args; end }.new
+
+    options[:hosts].each do |host|
+      host, user = host.split('@').reverse
+      user, pass = user.split(':') if user
+      user ||= options[:username] || ENV['USER']
+      output.print "#{user}@#{host}: "
+      if pass.nil? && password.nil? && (ask_password || options.has_key?(:password))
+        password = HighLine.new.ask($/ + "Password: ") { |q| q.echo = '*' }
+      end
+      pass ||= password
+
+      host, port = host.split(':')
+      Net::SSH.start(host, user, :port => port, :password => pass) do |ssh|
+        result = ssh.exec!(yield identity)
+        output.puts result
+      end
+    end
+  rescue Net::SSH::AuthenticationFailed
+    output.puts "Authentication failed."
+    exit 1
+  end
+
+  def self.parse_options filename, argv
+    {}.tap { |options|
+      opts = OptionParser.new { |opts|
+        opts.banner =
+          "usage: #{File.basename filename} [-i IDENDITY] [-u USER] [-p PASSWORD] [USER[:PASSWORD]@]HOSTNAME[:PORT] ..."
+        opts.separator ''
+
+        opts.on('-i', '--identity=IDENTITY', 'Identity file. Default: ~/.ssh/id_rsa.pub') do |v|
+          options[:identity] = v
+        end
+
+        opts.on('-u', '--username=USERNAME', 'Username') do |v|
+          options[:username] = v
+        end
+
+        opts.on('-p', '--password[=PASSWORD]', 'Password. Ask password if not given.') do |v|
+          options[:password] = v
+        end
+
+        opts.separator ''
+        opts.on_tail('-h', '--help', "Show this message") do
+          puts opts
+          exit
+        end
+      }
+      begin
+        opts.parse!(argv)
+      rescue SystemExit => e
+        exit e.status
+      rescue Exception => e
+        puts e.to_s
+        puts opts
+        exit 1
+      end
+
+      if argv.empty?
+        puts opts
+        exit 1
+      end
+    }
+  end
+end

data/lib/ssh-copy-id/version.rb

@@ -0,0 +1,3 @@
+module SSHCopyID
+  VERSION = "0.1.0"
+end

data/ssh-copy-id.rb.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ssh-copy-id/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "ssh-copy-id.rb"
+  gem.version       = SSHCopyID::VERSION
+  gem.authors       = ["Junegunn Choi"]
+  gem.email         = ["junegunn.c@gmail.com"]
+  gem.description   = %q{ssh-copy-id in Ruby. Supports copying to multiple servers.}
+  gem.summary       = %q{ssh-copy-id in Ruby. Supports copying to multiple servers.}
+  gem.homepage      = "https://github.com/junegunn/ssh-copy-id.rb"
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+  gem.add_runtime_dependency 'net-ssh',  '>= 2.6'
+  gem.add_runtime_dependency 'highline', '>= 1.6.15'
+end

data/test/bin_helper.rb

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '../lib')
+load File.join(File.dirname(__FILE__), '../bin/', File.basename(ARGV.shift))

metadata

@@ -0,0 +1,92 @@
+--- !ruby/object:Gem::Specification
+name: ssh-copy-id.rb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Junegunn Choi
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '2.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '2.6'
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.6.15
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.6.15
+description: ssh-copy-id in Ruby. Supports copying to multiple servers.
+email:
+- junegunn.c@gmail.com
+executables:
+- ssh-copy-id.rb
+- ssh-remove-id.rb
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/ssh-copy-id.rb
+- bin/ssh-remove-id.rb
+- lib/ssh-copy-id.rb
+- lib/ssh-copy-id/version.rb
+- ssh-copy-id.rb.gemspec
+- test/bin_helper.rb
+homepage: https://github.com/junegunn/ssh-copy-id.rb
+licenses: []
+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: ssh-copy-id in Ruby. Supports copying to multiple servers.
+test_files:
+- test/bin_helper.rb
+has_rdoc: