better_record 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: b1f24408013032454f5f6384cfa1723d81fd5b741c544f35ce95352fe4c334a8
+  data.tar.gz: 5476435865403fef0645f2502eaf10a412fe27d222b43568e5eb7f557f60f558
+SHA512:
+  metadata.gz: 307dc6781a0ae40f1d66ebb7f9e8ea7a4e4fec739554a314c0dceb4302b0f3bbaa928dfacf7b8649ec99b87bebd243ac107446d4ad4b61d5e91ef054d7156e6c
+  data.tar.gz: 3fc5c79c31830dacda3dc341c18584ed0c30ede6aa915011c40bcad19acbb78a3c65c97b4ea3754324077c1edce7215bb6c3837a4e35f0382de9036e3d1199b6

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright 2018 Sampson Crowley
+
+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.md

@@ -0,0 +1,28 @@
+# BetterRecord
+Short description and motivation.
+
+## Usage
+How to use my plugin.
+
+## Installation
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'better_record'
+```
+
+And then execute:
+```bash
+$ bundle
+```
+
+Or install it yourself as:
+```bash
+$ gem install better_record
+```
+
+## Contributing
+Contribution directions go here.
+
+## License
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,27 @@
+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    = 'BetterRecord'
+  rdoc.options << '--line-numbers'
+  rdoc.rdoc_files.include('README.md')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+require 'bundler/gem_tasks'
+
+require 'rake/testtask'
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << 'test'
+  t.pattern = 'test/**/*_test.rb'
+  t.verbose = false
+end
+
+task default: :test

data/app/assets/config/better_record_manifest.js

@@ -0,0 +1,2 @@
+//= link_directory ../javascripts/better_record .js
+//= link_directory ../stylesheets/better_record .css

data/app/controllers/better_record/application_controller.rb

@@ -0,0 +1,5 @@
+module BetterRecord
+  class ApplicationController < ActionController::Base
+    protect_from_forgery with: :exception
+  end
+end

data/app/helpers/better_record/application_helper.rb

@@ -0,0 +1,4 @@
+module BetterRecord
+  module ApplicationHelper
+  end
+end

data/app/jobs/better_record/application_job.rb

@@ -0,0 +1,4 @@
+module BetterRecord
+  class ApplicationJob < ActiveJob::Base
+  end
+end

data/app/mailers/better_record/application_mailer.rb

@@ -0,0 +1,6 @@
+module BetterRecord
+  class ApplicationMailer < ActionMailer::Base
+    default from: 'from@example.com'
+    layout 'mailer'
+  end
+end

data/app/models/better_record/audit.rb

@@ -0,0 +1,50 @@
+module BetterRecord
+  class Audit < Base
+    # == Constants ============================================================
+    ACTIONS = {
+      D: 'DELETE',
+      I: 'INSERT',
+      U: 'UPDATE',
+      T: 'TRUNCATE',
+    }.with_indifferent_access
+
+    # == Attributes ===========================================================
+    self.table_name = 'audit.logged_actions'
+
+    # == Extensions ===========================================================
+
+    # == Relationships ========================================================
+    belongs_to :audited,
+      polymorphic: :true,
+      primary_type: :table_name,
+      foreign_key: :row_id,
+      foreign_type: :table_name
+    # == Validations ==========================================================
+
+    # == Scopes ===============================================================
+
+    # == Callbacks ============================================================
+
+    # == Class Methods ========================================================
+    def self.default_print
+      [
+        :event_id,
+        :row_id,
+        :table_name,
+        :app_user_id,
+        :app_user_type,
+        :action_type,
+        :changed_columns
+      ]
+    end
+
+    # == Instance Methods =====================================================
+    def changed_columns
+      (self.changed_fields || {}).keys.join(', ').presence || 'N/A'
+    end
+
+    def action_type
+      ACTIONS[action] || 'UNKNOWN'
+    end
+  end
+end

data/app/models/better_record/base.rb

@@ -0,0 +1,157 @@
+module BetterRecord
+  class Base < ActiveRecord::Base
+    self.abstract_class = true
+
+    # == Constants ============================================================
+
+    # == Attributes ===========================================================
+
+    # == Extensions ===========================================================
+
+    # == Relationships ========================================================
+    if (ha = BetterRecord.has_audits_by_default)
+      has_many :audits,
+        class_name: 'BetterRecord::Audit',
+        primary_type: :table_name,
+        foreign_key: :row_id,
+        foreign_type: :table_name,
+        as: :audits
+    end
+
+    # == Validations ==========================================================
+    before_validation :set_booleans
+
+    # == Scopes ===============================================================
+
+    # == Callbacks ============================================================
+
+    # == Class Methods ========================================================
+    def self.boolean_columns
+      []
+    end
+
+    def self.column_comments(prefix = '')
+      longest_name = 0
+      column_names.each {|nm| longest_name = nm.length if nm.length > longest_name}
+      longest_name += 1
+      str = ''
+      columns.each do |col|
+        unless col.name == 'id'
+          spaces = "#{' ' * (longest_name - col.name.length)}"
+          is_required = "#{col.null == false ? ', required' : ''}"
+          is_default = "#{col.default ? ", default: #{col.default}" : ''}"
+          str << "#{prefix}##{spaces}#{col.name}: :#{col.type}#{is_required}\n"
+        end
+      end
+      str
+    end
+
+    def self.default_print
+      column_names
+    end
+
+    def self.find_or_retry_by(*args)
+      found = nil
+      retries = 0
+      begin
+        raise ActiveRecord::RecordNotFound unless found = find_by(*args)
+        return found
+      rescue
+        sleep retries += 1
+        retry if (retries) < 5
+      end
+      found
+    end
+
+    def self.queue_adapter_inline?
+      @@queue_adapter ||= Rails.application.config.active_job.queue_adapter
+      @@queue_adapter == :inline
+    end
+
+    def self.table_name_has_schema?
+      @@table_name_has_schema ||= (table_name =~ /\w+\.\w+/)
+    end
+
+    def self.table_name_without_schema
+      @@table_name_without_schema ||= (table_name =~ /\w+\.\w+/) ? table_name.split('.').last : table_name
+    end
+
+    def self.table_schema
+      @@table_schema ||= table_name_has_schema? ? table_name.split('.').first : 'public'
+    end
+
+    def self.table_size
+      BetterRecord::TableSize.unscoped.find_by(name: table_name_without_schema, schema: table_schema)
+    end
+
+    def self.transaction(*args)
+      super(*args) do
+        if Current.user
+          ip = Current.ip_address ? "'#{Current.ip_address}'" : 'NULL'
+
+          ActiveRecord::Base.connection.execute <<-SQL
+            CREATE TEMP TABLE IF NOT EXISTS
+              "_app_user" (user_id integer, user_type text, ip_address inet);
+
+            UPDATE
+              _app_user
+            SET
+              user_id=#{Current.user.id},
+              user_type='#{Current.user.class.to_s}',
+              ip_address=#{ip};
+
+            INSERT INTO
+              _app_user (user_id, user_type, ip_address)
+            SELECT
+              #{Current.user.id},
+              '#{Current.user.class.to_s}',
+              #{ip}
+            WHERE NOT EXISTS (SELECT * FROM _app_user);
+          SQL
+        end
+
+        yield
+      end
+    end
+
+    # == Instance Methods =====================================================
+    def queue_adapter_inline?
+      self.class.queue_adapter_inline?
+    end
+
+    %w(path url).each do |cat|
+      self.__send__ :define_method, :"rails_blob_#{cat}" do |*args|
+        Rails.application.routes.url_helpers.__send__ :"rails_blob_#{cat}", *args
+      end
+    end
+
+    def purge(attached)
+      attached.__send__ queue_adapter_inline? ? :purge : :purge_later
+    end
+
+    def boolean_columns
+      self.class.boolean_columns
+    end
+
+    private
+      # def table_name_has_schema?
+      #   self.class.table_name_has_schema?
+      # end
+      #
+      # def table_schema
+      #   self.class.table_schema
+      # end
+      #
+      # def table_name_without_schema
+      #   self.class.table_name_without_schema
+      # end
+
+      def set_booleans
+        self.class.boolean_columns.each do |nm|
+          self.__send__("#{nm}=", __send__("#{nm}=", !!Boolean.parse(__send__ nm)))
+        end
+        true
+      end
+
+  end
+end

data/app/models/better_record/table_size.rb

@@ -0,0 +1,77 @@
+module BetterRecord
+  class TableSize < Base
+    # == Constants ============================================================
+    CREATE_TABLE_SQL = <<-SQL
+      DROP TABLE IF EXISTS table_sizes;
+      CREATE TEMPORARY TABLE table_sizes AS (
+        SELECT
+          *,
+          pg_size_pretty(total_bytes) AS total,
+          pg_size_pretty(idx_bytes) AS idx,
+          pg_size_pretty(toast_bytes) AS toast,
+          pg_size_pretty(tbl_bytes) AS tbl
+        FROM (
+          SELECT
+            *,
+            total_bytes - idx_bytes - COALESCE(toast_bytes,0) AS tbl_bytes
+          FROM (
+            SELECT c.oid,nspname AS schema, relname AS name
+            , c.reltuples AS row_estimate
+            , pg_total_relation_size(c.oid) AS total_bytes
+            , pg_indexes_size(c.oid) AS idx_bytes
+            , pg_total_relation_size(reltoastrelid) AS toast_bytes
+            FROM pg_class c
+            LEFT JOIN pg_namespace n ON n.oid = c.relnamespace
+            WHERE relkind = 'r'
+          ) table_sizes
+        ) table_sizes
+      )
+    SQL
+
+    # == Attributes ===========================================================
+    self.primary_key = :oid
+    self.table_name = 'table_sizes'
+
+    # == Extensions ===========================================================
+
+    # == Relationships ========================================================
+
+    # == Validations ==========================================================
+
+    # == Scopes ===============================================================
+    default_scope { where(schema: [ :public ]) }
+    # == Callbacks ============================================================
+
+    # == Class Methods ========================================================
+    def self.load_schema(reload = false)
+      unless @loaded_schema && !reload
+        connection.execute CREATE_TABLE_SQL
+        @loaded_schema = true
+      end
+
+      super()
+    end
+
+    def self.find_by(*args)
+      load_schema(true)
+      super *args
+    end
+
+    # def self.default_print
+    #   [
+    #     :table_schema,
+    #     :table_name,
+    #     :
+    #   ]
+    # end
+
+    # == Instance Methods =====================================================
+    def changed_columns
+      (self.changed_fields || {}).keys.join(', ').presence || 'N/A'
+    end
+
+    def action_type
+      ACTIONS[action] || 'UNKNOWN'
+    end
+  end
+end

data/config/routes.rb

@@ -0,0 +1,2 @@
+BetterRecord::Engine.routes.draw do
+end

data/lib/better_record.rb

@@ -0,0 +1,37 @@
+require "active_support"
+
+class Boolean
+  def self.parse(value)
+    ActiveRecord::Type::Boolean.new.cast(value)
+  end
+end
+
+class Object
+  def yes_no_to_s
+    !!self == self ? (self ? 'yes' : 'no') : to_s
+  end
+end
+
+module BetterRecord
+  class << self
+    attr_accessor :default_polymorphic_method, :db_audit_schema, :has_audits_by_default
+  end
+  self.default_polymorphic_method = :polymorphic_name
+  self.db_audit_schema = ENV.fetch('DB_AUDIT_SCHEMA') { 'auditing' }
+  self.has_audits_by_default = Boolean.parse(ENV.fetch('BR_ADD_HAS_MANY') { false })
+end
+
+Dir.glob("#{File.expand_path(__dir__)}/better_record/*").each do |d|
+  require d
+end
+
+ActiveSupport.on_load(:active_record) do
+  module ActiveRecord
+    include BetterRecord::Associations
+    include BetterRecord::Batches
+    include BetterRecord::Migration
+    include BetterRecord::Reflection
+    include BetterRecord::Relation
+  end
+  include BetterRecord::NullifyBlankAttributes
+end

data/lib/better_record/associations.rb

@@ -0,0 +1,75 @@
+module BetterRecord
+  module Associations
+    class AssociationScope #:nodoc:
+      def self.get_bind_values(owner, chain)
+        binds = []
+        last_reflection = chain.last
+
+        binds << last_reflection.join_id_for(owner)
+        if last_reflection.type
+          binds << owner.class.__send__(last_reflection.options[:primary_type].presence || BetterRecord.default_polymorphic_method.presence || :polymorphic_name)
+        end
+
+        chain.each_cons(2).each do |reflection, next_reflection|
+          if reflection.type
+            binds << next_reflection.klass.__send__(reflection.options[:primary_type].presence || next_reflection[:primary_type].presence || BetterRecord.default_polymorphic_method.presence || :polymorphic_name)
+          end
+        end
+        binds
+      end
+
+      private
+        def last_chain_scope(scope, reflection, owner)
+          join_keys = reflection.join_keys
+          key = join_keys.key
+          foreign_key = join_keys.foreign_key
+
+          table = reflection.aliased_table
+          value = transform_value(owner[foreign_key])
+          scope = apply_scope(scope, table, key, value)
+
+          if reflection.type
+            polymorphic_type = transform_value(owner.class.__send__(reflection.options[:primary_type].presence || BetterRecord.default_polymorphic_method.presence || :polymorphic_name))
+            scope = apply_scope(scope, table, reflection.type, polymorphic_type)
+          end
+
+          scope
+        end
+
+        def next_chain_scope(scope, reflection, next_reflection)
+          join_keys = reflection.join_keys
+          key = join_keys.key
+          foreign_key = join_keys.foreign_key
+
+          table = reflection.aliased_table
+          foreign_table = next_reflection.aliased_table
+          constraint = table[key].eq(foreign_table[foreign_key])
+
+          if reflection.type
+            value = transform_value(next_reflection.klass.__send__(reflection.options[:primary_type].presence || BetterRecord.default_polymorphic_method.presence || :polymorphic_name))
+            scope = apply_scope(scope, table, reflection.type, value)
+          end
+
+          scope.joins!(join(foreign_table, constraint))
+        end
+    end
+
+    module Builder
+      class Association
+        def self.valid_options(options)
+          VALID_OPTIONS + [ :primary_type ] + Association.extensions.flat_map(&:valid_options)
+        end
+      end
+    end
+
+    class BelongsToAssociation
+    end
+
+    class BelongsToPolymorphicAssociation < BelongsToAssociation
+      def klass
+        type = owner[reflection.foreign_type]
+        type.presence && type.capitalize.singularize.constantize
+      end
+    end
+  end
+end

data/lib/better_record/batches.rb

@@ -0,0 +1,27 @@
+module BetterRecord
+  module Batches
+    def split_batches(options = {}, &block)
+      options.assert_valid_keys(:start, :batch_size, :preserve_order)
+      if block_given? && arel.orders.present? && options[:preserve_order]
+        relation = self
+        offset = options[:start] || 0
+        batch_size = options[:batch_size] || 1000
+
+        total = relation.count(:*)
+        records = relation.limit(batch_size).offset(offset).to_a
+        while records.any?
+          records_size = records.size
+
+          block.call records
+
+          break if records_size < batch_size
+          offset += batch_size
+          records = relation.limit(batch_size).offset(offset).to_a
+        end
+        nil
+      else
+        find_in_batches(options.except(:preserve_order), &block)
+      end
+    end
+  end
+end

data/lib/better_record/engine.rb

@@ -0,0 +1,5 @@
+module BetterRecord
+  class Engine < ::Rails::Engine
+    isolate_namespace BetterRecord
+  end
+end

data/lib/better_record/migration.rb

@@ -0,0 +1,101 @@
+module BetterRecord
+  module Migration
+    def audit_table(table_name, rows = nil, query_text = nil, skip_columns = nil)
+      reversible do |d|
+        d.up do
+          if(rows && rows.is_a?(Array))
+            skip_columns = rows
+            rows = true
+            query_text = true
+          end
+
+          if(rows.nil?)
+            execute "SELECT #{BetterRecord.db_audit_schema}.audit_table('#{table_name}');"
+          else
+            rows = !!rows ? 't' : 'f'
+            query_text = !!query_text ? 't' : 'f'
+            skip_columns = skip_columns.present? ? "'{#{skip_columns.join(',')}}'" : 'ARRAY[]'
+            execute "SELECT #{BetterRecord.db_audit_schema}.audit_table('#{table_name}', BOOLEAN '#{rows}', BOOLEAN '#{query_text}', #{skip_columns}::text[]);"
+          end
+        end
+
+        d.down do
+          execute "DROP TRIGGER audit_trigger_row ON #{table_name};"
+          execute "DROP TRIGGER audit_trigger_stm ON #{table_name};"
+        end
+      end
+    end
+
+    def login_triggers(table_name, password_col = 'password', email_col = 'email')
+      table_name = table_name.to_s
+
+      reversible do |d|
+        d.up do
+          password_text = ''
+
+          if !!password_col
+            password_text = <<-SQL
+              IF (NEW.#{password_col} IS NOT NULL)
+              AND (
+                (TG_OP = 'INSERT') OR ( NEW.#{password_col} IS DISTINCT FROM OLD.#{password_col} )
+              ) THEN
+                NEW.#{password_col} = hash_password(NEW.#{password_col});
+              ELSE
+                IF (TG_OP IS DISTINCT FROM 'INSERT') THEN
+                  NEW.#{password_col} = OLD.#{password_col};
+                ELSE
+                  NEW.#{password_col} = NULL;
+                END IF;
+              END IF;
+
+            SQL
+          end
+
+          email_text = ''
+
+          if !!email_col
+            email_text = <<-SQL
+              IF (TG_OP = 'INSERT') OR ( NEW.#{email_col} IS DISTINCT FROM OLD.#{email_col} ) THEN
+                NEW.#{email_col} = validate_email(NEW.#{email_col});
+              END IF;
+
+            SQL
+          end
+
+          execute <<-SQL
+            CREATE OR REPLACE FUNCTION #{table_name.singularize}_changed()
+              RETURNS TRIGGER AS
+            $BODY$
+            BEGIN
+              #{password_text}
+              #{email_text}
+              RETURN NEW;
+            END;
+            $BODY$
+            language 'plpgsql';
+          SQL
+
+          execute <<-SQL
+            CREATE TRIGGER #{table_name}_on_insert
+            BEFORE INSERT ON #{table_name}
+            FOR EACH ROW
+            EXECUTE PROCEDURE #{table_name.singularize}_changed();
+          SQL
+
+          execute <<-SQL
+            CREATE TRIGGER #{table_name}_on_update
+            BEFORE UPDATE ON #{table_name}
+            FOR EACH ROW
+            EXECUTE PROCEDURE #{table_name.singularize}_changed();
+
+          SQL
+        end
+
+        d.down do
+          execute "DROP TRIGGER IF EXISTS #{table_name}_on_insert ON #{table_name};"
+          execute "DROP TRIGGER IF EXISTS #{table_name}_on_update ON #{table_name};"
+        end
+      end
+    end
+  end
+end

data/lib/better_record/nullify_blank_attributes.rb

@@ -0,0 +1,8 @@
+module BetterRecord
+  module NullifyBlankAttributes
+    def write_attribute(attr_name, value)
+      new_value = value == false ? false : value.presence
+      super(attr_name, new_value)
+    end
+  end
+end

data/lib/better_record/railtie.rb

@@ -0,0 +1,4 @@
+module BetterRecord
+  class Railtie < ::Rails::Railtie
+  end
+end

data/lib/better_record/reflection.rb

@@ -0,0 +1,21 @@
+module BetterRecord
+  module Reflection
+    class AbstractReflection
+      def join_scope(table, foreign_klass)
+        predicate_builder = predicate_builder(table)
+        scope_chain_items = join_scopes(table, predicate_builder)
+        klass_scope       = klass_join_scope(table, predicate_builder)
+
+        if type
+          klass_scope.where!(type => foreign_klass.__send__(options[:primary_type] || :table_name))
+        end
+
+        scope_chain_items.inject(klass_scope, &:merge!)
+      end
+    end
+
+    class RuntimeReflection < AbstractReflection # :nodoc:
+      delegate :scope, :type, :constraints, :get_join_keys, :options, to: :@reflection
+    end
+  end
+end

data/lib/better_record/relation.rb

@@ -0,0 +1,62 @@
+module BetterRecord
+  module Relation
+    # pluck_in_batches:  yields an array of *columns that is at least size
+    #                    batch_size to a block.
+    #
+    #                    Special case: if there is only one column selected than each batch
+    #                                  will yield an array of columns like [:column, :column, ...]
+    #                                  rather than [[:column], [:column], ...]
+    # Arguments
+    #   columns      ->  an arbitrary selection of columns found on the table.
+    #   batch_size   ->  How many items to pluck at a time
+    #   &block       ->  A block that processes an array of returned columns.
+    #                    Array is, at most, size batch_size
+    #
+    # Returns
+    #   nothing is returned from the function
+    def pluck_in_batches(*columns, batch_size: 1000)
+      if columns.empty?
+        raise "There must be at least one column to pluck"
+      end
+
+      # the :id to start the query at
+      batch_start = nil
+
+      # It's cool. We're only taking in symbols
+      # no deep clone needed
+      select_columns = columns.dup
+
+      # Find index of :id in the array
+      remove_id_from_results = false
+      id_index = columns.index(primary_key.to_sym)
+
+      # :id is still needed to calculate offsets
+      # add it to the front of the array and remove it when yielding
+      if id_index.nil?
+        id_index = 0
+        select_columns.unshift(primary_key)
+
+        remove_id_from_results = true
+      end
+
+      loop do
+        relation = self.reorder(table[primary_key].asc).limit(batch_size)
+        relation = relation.where(table[primary_key].gt(batch_start)) if batch_start
+        items = relation.pluck(*select_columns)
+
+        break if items.empty?
+
+        # Use the last id to calculate where to offset queries
+        last_item = items.last
+        batch_start = last_item.is_a?(Array) ? last_item[id_index] : last_item
+
+        # Remove :id column if not in *columns
+        items.map! { |row| row[1..-1] } if remove_id_from_results
+
+        yield items
+
+        break if items.size < batch_size
+      end
+    end
+  end
+end

data/lib/better_record/version.rb

@@ -0,0 +1,3 @@
+module BetterRecord
+  VERSION = '0.1.0'
+end

data/lib/generators/create_helper_functions/USAGE

@@ -0,0 +1,9 @@
+Description:
+    Explain the generator
+
+Example:
+    rails generate create_helper_functions --schema_name audit
+
+    This will create:
+      db/migrate/*_create_database_helper_functions.rb
+      db/postgres-audit-trigger.psql

data/lib/generators/create_helper_functions/create_helper_functions_generator.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+require 'rails/generators/active_record'
+
+class CreateHelperFunctionsGenerator < ActiveRecord::Generators::Base
+  source_root File.expand_path('templates', __dir__)
+  argument :name, type: :string, default: 'testes'
+  class_option :audit_schema, type: :string, default: 'audit'
+  class_option :eject, type: :boolean, default: false
+
+  def copy_initializer
+    template 'gitattributes', '.gitattributes'
+    template 'gitignore', '.gitignore'
+    template 'jsbeautifyrc', '.jsbeautifyrc'
+    template 'pryrc', '.pryrc'
+    template 'ruby-version', '.ruby-version'
+    template 'postgres-audit-trigger.psql', 'db/postgres-audit-trigger.psql'
+    if !!options['eject']
+      template 'initializer.rb', 'config/initializers/better_record.rb'
+    else
+      template 'initializer.rb', 'config/initializers/better_record.rb'
+    end
+    migration_template "migration.rb", "#{migration_path}/create_database_helper_functions.rb", migration_version: migration_version, force: true
+    application ''
+
+    eager_line = 'config.eager_load_paths += Dir["#{config.root}/lib/modules/**/"]'
+
+    gsub_file 'config/application.rb', /([ \t]*?#{Regexp.escape(eager_line)}[ \t0-9\.]*?)\n/mi do |match|
+      ""
+    end
+
+    gsub_file 'config/application.rb', /#{Regexp.escape("config.load_defaults")}[ 0-9\.]+\n/mi do |match|
+      "#{match}    #{'config.eager_load_paths += Dir["#{config.root}/lib/modules/**/"]'}\n"
+    end
+
+    gsub_file 'app/models/application_record.rb', /(#{Regexp.escape("class ApplicationRecord < ActiveRecord::Base")})/mi do |match|
+      p match
+      "class ApplicationRecord < BetterRecord::Base"
+    end
+  end
+
+  def audit_schema
+    @audit_schema ||= options['audit_schema'].presence || 'audit'
+  end
+
+  private
+
+    def migration_path
+      if Rails.version >= '5.0.3'
+        db_migrate_path
+      else
+        @migration_path ||= File.join("db", "migrate")
+      end
+    end
+
+    def migration_version
+      if Rails.version.start_with? '5'
+        "[#{Rails::VERSION::MAJOR}.#{Rails::VERSION::MINOR}]"
+      end
+    end
+end

metadata

@@ -0,0 +1,102 @@
+--- !ruby/object:Gem::Specification
+name: better_record
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Sampson Crowley
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-07-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.2.0
+- !ruby/object:Gem::Dependency
+  name: pg
+  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'
+description: |2
+    This app extends active record to allow you to change the polymorphic type value in relationships.
+    It also extends active record batching functions to be able to order records while batching.
+      - As a bonus. the same 'split_batches' function is made available to Arrays
+    It also adds optional auditing functions and password hashing functions, as well as migration helpers
+email:
+- sampsonsprojects@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- MIT-LICENSE
+- README.md
+- Rakefile
+- app/assets/config/better_record_manifest.js
+- app/controllers/better_record/application_controller.rb
+- app/helpers/better_record/application_helper.rb
+- app/jobs/better_record/application_job.rb
+- app/mailers/better_record/application_mailer.rb
+- app/models/better_record/audit.rb
+- app/models/better_record/base.rb
+- app/models/better_record/table_size.rb
+- config/routes.rb
+- lib/better_record.rb
+- lib/better_record/associations.rb
+- lib/better_record/batches.rb
+- lib/better_record/engine.rb
+- lib/better_record/migration.rb
+- lib/better_record/nullify_blank_attributes.rb
+- lib/better_record/railtie.rb
+- lib/better_record/reflection.rb
+- lib/better_record/relation.rb
+- lib/better_record/version.rb
+- lib/generators/create_helper_functions/USAGE
+- lib/generators/create_helper_functions/create_helper_functions_generator.rb
+- lib/tasks/db/create_audits.rake
+homepage: https://github.com/SampsonCrowley/multi_app_active_record
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.7.6
+signing_key: 
+specification_version: 4
+summary: Fix functions that are lacking in Active record to be compatible with multi-app
+  databases
+test_files: []