ssh_bro 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1b088e5bb6d1e18b88d40b54a2c5c7cb270e2a7c
+  data.tar.gz: c7e31787f2851853e31b5c98e79d95ab590f1f5d
+SHA512:
+  metadata.gz: 283e4c5c68dc26273b1d405300f6f756a2704bed6dd56d1fb8405f555558604896238d4d84a8b143a8dcaf1fb59eda6a8081bb3ff0d88e43473d745339608cf2
+  data.tar.gz: db7e924e5754cefa2999a406ac111677aa70bb241a12d85a0f9f02a19f41d8f774da062b89e3c3a1894885dd685157dfb4434282b2fa6cb75bf9a9a0f848b5fe

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_bro.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Daniel Leavitt
+
+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 @@
+# SSHBro
+
+Replaces a fella's SSH Config (~/.ssh/config) with some stuff from a Google doc.
+
+This gives you handy bookmarks for all them servers and tab-completion if you're using a decent shell.
+
+## Installation
+
+Like this:
+
+    $ gem install ssh_bro
+
+Ruby 1.9.3+ blah blah.
+
+## Usage
+
+Run it:
+
+  $ sshbro
+
+It will prompt you for things. To update your SSH config file:
+
+  $ sshbro > ~/.ssh/config
+
+## TODO:
+
+- Allow config file templates.
+- Allow per-host templates.

data/Rakefile

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

data/bin/sshbro

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+Signal.trap("INT") { exit 1 }
+
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', 'lib')
+
+require 'ssh_bro/cli'
+
+SSHBro::CLI.new.run

data/lib/ssh_bro.rb

@@ -0,0 +1,94 @@
+require 'csv'
+
+require 'google_doc_seed'
+
+require "ssh_bro/version"
+
+class String
+  def yaml_truthy?
+    ['y', 'yes', 'true'].include?(self.downcase)
+  end
+
+  def yaml_falsey?
+    ['n', 'no', 'false'].include?(self.downcase)
+  end
+
+  def to_yaml_boolean
+    if self.yaml_truthy?
+      return true
+    elsif self.yaml_falsey?
+      return false
+    else
+      self
+    end
+  end
+end
+
+module SSHBro
+  class Retriever
+    def initialize(token)
+      @seeder = GoogleDocSeed.new(token)
+    end
+
+    def retrieve(doc_id)
+      @csv = CSV.parse(@seeder.to_csv_string(doc_id), {
+        headers: true,
+        header_converters: :symbol,
+        converters: [
+          -> (f) { f.respond_to?(:empty?) && f.empty? ? nil : f },
+          -> (f) { f.respond_to?(:gsub) ? f.gsub(/\s+$/, '') : f },
+          -> (f) { f.respond_to?(:to_yaml_boolean) ? f.to_yaml_boolean : f },
+          :all
+        ]
+      }).map(&:to_hash)
+        .reject { |h| h[:aliases].nil? || h[:aliases].empty? || h[:use] != true }
+    end
+
+    def to_ssh_hosts
+      # TODO: validate
+      hosts = @csv.map { |h| h.merge(text: apply_ssh_template(h)) }
+      groups = hosts.group_by { |h| [h[:provider], h[:owner]].join(" - ") }
+
+      groups.map do |group_name, group|
+        header = ('#' * 45) + "\n# #{group_name}\n" + ('#' * 45) + "\n\n"
+        body = group.map { |h| h[:text] }.join("\n\n")
+        header + body
+      end.join("\n\n")
+    end
+
+    def apply_ssh_template(h)
+      lines = [ "Host #{h[:aliases]}",
+                "Hostname #{h[:hostname]}",
+                "User #{h[:user]}",
+                "RemoteForward 52698 localhost:52698" ].join("\n  ")
+
+
+      lines.prepend("# #{h[:comment]}\n\n") if h[:comment]
+
+      lines
+    end
+
+    def to_ansible_hosts
+      hosts = @csv.map { |h| h.merge(text: apply_ansible_template(h)) }
+      groups = hosts.group_by do |h| 
+        "; " + [h[:provider], h[:owner]].join(" - ") + "\n[#{h[:group]}]\n"
+      end
+
+      groups.map do |group_name, group|
+        body = group.map { |h| h[:text] }.join("\n")
+        group_name + body
+      end.join("\n\n")
+    end
+
+    def apply_ansible_template(h)
+      words = [ h[:aliases].split(' ').first,
+                "ansible_ssh_host=#{h[:hostname]}",
+                "ansible_ssh_user=#{h[:user]}" ]
+      words.join(" ")
+    end
+
+    def to_yaml
+      @csv.map { |h| h.reduce({}) { |memo,(k,v)| memo.merge(k.to_s => v) } }.to_yaml
+    end
+  end
+end

data/lib/ssh_bro/cli.rb

@@ -0,0 +1,79 @@
+require 'cgi'
+require 'uri'
+require 'yaml'
+require 'json'
+require 'net/http'
+require 'fileutils'
+
+require 'ssh_bro'
+
+module SSHBro
+  class CLI
+    CLIENT_ID     = '537798680864-nmntkirifcaso227mu8svaonjoo3q5i9.apps.googleusercontent.com'
+    CLIENT_SECRET = '1VSrX4z1f0gM_KR9Y4oQSoke'
+
+    def run
+      creds_path = File.join(File.expand_path('~'), '.ssh', '.sshbro.yml')
+
+      begin
+        creds = YAML.load_file(creds_path)
+        refresh_token = creds['refresh_token']
+        doc_id = creds['doc_id']
+        STDERR.puts "Loading credentials from #{creds_path}"
+      rescue Errno::ENOENT
+      end
+
+      if refresh_token
+        uri = URI('https://accounts.google.com/o/oauth2/token')
+        res = Net::HTTP.post_form(uri, refresh_token:  refresh_token,
+                                       client_id:      CLIENT_ID,
+                                       client_secret:  CLIENT_SECRET,
+                                       grant_type:     'refresh_token')
+
+        access_token = JSON.parse(res.body)['access_token']
+      else
+        token = GoogleDocSeed.get_access_token(CLIENT_ID, CLIENT_SECRET)
+        refresh_token = token['refresh_token']
+        access_token = token['access_token']
+        save = true
+      end
+
+      unless doc_id
+        STDERR.puts 'Please enter the Google Spreadsheet URL: '
+        doc_url = STDIN.gets.chomp
+        doc_id = CGI.parse(URI.parse(doc_url).query)['key'].first
+        save = true
+      end
+
+      if save
+        STDERR.puts "Okay to save your refresh to #{creds_path}? (y\\N)"
+        if gets.chomp.yaml_truthy?
+          File.write(creds_path, {
+            'refresh_token' => refresh_token, 'doc_id' => doc_id
+          }.to_yaml)
+        end
+      end
+
+      retriever = Retriever.new(access_token)
+      retriever.retrieve(doc_id)
+
+      STDERR.puts "Select an output format"
+      STDERR.puts "1: SSH Config"
+      STDERR.puts "2: Ansible Hosts"
+      STDERR.puts "3: YAML"
+
+      hosts_string = case STDIN.gets.chomp
+      when "1" then retriever.to_ssh_hosts
+      when "2" then retriever.to_ansible_hosts
+      when "3" then retriever.to_yaml
+      else raise "Invalid selection"
+      end
+
+      
+
+      STDERR.puts "\n\n"
+
+      STDOUT.puts hosts_string
+    end
+  end
+end

data/lib/ssh_bro/version.rb

@@ -0,0 +1,3 @@
+module SSHBro
+  VERSION = "0.0.1"
+end

data/ssh_bro.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ssh_bro/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "ssh_bro"
+  spec.version       = SSHBro::VERSION
+  spec.authors       = ["Daniel Leavitt"]
+  spec.email         = ["daniel.leavitt@gmail.com"]
+  spec.summary       = %q{Load SSH config from a Google Doc.}
+  spec.description   = %q{Load SSH config from a Google Doc.}
+  spec.homepage      = "https://github.com/dleavitt/ssh_bro"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "rake"
+
+  spec.add_dependency 'launchy'
+  spec.add_dependency 'google-api-client'
+  spec.add_dependency 'google_doc_seed', '~> 0.0.4'
+end

metadata

@@ -0,0 +1,125 @@
+--- !ruby/object:Gem::Specification
+name: ssh_bro
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Daniel Leavitt
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-04-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: launchy
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: google-api-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: google_doc_seed
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.4
+description: Load SSH config from a Google Doc.
+email:
+- daniel.leavitt@gmail.com
+executables:
+- sshbro
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/sshbro
+- lib/ssh_bro.rb
+- lib/ssh_bro/cli.rb
+- lib/ssh_bro/version.rb
+- ssh_bro.gemspec
+homepage: https://github.com/dleavitt/ssh_bro
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Load SSH config from a Google Doc.
+test_files: []