Validate key uniqueness across Key and DeployKey tables

miks
1 parent c973fce6
Showing 3 changed files with 15 additions and 0 deletions Show diff stats
app/models/deploy_key.rb
app/models/key.rb
lib/unique_public_key_validator.rb
+require 'unique_public_key_validator'
+
 class DeployKey < ActiveRecord::Base
   belongs_to :project
@@ -10,6 +12,8 @@ class DeployKey &lt; ActiveRecord::Base
             :uniqueness => true,
             :length   => { :within => 0..5000 }
+  validates_with UniquePublicKeyValidator
+
   before_save :set_identifier
   after_save :update_repository
   after_destroy :repository_delete_key
+require 'unique_public_key_validator'
+
 class Key < ActiveRecord::Base
   belongs_to :user
@@ -10,6 +12,8 @@ class Key &lt; ActiveRecord::Base
             :uniqueness => true,
             :length   => { :within => 0..5000 }
+  validates_with UniquePublicKeyValidator
+
   before_save :set_identifier
   after_save :update_repository
   after_destroy :repository_delete_key
@@ -0,0 +1,7 @@
+class UniquePublicKeyValidator < ActiveModel::Validator
+  def validate(record)
+    if (DeployKey.where('key = ? AND id !=?', record.key , record.id).count > 0 || Key.where('key = ? AND id !=?', record.key , record.id).count > 0)
+      record.errors.add :key, 'already exist.'
+    end
+  end
+end
@@ -0,0 +1,7 @@		@@ -0,0 +1,7 @@
	1	+class UniquePublicKeyValidator < ActiveModel::Validator
	2	+ def validate(record)
	3	+ if (DeployKey.where('key = ? AND id !=?', record.key , record.id).count > 0 \|\| Key.where('key = ? AND id !=?', record.key , record.id).count > 0)
	4	+ record.errors.add :key, 'already exist.'
	5	+ end
	6	+ end
	7	+end