steamy 0.0.3 → 0.0.4

data/.gitignore

@@ -2,3 +2,4 @@
 .bundle
 Gemfile.lock
 pkg/*
+vendor

data/bin/steamy

@@ -1,7 +1,7 @@
 #!/usr/bin/env ruby
 #/ Usage: steamy HOST DATABASE
 #/
-require 'steamy'
+require 'boot'
 require 'optparse'
 
 def usage
@@ -20,6 +20,12 @@ optparse = OptionParser.new do |opts|
     exit 0
   end
 
+  opts.on('-d', '--databases HOST') do |host|
+		instance = Steamy::App.new(host)
+		instance.available_databases
+		exit 0
+  end
+
   opts.on('-h', '--help', 'Display help info') do
     usage
     exit 0
@@ -28,7 +34,7 @@ end
 
 begin
   optparse.parse!
-  
+
   # The url has no flag and will be in ARGV[0] if it's set
   options[:host] = ARGV[0] || nil
   options[:database] = ARGV[1] || nil
@@ -42,5 +48,5 @@ rescue OptionParser::InvalidOption, OptionParser::MissingArgument
   exit 1
 end
 
-preview = Steamy::Steamy.new(options[:host], options[:database])
-preview.mysqldump
+instance = Steamy::App.new(options[:host], options[:database])
+instance.dump

data/lib/boot.rb

@@ -0,0 +1,8 @@
+require 'rubygems'
+require 'bundler'
+Bundler.setup
+
+require "steamy"
+require "steamy/version"
+require "steamy/sequel_pro"
+require "steamy/app"

data/lib/steamy/app.rb

@@ -0,0 +1,88 @@
+module Steamy  
+  class App
+    
+    def initialize(host = nil, database = nil)
+        @host = host
+        @database = database
+        
+        @user = ''
+        @password = ''
+      
+        @sequel_pro = SequelPro.new()
+        @connections = @sequel_pro.connections
+    end
+    
+    def dump
+        setup_connection
+        
+        if @database.nil?
+          @database = get_database
+        end
+        
+        timestamp = Time.now.strftime("%Y-%m-%d_%H%M%S")
+        name = "#{@host}_#{@database}_#{timestamp}.sql.gz"
+        remote_exec("'mysqldump #{@user} #{@password} #{@database} | gzip -c' > ~/Desktop/#{name}")
+    end
+    
+    def get_database
+      databases = show_databases.split("\n")
+      
+      puts "Choose a database to dump:"
+      databases.each_with_index do |db, i|
+        puts i.to_s + ': ' + db
+      end
+    
+      # Get index from user
+      index = gets.to_i
+      databases[index]
+    end
+    
+    def setup_connection
+      if @host.nil? || @connections[@host].nil?
+        if @host.nil?
+          puts "no host, available hosts:"
+        else 
+          puts "no host #{@host}, available hosts:"
+        end
+        @sequel_pro.list
+        exit 1
+      end
+      
+      unless @connections[@host]['user'].empty?
+        @user = "-u #{@connections[@host]['user']}"
+      end
+      
+      unless @connections[@host]['password'].empty?
+        @password = "-p\"#{@connections[@host]['password']}\""
+      end
+
+      @database = set_database(@host, @database)
+    end
+    
+    def set_database(host, database)
+      if host.nil? || @connections[@host]['database'].nil? || database
+        database = database
+      else
+        puts "This user can only access '#{@connections[@host]['database']}', update credentials to access other databases or specify database to dump."
+        exit 1
+      end
+      
+      database
+    end
+    
+    def show_databases
+      remote_exec(sprintf('\'mysql -e "show databases" %s %s\'', @user, @password))
+    end
+    
+    def remote_exec(command)
+      command =  "ssh #{@host} #{command}"
+      `#{command}`
+    end
+    
+    def available_databases
+      setup_connection
+      puts show_databases
+    end
+    
+  end
+end

data/lib/steamy/sequel_pro.rb

@@ -1,26 +1,25 @@
-require 'rubygems'
 require 'plist'
-require 'pp'
 
 module Steamy
   class SequelPro
     
     attr_reader :connections
     
-    def initialize(settings)
-      @settings = settings
+    def initialize
       @connections = connections
     end
     
     def connections
       connections = {}
       
-      Dir.chdir(@settings[:saved_connections])
+      Dir.chdir(Steamy.config[:saved_connections])
 
       # For each vhost replace the DOCUMENT_ROOT and write back out to sites-enabled
       Dir.glob("*.spf") do |file|
          connection = Plist::parse_xml(file)
-         connections[connection['data']['connection']['ssh_host']] = connection['data']['connection']
+         unless connection['data']['connection']['ssh_host'].nil?
+           connections[connection['data']['connection']['ssh_host']] = connection['data']['connection']
+         end
       end
       
       connections
@@ -28,10 +27,12 @@ module Steamy
     
     def available_connections
       keys = @connections.keys
-      keys.reject! { |name| name.nil? }
       keys.sort!
-      keys.each do |name|
-        puts name
+    end
+    
+    def list
+      available_connections.each do |connection|
+        puts connection
       end
     end
     

data/lib/steamy/version.rb

@@ -1,3 +1,3 @@
 module Steamy
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/lib/steamy.rb

@@ -1,76 +1,17 @@
-require "steamy/version"
-require "steamy/config"
-require "steamy/sequel_pro"
-
 module Steamy
-  class Steamy
-    
-    def initialize(host = nil, database = nil)
-      
-      @settings = Config.config
-      @settings[:saved_connections] = File.expand_path(@settings[:saved_connections])
-      @host = host
-      
-      @sequel_pro = SequelPro.new(@settings)
-      @connections = @sequel_pro.connections
-
-      if @host.nil? || @connections[@host].nil?
-        puts "no host #{@host}, available hosts:"
-        list
-        exit 1
-      end
-      
-      @ssh      = "ssh #{@host}"
-      @user     = (@connections[@host]['user'].empty?) ? '' :  "-u #{@connections[@host]['user']}"
-      @password = (@connections[@host]['password'].empty?) ? '' :  "-p\"#{@connections[@host]['password']}\""
-      @database = set_database(host, database)
-    end
-    
-    def set_database(host, database)
-      if host.nil? || @connections[@host]['database'].nil? || database
-        database = database
-      else
-        puts "This user can only access '#{@connections[@host]['database']}', update credentials to access other databases or specify database to dump."
-        exit 1
-      end
-      
-      database
-    end
-    
-    def remote_exec(command)
-      command =  "#{@ssh} #{command}"
-      `#{command}`
-    end
-    
-    def show_databases
-      remote_exec(sprintf('\'mysql -e "show databases" %s %s\'', @user, @password))
-    end
-    
-    def list
-      @sequel_pro.available_connections
-    end
-    
-    def get_database
-      databases = show_databases.split("\n")
-      
-      puts "Choose a database to dump:"
-      databases.each_with_index do |db, i|
-        puts i.to_s + ': ' + db
-      end
+  
+  def self.config
+    {
+      :saved_connections => File.expand_path(yaml[:saved_connections])
+    }
+  end
 
-      # Get index from user
-      index = gets.to_i
-      databases[index]
+  def self.yaml
+    if File.exist?("#{ENV['HOME']}/.steamy")
+      @yaml ||= YAML.load_file("#{ENV['HOME']}/.steamy")
+    else
+      {}
     end
-    
-    def mysqldump
-      if @database.nil?
-        @database = get_database
-      end
-      timestamp = Time.now.strftime("%Y-%m-%d_%H%M%S")
-      name = "#{@host}_#{@database}_#{timestamp}.sql.gz"
-      remote_exec("'mysqldump #{@user} #{@password} #{@database} | gzip -c' > ~/Desktop/#{name}")
-    end
-    
   end
+
 end

metadata

@@ -1,88 +1,69 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: steamy
-version: !ruby/object:Gem::Version 
-  hash: 25
+version: !ruby/object:Gem::Version
+  version: 0.0.4
   prerelease: 
-  segments: 
-  - 0
-  - 0
-  - 3
-  version: 0.0.3
 platform: ruby
-authors: 
+authors:
 - Ben Ubois
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2011-09-27 00:00:00 Z
-dependencies: 
-- !ruby/object:Gem::Dependency 
+date: 2012-05-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
   name: plist
-  prerelease: false
-  requirement: &id001 !ruby/object:Gem::Requirement 
+  requirement: &70262401977160 !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id001
+  prerelease: false
+  version_requirements: *70262401977160
 description: steamy exports remote MySQL databases using SequelPro saved connections.
-email: 
+email:
 - ben@benubois.com
-executables: 
+executables:
 - steamy
 extensions: []
-
 extra_rdoc_files: []
-
-files: 
+files:
 - .gitignore
 - Gemfile
 - README.md
 - Rakefile
 - bin/steamy
+- lib/boot.rb
 - lib/steamy.rb
+- lib/steamy/app.rb
 - lib/steamy/config.rb
 - lib/steamy/sequel_pro.rb
 - lib/steamy/version.rb
 - steamy.gemspec
-homepage: ""
+homepage: ''
 licenses: []
-
 post_install_message: 
 rdoc_options: []
-
-require_paths: 
+require_paths:
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
+required_ruby_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      hash: 3
-      segments: 
-      - 0
-      version: "0"
-required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      hash: 3
-      segments: 
-      - 0
-      version: "0"
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
 requirements: []
-
 rubyforge_project: steamy
-rubygems_version: 1.8.6
+rubygems_version: 1.8.11
 signing_key: 
 specification_version: 3
 summary: Export remote MySQL Databases
 test_files: []
-