dm-checked-types 0.0.1

data/Rakefile

@@ -0,0 +1,13 @@
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gemspec|
+    gemspec.name = "dm-checked-types"
+    gemspec.summary = "CHECK constraints in your DB"
+    gemspec.description = "Property types that add CHECK constraints to your DB and validations to your model. Data integrity FTW."
+    gemspec.email = "contact@rhnh.net"
+    gemspec.homepage = "http://github.com/xaviershay/dm-checked-types"
+    gemspec.authors = ["Xavier Shay"]
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: gem install jeweler"
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/lib/dm-checked-types.rb

@@ -0,0 +1,13 @@
+require 'dm-checked-types/checked_integer'
+require 'dm-checked-types/migrations'
+
+module DataMapper
+  module Migrations
+    constants.each do |const_name|
+      if CheckedTypes.const_defined?(const_name)
+        mod = const_get(const_name)
+        mod.send(:include, CheckedTypes.const_get(const_name))
+      end
+    end
+  end
+end

data/lib/dm-checked-types/checked_integer.rb

@@ -0,0 +1,53 @@
+class CheckedInteger < DataMapper::Type
+  primitive Integer
+
+  def self.inherited(target)
+    target.instance_variable_set('@primitive', self.primitive)
+  end
+
+  def self.range
+    @range || {}
+  end
+
+  def self.range=(value)
+    @range = value
+  end
+
+  def self.new(range)
+    type = generated_types[range] || Class.new(CheckedInteger)
+    type.range = range
+    generated_types[range] = type
+    type
+  end
+
+  def self.generated_types
+    @generated_types ||= {}
+  end
+
+  def self.[](range = {})
+    new(range)
+  end
+
+  def self.bind(property)
+    if defined?(::DataMapper::Validate)
+      model = property.model
+
+      unless model.skip_auto_validation_for?(property)
+        if property.type.ancestors.include?(CheckedInteger)
+          range = self.range
+          model.class_eval do
+            lower_bound = range[:gte]
+            lower_bound ||= range[:gt] + 1 if range[:gt]
+            lower_bound ||= -n
+
+            upper_bound = range[:lt] - 1 if range[:lt]
+            upper_bound ||= range[:lte]
+            upper_bound ||= n
+
+            validates_within property.name, :set => (lower_bound..upper_bound)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/dm-checked-types/migrations.rb

@@ -0,0 +1,92 @@
+module CheckedTypes
+  module DataObjectsAdapter
+    def create_custom_type_statements(repository_name, model)
+      model.properties.select {|x| x.type.ancestors.include?(CheckedInteger) }.each do |property|
+        table_name      = model.storage_name(repository_name)
+        property.type.range.each do |key, value|
+          comparator = {
+            :gt  => '>',
+            :gte => '>=',
+            :lt  => '<',
+            :lte => '<='
+          }[key] || raise(ArgumentError.new("Unsupported comparator: #{key}"))
+
+          sql = <<-EOS.compress_lines
+            ALTER TABLE #{quote_name(table_name)}
+            ADD CONSTRAINT #{quote_name(check_constraint_name(table_name, property.name, key))}
+            CHECK (#{quote_name(property.field)} #{comparator} #{value})
+          EOS
+          execute(sql)
+        end
+      end
+    end
+
+    def destroy_custom_type_statements(repository_name, model)
+      model.properties.select {|x| x.type.ancestors.include?(CheckedInteger) }.each do |property|
+        table_name = model.storage_name(repository_name)
+
+        property.type.range.each do |key, value|
+          constraint_name = check_constraint_name(table_name, property.name, key)
+
+          next unless constraint_exists?(model.storage_name, constraint_name)
+
+          sql = <<-EOS.compress_lines
+            ALTER TABLE #{quote_name(model.storage_name(repository_name))}
+            DROP CONSTRAINT #{quote_name(constraint_name)}
+          EOS
+          execute(sql)
+        end
+      end.compact
+    end
+
+    private
+
+    def check_constraint_name(table_name, relationship_name, comparator_name)
+      "#{table_name}_#{relationship_name}_#{comparator_name}"
+    end
+
+    def quote_constraint_name(foreign_key)
+      quote_table_name(foreign_key)
+    end
+  end
+
+  module SingletonMethods
+    def self.included(base)
+      # TODO: figure out how to make this work without AMC
+      base.class_eval <<-RUBY, __FILE__, __LINE__ + 1
+        alias_method :auto_migrate_down_without_custom_types!, :auto_migrate_down!
+        alias_method :auto_migrate_down!, :auto_migrate_down_with_custom_types!
+
+        alias_method :auto_migrate_up_without_custom_types!, :auto_migrate_up!
+        alias_method :auto_migrate_up!, :auto_migrate_up_with_custom_types!
+      RUBY
+    end
+
+    def auto_migrate_down_with_custom_types!(repository_name = nil)
+      repository_execute(:auto_migrate_down_with_custom_types!, repository_name)
+      auto_migrate_down_without_custom_types!(repository_name)
+    end
+
+    def auto_migrate_up_with_custom_types!(repository_name = nil)
+      auto_migrate_up_without_custom_types!(repository_name)
+      repository_execute(:auto_migrate_up_with_custom_types!, repository_name)
+    end
+  end
+
+  module Model
+    def auto_migrate_down_with_custom_types!(repository_name = self.repository_name)
+      return unless storage_exists?(repository_name)
+      return if self.respond_to?(:is_remixable?) && self.is_remixable?
+
+      adapter = DataMapper.repository(repository_name).adapter
+      adapter.destroy_custom_type_statements(repository_name, self)
+    end
+
+    def auto_migrate_up_with_custom_types!(repository_name = self.repository_name)
+      return if self.respond_to?(:is_remixable?) && self.is_remixable?
+
+      adapter = DataMapper.repository(repository_name).adapter
+      adapter.create_custom_type_statements(repository_name, self)
+    end
+  end
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification 
+name: dm-checked-types
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Xavier Shay
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-03-06 00:00:00 +13:00
+default_executable: 
+dependencies: []
+
+description: Property types that add CHECK constraints to your DB and validations to your model. Data integrity FTW.
+email: contact@rhnh.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- Rakefile
+- VERSION
+- lib/dm-checked-types.rb
+- lib/dm-checked-types/checked_integer.rb
+- lib/dm-checked-types/migrations.rb
+has_rdoc: true
+homepage: http://github.com/xaviershay/dm-checked-types
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: CHECK constraints in your DB
+test_files: []
+