rails_db_objects 0.1.1 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 236a510b06b887a06f6b4ea958ed92cfb183d401
-  data.tar.gz: 7af1e835809cf8108a8b93a743ae8cbca821cb28
+  metadata.gz: a34d7b5731efea0e2ae83c0e0a66f62eafbe7ab9
+  data.tar.gz: 6e9758078a65334f7d408ed25d3cc60bd1b8b3b9
 SHA512:
-  metadata.gz: e8749705009d45ebd899118e3f803587f6061df240dd1508054a9e3a3cb0e00c92783de815686621962b98b03c2ab1da01ae008a602dafc15498bc9026b4c46e
-  data.tar.gz: eb95b8974ffd4554d838252f4979ec1138654929792b500a8010f1db0fd2013459a84cc8a4b3ce059e20e1137470cfcf2e798d80406d6c218b9e995a48fc68aa
+  metadata.gz: 95fa51efaa72f6dffefa6600a6c02838f0739b093cd2d0b913c6f01140364498cbecaa7eb8e9783befaef0527e832b265a2f81f02571c3b3746aa5ed1c34dbfc
+  data.tar.gz: 9e506abd8ecc4c071100981d93d39f4ffa05f4afe39e80210a42e587c4c9a6fd39b884714d8e64d9ec77229ed70b6433e3f650ce102e0919c9eb3c74c454f5f4

data/Rakefile

@@ -1,28 +1,28 @@
-begin
-  require 'bundler/setup'
-rescue LoadError
-  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
-end
-
-require 'rdoc/task'
-
-RDoc::Task.new(:rdoc) do |rdoc|
-  rdoc.rdoc_dir = 'rdoc'
-  rdoc.title    = 'RailsDbObjects'
-  rdoc.options << '--line-numbers'
-  rdoc.rdoc_files.include('README.rdoc')
-  rdoc.rdoc_files.include('lib/**/*.rb')
-end
-
-Bundler::GemHelper.install_tasks
-
-require 'rake/testtask'
-
-Rake::TestTask.new(:test) do |t|
-  t.libs << 'lib'
-  t.libs << 'test'
-  t.pattern = 'test/**/*_test.rb'
-  t.verbose = false
-end
-
-task default: :test
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
+
+require 'rdoc/task'
+
+RDoc::Task.new(:rdoc) do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title    = 'RailsDbObjects'
+  rdoc.options << '--line-numbers'
+  rdoc.rdoc_files.include('README.rdoc')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+Bundler::GemHelper.install_tasks
+
+require 'rake/testtask'
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << 'lib'
+  t.libs << 'test'
+  t.pattern = 'test/**/*_test.rb'
+  t.verbose = false
+end
+
+task default: :test

data/lib/rails_db_objects/db_objects_creator.rb

@@ -1,231 +1,318 @@
-module RailsDbObjects
-  class DbObjectsCreator
-    attr_reader :objects
-
-    def initialize
-      @objects = {}
-    end
-
-    def register_files(files)
-      files.each do |file|
-        object_name = File.basename(file, File.extname(file))
-        object_type = File.dirname(file.to_s).to_s.split('/').last.upcase
-
-        @objects[object_type] ||= {}
-
-        content = File.read(file)
-        content_lines = content.split("\n")
-
-        # Reject the commented lines from the file
-        sql_content = content_lines.reject { |x| x.strip =~ /^--/ || x.strip =~ /^#/ }.join("\n")
-
-        file_obj = {
-          path: file,
-          sql_content: sql_content,
-          status: :none,
-          requires: [],
-          silent: false,
-          keep: false,
-          deleted: false,
-          dbschema: Rails.configuration.rails_db_objects[:objects_dbschema],
-          dropsql: [],
-          createsql: [],
-          vanilla: false,
-          condition: []
-        }
-
-        # Detect directives in commentary
-        directives = extract_from_comments(content_lines)
-
-        # puts "directives: #{directives.inspect}"
-        directives.each { |directive| prepare_directive(file_obj, directive) }
-
-        @objects[object_type][object_name] = file_obj
-      end
-    end
-
-    def drop_objects
-      reset_objects_status!
-      @objects.keys.each do |object_type|
-        @objects[object_type].each do |name, object|
-          drop_object(object_type, name, object) unless object[:skip_pre]
-        end
-      end
-    end
-
-    def create_objects
-      reset_objects_status!
-      @objects.keys.each do |object_type|
-        @objects[object_type].each do |name, object|
-          create_object(object_type, name, object) unless object[:skip_post]
-        end
-      end
-    end
-
-    private
-
-    def full(name)
-      prefix = @dbschema.blank? ? '' : "#{@dbschema}."
-      "#{prefix}#{wrap_name(name)}"
-    end
-
-    def wrap_name(name)
-      adapter_name = ActiveRecord::Base.connection.adapter_name
-
-      case adapter_name
-      when 'PostgreSQL'
-        "\"#{name}\""
-      when 'MySQL'
-        "`#{name}`"
-      when 'SQLServer'
-        "[#{name}]"
-      end
-    end
-
-    def extract_from_comments(content_lines)
-      dir_lines = content_lines.select { |x| x.strip =~ /^--/ || x.strip =~ /^#/ }.map(&:strip)
-      dir_lines.map { |x| /^--/.match?(x) ? x[2..-1] : x[1..-1] }.select { |x| x =~ /^!/ }
-    end
-
-    def prepare_directive(file_obj, directive)
-      file_obj[:requires] += directive.split(' ')[1..-1] if /^!require /.match?(directive)
-      file_obj[:vanilla] = /^!vanilla/.match?(directive)
-      file_obj[:skip_post] = /^!deleted/.match?(directive)
-      file_obj[:silent] = /^!silent/.match?(directive)
-      file_obj[:skip_pre] = /^!keep/.match?(directive)
-      file_obj[:dbschema] = directive.split(' ')[1..-1] if /^!schema/.match?(directive)
-      file_obj[:condition] << directive.split(' ')[1..-1].join(' ') if /^!condition /.match?(directive)
-      file_obj[:dropsql] << directive.split(' ')[1..-1].join(' ') if /^!dropsql /.match?(directive)
-      file_obj[:createsql] << directive.split(' ')[1..-1].join(' ') if /^!createsql /.match?(directive)
-    end
-
-    def reset_objects_status!
-      @objects.keys.each do |object_type|
-        @objects[object_type].each do |_name, object|
-          object[:status] = :none
-        end
-      end
-    end
-
-    def drop_object(object_type, name, object)
-      return if object[:status] == :loaded
-
-      if object[:status] == :inprogress
-        raise "Error: Circular file reference! (#{object_type} #{name})"
-      end
-
-      object[:requires].each do |requirement|
-        requires = requirement.split('/', 2)
-        required_object = requires.last
-        required_type = requires.first.upcase if requires.length == 2
-        required_type ||= object_type
-        drop_object required_type, required_object, @objects[required_type.upcase][required_object]
-      end
-
-      @dbschema = (object[:dbschema] || []).clone
-      full_name = full(name)
-
-      sql = if object[:vanilla]
-              object[:sql_content]
-            elsif object[:dropsql].compact.empty?
-              "DROP #{object_type} #{full_name};"
-            else
-              object[:dropsql].compact.join(";\n")
-            end
-
-      begin
-        unless object[:condition].compact.empty?
-          condition = !ActiveRecord::Base.connection.select_rows(object[:condition].join("\n")).empty?
-          if condition
-            puts "CONDITION MET FOR #{object_type} #{full_name}"
-          else
-            puts "CONDITION NOT MET FOR #{object_type} #{full_name}"
-            return
-          end
-        end
-        ActiveRecord::Base.connection.execute(sql)
-        puts "#{sql}... OK"
-      rescue StandardError => e
-        unless object[:silent]
-          puts '#' * 80
-          puts e.message.to_s
-          # puts "#"*80
-          # puts "#{e.backtrace}"
-          puts '#' * 80
-          puts "WARNING: #{sql}... ERROR"
-          puts '#' * 80
-          # else
-          #   puts "WARNING: #{sql}... SILENT"
-        end
-      end
-
-      object[:status] = :loaded
-    end
-
-    def create_object(object_type, name, object)
-      # skip empty sql content
-      if object[:sql_content].strip.blank?
-        puts "#{object_type} #{name} EMPTY... SKIPPING"
-        return
-      end
-
-      # object already loaded.
-      return if object[:status] == :loaded
-
-      raise "Error: Circular file reference! (#{object_type} #{name})" if object[:status] == :inprogress
-
-      object[:status] = :inprogress
-
-      create_dependencies(object, object_type)
-
-      @dbschema = (object[:dbschema] || []).clone
-      full_name = full(name)
-
-      sql = if object[:vanilla]
-              object[:sql_content]
-            elsif object[:createsql].compact.empty?
-              "CREATE #{object_type} #{full_name}\n#{object[:sql_content]}"
-            else
-              object[:createsql].compact.join("\n")
-            end
-
-      begin
-        unless object[:condition].compact.empty?
-          condition = ActiveRecord::Base.connection.select_rows(object[:condition].join("\n")).empty?
-          if condition
-            puts "CONDITION MET FOR #{object_type} #{full_name}"
-          else
-            puts "CONDITION NOT MET FOR #{object_type} #{full_name}"
-            return
-          end
-        end
-        ActiveRecord::Base.connection.execute(sql.to_s)
-        puts "CREATE #{object_type} #{full_name}... OK"
-      rescue StandardError => e
-        unless object[:silent]
-          puts '#' * 80
-          puts e.message.to_s
-          # puts "#"*80
-          # puts "#{e.backtrace}"
-          puts '#' * 80
-          puts "WARNING: CREATE #{object_type} #{full_name}... ERROR"
-          puts '#' * 80
-          # else
-          #   puts "WARNING: CREATE #{object_type} #{full_name}... SILENT"
-        end
-      end
-
-      object[:status] = :loaded
-    end
-
-    def create_dependencies(object, object_type)
-      object[:requires].each do |requirement|
-        requires = requirement.split('/', 2)
-        required_object = requires.last
-        required_type = requires.first.upcase if requires.length == 2
-        required_type ||= object_type
-        create_object required_type, required_object, @objects[required_type.upcase][required_object]
-      end
-    end
-  end
-end
+module RailsDbObjects
+  class DbObjectsCreator
+    attr_reader :objects
+
+    def initialize
+      @objects = {}
+    end
+
+    def register_files(files)
+      files.each do |file|
+        object_name = File.basename(file, File.extname(file))
+        object_type = File.dirname(file.to_s).to_s.split('/').last.upcase
+
+        @objects[object_type] ||= {}
+
+        content = File.read(file)
+        content_lines = content.split("\n")
+
+        # Reject the commented lines from the file
+        sql_content = content_lines.reject { |x| x.strip =~ /^--/ || x.strip =~ /^#/ }.join("\n")
+
+        file_obj = {
+          name: object_name,
+          type: object_type,
+          debug: false,
+          directives: [],
+          path: file,
+          sql_content: sql_content,
+          status: :none,
+          requires: [],
+          silent: false,
+          keep: false,
+          nodrop: false,
+          deleted: false,
+          nocreate: false,
+          dbschema: Rails.configuration.rails_db_objects[:objects_dbschema],
+          dropsql: [],
+          createsql: [],
+          vanilla: false,
+          condition: [],
+          dropconditionruby: [],
+          createconditionruby: [],
+          beforedropruby: [],
+          beforedropsql: [],
+          afterdropruby: [],
+          afterdropsql: [],
+          beforecreateruby: [],
+          beforecreatesql: [],
+          aftercreateruby: [],
+          aftercreatesql: []
+        }
+
+        # Detect directives in commentary
+        directives = extract_from_comments(content_lines)
+
+        # puts "directives: #{directives.inspect}"
+        directives.each { |directive| prepare_directive(file_obj, directive) }
+
+        @objects[object_type][object_name] = file_obj
+      end
+    end
+
+    def drop_objects
+      reset_objects_status!
+      @objects.keys.each do |object_type|
+        @objects[object_type].each do |_name, object|
+          drop_object(object)
+        end
+      end
+    end
+
+    def create_objects
+      reset_objects_status!
+      @objects.keys.each do |object_type|
+        @objects[object_type].each do |_name, object|
+          create_object(object)
+        end
+      end
+    end
+
+    private
+
+    def full(name)
+      prefix = @dbschema.blank? ? '' : "#{@dbschema}."
+      "#{prefix}#{wrap_name(name)}"
+    end
+
+    def wrap_name(name)
+      adapter_name = ActiveRecord::Base.connection.adapter_name
+
+      case adapter_name
+      when 'PostgreSQL'
+        "\"#{name}\""
+      when 'MySQL'
+        "`#{name}`"
+      when 'SQLServer'
+        "[#{name}]"
+      end
+    end
+
+    def extract_from_comments(content_lines)
+      dir_lines = content_lines.select { |x| x.strip =~ /^--/ || x.strip =~ /^#/ }.map(&:strip)
+      dir_lines.map { |x| /^--/.match?(x) ? x[2..-1] : x[1..-1] }.select { |x| x =~ /^!/ }
+    end
+
+    # rubocop:disable Metrics/AbcSize
+    def prepare_directive(file_obj, directive)
+      file_obj[:debug] = /^!debug/.match?(directive) unless file_obj[:debug]
+      file_obj[:directives] << directive
+      file_obj[:requires] += directive.split(' ')[1..-1] if /^!require /.match?(directive)
+      file_obj[:vanilla] = /^!vanilla/.match?(directive) unless file_obj[:vanilla]
+      file_obj[:nocreate] = file_obj[:deleted] = /^!deleted/.match?(directive) unless file_obj[:deleted]
+      file_obj[:silent] = /^!silent/.match?(directive) unless file_obj[:silent]
+      file_obj[:nodrop] = file_obj[:keep] = /^!keep/.match?(directive) unless file_obj[:keep]
+      file_obj[:dbschema] = directive.split(' ')[1..-1] if /^!schema/.match?(directive)
+      file_obj[:dropconditionruby] << directive.split(' ')[1..-1].join(' ') if /^!dropconditionruby /.match?(directive)
+      file_obj[:createconditionruby] << directive.split(' ')[1..-1].join(' ') if /^!createconditionruby /.match?(directive)
+      file_obj[:condition] << directive.split(' ')[1..-1].join(' ') if /^!condition /.match?(directive)
+      file_obj[:beforedropsql] << directive.split(' ')[1..-1].join(' ') if /^!beforedropsql /.match?(directive)
+      file_obj[:dropsql] << directive.split(' ')[1..-1].join(' ') if /^!dropsql /.match?(directive)
+      file_obj[:afterdropsql] << directive.split(' ')[1..-1].join(' ') if /^!afterdropsql /.match?(directive)
+      file_obj[:beforecreatesql] << directive.split(' ')[1..-1].join(' ') if /^!beforecreatesql /.match?(directive)
+      file_obj[:createsql] << directive.split(' ')[1..-1].join(' ') if /^!createsql /.match?(directive)
+      file_obj[:aftercreatesql] << directive.split(' ')[1..-1].join(' ') if /^!aftercreatesql /.match?(directive)
+    end
+    # rubocop:enable Metrics/AbcSize
+
+    def reset_objects_status!
+      @objects.keys.each do |object_type|
+        @objects[object_type].each do |_name, object|
+          object[:status] = :none
+        end
+      end
+    end
+
+    def conditional_ruby(condition, object)
+      interpolated_command = object.instance_eval('"' + condition.gsub(/\"/, '\"') + '"')
+      condition = object.instance_eval(interpolated_command)
+      if object[:debug]
+        puts '=' * 80
+        puts "interpolated_command: #{interpolated_command}"
+        puts '=' * 80
+        puts "condition: #{condition.inspect}"
+        puts '=' * 80
+      end
+      condition ? true : false
+    end
+
+    def interpolate_sql(sql, object)
+      object.instance_eval('"' + sql.gsub(/\"/, '\"') + '"')
+    end
+
+    # rubocop:disable Metrics/AbcSize
+    def drop_object(object)
+      return if object[:nodrop]
+
+      return if object[:status] == :loaded
+
+      object_type = object[:type]
+      name = object[:name]
+      @dbschema = (object[:dbschema] || []).clone
+
+      raise "Error: Circular file reference! (#{object_type} #{name})" if object[:status] == :inprogress
+
+      full_name = full(name)
+
+      if object[:dropconditionruby].compact.any?
+        condition = conditional_ruby(object[:dropconditionruby].join("\n"), object)
+        if condition
+          puts "RUBY CONDITION MET FOR #{object_type} #{full_name} #{condition}"
+        else
+          puts "RUBY CONDITION NOT MET FOR #{object_type} #{full_name} #{condition}"
+          return
+        end
+      elsif object[:condition].compact.any?
+        condition = !ActiveRecord::Base.connection.select_rows(object[:condition].join("\n")).empty?
+        if condition
+          puts "SQL CONDITION MET FOR #{object_type} #{full_name}"
+        else
+          puts "SQL CONDITION NOT MET FOR #{object_type} #{full_name}"
+          return
+        end
+      end
+
+      if object[:debug]
+        puts '=' * 80
+        puts "DROP OBJECT #{name} / #{object[:path]}"
+        puts '=' * 80
+        puts object.to_json
+        puts '=' * 80
+      end
+
+      object[:requires].each do |requirement|
+        requires = requirement.split('/', 2)
+        required_object = requires.last
+        required_type = requires.first.upcase if requires.length == 2
+        required_type ||= object_type
+        drop_object @objects[required_type.upcase][required_object]
+      end
+
+      sql = if object[:dropsql].compact.empty? && !object[:vanilla]
+              "DROP #{object_type} #{full_name};"
+            else
+              interpolate_sql(object[:dropsql].compact.join(";\n"), object)
+            end
+
+      begin
+        conditional_ruby(object[:beforedropruby].join("\n"), object) unless object[:beforedropruby].empty?
+        ActiveRecord::Base.connection.execute(object[:beforedropsql].join("\n")) unless object[:beforedropsql].empty?
+        ActiveRecord::Base.connection.execute(sql)
+        ActiveRecord::Base.connection.execute(object[:afterdropsql].join("\n")) unless object[:afterdropsql].empty?
+        puts "DROP #{object_type} #{full_name}... OK"
+        conditional_ruby(object[:afterdropruby].join("\n"), object) unless object[:afterdropruby].empty?
+      rescue StandardError => e
+        unless object[:debug]
+          puts '#' * 80
+          puts e.message.to_s
+          # puts "#"*80
+          # puts "#{e.backtrace}"
+          puts '#' * 80
+          puts "WARNING: #{sql}... ERROR"
+          puts '#' * 80
+        end
+      end
+
+      object[:status] = :loaded
+    end
+
+    def create_object(object)
+      return if object[:nocreate]
+
+      return if object[:status] == :loaded
+
+      object_type = object[:type]
+      name = object[:name]
+
+      raise "Error: Circular file reference! (#{object_type} #{name})" if object[:status] == :inprogress
+
+      object[:status] = :inprogress
+      @dbschema = (object[:dbschema] || []).clone
+
+      full_name = full(name)
+
+      if object[:createconditionruby].compact.any?
+        condition = conditional_ruby(object[:createconditionruby].join("\n"), object)
+        if condition
+          puts "RUBY CONDITION MET FOR #{object_type} #{full_name} / #{condition}"
+        else
+          puts "RUBY CONDITION NOT MET FOR #{object_type} #{full_name}"
+          return
+        end
+      elsif object[:condition].compact.any?
+        condition = ActiveRecord::Base.connection.select_rows(object[:condition].join("\n")).empty?
+        if condition
+          puts "CONDITION MET FOR #{object_type} #{full_name}"
+        else
+          puts "CONDITION NOT MET FOR #{object_type} #{full_name}"
+          return
+        end
+      end
+
+      if object[:debug]
+        puts '=' * 80
+        puts "CREATE OBJECT #{name} / #{object[:path]}"
+        puts '=' * 80
+        puts object.to_json
+        puts '=' * 80
+      end
+      # skip empty sql content
+      if object[:sql_content].strip.blank?
+        puts "#{object_type} #{name} EMPTY... SKIPPING"
+        return
+      end
+
+      create_dependencies(object, object_type)
+
+      sql = if object[:vanilla]
+              interpolate_sql(object[:sql_content], object)
+            elsif object[:createsql].compact.empty?
+              "CREATE #{object_type} #{full_name}\n#{object[:sql_content]}"
+            else
+              interpolate_sql(object[:createsql].compact.join(";\n"), object)
+            end
+
+      begin
+        conditional_ruby(object[:beforecreateruby].join("\n"), object) unless object[:beforecreateruby].empty?
+        ActiveRecord::Base.connection.execute(object[:beforecreatesql].join("\n")) unless object[:beforecreatesql].empty?
+        ActiveRecord::Base.connection.execute(sql.to_s)
+        ActiveRecord::Base.connection.execute(object[:aftercreatesql].join("\n")) unless object[:aftercreatesql].empty?
+        conditional_ruby(object[:aftercreateruby].join("\n"), object) unless object[:aftercreateruby].empty?
+        puts "CREATE #{object_type} #{full_name}... OK"
+      rescue StandardError => e
+        unless object[:silent]
+          puts '#' * 80
+          puts e.message.to_s
+          # puts '#' * 80
+          # puts "#{e.backtrace}"
+          puts '#' * 80
+          puts "WARNING: CREATE #{object_type} #{full_name}... ERROR"
+          puts '#' * 80
+        end
+      end
+
+      object[:status] = :loaded
+    end
+    # rubocop:enable Metrics/AbcSize
+
+    def create_dependencies(object, object_type)
+      object[:requires].each do |requirement|
+        requires = requirement.split('/', 2)
+        required_object = requires.last
+        required_type = requires.first.upcase if requires.length == 2
+        required_type ||= object_type
+        create_object(@objects[required_type.upcase][required_object])
+      end
+    end
+  end
+end

data/lib/rails_db_objects/railtie.rb

@@ -1,15 +1,15 @@
-class Railtie < Rails::Railtie
-  railtie_name :rails_db_objects
-
-  config.rails_db_objects = ActiveSupport::OrderedHash.new
-
-  initializer 'rails_db_objects.initialize' do |app|
-    app.config.rails_db_objects[:objects_path] = %w[db/objects/**/]
-    app.config.rails_db_objects[:objects_ext] = '*.sql'
-    app.config.rails_db_objects[:objects_dbschema] = ['dbo']
-  end
-
-  rake_tasks do
-    load 'tasks/rails_db_objects_tasks.rake'
-  end
-end
+class Railtie < Rails::Railtie
+  railtie_name :rails_db_objects
+
+  config.rails_db_objects = ActiveSupport::OrderedHash.new
+
+  initializer 'rails_db_objects.initialize' do |app|
+    app.config.rails_db_objects[:objects_path] = %w[db/objects/**/]
+    app.config.rails_db_objects[:objects_ext] = '*.sql'
+    app.config.rails_db_objects[:objects_dbschema] = ['dbo']
+  end
+
+  rake_tasks do
+    load 'tasks/rails_db_objects_tasks.rake'
+  end
+end

data/lib/rails_db_objects/version.rb

@@ -1,3 +1,3 @@
-module RailsDbObjects
-  VERSION = '0.1.1'.freeze
-end
+module RailsDbObjects
+  VERSION = '1.0.0'.freeze
+end

data/lib/tasks/rails_db_objects_tasks.rake

@@ -1,45 +1,45 @@
-namespace :db do
-  desc 'Generate all the database objects of the current project'
-  task create_objects: :environment do
-    creator = RailsDbObjects::DbObjectsCreator.new
-
-    objects_path = Rails.configuration.rails_db_objects[:objects_path]
-    objects_ext = Rails.configuration.rails_db_objects[:objects_ext]
-
-    objects_path.each do |path|
-      creator.register_files(Dir[File.join(path, objects_ext)].map { |x| File.expand_path(x) })
-    end
-
-    creator.create_objects
-  end
-
-  desc 'Drop all the database objects of the current project'
-  task drop_objects: :environment do
-    creator = RailsDbObjects::DbObjectsCreator.new
-
-    objects_path = Rails.configuration.rails_db_objects[:objects_path]
-    objects_ext = Rails.configuration.rails_db_objects[:objects_ext]
-
-    objects_path.each do |path|
-      creator.register_files(Dir[File.join(path, objects_ext)].map { |x| File.expand_path(x) })
-    end
-
-    creator.drop_objects
-  end
-end
-
-require 'rake/hooks'
-
-before 'db:migrate' do
-  Rake::Task['db:drop_objects'].invoke
-end
-before 'db:rollback' do
-  Rake::Task['db:drop_objects'].invoke
-end
-
-after 'db:migrate' do
-  Rake::Task['db:create_objects'].invoke
-end
-after 'db:rollback' do
-  Rake::Task['db:create_objects'].invoke
-end
+namespace :db do
+  desc 'Generate all the database objects of the current project'
+  task create_objects: :environment do
+    creator = RailsDbObjects::DbObjectsCreator.new
+
+    objects_path = Rails.configuration.rails_db_objects[:objects_path]
+    objects_ext = Rails.configuration.rails_db_objects[:objects_ext]
+
+    objects_path.each do |path|
+      creator.register_files(Dir[File.join(path, objects_ext)].map { |x| File.expand_path(x) })
+    end
+
+    creator.create_objects
+  end
+
+  desc 'Drop all the database objects of the current project'
+  task drop_objects: :environment do
+    creator = RailsDbObjects::DbObjectsCreator.new
+
+    objects_path = Rails.configuration.rails_db_objects[:objects_path]
+    objects_ext = Rails.configuration.rails_db_objects[:objects_ext]
+
+    objects_path.each do |path|
+      creator.register_files(Dir[File.join(path, objects_ext)].map { |x| File.expand_path(x) })
+    end
+
+    creator.drop_objects
+  end
+end
+
+require 'rake/hooks'
+
+before 'db:migrate' do
+  Rake::Task['db:drop_objects'].invoke
+end
+before 'db:rollback' do
+  Rake::Task['db:drop_objects'].invoke
+end
+
+after 'db:migrate' do
+  Rake::Task['db:create_objects'].invoke
+end
+after 'db:rollback' do
+  Rake::Task['db:create_objects'].invoke
+end

data/test/dummy/Rakefile

@@ -1,6 +1,6 @@
-# Add your own tasks in files placed in lib/tasks ending in .rake,
-# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
-
-require File.expand_path('config/application', __dir__)
-
-Rails.application.load_tasks
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
+require File.expand_path('config/application', __dir__)
+
+Rails.application.load_tasks

data/test/dummy/bin/bundle

@@ -1,3 +1,3 @@
-#!/usr/bin/env ruby.exe
-ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../Gemfile', __dir__)
-load Gem.bin_path('bundler', 'bundle')
+#!/usr/bin/env ruby.exe
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../Gemfile', __dir__)
+load Gem.bin_path('bundler', 'bundle')

data/test/dummy/bin/rails

@@ -1,4 +1,4 @@
-#!/usr/bin/env ruby.exe
-APP_PATH = File.expand_path('../config/application', __dir__)
-require_relative '../config/boot'
-require 'rails/commands'
+#!/usr/bin/env ruby.exe
+APP_PATH = File.expand_path('../config/application', __dir__)
+require_relative '../config/boot'
+require 'rails/commands'

data/test/dummy/bin/rake

@@ -1,4 +1,4 @@
-#!/usr/bin/env ruby.exe
-require_relative '../config/boot'
-require 'rake'
-Rake.application.run
+#!/usr/bin/env ruby.exe
+require_relative '../config/boot'
+require 'rake'
+Rake.application.run

data/test/dummy/bin/setup

@@ -1,29 +1,29 @@
-#!/usr/bin/env ruby.exe
-require 'pathname'
-
-# path to your application root.
-APP_ROOT = Pathname.new File.expand_path('..', __dir__)
-
-Dir.chdir APP_ROOT do
-  # This script is a starting point to setup your application.
-  # Add necessary setup steps to this file:
-
-  puts '== Installing dependencies =='
-  system 'gem install bundler --conservative'
-  system 'bundle check || bundle install'
-
-  # puts "\n== Copying sample files =="
-  # unless File.exist?("config/database.yml")
-  #   system "cp config/database.yml.sample config/database.yml"
-  # end
-
-  puts "\n== Preparing database =="
-  system 'bin/rake db:setup'
-
-  puts "\n== Removing old logs and tempfiles =="
-  system 'rm -f log/*'
-  system 'rm -rf tmp/cache'
-
-  puts "\n== Restarting application server =="
-  system 'touch tmp/restart.txt'
-end
+#!/usr/bin/env ruby.exe
+require 'pathname'
+
+# path to your application root.
+APP_ROOT = Pathname.new File.expand_path('..', __dir__)
+
+Dir.chdir APP_ROOT do
+  # This script is a starting point to setup your application.
+  # Add necessary setup steps to this file:
+
+  puts '== Installing dependencies =='
+  system 'gem install bundler --conservative'
+  system 'bundle check || bundle install'
+
+  # puts "\n== Copying sample files =="
+  # unless File.exist?("config/database.yml")
+  #   system "cp config/database.yml.sample config/database.yml"
+  # end
+
+  puts "\n== Preparing database =="
+  system 'bin/rake db:setup'
+
+  puts "\n== Removing old logs and tempfiles =="
+  system 'rm -f log/*'
+  system 'rm -rf tmp/cache'
+
+  puts "\n== Restarting application server =="
+  system 'touch tmp/restart.txt'
+end

data/test/dummy/config/application.rb

@@ -1,25 +1,25 @@
-require File.expand_path('boot', __dir__)
-
-require 'rails/all'
-
-Bundler.require(*Rails.groups)
-require 'rails_db_objects'
-
-module Dummy
-  class Application < Rails::Application
-    # Settings in config/environments/* take precedence over those specified here.
-    # Application configuration should go into files in config/initializers
-    # -- all .rb files in that directory are automatically loaded.
-
-    # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
-    # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
-    # config.time_zone = 'Central Time (US & Canada)'
-
-    # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
-    # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]
-    # config.i18n.default_locale = :de
-
-    # Do not swallow errors in after_commit/after_rollback callbacks.
-    config.active_record.raise_in_transactional_callbacks = true
-  end
-end
+require File.expand_path('boot', __dir__)
+
+require 'rails/all'
+
+Bundler.require(*Rails.groups)
+require 'rails_db_objects'
+
+module Dummy
+  class Application < Rails::Application
+    # Settings in config/environments/* take precedence over those specified here.
+    # Application configuration should go into files in config/initializers
+    # -- all .rb files in that directory are automatically loaded.
+
+    # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
+    # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
+    # config.time_zone = 'Central Time (US & Canada)'
+
+    # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
+    # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]
+    # config.i18n.default_locale = :de
+
+    # Do not swallow errors in after_commit/after_rollback callbacks.
+    config.active_record.raise_in_transactional_callbacks = true
+  end
+end

data/test/dummy/config/boot.rb

@@ -1,5 +1,5 @@
-# Set up gems listed in the Gemfile.
-ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../../Gemfile', __dir__)
-
-require 'bundler/setup' if File.exist?(ENV['BUNDLE_GEMFILE'])
-$LOAD_PATH.unshift File.expand_path('../../../lib', __dir__)
+# Set up gems listed in the Gemfile.
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../../Gemfile', __dir__)
+
+require 'bundler/setup' if File.exist?(ENV['BUNDLE_GEMFILE'])
+$LOAD_PATH.unshift File.expand_path('../../../lib', __dir__)

data/test/dummy/config/environment.rb

@@ -1,5 +1,5 @@
-# Load the Rails application.
-require File.expand_path('application', __dir__)
-
-# Initialize the Rails application.
-Rails.application.initialize!
+# Load the Rails application.
+require File.expand_path('application', __dir__)
+
+# Initialize the Rails application.
+Rails.application.initialize!

data/test/rails_db_objects_test.rb

@@ -1,7 +1,7 @@
-require 'test_helper'
-
-class RailsDbObjectsTest < ActiveSupport::TestCase
-  test 'truth' do
-    assert_kind_of Module, RailsDbObjects
-  end
-end
+require 'test_helper'
+
+class RailsDbObjectsTest < ActiveSupport::TestCase
+  test 'truth' do
+    assert_kind_of Module, RailsDbObjects
+  end
+end

data/test/test_helper.rb

@@ -1,20 +1,20 @@
-# Configure Rails Environment
-ENV['RAILS_ENV'] = 'test'
-
-require File.expand_path('../test/dummy/config/environment.rb', __dir__)
-ActiveRecord::Migrator.migrations_paths = [File.expand_path('../test/dummy/db/migrate', __dir__)]
-require 'rails/test_help'
-
-# Filter out Minitest backtrace while allowing backtrace from other libraries
-# to be shown.
-Minitest.backtrace_filter = Minitest::BacktraceFilter.new
-
-# Load support files
-Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each { |f| require f }
-
-# Load fixtures from the engine
-if ActiveSupport::TestCase.respond_to?(:fixture_path=)
-  ActiveSupport::TestCase.fixture_path = File.expand_path('fixtures', __dir__)
-  ActionDispatch::IntegrationTest.fixture_path = ActiveSupport::TestCase.fixture_path
-  ActiveSupport::TestCase.fixtures :all
-end
+# Configure Rails Environment
+ENV['RAILS_ENV'] = 'test'
+
+require File.expand_path('../test/dummy/config/environment.rb', __dir__)
+ActiveRecord::Migrator.migrations_paths = [File.expand_path('../test/dummy/db/migrate', __dir__)]
+require 'rails/test_help'
+
+# Filter out Minitest backtrace while allowing backtrace from other libraries
+# to be shown.
+Minitest.backtrace_filter = Minitest::BacktraceFilter.new
+
+# Load support files
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each { |f| require f }
+
+# Load fixtures from the engine
+if ActiveSupport::TestCase.respond_to?(:fixture_path=)
+  ActiveSupport::TestCase.fixture_path = File.expand_path('fixtures', __dir__)
+  ActionDispatch::IntegrationTest.fixture_path = ActiveSupport::TestCase.fixture_path
+  ActiveSupport::TestCase.fixtures :all
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails_db_objects
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 1.0.0
 platform: ruby
 authors:
 - ''
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-10-18 00:00:00.000000000 Z
+date: 2019-01-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-hooks