1 files changed, 29 insertions, 0 deletions
diff --git a/app/models/invite_code_validator.rb b/app/models/invite_code_validator.rb
new file mode 100644
index 0000000..676e4fa
--- /dev/null
+++ b/app/models/invite_code_validator.rb
@@ -0,0 +1,29 @@
+class InviteCodeValidator < ActiveModel::Validator
+
+  def validate(user)
+
+    user_invite_code = InviteCode.find_by_invite_code user.invite_code
+
+    if not_existent?(user_invite_code)
+      add_error_to_user("This is not a valid code", user)
+
+    elsif has_no_uses_left?(user_invite_code)
+      add_error_to_user("This code has already been used", user)
+    end
+  end
+
+  private
+  def not_existent?(code)
+    code == nil
+  end
+
+  def has_no_uses_left?(code)
+    code.invite_count >= code.max_uses
+  end
+
+  def add_error_to_user(error, user)
+    user.errors[:invite_code] << error
+  end
+end
+
+