Contents

require 'pg_inheritance/active_record/inheritable'

module PGInheritance
  module ActiveRecord
    module SchemaStatements
      include PGInheritance::ActiveRecord::Inheritable

      INDEX_OPTIONS = %i[unique using where orders name].freeze

      def create_table(table_name, options = {})
        parent_table = options.delete(:inherits)

        prepare_options(parent_table, options)

        table_name = prepare_table_name(table_name, options)

        super(table_name, options)
      end

      private

      def prepare_options(inherited_table, options)
        remove_pk_option(options)
        add_inheritance_query(inherited_table, options)
      end

      def remove_pk_option(options)
        return unless options[:inherits]

        options[:id] = false
        options.delete(:primary_key)
      end

      def add_inheritance_query(inherited_table, options)
        return unless inherited_table

        options[:options] = build_inherits_option(inherited_table)
      end

      def prepare_table_name(table_name, options)
        schema = options.delete(:schema)

        return table_name unless schema

        "#{schema}.#{table_name}"
      end

      def build_inherits_option(inherited_table)
        "INHERITS (#{inherited_table})"
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
pg_inheritance-0.1.2	lib/pg_inheritance/active_record/schema_statements.rb
pg_inheritance-0.1.1	lib/pg_inheritance/active_record/schema_statements.rb