db_rocket 0.0.2

data/LICENSE

@@ -0,0 +1,21 @@
+Copyright (c) 2010 Mauro Torres
+
+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.rdoc

@@ -0,0 +1,36 @@
+== DB ROCKET
+
+A simple database agnostic import/export app to transfer data to/from a remote database from Ruby on Rails APP based on TAPS.
+
+== Usage: Server
+
+The first thing that you need is create yaml configuration for add the data for access to the server, so into your app rails run:
+
+#db_rocket create
+
+this going to create yaml called config/db_rocket.yml. Fill this file and run
+
+#db_rocket server:start
+
+and this is all
+
+== Usage: Client
+
+for push your db to the server you can do:
+
+#db_rocket push
+
+or for get the db from your server you can do:
+
+#db_rocket pull
+
+== Options
+
+--environment production - by default db_rocket get from RAILS_ENV
+--tables logs,tags - specify the tables
+--filter '^log_'
+
+=== Copyright
+
+Copyright (c) 2010 Mauro Torres. See LICENSE for details.
+

data/bin/db_rocket

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift(File.dirname(__FILE__) + '/../lib/db_rockets/cli/')
+
+require 'rubygems'
+require 'init'
+
+args = ARGV.dup
+ARGV.clear
+command = args.shift.strip rescue 'help'
+
+if File.exists?('config/environment.rb')
+  DBRocket::Command.run(command, args)
+else
+  puts "app rails not found!, you need stay inside on the root of one rails app"
+end
+

data/lib/db_rocket.rb

@@ -0,0 +1,2 @@
+require "db_rockets/db_rocket"
+

data/lib/db_rockets/cli/command.rb

@@ -0,0 +1,50 @@
+module DBRocket
+  module Command
+    class InvalidCommand < RuntimeError; end
+    class CommandFailed  < RuntimeError; end
+
+    class << self
+      def run(command, args)
+        run_internal(command, args)
+      rescue InvalidCommand
+        display "Unknown command. Run 'db_rocket help' for usage information."
+      end
+
+      def run_internal(command, args)
+        namespace, command = parse(command)
+        require "#{namespace}"
+        klass = DBRocket::Command.const_get(namespace.capitalize).new(args)
+        raise InvalidCommand unless klass.respond_to?(command)
+        klass.send(command)
+      end
+
+      def display(msg)
+        puts(msg)
+      end
+
+      def parse(command)
+        parts = command.split(':')
+        case parts.size
+          when 1
+            if namespaces.include? command
+              return command, 'index'
+            else
+              return 'app', command
+            end
+          when 2
+            raise InvalidCommand unless namespaces.include? parts[0]
+            return parts
+          else
+            raise InvalidCommand
+        end
+      end
+
+      def namespaces
+        @@namespaces ||= Dir["#{File.dirname(__FILE__)}/commands/*"].map do |namespace|
+          namespace.gsub(/.*\//, '').gsub(/\.rb/, '')
+        end
+      end
+    end
+  end
+end
+

data/lib/db_rockets/cli/commands/app.rb

@@ -0,0 +1,36 @@
+require 'yaml'
+require 'logger'
+
+module DBRocket::Command
+  class App < Base
+    def create
+      name = args.shift.downcase.strip rescue nil
+      if make_config_file == "y"
+        display "You can configurate db_rocket on config/db_rocket.yml"
+      end
+    end
+
+    def push
+      load_taps
+      opts = parse_taps_opts
+
+      display("Warning: Data in the app will be overwritten and will not be recoverable.")
+
+      if extract_option("--force") || confirm
+        taps_client(:push, opts)
+      end
+    end
+
+    def pull
+      load_taps
+      opts = parse_taps_opts
+
+      display("Warning: Data in the database '#{opts[:database_url]}' will be overwritten and will not be recoverable.")
+
+      if extract_option("--force") || confirm
+        taps_client(:pull, opts)
+      end
+    end
+  end
+end
+

data/lib/db_rockets/cli/commands/base.rb

@@ -0,0 +1,226 @@
+require 'fileutils'
+
+module DBRocket::Command
+  class Base
+    attr_accessor :args
+    def initialize(args)
+      @args = args
+    end
+
+    def extract_option(options, default=true)
+      values = options.is_a?(Array) ? options : [options]
+      return unless opt_index = args.select { |a| values.include? a }.first
+      opt_position = args.index(opt_index) + 1
+      if args.size > opt_position && opt_value = args[opt_position]
+        if opt_value.include?('--')
+          opt_value = nil
+        else
+          args.delete_at(opt_position)
+        end
+      end
+      opt_value ||= default
+      args.delete(opt_index)
+      block_given? ? yield(opt_value) : opt_value
+    end
+
+    def display(msg, newline=true)
+      if newline
+        puts(msg)
+      else
+        print(msg)
+        STDOUT.flush
+      end
+    end
+
+    def confirm(message="Are you sure you wish to continue? (y/n)?")
+      display("#{message} ", false)
+      ask.downcase == 'y'
+    end
+
+    def error(msg)
+      STDERR.puts(msg)
+      exit 1
+    end
+
+
+    def ask
+      gets.strip
+    end
+
+    def shell(cmd)
+      `#{cmd}`
+    end
+
+    def home_directory
+      running_on_windows? ? ENV['USERPROFILE'] : ENV['HOME']
+    end
+
+    def running_on_windows?
+      RUBY_PLATFORM =~ /mswin32/
+    end
+
+    def config_file
+      'config/db_rocket.yml'
+    end
+
+    def ask_for_config_file
+      if File.exists?(config_file)
+        print "The file config/db_rocket.yml exists, do you want overwrite this? (y/n): "
+        ask
+      else
+        "y"
+      end
+    end
+
+    def parse_database_yml(environment = nil)
+      return "" unless File.exists?(Dir.pwd + '/config/database.yml')
+
+      environment = ENV['RAILS_ENV'] || ENV['MERB_ENV'] || ENV['RACK_ENV'] if environment.nil?
+      environment = 'development' if environment.nil? or environment.empty?
+
+      conf = YAML.load(File.read(Dir.pwd + '/config/database.yml'))[environment]
+      case conf['adapter']
+        when 'sqlite3'
+          return "sqlite://#{conf['database']}"
+        when 'postgresql'
+          uri_hash = conf_to_uri_hash(conf)
+          uri_hash['scheme'] = 'postgres'
+          return uri_hash_to_url(uri_hash)
+        else
+          return uri_hash_to_url(conf_to_uri_hash(conf))
+      end
+    rescue Exception => ex
+      puts "Error parsing database.yml: #{ex.message}"
+      puts ex.backtrace
+      ""
+    end
+
+    def conf_to_uri_hash(conf)
+      uri = {}
+      uri['scheme'] = conf['adapter']
+      uri['username'] = conf['user'] || conf['username']
+      uri['password'] = conf['password']
+      uri['host'] = conf['host'] || conf['hostname']
+      uri['port'] = conf['port']
+      uri['path'] = conf['database']
+
+      conf['encoding'] = 'utf8' if conf['encoding'] == 'unicode' or conf['encoding'].nil?
+      uri['query'] = "encoding=#{conf['encoding']}"
+
+      uri
+    end
+
+    def userinfo_from_uri(uri)
+      username = uri['username'].to_s
+      password = uri['password'].to_s
+      return nil if username == ''
+
+      userinfo  = ""
+      userinfo << username
+      userinfo << ":" << password if password.length > 0
+      userinfo
+    end
+
+    def uri_hash_to_url(uri)
+      uri_parts = {
+        :scheme   => uri['scheme'],
+        :userinfo => userinfo_from_uri(uri),
+        :password => uri['password'],
+        :host     => uri['host'] || '127.0.0.1',
+        :port     => uri['port'],
+        :path     => "/%s" % uri['path'],
+        :query    => uri['query'],
+      }
+
+      URI::Generic.build(uri_parts).to_s
+    end
+
+    def parse_taps_opts
+      opts = {}
+      opts[:default_chunksize] = extract_option("--chunksize") || 1000
+      opts[:default_chunksize] = opts[:default_chunksize].to_i rescue 1000
+      opts[:environment] = extract_option("--environment") || ENV['RAILS_ENV'] || ENV['MERB_ENV'] || ENV['RACK_ENV']  || 'development'
+
+      if filter = extract_option("--filter")
+        opts[:table_filter] = filter
+      elsif tables = extract_option("--tables")
+        r_tables = tables.split(",").collect { |t| "^#{t.strip}$" }
+        opts[:table_filter] = "(#{r_tables.join("|")})"
+      end
+
+      if extract_option("--disable-compression")
+        opts[:disable_compression] = true
+      end
+
+      if resume_file = extract_option("--resume-filename")
+        opts[:resume_filename] = resume_file
+      end
+
+      opts[:indexes_first] = !extract_option("--indexes-last")
+
+      opts[:database_url] = args.shift.strip rescue ''
+      if opts[:database_url] == ''
+        opts[:database_url] = parse_database_yml(opts[:environment])
+        display "Auto-detected local database: #{opts[:database_url]}" if opts[:database_url] != ''
+      end
+      raise(CommandFailed, "Invalid database url") if opts[:database_url] == ''
+
+      if extract_option("--debug")
+        Taps.log.level = Logger::DEBUG
+      end
+
+      #ENV['TZ'] = 'America/Los_Angeles'
+      opts
+    end
+
+    def taps_client(op, opts)
+      Taps::Config.verify_database_url(opts[:database_url])
+      if opts[:resume_filename]
+        Taps::Cli.new([]).clientresumexfer(op, opts)
+      else
+        opts[:remote_url] = make_url_from_config(opts[:environment])
+        Taps::Cli.new([]).clientxfer(op, opts)
+      end
+    end
+
+    def make_url_from_config environment
+      conf = YAML.load(File.read(Dir.pwd + "/#{config_file}"))[environment]
+      "http://#{conf['http_user']}:#{conf['http_password']}@#{conf['server']}:#{conf['port']}"
+    end
+
+    def load_taps
+      require 'taps/operation'
+      require 'taps/cli'
+      error "The db rocket gem requires taps 0.3" unless Taps.version =~ /^0.3/
+      display "Loaded Taps v#{Taps.version}"
+    rescue LoadError
+      message  = "Taps 0.3 Load Error: #{$!.message}\n"
+      message << "You may need to install or update the taps gem to use db commands.\n"
+      message << "On most systems this will be:\n\nsudo gem install taps"
+      error message
+    end
+
+
+    def make_config_file
+      overwrite_or_create_file = ask_for_config_file
+      if overwrite_or_create_file == "y"
+        config_file_hash = <<EOFILE
+common: &common
+  server: 127.0.0.1
+  port: 5555
+  http_user: user
+  http_password: password
+development:
+  <<: *common
+production:
+  <<: *common
+EOFILE
+        File.open(config_file, 'w') do |f|
+          f.puts config_file_hash
+        end
+      end
+      overwrite_or_create_file
+    end
+  end
+end
+

data/lib/db_rockets/cli/commands/help.rb

@@ -0,0 +1,31 @@
+module DBRocket::Command
+  class Help < Base
+    def index
+      display usage
+    end
+
+    def usage
+      usage = <<EOTXT
+=== General Commands
+
+ help                     # show this usage
+ create                   # create config file for your app
+ push
+ pull
+
+=== Server
+
+ server:start             #run db_rocket server
+ server:stop              #stop db_rocket server
+
+=== Example story:
+
+ rails myapp
+ cd myapp
+ (...make edits...)
+ db_rocket create
+EOTXT
+    end
+  end
+end
+

data/lib/db_rockets/cli/commands/server.rb

@@ -0,0 +1,64 @@
+require 'yaml'
+require 'logger'
+
+module DBRocket::Command
+  class Server < Base
+    def start
+      if File.exists?(pid_file)
+        puts "db_rocket server is running...."
+      else
+        load_taps
+        opts = parse_server_taps_opts
+        Taps.log.level = Logger::DEBUG if opts[:debug]
+        Taps::Config.database_url = opts[:database_url]
+        Taps::Config.login = opts[:login]
+        Taps::Config.password = opts[:password]
+
+        Taps::Config.verify_database_url
+        require 'taps/server'
+        pid = fork do
+          Taps::Server.run!({
+            :port => opts[:port],
+            :environment => :production,
+            :logging => true,
+            :dump_errors => true,
+          })
+        end
+        File.open(pid_file, 'w') {|f| f.write(pid) }
+      end
+    end
+
+    def stop
+      if File.exists?(pid_file)
+        process_id = File.open(pid_file,'r').readline
+        Process.kill 9, process_id.to_i
+        FileUtils.rm(pid_file)
+      end
+    end
+
+    def pid_file
+      '/tmp/db_rockets.pid'
+    end
+
+    def parse_server_taps_opts
+      opts = {}
+      opts[:environment] = extract_option("--environment") || ENV['RAILS_ENV'] || ENV['MERB_ENV'] || ENV['RACK_ENV']  || 'development'
+      opts[:database_url] = args.shift.strip rescue ''
+      if opts[:database_url] == ''
+        opts[:database_url] = parse_database_yml(opts[:environment])
+        display "Auto-detected local database: #{opts[:database_url]}" if opts[:database_url] != ''
+      end
+      raise(CommandFailed, "Invalid database url") if opts[:database_url] == ''
+
+      if extract_option("--debug")
+        opts[:debug] = true
+      end
+      conf = YAML.load(File.read(Dir.pwd + "/#{config_file}"))[opts[:environment]]
+      opts[:login]    = conf['http_user']
+      opts[:password] = conf['http_password']
+      opts[:port]     = conf['port']
+      opts
+    end
+  end
+end
+

data/lib/db_rockets/cli/init.rb

@@ -0,0 +1,8 @@
+module DBRocket; end
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.dirname(__FILE__) + '/commands')
+
+require "command"
+require "base"
+

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification 
+name: db_rocket
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 2
+  version: 0.0.2
+platform: ruby
+authors: 
+- Mauro Torres
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-10-09 00:00:00 -03:00
+default_executable: db_rocket
+dependencies: []
+
+description: Db Rocketa allows make your dump db very easy!
+email: maurotorres@gmail.com
+executables: 
+- db_rocket
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- lib/db_rocket.rb
+- lib/db_rockets/cli/command.rb
+- lib/db_rockets/cli/commands/app.rb
+- lib/db_rockets/cli/commands/base.rb
+- lib/db_rockets/cli/commands/help.rb
+- lib/db_rockets/cli/commands/server.rb
+- lib/db_rockets/cli/init.rb
+- LICENSE
+- README.rdoc
+has_rdoc: true
+homepage: http://github.com/chebyte/db_rocket
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: make your dump db very easy!
+test_files: []
+