db_schema-definitions 0.1

data/lib/db_schema/definitions/index.rb

@@ -0,0 +1,56 @@
+module DbSchema
+  module Definitions
+    class Index
+      include Dry::Equalizer(:name, :columns, :unique?, :type, :condition)
+      attr_reader :name, :columns, :type, :condition
+
+      def initialize(name:, columns:, unique: false, type: :btree, condition: nil)
+        @name      = name.to_sym
+        @columns   = columns
+        @unique    = unique
+        @type      = type
+        @condition = condition
+      end
+
+      def unique?
+        @unique
+      end
+
+      def btree?
+        type == :btree
+      end
+
+      def columns_to_sequel
+        if btree?
+          columns.map(&:ordered_expression)
+        else
+          columns.map(&:to_sequel)
+        end
+      end
+
+      def has_expressions?
+        !condition.nil? || columns.any?(&:expression?)
+      end
+
+      def with_name(new_name)
+        Index.new(
+          name:      new_name,
+          columns:   columns,
+          unique:    unique?,
+          type:      type,
+          condition: condition
+        )
+      end
+    end
+
+    class NullIndex < Index
+      def initialize
+        @columns = []
+      end
+    end
+  end
+end
+
+require_relative 'index/column'
+require_relative 'index/table_field'
+require_relative 'index/expression'

data/lib/db_schema/definitions/index/column.rb

@@ -0,0 +1,32 @@
+module DbSchema
+  module Definitions
+    class Index
+      class Column
+        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 ordered_expression
+          if asc?
+            Sequel.asc(to_sequel, nulls: nulls)
+          else
+            Sequel.desc(to_sequel, nulls: nulls)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/index/expression.rb

@@ -0,0 +1,19 @@
+module DbSchema
+  module Definitions
+    class Index
+      class Expression < Column
+        def expression?
+          true
+        end
+
+        def index_name_segment
+          name.scan(/\b[A-Za-z0-9_]+\b/).join('_')
+        end
+
+        def to_sequel
+          Sequel.lit("(#{name})")
+        end
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/index/table_field.rb

@@ -0,0 +1,19 @@
+module DbSchema
+  module Definitions
+    class Index
+      class TableField < Column
+        def expression?
+          false
+        end
+
+        def index_name_segment
+          name
+        end
+
+        def to_sequel
+          name
+        end
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/schema.rb

@@ -0,0 +1,36 @@
+module DbSchema
+  module Definitions
+    class Schema
+      include Dry::Equalizer(:tables, :enums, :extensions)
+      attr_reader :tables, :enums, :extensions
+      attr_writer :tables
+
+      def initialize(tables: [], enums: [], extensions: [])
+        @tables     = tables
+        @enums      = enums
+        @extensions = extensions
+      end
+
+      def table(table_name)
+        tables.find { |table| table.name == table_name } || NullTable.new
+      end
+      alias_method :[], :table
+
+      def has_table?(table_name)
+        !table(table_name).is_a?(NullTable)
+      end
+
+      def enum(enum_name)
+        enums.find { |enum| enum.name == enum_name } || NullEnum.new
+      end
+
+      def has_enum?(enum_name)
+        !enum(enum_name).is_a?(NullEnum)
+      end
+
+      def has_extension?(extension_name)
+        extensions.any? { |extension| extension.name == extension_name }
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/table.rb

@@ -0,0 +1,120 @@
+module DbSchema
+  module Definitions
+    class Table
+      include Dry::Equalizer(:name, :fields, :indexes, :checks, :foreign_keys)
+      attr_reader :name, :fields, :indexes, :checks, :foreign_keys
+
+      def initialize(name, fields: [], indexes: [], checks: [], foreign_keys: [])
+        @name         = name.to_sym
+        @fields       = fields
+        @indexes      = indexes
+        @checks       = checks
+        @foreign_keys = foreign_keys
+      end
+
+      def has_expressions?
+        fields.any?(&:default_is_expression?) ||
+          indexes.any?(&:has_expressions?) ||
+          checks.any?
+      end
+
+      def field(field_name)
+        fields.find { |field| field.name == field_name } || NullField.new
+      end
+      alias_method :[], :field
+
+      def has_field?(field_name)
+        !field(field_name).is_a?(NullField)
+      end
+
+      def index(index_name)
+        indexes.find { |index| index.name == index_name } || NullIndex.new
+      end
+
+      def has_index?(index_name)
+        !index(index_name).is_a?(NullIndex)
+      end
+
+      def has_index_on?(*field_names)
+        indexes.any? do |index|
+          index.columns.none?(&:expression?) && index.columns.map(&:name) == field_names
+        end
+      end
+
+      def has_unique_index_on?(*field_names)
+        indexes.any? do |index|
+          index.unique? && index.columns.none?(&:expression?) && index.columns.map(&:name) == field_names
+        end
+      end
+
+      def check(check_name)
+        checks.find { |check| check.name == check_name } || NullCheckConstraint.new
+      end
+
+      def has_check?(check_name)
+        !check(check_name).is_a?(NullCheckConstraint)
+      end
+
+      def foreign_key(fkey_name)
+        foreign_keys.find { |fkey| fkey.name == fkey_name } || NullForeignKey.new
+      end
+
+      def has_foreign_key?(fkey_name)
+        !foreign_key(fkey_name).is_a?(NullForeignKey)
+      end
+
+      def has_foreign_key_to?(other_table_name)
+        foreign_keys.any? { |fkey| fkey.table == other_table_name }
+      end
+
+      def with_name(new_name)
+        Table.new(
+          new_name,
+          fields:       fields,
+          indexes:      indexes,
+          checks:       checks,
+          foreign_keys: foreign_keys
+        )
+      end
+
+      def with_fields(new_fields)
+        Table.new(
+          name,
+          fields:       new_fields,
+          indexes:      indexes,
+          checks:       checks,
+          foreign_keys: foreign_keys
+        )
+      end
+
+      def with_indexes(new_indexes)
+        Table.new(
+          name,
+          fields:       fields,
+          indexes:      new_indexes,
+          checks:       checks,
+          foreign_keys: foreign_keys
+        )
+      end
+
+      def with_foreign_keys(new_foreign_keys)
+        Table.new(
+          name,
+          fields:       fields,
+          indexes:      indexes,
+          checks:       checks,
+          foreign_keys: new_foreign_keys
+        )
+      end
+    end
+
+    class NullTable < Table
+      def initialize
+        @fields       = []
+        @indexes      = []
+        @checks       = []
+        @foreign_keys = []
+      end
+    end
+  end
+end

data/lib/db_schema/definitions/version.rb

@@ -0,0 +1,5 @@
+module DbSchema
+  module Definitions
+    VERSION = '0.1'
+  end
+end

metadata

@@ -0,0 +1,218 @@
+--- !ruby/object:Gem::Specification
+name: db_schema-definitions
+version: !ruby/object:Gem::Version
+  version: '0.1'
+platform: ruby
+authors:
+- Vsevolod Romashov
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2017-11-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: dry-equalizer
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: awesome_print
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: guard-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: terminal-notifier
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: terminal-notifier-guard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email:
+- 7@7vn.ru
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Guardfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- db_schema-definitions.gemspec
+- lib/db_schema/definitions.rb
+- lib/db_schema/definitions/check_constraint.rb
+- lib/db_schema/definitions/enum.rb
+- lib/db_schema/definitions/extension.rb
+- lib/db_schema/definitions/field.rb
+- lib/db_schema/definitions/field/array.rb
+- lib/db_schema/definitions/field/base.rb
+- lib/db_schema/definitions/field/binary.rb
+- lib/db_schema/definitions/field/bit_string.rb
+- lib/db_schema/definitions/field/boolean.rb
+- lib/db_schema/definitions/field/character.rb
+- lib/db_schema/definitions/field/custom.rb
+- lib/db_schema/definitions/field/datetime.rb
+- lib/db_schema/definitions/field/extensions/chkpass.rb
+- lib/db_schema/definitions/field/extensions/citext.rb
+- lib/db_schema/definitions/field/extensions/cube.rb
+- lib/db_schema/definitions/field/extensions/hstore.rb
+- lib/db_schema/definitions/field/extensions/isn.rb
+- lib/db_schema/definitions/field/extensions/ltree.rb
+- lib/db_schema/definitions/field/extensions/seg.rb
+- lib/db_schema/definitions/field/geometric.rb
+- lib/db_schema/definitions/field/json.rb
+- lib/db_schema/definitions/field/monetary.rb
+- lib/db_schema/definitions/field/network.rb
+- lib/db_schema/definitions/field/numeric.rb
+- lib/db_schema/definitions/field/range.rb
+- lib/db_schema/definitions/field/text_search.rb
+- lib/db_schema/definitions/field/uuid.rb
+- lib/db_schema/definitions/foreign_key.rb
+- lib/db_schema/definitions/index.rb
+- lib/db_schema/definitions/index/column.rb
+- lib/db_schema/definitions/index/expression.rb
+- lib/db_schema/definitions/index/table_field.rb
+- lib/db_schema/definitions/schema.rb
+- lib/db_schema/definitions/table.rb
+- lib/db_schema/definitions/version.rb
+homepage: https://github.com/db-schema/definitions
+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.6.13
+signing_key: 
+specification_version: 4
+summary: Database object definitions for DbSchema
+test_files: []