db_stager 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b61c5b179445a96bd7fffc53b1bfbbe37e7cbd70
-  data.tar.gz: 236e74f0283dcfdbdacb4c548d463c7b353c1afc
+  metadata.gz: 5596cc698905e2cb5c03fbcebee2a272403360f4
+  data.tar.gz: 2e27f20781268cc10f257c000cec3307d8b043cf
 SHA512:
-  metadata.gz: a6cc24928a5f0f331e406e50d7edb2d22b07c1d1f518e7a7167662010f4048ff69856ce0d2a4781602e7c0677ce3ee389635751f4166cdccd44bf2ebc0b3f3ec
-  data.tar.gz: 2604bfb89e98d8c897a3651e41afb4ae281186184dc5bb2e0dd57034071c9e928774b23ba1438b1e4829f3436b416964ae959977056f5034fd01935df48c107f
+  metadata.gz: ecd312cb600acc15779f8336443da10cd65997fd80a18a08744a5d3f0d7a88fde9aa99c253d972b128ab7779633b44b29654d2ca1830eefb44d69865cd9c3226
+  data.tar.gz: 9acec0d9ffaf4eaff6dcbd7b034b00828a93c0c31e2c2a1a8b53f3b24cd629dcf213b0e8af4efad56c93c0dc2ea65d015d52a932e07a46627bc6ae4601e668a6

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    db_stager (0.1.1)
+    db_stager (0.2.0)
       dotenv (~> 2.2)
       sequel (~> 4.41.0)
       thor (~> 0.19)

data/lib/db_stager/cli.rb

@@ -23,14 +23,10 @@ module DbStager
     def capture(dir=".")
       Dir.chdir(dir)
 
-      environment = options[:environment] || ENV['RAILS_ENV'] || ENV['RACK_ENV']
-      # Get connection info
-
-      db = Sequel.connect(YAML.load(ERB.new(File.read(File.join(Dir.pwd, 'config', 'database.yml'))).result)[environment])
+      sh 'rake db:structure:dump'
 
       capture_file = options[:file] || File.join(Dir.pwd, "db", "#{environment}_capture.sql")
       File.open(capture_file, "w") do |file|
-
         db.tables.each do |table_name|
           items = db[table_name]
           items.each do |item|
@@ -45,17 +41,25 @@ module DbStager
     desc "upload", "upload your most recent capture to S3"
     desc "download", "download your most recent capture from S3"
 
+    method_option :environment, type: :string, aliases: '-e'
+    method_option :structure_file, type: :string, aliases: '-s'
+    method_option :file, type: :string, aliases: '-f'
     desc "load", "overwrite the current db with the most recently downloaded stager db"
     def load(dir=".")
       Dir.chdir(dir)
-
-      environment = options[:environment] || ENV['RAILS_ENV'] || ENV['RACK_ENV']
-      # Get connection info
-
-      db = Sequel.connect(YAML.load(ERB.new(File.read(File.join(Dir.pwd, 'config', 'database.yml'))).result)[environment])
-
+      structure_file = options[:structure_file] || File.join(Dir.pwd, "db", "structure.sql")
+      db.run File.read(struture_file)
       capture_file = options[:file] || File.join(Dir.pwd, "db", "#{environment}_capture.sql")
       db.run File.read(capture_file)
     end
   end
+
+  protected
+  def environment
+    @environment ||= options[:environment] || ENV['RAILS_ENV'] || ENV['RACK_ENV'] || "development"
+  end
+
+  def db
+    @db ||= Sequel.connect(YAML.load(ERB.new(File.read(File.join(Dir.pwd, 'config', 'database.yml'))).result)[environment])
+  end
 end

data/lib/db_stager/version.rb

@@ -1,3 +1,3 @@
 module DbStager
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: db_stager
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Rob Kaufman
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-07-20 00:00:00.000000000 Z
+date: 2018-08-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler