batteries 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1b72dbe55d0b2e75ac7f7252523cc0a6e9aa0e8c
-  data.tar.gz: 14941b0f5403a9f8d808dab344a1311ba478d5e6
+  metadata.gz: a0be368a6cfd39c55ff6d08907ac0a9b64d8dc1a
+  data.tar.gz: 7db889643e97db81a6a547a0850da3d071183820
 SHA512:
-  metadata.gz: a580120bb878da83920740b14f0897633026a4fbdb080df8a206fd8ade2c7351e7ea461e7bebd4220b9d301f50c7e88060479e540fff04dd6b07f4fa4a5abf28
-  data.tar.gz: a2d3ed6ab93bed2e952dd51e387be6c3a6cdb549521a8e2057649390a540b7d4397e89ad1f65be24a9809a93e1689bce5810ed3916ab7ed45102ed15f245adb4
+  metadata.gz: fa4a786620129b69b068b746bb357289450daef25539424c6ca9cd53681cf2af57c1e61d4ecbcd6a64c54d2df0f4dacdda563fdfb7586453a4e1eb14020ee819
+  data.tar.gz: 42dc68a0e112ac93ed1c13e8305461c4e05a462324b4f6eeb1dd6de4bed9c5fed23e31dc72893c82dbbfed16eb387106c9ef69639e345078eed7c269096b781e

data/lib/batteries/tasks/migrations.rb

@@ -6,12 +6,14 @@ require "logger"
 module Batteries
   module Tasks
     class Migrations < ::Rake::TaskLib
-      attr_accessor :migrations_path, :database, :logger
+      attr_accessor :migrations_path, :logger, :setup_hook
+      attr_writer :database
 
       def initialize(options: {})
-        @migrations_path = options.fetch(:migrations_path) { "migrate" }
-        @database = options.fetch(:database) { DB if defined?(DB) }
-        @logger = options.fetch(:logger) { Logger.new($stdout) }
+        self.migrations_path = options.fetch(:migrations_path) { "migrate" }
+        self.database = options.fetch(:database) { DB if defined?(DB) }
+        self.logger = options.fetch(:logger) { Logger.new($stdout) }
+        self.setup_hook = options[:setup_hook]
 
         yield self if block_given?
 
@@ -62,7 +64,16 @@ module Batteries
         end
       end
 
+      def database
+        case @database
+        when String then Object.const_get(@database)
+        else @database
+        end
+      end
+
       def migrate(env, version)
+        ENV["RACK_ENV"] = env
+        setup_hook.call if setup_hook.respond_to?(:call)
         Sequel.extension :migration
         database.loggers << logger if logger
         Sequel::Migrator.apply(database, migrations_path, version)

data/lib/batteries/version.rb

@@ -1,3 +1,3 @@
 module Batteries
-  VERSION = "0.3.0"
+  VERSION = "0.4.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: batteries
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Adam Daniels
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-07-06 00:00:00.000000000 Z
+date: 2017-11-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rubygems-tasks
@@ -87,7 +87,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.12
+rubygems_version: 2.6.13
 signing_key: 
 specification_version: 4
 summary: Powering your Roda/Sinatra applications