backhoe 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d019d5b66f418151f88ebf3d897d0272a32d173e
-  data.tar.gz: f6e7f9c5df280e9730f6d48da8b37d756d54165e
+  metadata.gz: 56b4732a51c97a122c1ceb8777b8c1063e12465b
+  data.tar.gz: b4e047fc5ceabbe0a6ca030ceea674ba7ad71ff8
 SHA512:
-  metadata.gz: 6750d9c40ce7993f43a5a14b183833d533a475adbab48d249f7f7ac6295ef81fac6e5f046c676166d4a75cb38a8c3f5eafe9895695525813131ee1081bdd51f5
-  data.tar.gz: 14f0271c2c51661fd1822655bfb15c47c2eba52ac878cbd5b097be601876c86d470b79610cb3bddb0fc4737d9289b29cd8b9054de24230ce27c8f3b8e37ecfe7
+  metadata.gz: 1cc3c5f7edc01601af3c9c6e7a28b86e8e569fe0ab48ef378c577b73e6287580ec394380fbc629675003b608ace0c8d856634b0db2d1b4ffebcfdec021b4ca52
+  data.tar.gz: 3c39d974916a71d3ad7f725f40d9e23c6829f8a68d558ffb41d04e8ceac520b24e13a818fe0fdf30b41bfee04a32d0114522a223c56bc5910c020e2b3bb7be51

data/.travis.yml

@@ -1,5 +1,5 @@
 sudo: false
 language: ruby
-rvm:
-  - 2.2.1
-before_install: gem install bundler -v 1.16.1
+cache:
+  bundler: true
+

data/Gemfile

@@ -4,3 +4,4 @@ git_source(:github) {|repo_name| "https://github.com/#{repo_name}" }
 
 # Specify your gem's dependencies in backhoe.gemspec
 gemspec
+

data/README.md

@@ -1,8 +1,7 @@
 # Backhoe
+[![Build Status](https://travis-ci.org/botandrose/backhoe.svg)](https://travis-ci.org/botandrose/backhoe)
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/backhoe`. To experiment with that code, run `bin/console` for an interactive prompt.
-
-TODO: Delete this and the text above, and describe your gem
+Dump and load current ActiveRecord database to and from a file.
 
 ## Installation
 
@@ -32,7 +31,7 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/backhoe.
+Bug reports and pull requests are welcome on GitHub at https://github.com/botandrose/backhoe.
 
 ## License
 

data/backhoe.gemspec

@@ -21,7 +21,10 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
+  spec.add_dependency "activerecord"
+
   spec.add_development_dependency "bundler", "~> 1.16"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "mysql2"
 end

data/lib/backhoe.rb

@@ -1,6 +1,8 @@
 require "backhoe/version"
-require "rake"
-require "fileutils"
+require "backhoe/mysql"
+require "backhoe/postgresql"
+require "backhoe/sqlite"
+require "active_record"
 
 module Backhoe
   mattr_accessor(:file_path) { "db/data.sql" }
@@ -24,71 +26,5 @@ module Backhoe
       ActiveRecord::Base.configurations[Rails.env || "development"]
     end
   end
-
-  class Base < Struct.new(:config, :file_path)
-    include Rake::DSL
-
-    private
-
-    def database
-      config["database"]
-    end
-  end
-
-  class Mysql < Base
-    def dump skip_tables: []
-      mysqldump = `which mysqldump`.strip
-      raise RuntimeError, "Cannot find mysqldump." if mysqldump.blank?
-      sh "#{mysqldump} --no-create-db --single-transaction --quick -e #{skip_table_options(skip_tables)} #{mysql_options} > #{file_path}"
-    end
-
-    def load
-      mysql = `which mysql`.strip
-      raise RuntimeError, "Cannot find mysql." if mysql.blank?
-      sh "#{mysql} #{mysql_options} < #{file_path}"
-    end
-
-    private
-
-    def skip_table_options skip_tables
-      skip_tables.map do |table|
-        "--ignore-table=#{config["database"]}.#{table}"
-      end.join(" ")
-    end
-    
-    def mysql_options
-      options =  " -u #{config["username"]}"
-      options += " -p'#{config["password"]}'" if config["password"]
-      options += " -h #{config["host"]}"      if config["host"]
-      options += " -S #{config["socket"]}"    if config["socket"]
-      options += " '#{config["database"]}'"
-    end
-  end
-
-  Mysql2 = Mysql
-
-  class Sqlite3 < Base
-    def dump **_
-      FileUtils.cp database, file_path
-    end
-
-    def load
-      FileUtils.cp file_path, database
-    end
-  end
-
-  class Postgresql < Base
-    def dump **_
-      pg_dump = `which pg_dump`.strip
-      raise RuntimeError, "Cannot find pg_dump." if pg_dump.blank?
-      sh "#{pg_dump} -c -f#{file_path} #{database}"
-    end
-
-    def load
-      psql = `which psql`.strip
-      raise RuntimeError, "Cannot find psql." if psql.blank?
-      sh "#{psql} -q -d#{database} -f#{file_path}"
-    end
-  end
 end
 

data/lib/backhoe/base.rb

@@ -0,0 +1,14 @@
+require "rake"
+
+module Backhoe
+  class Base < Struct.new(:config, :file_path)
+    include Rake::DSL
+
+    private
+
+    def database
+      config["database"]
+    end
+  end
+end
+

data/lib/backhoe/mysql.rb

@@ -0,0 +1,96 @@
+require "backhoe/base"
+
+module Backhoe
+  class Mysql < Base
+    def dump skip_tables: []
+      mysqldump = `which mysqldump`.strip
+      raise RuntimeError, "Cannot find mysqldump." if mysqldump.blank?
+      sh "#{mysqldump} --no-create-db --single-transaction --quick -e #{skip_table_options(skip_tables)} #{mysql_options} > #{file_path}"
+    end
+
+    def load
+      mysql = `which mysql`.strip
+      raise RuntimeError, "Cannot find mysql." if mysql.blank?
+      sh "#{mysql} #{mysql_options} < #{file_path}"
+    end
+
+    private
+
+    def skip_table_options skip_tables
+      skip_tables.map do |table|
+        "--ignore-table=#{config["database"]}.#{table}"
+      end.join(" ")
+    end
+    
+    def mysql_options
+      options =  " -u #{config["username"]}"
+      options += " -p'#{config["password"]}'" if config["password"]
+      options += " -h #{config["host"]}"      if config["host"]
+      options += " -S #{config["socket"]}"    if config["socket"]
+      options += " '#{config["database"]}'"
+    end
+  end
+
+  module MysqlSkipColumns
+    def dump **options
+      if skip_columns = options.delete(:skip_columns)
+        SanitizedDatabase.new(skip_columns, file_path).dump do |skip_tables|
+          options[:skip_tables] ||= []
+          options[:skip_tables] += skip_tables
+          super options
+        end
+      else
+        super
+      end
+    end
+
+    private
+
+    class SanitizedDatabase < Struct.new(:config, :file_path)
+      def dump
+        with_sanitized_tables do
+          yield skip_tables
+        end
+        skip_tables.each do |table|
+          File.write file_path, "RENAME TABLE `sanitized_#{table}` TO `#{table}`;", mode: "a"
+        end
+      end
+
+      private
+
+      def skip_tables
+        config.keys
+      end
+
+      def with_sanitized_tables
+        ActiveRecord::Base.transaction do
+          config.each do |table, columns|
+            sanitized_table = "sanitized_#{table}"
+            sql <<-SQL
+              DROP TABLE IF EXISTS `#{sanitized_table}`;
+              CREATE TABLE `#{sanitized_table}` LIKE `#{table}`;
+              INSERT INTO `#{sanitized_table}` SELECT * FROM `#{table}`;
+              ALTER TABLE `#{sanitized_table}` #{columns.map { |column| "DROP `#{column}`" }.join(", ")};
+            SQL
+          end
+
+          yield
+
+          config.each do |table, _|
+            sql "DROP TABLE `sanitized_#{table}`"
+          end
+        end
+      end
+
+      def sql queries
+        queries.split(";").select(&:present?).each do |query|
+          ActiveRecord::Base.connection.execute query
+        end
+      end
+    end
+  end
+  Mysql.prepend MysqlSkipColumns
+
+  Mysql2 = Mysql
+end
+

data/lib/backhoe/postgresql.rb

@@ -0,0 +1,18 @@
+require "backhoe/base"
+
+module Backhoe
+  class Postgresql < Base
+    def dump **_
+      pg_dump = `which pg_dump`.strip
+      raise RuntimeError, "Cannot find pg_dump." if pg_dump.blank?
+      sh "#{pg_dump} -c -f#{file_path} #{database}"
+    end
+
+    def load
+      psql = `which psql`.strip
+      raise RuntimeError, "Cannot find psql." if psql.blank?
+      sh "#{psql} -q -d#{database} -f#{file_path}"
+    end
+  end
+end
+

data/lib/backhoe/sqlite.rb

@@ -0,0 +1,15 @@
+require "backhoe/base"
+require "fileutils"
+
+module Backhoe
+  class Sqlite3 < Base
+    def dump **_
+      FileUtils.cp database, file_path
+    end
+
+    def load
+      FileUtils.cp file_path, database
+    end
+  end
+end
+

data/lib/backhoe/version.rb

@@ -1,3 +1,3 @@
 module Backhoe
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: backhoe
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Micah Geisel
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-05-12 00:00:00.000000000 Z
+date: 2018-05-13 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  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: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -52,6 +66,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  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'
 description: Dump and load current database to and from a file.
 email:
 - micah@botandrose.com
@@ -70,6 +98,10 @@ files:
 - bin/console
 - bin/setup
 - lib/backhoe.rb
+- lib/backhoe/base.rb
+- lib/backhoe/mysql.rb
+- lib/backhoe/postgresql.rb
+- lib/backhoe/sqlite.rb
 - lib/backhoe/version.rb
 homepage: https://github.com/botandrose/backhoe
 licenses: