mysql_rake_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 mysql_rake_tasks.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2011 Gregory Ostermayr
+
+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,43 @@
+= mysql_rake_tasks
+
+A collection of rails rake tasks for mysql
+1. create_users - creates localhost mysql user accounts for each database listing in the database.yml
+
+
+== Install
+
+1. Add mysql_rake_tasks to your gem file
+
+  gem 'mysql_rake_tasks', '~> 0.0.1'
+
+2. Run the bundle command
+
+  bundle install
+
+== Examples
+
+=== create_users
+
+To create mysql users in interactive mode run:
+
+  rake db:mysql:create_users
+  mysql user: root
+  mysql pass:
+
+You can also specify your root username and password on the command line:
+
+  rake db:mysel:create_users[root,mypassword]
+
+== License
+
+MIT
+
+== Credits
+
+Author: Gregory Ostermayr gregory.ostermayr@gmail.com
+
+Contributed code and/or ideas:
+
+Kevin Woods kwoods@privateergroup.com
+Travis Herrick
+

data/Rakefile

@@ -0,0 +1,12 @@
+require 'bundler/gem_helper'
+Bundler::GemHelper.install_tasks
+
+
+task :default => :test
+
+require 'rake/testtask'
+Rake::TestTask.new do |t|
+  t.test_files = FileList['test/*_test.rb']
+  t.verbose = true
+  t.warning = false
+end

data/lib/mysql_rake_tasks/mysql_rake_tasks.rake

@@ -0,0 +1,12 @@
+namespace :db do
+  namespace :mysql do
+
+    desc "Create MySQL users from database.yml (localhost only). Run without parameters for interactive mode." 
+    task :create_users, [:root_user, :pass] => [:environment] do |rake_task, args|
+      MysqlRakeTasks::Tasks::create_users(args)
+    end
+
+  end
+end
+
+

data/lib/mysql_rake_tasks/railtie.rb

@@ -0,0 +1,10 @@
+require 'rails'
+
+module MysqlRakeTasks
+  class Railtie < Rails::Railtie
+
+    rake_tasks do
+      load "mysql_rake_tasks/mysql_rake_tasks.rake"
+    end
+  end
+end

data/lib/mysql_rake_tasks/tasks.rb

@@ -0,0 +1,77 @@
+require 'rails'
+require 'mysql2'
+
+module MysqlRakeTasks
+  class Tasks
+
+    # Parses input args for username and password, if not given
+    # it will prompt the user
+    def self.get_input(*args)
+      if !args[0].nil? then
+        root_user = args[0][:root_user]
+        root_pass = args[0][:pass]
+      end
+
+      if root_user.nil? or root_pass.nil? then
+        $stdout.puts 'mysql user:'
+        root_user = $stdin.gets.chomp
+
+	$stdout.puts 'mysql password:'
+        system 'stty -echo'
+        root_pass = $stdin.gets.chomp
+        system 'stty echo'
+      end
+
+      {:root_user => root_user, :pass => root_pass}
+    end
+
+    # creates user permissions for mysql database for localhost only 
+    def self.create_users(args)
+      args = self.get_input(args)
+      @root_user = args[:root_user]
+      @pass = args[:pass]
+
+      # create a mysql user for each listing in database.yml file
+      Rails::configuration.database_configuration.each do |listing|
+        begin
+  	  @config = listing[1]
+          db = Mysql2::Client.new(
+               :host => 'localhost',
+               :username => @root_user,
+               :password => @pass,
+               :socket => @config['socket'])
+
+          sql = self.create_user_sql(@config)
+          db.query sql
+	  $stdout.puts "Created #{@config['username']} on #{@config['database']}\n"
+        rescue Mysql2::Error => e
+          $stdout.puts "Error code: #{e.errno}"
+          $stdout.puts "Error message: #{e.error}"
+          $stdout.puts "Error SQLSTATE: #{e.sqlstate}" if e.respond_to?("sqlstate")
+        ensure # disconnect from server 
+          db.close if db
+        end
+      end
+    end
+
+    def self.create_user_sql(config)
+      if config.nil? then
+        return ""
+      end
+
+      if config['username'].nil? then
+        puts 'Error code: missing username entry'
+      end
+
+      sql  = <<-SQL
+        GRANT
+        ALL PRIVILEGES
+        ON #{config['database']}.*
+        TO #{config['username']}@localhost
+        IDENTIFIED BY '#{config['password']}';
+      SQL
+    end
+
+  end
+end
+

data/lib/mysql_rake_tasks/version.rb

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

data/lib/mysql_rake_tasks.rb

@@ -0,0 +1,7 @@
+require 'mysql_rake_tasks/version'
+
+module MysqlRakeTasks
+  require "mysql_rake_tasks/railtie" if defined?(Rails)
+  require "mysql_rake_tasks/tasks"
+  require 'mysql2'
+end

data/mysql_rake_tasks.gemspec

@@ -0,0 +1,30 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "mysql_rake_tasks/version"
+
+Gem::Specification.new do |s|
+  s.name        = "mysql_rake_tasks"
+  s.version     = MysqlRakeTasks::VERSION
+  s.authors     = ["Gregory Ostermayr"]
+  s.email       = ["gregory.ostermayr@gmail.com"]
+  s.homepage    = "https://github.com/gregors/mysql_rake_tasks"
+  s.summary     = %q{Rake tasks for mysql}
+  s.description = %q{A collection of rails rake tasks for mysql.}
+  s.license = "MIT"
+  s.extra_rdoc_files = [
+   "LICENSE",
+   "README.rdoc"
+  ]
+
+  s.rubyforge_project = "mysql_rake_tasks"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = Dir['test/*.rb'] 
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  s.add_development_dependency "rake"
+  s.add_runtime_dependency "mysql2"
+end

data/test/tasks_test.rb

@@ -0,0 +1,116 @@
+require File.expand_path('../../lib/mysql_rake_tasks/tasks',__FILE__)
+
+require 'test/unit'
+require 'stringio'
+require 'mocha'
+
+class TasksTest < Test::Unit::TestCase
+
+  def test_two_args_returns_two_values
+    output = MysqlRakeTasks::Tasks::get_input({:root_user => 'user', :pass => 'pass'})
+
+    assert_equal 2, output.length
+    assert_equal 'user', output[:root_user]
+    assert_equal 'pass', output[:pass]
+  end
+
+  def test_lack_of_args_invokes_cli_interface
+    output = ""
+    screen = io_mock do |input|
+               input.string = "root\nmypassword\n"
+               output = MysqlRakeTasks::Tasks.get_input
+             end
+
+    assert_equal 'mysql user:', screen[0]
+    assert_equal 'mysql password:', screen[1]
+    assert_equal 'root', output[:root_user]
+    assert_equal 'mypassword', output[:pass]
+  end
+
+
+  def test_unsuccessful_authentication
+    Rails.stubs(:configuration).returns(Rails::Application::Configuration.allocate) 
+    Rails.configuration.stubs(:database_configuration).returns(stub_config)
+
+    screen = io_mock do |input|
+      MysqlRakeTasks::Tasks.create_users(:root_user => 'root', :pass => 'wrong')
+    end
+   
+    assert_equal 'Error code: 1045', screen[0]
+    assert_equal "Error message: Access denied for user 'root'@'localhost' (using password: YES)", screen[1]
+    assert_equal 'Error code: 1045', screen[2]
+    assert_equal "Error message: Access denied for user 'root'@'localhost' (using password: YES)", screen[3]
+    assert_equal 'Error code: 1045', screen[4]
+    assert_equal "Error message: Access denied for user 'root'@'localhost' (using password: YES)", screen[5]
+  end
+
+  def test_successful_creation
+    Rails.stubs(:configuration).returns(Rails::Application::Configuration.allocate)
+    Rails.configuration.stubs(:database_configuration).returns(stub_config)
+
+    screen = io_mock do |input|
+      # :pass needs to be set to mysql root in order to pass
+      MysqlRakeTasks::Tasks.create_users(:root_user => 'root', :pass => 'myrootpass') 
+    end
+
+    assert_equal "Created dev on task_development", screen[0], 'Note: ***check test machine password***'
+    assert_equal "Created test on task_test", screen[1]
+    assert_equal "Created prod on task_production", screen[2]
+  end
+
+  def test_lack_of_user_throws_error
+    config = stub_config
+    config["development"].delete "username"
+
+    Rails.stubs(:configuration).returns(Rails::Application::Configuration.allocate)
+    Rails.configuration.stubs(:database_configuration).returns(config)
+
+    screen = io_mock do |input|
+      # :pass needs to be set to mysql root in order to pass
+      MysqlRakeTasks::Tasks.create_users(:root_user => 'root', :pass => 'myrootpass')
+    end
+
+    assert_equal 'Error code: missing username entry', screen[0], 'Note: ***check test machine password***'
+    assert_equal 'Error code: 1064', screen[1]
+  end
+
+  def stub_config
+   {"development"=>{"adapter"=>"mysql2",
+                    "encoding"=>"utf8",
+                    "reconnect"=>false,
+                    "database"=>"task_development",
+                    "pool"=>5,
+                    "username"=>"dev",
+                    "password"=>"devpassword"},
+   "test"=>{"adapter"=>"mysql2",
+            "encoding"=>"utf8",
+            "reconnect"=>false,
+            "database"=>"task_test",
+            "pool"=>5,
+            "username"=>"test",
+            "password"=>"taskpassword"},
+   "production"=>{"adapter"=>"mysql2",
+                  "encoding"=>"utf8",
+                  "reconnect"=>false,
+                  "database"=>"task_production",
+                  "pool"=>5,
+                  "username"=>"prod",
+                  "password"=>"prodpassword"}} 
+  end
+
+
+  def io_mock
+    org_stdin = $stdin
+    org_stdin = $stdout
+
+    $stdin = StringIO.new
+    $stdout = StringIO.new
+
+    yield $stdin
+
+    return $stdout.string.split("\n")
+  ensure
+    $stdin = org_stdin
+    $stdout = org_stdin
+  end
+end

metadata

@@ -0,0 +1,85 @@
+--- !ruby/object:Gem::Specification
+name: mysql_rake_tasks
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Gregory Ostermayr
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-09-05 00:00:00.000000000 -04:00
+default_executable: 
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: &16396240 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *16396240
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  requirement: &16395820 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *16395820
+description: A collection of rails rake tasks for mysql.
+email:
+- gregory.ostermayr@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE
+- README.rdoc
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.rdoc
+- Rakefile
+- lib/mysql_rake_tasks.rb
+- lib/mysql_rake_tasks/mysql_rake_tasks.rake
+- lib/mysql_rake_tasks/railtie.rb
+- lib/mysql_rake_tasks/tasks.rb
+- lib/mysql_rake_tasks/version.rb
+- mysql_rake_tasks.gemspec
+- test/tasks_test.rb
+has_rdoc: true
+homepage: https://github.com/gregors/mysql_rake_tasks
+licenses:
+- MIT
+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: mysql_rake_tasks
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: Rake tasks for mysql
+test_files:
+- test/tasks_test.rb