notification scaffold

app/models/notification.rb

@@ -0,0 +1,30 @@
+class Notification
+  #
+  # Notification levels
+  #
+  N_DISABLED = 0
+  N_PARTICIPATING = 1
+  N_WATCH = 2
+
+  attr_accessor :user
+
+  def self.notification_levels
+    [N_DISABLED, N_PARTICIPATING, N_WATCH]
+  end
+
+  def initialize(user)
+    @user = user
+  end
+
+  def disabled?
+    user.notification_level == N_DISABLED
+  end
+
+  def participating?
+    user.notification_level == N_PARTICIPATING
+  end
+
+  def watch?
+    user.notification_level == N_WATCH
+  end
+end

app/models/user.rb

@@ -54,13 +54,6 @@ class User < ActiveRecord::Base
   attr_accessible :login
 
 
-  #
-  # Notification levels
-  #
-  N_DISABLED = 0
-  N_PARTICIPATING = 1
-  N_WATCH = 2
-
   #
   # Relations
   #
@@ -116,6 +109,9 @@ class User < ActiveRecord::Base
             format: { with: Gitlab::Regex.username_regex,
                       message: "only letters, digits & '_' '-' '.' allowed. Letter should be first" }
 
+  validates :notification_level,
+    inclusion: { in: Notification.notification_levels },
+    presence: true
 
   validate :namespace_uniq, if: ->(user) { user.username_changed? }
 
@@ -216,6 +212,10 @@ class User < ActiveRecord::Base
     username
   end
 
+  def notification
+    @notification ||= Notification.new(self)
+  end
+
   def generate_password
     if self.force_random_password
       self.password = self.password_confirmation = Devise.friendly_token.first(8)