sql_tasks 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in sql_tasks.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2011 Within3
+
+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/Rakefile

@@ -0,0 +1,4 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+load 'lib/sql_tasks/tasks/sql_tasks.rake'

data/lib/sql_tasks.rb

@@ -0,0 +1,9 @@
+module SqlTasks
+
+  def self.install_tasks
+    load File.join(File.dirname(__FILE__), 'sql_tasks/tasks/sql_tasks.rake')
+  end
+
+  require 'sql_tasks/railtie' if defined?(Rails)
+
+end

data/lib/sql_tasks/railtie.rb

@@ -0,0 +1,9 @@
+module SqlTasks
+  class Railtie < ::Rails::Railtie
+
+    rake_tasks do
+      load 'sql_tasks/tasks/sql_tasks.rake'
+    end
+
+  end
+end

data/lib/sql_tasks/sql_command.rb

@@ -0,0 +1,129 @@
+require 'tempfile'
+require 'shellwords'
+
+class ShellCommand
+
+  def initialize(command, *args)
+    @arguments = args
+    @command = command
+  end
+
+  def to_s
+    command_string = @command
+    command_string << " " + arguments if arguments?
+    command_string
+  end
+
+  def arguments
+    @arguments.map do |options|
+      if Hash === options
+        if options[:password] == nil
+          options[:password] = ''
+        end
+        options.map do |args|
+          key, value = args.first, escaped(args.last)
+          "--%s=%s" % [key, value]
+        end
+      else
+        options
+      end
+    end.join(" ")
+  end
+
+  def execute
+    `#{to_s}`
+  end
+
+  def arguments?
+    @arguments.any?
+  end
+
+  private
+
+  def escaped(val)
+    Shellwords.shellescape(val)
+  end
+
+end
+
+class SqlCommand
+
+  def initialize(path, database_config)
+    @config = database_config
+    @path = path
+  end
+
+  def dump_data
+    command = ShellCommand.new("mysqldump", config_hash, "--skip-dump-date --skip-comments", @config["database"], { :tab => @path })
+  end
+
+  def load_data
+    ShellCommand.new("mysqlimport", config_hash, @config["database"], @path)
+  end
+
+  def import_schema
+    ShellCommand.new("mysql", config_hash, @config["database"], "<#{@path}")
+  end
+
+  def truncate_data
+    file = write_truncate_script
+    if @config["password"] == nil
+      ShellCommand.new(file.path, @config["username"], 'nopwd',  escaped(@config["database"]))
+    else
+      ShellCommand.new(file.path, @config["username"], escaped(@config["password"]), escaped(@config["database"]))
+    end
+  end
+
+  def drop_db
+    ShellCommand.new("mysql", config_hash, '--execute="drop database %s;"' % @config["database"])
+  end
+
+  def create_db
+    ShellCommand.new("mysql", config_hash, '--execute="create database %s;"' % @config["database"])
+  end
+
+  private
+
+  def config_hash
+    { "user" => @config["username"], :password => @config["password"], :host => @config["host"] }
+  end
+
+  def escaped(val)
+    Shellwords.shellescape(val)
+  end
+
+  def write_truncate_script
+    script = <<-EOS
+#!/usr/bin/env bash
+
+user=$1
+PASS=$2
+dbname=$3
+if [ "$PASS" == "nopwd" ]; then
+  basecmd="mysql --user=${user} ${dbname}"
+else
+  basecmd="mysql --user=${user} --password=$PASS ${dbname}"
+fi
+
+tables=$(${basecmd} -e "SHOW TABLES;" | grep -v "+--" | grep -v "Tables_in_${dbname}")
+if [ $? -ne 0 ]; then
+  echo "Unable to retrieve the table names." >&2
+  exit 1
+fi
+
+cmd=""
+
+for table in ${tables}; do
+  cmd="${cmd} TRUNCATE ${table};"
+done
+
+$(${basecmd} -e "${cmd}")
+EOS
+    Tempfile.new("mysqltruncate").tap do |file|
+      file.write(script)
+      file.rewind
+      file.chmod(0777)
+    end
+  end
+
+end

data/lib/sql_tasks/tasks/sql_tasks.rake

@@ -0,0 +1,66 @@
+require File.dirname(__FILE__) + "/../sql_command"
+require 'yaml'
+
+namespace :sql do
+
+  task :prereqs do
+    @rails_env = ENV['RAILS_ENV'] || 'development'
+    @root = Dir.pwd
+    @config = YAML.load_file("#{@root}/config/database.yml")
+    @pathname = File.join(@root, "db", "#{@rails_env}_sql")
+  end
+
+  desc "Dump SQL data and schema to files"
+  task :dump => :prereqs do
+    command = SqlCommand.new(@pathname, @config[@rails_env])
+    command.dump_data.execute
+
+  end
+
+  task :load => :prereqs do
+
+    command = SqlCommand.new(File.join(@pathname, "*.txt"), @config[@rails_env])
+    command.load_data.execute
+
+  end
+
+  desc "Truncates all tables in config/database.yml"
+  task :truncate => :prereqs do
+
+    command = SqlCommand.new("", @config[@rails_env])
+    command.truncate_data.execute
+
+  end
+
+  task :refresh => [:truncate, :load]
+
+  namespace :db do
+
+    task :drop => :prereqs do
+
+      command = SqlCommand.new("", @config[@rails_env])
+      command.drop_db.execute
+
+    end
+
+    task :create => :prereqs do
+
+      command = SqlCommand.new("", @config[@rails_env])
+      command.create_db.execute
+    end
+
+    task :setup => [:drop, :create, "sql:schema:load", "sql:load"]
+
+  end
+
+  namespace :schema do
+
+    task :load => :prereqs do
+      Dir[File.join(@pathname, "*.sql")].each do |sql|
+        SqlCommand.new(sql, @config[@rails_env]).import_schema.execute
+      end
+    end
+
+  end
+
+end

data/lib/sql_tasks/version.rb

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

data/spec/sql_command_spec.rb

@@ -0,0 +1,163 @@
+require 'shellwords'
+require 'sql_tasks/sql_command'
+
+describe ShellCommand do
+
+  it "takes the command" do
+    ShellCommand.new("mysqlimport").tap do |output|
+      output.should have_command("mysqlimport")
+    end
+  end
+
+  it "converts user to arguments" do
+    ShellCommand.new("mysqlimport", :user => "root").tap do |output|
+      output.should have_command "mysqlimport"
+      output.should have_parameter "--user=root"
+    end
+  end
+
+  it "converts user and password to arguments" do
+    ShellCommand.new("mysqlimport", :user => "root", :password => "blah").tap do |output|
+      output.should have_command "mysqlimport"
+      output.should have_parameter "--user=root"
+      output.should have_parameter "--password=blah"
+    end
+  end
+
+  it "does not have the password as a parameter when it's an empty string" do
+    ShellCommand.new("mysqlimport", :user => "root", :password => nil).tap do |output|
+      output.should have_command "mysqlimport"
+      output.should have_parameter "--user=root"
+      output.should have_parameter "--password="
+    end
+  end
+
+  it "takes single word arguments" do
+    ShellCommand.new("mysqlimport", "database_name").tap do |output|
+      output.should have_command "mysqlimport"
+      output.should have_parameter "database_name"
+    end
+  end
+
+  it "takes two single words arguments" do
+    ShellCommand.new("mysqlimport", "database_name", "file_path").tap do |output|
+      output.should have_command "mysqlimport"
+      output.should have_parameter "database_name"
+      output.should have_parameter "file_path"
+    end
+  end
+
+  it "takes a single word argument with the user argument" do
+    ShellCommand.new("mysqlimport", { :user => "root" }, "database_name").tap do |output|
+      output.should have_command "mysqlimport"
+      output.should have_parameter "--user=root"
+      output.should have_parameter "database_name"
+    end
+  end
+
+  it "executes a given command" do
+    command = ShellCommand.new("command")
+    command.should_receive(:`).with("command")
+    command.execute
+  end
+
+  describe SqlCommand do
+
+    it "executes a mysqldump command" do
+      config = {"username" => 'root', "password" => 'blah', "host" => 'hostname', "database" => 'database'}
+
+      SqlCommand.new("path", config).dump_data.tap do |output|
+        output.should have_command "mysqldump"
+        output.should have_parameter "--user=root"
+        output.should have_parameter "--password=blah"
+        output.should have_parameter "--host=hostname"
+        output.should have_parameter "--tab=path"
+        output.should have_parameter "database"
+      end
+
+    end
+
+    it "executes a mysqlimport command" do
+      config = {"username" => 'root', "password" => 'blah', "host" => 'hostname', "database" => 'database'}
+
+      SqlCommand.new("path", config).load_data.tap do |output|
+        output.should have_command "mysqlimport"
+        output.should have_parameter "database"
+        output.should have_parameter "path"
+      end
+    end
+
+    it "sets up the schema" do
+      config = {"username" => 'root', "password" => 'blah', "host" => 'hostname', "database" => 'database'}
+
+      SqlCommand.new("test.sql", config).import_schema.tap do |output|
+        output.should have_command "mysql"
+        output.should have_parameter "database"
+        output.should have_parameter "<test.sql"
+      end
+    end
+
+    it "executes truncate data command" do
+      config = {"username" => 'root', "password" => 'blah', "host" => 'hostname', "database" => 'database'}
+
+      SqlCommand.new("", config).truncate_data.tap do |output|
+        output.should have_command "mysqltruncate"
+        output.should have_parameter "database"
+      end
+    end
+
+    it "creates the truncate data command without the password" do
+      config = {"username" => 'root', "password" => nil, "host" => 'hostname', "database" => 'database'}
+
+      SqlCommand.new("", config).truncate_data.tap do |output|
+        output.should have_command "mysqltruncate"
+        output.should have_parameter "root"
+        output.should have_parameter "nopwd"
+        output.should have_parameter "database"
+        output.should_not have_parameter "hostname"
+      end
+    end
+
+    it "executes drop database command" do
+      config = {"username" => 'root', "password" => 'blah', "host" => 'hostname', "database" => 'database'}
+
+      SqlCommand.new("", config).drop_db.tap do |output|
+        output.should have_command "mysql"
+        output.should have_parameter '--execute=drop database database;'
+      end
+    end
+
+    it "executes create database command" do
+      config = {"username" => 'root', "password" => 'blah', "host" => 'hostname', "database" => 'database'}
+
+      SqlCommand.new("", config).create_db.tap do |output|
+        output.should have_command "mysql"
+        output.should have_parameter '--execute=create database database;'
+      end
+    end
+
+    it "escapes values" do
+      config = {"username" => 'root', "password" => 'pa$$word1', "host" => 'hostname', "database" => 'database'}
+
+      SqlCommand.new("", config).create_db.tap do |output|
+        output.to_s.should =~ %r{pa\\\$\\\$word1}
+      end
+    end
+
+  end
+
+  RSpec::Matchers.define :have_command do |expected|
+    match do |actual|
+      parts = actual.to_s.split(" ")
+      File.basename(parts.first) =~ /^#{expected}/
+    end
+  end
+
+  RSpec::Matchers.define :have_parameter do |expected|
+    match do |actual|
+      parts = Shellwords.shellsplit(actual.to_s)
+      parts.include?(expected)
+    end
+  end
+
+end

data/sql_tasks.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "sql_tasks/version"
+
+Gem::Specification.new do |s|
+  s.name        = "sql_tasks"
+  s.version     = SqlTasks::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Joe Fiorini", "Attila Domokos"]
+  s.email       = ["joe@joefiorini.com", "adomokos@within3.com"]
+  s.homepage    = ""
+  s.summary     = %q{Faster loading and dumping for your MySQL database fixtures.}
+  s.description = %q{Uses built-in mysql commands for dumping/loading of plain text fixtures and schema.}
+
+  s.rubyforge_project = "sql_tasks"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+
+  s.add_development_dependency "rspec"
+
+end

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification 
+name: sql_tasks
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Joe Fiorini
+- Attila Domokos
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-03-11 00:00:00 -05:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+description: Uses built-in mysql commands for dumping/loading of plain text fixtures and schema.
+email: 
+- joe@joefiorini.com
+- adomokos@within3.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- LICENSE
+- Rakefile
+- lib/sql_tasks.rb
+- lib/sql_tasks/railtie.rb
+- lib/sql_tasks/sql_command.rb
+- lib/sql_tasks/tasks/sql_tasks.rake
+- lib/sql_tasks/version.rb
+- spec/sql_command_spec.rb
+- sql_tasks.gemspec
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+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 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: sql_tasks
+rubygems_version: 1.5.0
+signing_key: 
+specification_version: 3
+summary: Faster loading and dumping for your MySQL database fixtures.
+test_files: 
+- spec/sql_command_spec.rb