db_schema 0.1

data/lib/db_schema/configuration.rb

@@ -0,0 +1,29 @@
+module DbSchema
+  class Configuration
+    attr_reader :adapter, :host, :port, :database, :user, :password
+
+    def initialize(adapter: 'postgres', host: 'localhost', port: 5432, database:, user: nil, password: '', log_changes: true, dry_run: false, post_check: true)
+      @adapter     = adapter
+      @host        = host
+      @port        = port
+      @database    = database
+      @user        = user
+      @password    = password
+      @log_changes = log_changes
+      @dry_run     = dry_run
+      @post_check  = post_check
+    end
+
+    def log_changes?
+      @log_changes
+    end
+
+    def dry_run?
+      @dry_run
+    end
+
+    def post_check_enabled?
+      @post_check
+    end
+  end
+end

data/lib/db_schema/definitions.rb

@@ -0,0 +1,122 @@
+require 'dry/equalizer'
+
+module DbSchema
+  module Definitions
+    class Index
+      include Dry::Equalizer(:name, :fields, :unique?, :type, :condition)
+      attr_reader :name, :fields, :type, :condition
+
+      def initialize(name:, fields:, unique: false, type: :btree, condition: nil)
+        @name      = name.to_sym
+        @fields    = fields
+        @unique    = unique
+        @type      = type
+        @condition = condition
+      end
+
+      def unique?
+        @unique
+      end
+
+      def btree?
+        type == :btree
+      end
+
+      class Field
+        include Dry::Equalizer(:name, :order, :nulls)
+        attr_reader :name, :order, :nulls
+
+        def initialize(name, order: :asc, nulls: order == :asc ? :last : :first)
+          @name  = name
+          @order = order
+          @nulls = nulls
+        end
+
+        def asc?
+          @order == :asc
+        end
+
+        def desc?
+          @order == :desc
+        end
+
+        def to_sequel
+          if asc?
+            Sequel.asc(name, nulls: nulls)
+          else
+            Sequel.desc(name, nulls: nulls)
+          end
+        end
+      end
+    end
+
+    class ForeignKey
+      include Dry::Equalizer(:name, :fields, :table, :keys, :on_update, :on_delete, :deferrable?)
+      attr_reader :name, :fields, :table, :keys, :on_update, :on_delete
+
+      def initialize(name:, fields:, table:, keys: [], on_update: :no_action, on_delete: :no_action, deferrable: false)
+        @name       = name
+        @fields     = fields
+        @table      = table
+        @keys       = keys
+        @on_update  = on_update
+        @on_delete  = on_delete
+        @deferrable = deferrable
+      end
+
+      def references_primary_key?
+        keys.empty?
+      end
+
+      def deferrable?
+        @deferrable
+      end
+
+      def options
+        {
+          deferrable: deferrable?,
+          name:       name,
+          on_delete:  on_delete,
+          on_update:  on_update
+        }.tap do |options|
+          options[:key] = keys unless references_primary_key?
+        end
+      end
+    end
+
+    class CheckConstraint
+      include Dry::Equalizer(:name, :condition)
+      attr_reader :name, :condition
+
+      def initialize(name:, condition:)
+        @name      = name
+        @condition = condition
+      end
+    end
+
+    class Table
+      include Dry::Equalizer(:name, :fields, :indices, :checks, :foreign_keys)
+      attr_reader :name, :fields, :indices, :checks, :foreign_keys
+
+      def initialize(name, fields: [], indices: [], checks: [], foreign_keys: [])
+        @name         = name.to_sym
+        @fields       = fields
+        @indices      = indices
+        @checks       = checks
+        @foreign_keys = foreign_keys
+      end
+    end
+
+    class Enum
+      include Dry::Equalizer(:name, :values)
+      attr_reader :name, :values
+
+      def initialize(name, values)
+        @name   = name
+        @values = values
+      end
+    end
+  end
+end
+
+require_relative 'definitions/field'

data/lib/db_schema/definitions/field.rb

@@ -0,0 +1,38 @@
+module DbSchema
+  module Definitions
+    module Field
+      class << self
+        def build(name, type, **options)
+          type_class_for(type).new(name, **options)
+        end
+
+        def type_class_for(type)
+          registry.fetch(type) do |type|
+            raise ArgumentError, "#{type.inspect} type is not supported."
+          end
+        end
+
+        def registry
+          @registry ||= {}
+        end
+      end
+    end
+  end
+end
+
+require_relative 'field/base'
+require_relative 'field/numeric'
+require_relative 'field/monetary'
+require_relative 'field/character'
+require_relative 'field/binary'
+require_relative 'field/datetime'
+require_relative 'field/boolean'
+require_relative 'field/geometric'
+require_relative 'field/network'
+require_relative 'field/bit_string'
+require_relative 'field/text_search'
+require_relative 'field/uuid'
+require_relative 'field/json'
+require_relative 'field/array'
+require_relative 'field/range'
+require_relative 'field/custom'

data/lib/db_schema/definitions/field/array.rb

@@ -0,0 +1,19 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Array < Base
+        register :array
+        attr_reader :element_type
+
+        def initialize(name, of:, **options)
+          super(name, **options)
+          @element_type = Field.type_class_for(of)
+        end
+
+        def attributes
+          super.merge(element_type: element_type.type)
+        end
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/field/base.rb

@@ -0,0 +1,90 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Base
+        include Dry::Equalizer(:name, :type, :primary_key?, :options)
+        attr_reader :name, :default
+
+        def initialize(name, primary_key: false, null: true, default: nil, **attributes)
+          @name        = name
+          @primary_key = primary_key
+          @null        = null
+          @default     = process_default(default)
+          @attributes  = attributes
+        end
+
+        def primary_key?
+          @primary_key
+        end
+
+        def null?
+          !primary_key? && @null
+        end
+
+        def options
+          attributes.tap do |options|
+            options[:null] = false unless null?
+            options[:default] = default unless default.nil?
+          end
+        end
+
+        def attributes
+          self.class.valid_attributes.reduce({}) do |hash, attr_name|
+            if attr_value = @attributes[attr_name]
+              hash.merge(attr_name => attr_value)
+            elsif default_value = self.class.default_attribute_values[attr_name]
+              hash.merge(attr_name => default_value)
+            else
+              hash
+            end
+          end
+        end
+
+        def custom_type?
+          false
+        end
+
+        def type
+          self.class.type
+        end
+
+        def process_default(default)
+          if default.is_a?(Symbol)
+            Sequel.function(default)
+          else
+            default
+          end
+        end
+
+        class << self
+          def register(*types)
+            types.each do |type|
+              Field.registry[type] = self
+            end
+          end
+
+          def attributes(*attr_names, **attributes_with_defaults)
+            valid_attributes.push(*attr_names)
+
+            attributes_with_defaults.each do |attr_name, default_value|
+              valid_attributes.push(attr_name)
+              default_attribute_values[attr_name] = default_value
+            end
+          end
+
+          def valid_attributes
+            @valid_attributes ||= []
+          end
+
+          def default_attribute_values
+            @default_attribute_values ||= {}
+          end
+
+          def type
+            Field.registry.key(self)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/field/binary.rb

@@ -0,0 +1,9 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Bytea < Base
+        register :bytea
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/field/bit_string.rb

@@ -0,0 +1,15 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Bit < Base
+        register :bit
+        attributes length: 1
+      end
+
+      class Varbit < Base
+        register :varbit, :'bit varying'
+        attributes :length
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/field/boolean.rb

@@ -0,0 +1,9 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Boolean < Base
+        register :boolean
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/field/character.rb

@@ -0,0 +1,19 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Char < Base
+        register :char, :character
+        attributes length: 1
+      end
+
+      class Varchar < Base
+        register :varchar, :'character varying'
+        attributes :length
+      end
+
+      class Text < Base
+        register :text
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/field/custom.rb

@@ -0,0 +1,22 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Custom < Base
+        attr_reader :type_name
+
+        def initialize(name, type_name:, **options)
+          super(name, **options)
+          @type_name = type_name
+        end
+
+        def custom_type?
+          true
+        end
+
+        def type
+          type_name
+        end
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/field/datetime.rb

@@ -0,0 +1,30 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Timestamp < Base
+        register :timestamp, :'timestamp without time zone'
+      end
+
+      class Timestamptz < Base
+        register :timestamptz, :'timestamp with time zone'
+      end
+
+      class Date < Base
+        register :date
+      end
+
+      class Time < Base
+        register :time, :'time without time zone'
+      end
+
+      class Timetz < Base
+        register :timetz, :'time with time zone'
+      end
+
+      class Interval < Base
+        register :interval
+        attributes :fields, :precision
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/field/geometric.rb

@@ -0,0 +1,33 @@
+module DbSchema
+  module Definitions
+    module Field
+      class Point < Base
+        register :point
+      end
+
+      class Line < Base
+        register :line
+      end
+
+      class Lseg < Base
+        register :lseg
+      end
+
+      class Box < Base
+        register :box
+      end
+
+      class Path < Base
+        register :path
+      end
+
+      class Polygon < Base
+        register :polygon
+      end
+
+      class Circle < Base
+        register :circle
+      end
+    end
+  end
+end