]> git.openstreetmap.org Git - rails.git/blob - app/models/user.rb
Use a state machine for user status
[rails.git] / app / models / user.rb
1 # == Schema Information
2 #
3 # Table name: users
4 #
5 #  email               :string           not null
6 #  id                  :bigint(8)        not null, primary key
7 #  pass_crypt          :string           not null
8 #  creation_time       :datetime         not null
9 #  display_name        :string           default(""), not null
10 #  data_public         :boolean          default(FALSE), not null
11 #  description         :text             default(""), not null
12 #  home_lat            :float
13 #  home_lon            :float
14 #  home_zoom           :integer          default(3)
15 #  nearby              :integer          default(50)
16 #  pass_salt           :string
17 #  email_valid         :boolean          default(FALSE), not null
18 #  new_email           :string
19 #  creation_ip         :string
20 #  languages           :string
21 #  status              :enum             default("pending"), not null
22 #  terms_agreed        :datetime
23 #  consider_pd         :boolean          default(FALSE), not null
24 #  auth_uid            :string
25 #  preferred_editor    :string
26 #  terms_seen          :boolean          default(FALSE), not null
27 #  description_format  :enum             default("markdown"), not null
28 #  changesets_count    :integer          default(0), not null
29 #  traces_count        :integer          default(0), not null
30 #  diary_entries_count :integer          default(0), not null
31 #  image_use_gravatar  :boolean          default(FALSE), not null
32 #  auth_provider       :string
33 #  home_tile           :bigint(8)
34 #  tou_agreed          :datetime
35 #
36 # Indexes
37 #
38 #  users_auth_idx                (auth_provider,auth_uid) UNIQUE
39 #  users_display_name_idx        (display_name) UNIQUE
40 #  users_display_name_lower_idx  (lower((display_name)::text))
41 #  users_email_idx               (email) UNIQUE
42 #  users_email_lower_idx         (lower((email)::text))
43 #  users_home_idx                (home_tile)
44 #
45
46 class User < ApplicationRecord
47   require "digest"
48   include AASM
49
50   has_many :traces, -> { where(:visible => true) }
51   has_many :diary_entries, -> { order(:created_at => :desc) }
52   has_many :diary_comments, -> { order(:created_at => :desc) }
53   has_many :diary_entry_subscriptions, :class_name => "DiaryEntrySubscription"
54   has_many :diary_subscriptions, :through => :diary_entry_subscriptions, :source => :diary_entry
55   has_many :messages, -> { where(:to_user_visible => true).order(:sent_on => :desc).preload(:sender, :recipient) }, :foreign_key => :to_user_id
56   has_many :new_messages, -> { where(:to_user_visible => true, :message_read => false).order(:sent_on => :desc) }, :class_name => "Message", :foreign_key => :to_user_id
57   has_many :sent_messages, -> { where(:from_user_visible => true).order(:sent_on => :desc).preload(:sender, :recipient) }, :class_name => "Message", :foreign_key => :from_user_id
58   has_many :friendships, -> { joins(:befriendee).where(:users => { :status => %w[active confirmed] }) }
59   has_many :friends, :through => :friendships, :source => :befriendee
60   has_many :tokens, :class_name => "UserToken", :dependent => :destroy
61   has_many :preferences, :class_name => "UserPreference"
62   has_many :changesets, -> { order(:created_at => :desc) }
63   has_many :changeset_comments, :foreign_key => :author_id
64   has_and_belongs_to_many :changeset_subscriptions, :class_name => "Changeset", :join_table => "changesets_subscribers", :foreign_key => "subscriber_id"
65   has_many :note_comments, :foreign_key => :author_id
66   has_many :notes, :through => :note_comments
67
68   has_many :client_applications
69   has_many :oauth_tokens, -> { order(:authorized_at => :desc).preload(:client_application) }, :class_name => "OauthToken"
70
71   has_many :oauth2_applications, :class_name => Doorkeeper.config.application_model.name, :as => :owner
72   has_many :access_grants, :class_name => Doorkeeper.config.access_grant_model.name, :foreign_key => :resource_owner_id
73   has_many :access_tokens, :class_name => Doorkeeper.config.access_token_model.name, :foreign_key => :resource_owner_id
74
75   has_many :blocks, :class_name => "UserBlock"
76   has_many :blocks_created, :class_name => "UserBlock", :foreign_key => :creator_id
77   has_many :blocks_revoked, :class_name => "UserBlock", :foreign_key => :revoker_id
78
79   has_many :roles, :class_name => "UserRole"
80
81   has_many :issues, :class_name => "Issue", :foreign_key => :reported_user_id
82   has_many :issue_comments
83
84   has_many :reports
85
86   scope :visible, -> { where(:status => %w[pending active confirmed]) }
87   scope :active, -> { where(:status => %w[active confirmed]) }
88   scope :identifiable, -> { where(:data_public => true) }
89
90   has_one_attached :avatar
91
92   validates :display_name, :presence => true, :length => 3..255,
93                            :exclusion => %w[new terms save confirm confirm-email go_public reset-password forgot-password suspended]
94   validates :display_name, :if => proc { |u| u.display_name_changed? },
95                            :uniqueness => { :case_sensitive => false }
96   validates :display_name, :if => proc { |u| u.display_name_changed? },
97                            :characters => { :url_safe => true },
98                            :whitespace => { :leading => false, :trailing => false }
99   validates :email, :presence => true, :confirmation => true, :characters => true
100   validates :email, :if => proc { |u| u.email_changed? },
101                     :uniqueness => { :case_sensitive => false }
102   validates :email, :if => proc { |u| u.email_changed? },
103                     :whitespace => { :leading => false, :trailing => false }
104   validates :pass_crypt, :confirmation => true, :length => 8..255
105   validates :home_lat, :allow_nil => true, :numericality => true, :inclusion => { :in => -90..90 }
106   validates :home_lon, :allow_nil => true, :numericality => true, :inclusion => { :in => -180..180 }
107   validates :home_zoom, :allow_nil => true, :numericality => { :only_integer => true }
108   validates :preferred_editor, :inclusion => Editors::ALL_EDITORS, :allow_nil => true
109   validates :auth_uid, :unless => proc { |u| u.auth_provider.nil? },
110                        :uniqueness => { :scope => :auth_provider }
111   validates :avatar, :if => proc { |u| u.attachment_changes["avatar"] },
112                      :image => true
113
114   validates_email_format_of :email, :if => proc { |u| u.email_changed? }
115   validates_email_format_of :new_email, :allow_blank => true, :if => proc { |u| u.new_email_changed? }
116
117   alias_attribute :created_at, :creation_time
118
119   before_save :encrypt_password
120   before_save :update_tile
121   after_save :spam_check
122
123   def to_param
124     display_name
125   end
126
127   def self.authenticate(options)
128     if options[:username] && options[:password]
129       user = find_by("email = ? OR display_name = ?", options[:username].strip, options[:username])
130
131       if user.nil?
132         users = where("LOWER(email) = LOWER(?) OR LOWER(display_name) = LOWER(?)", options[:username].strip, options[:username])
133
134         user = users.first if users.count == 1
135       end
136
137       if user && PasswordHash.check(user.pass_crypt, user.pass_salt, options[:password])
138         if PasswordHash.upgrade?(user.pass_crypt, user.pass_salt)
139           user.pass_crypt, user.pass_salt = PasswordHash.create(options[:password])
140           user.save
141         end
142       else
143         user = nil
144       end
145     elsif options[:token]
146       token = UserToken.find_by(:token => options[:token])
147       user = token.user if token
148     end
149
150     if user &&
151        (user.status == "deleted" ||
152          (user.status == "pending" && !options[:pending]) ||
153          (user.status == "suspended" && !options[:suspended]))
154       user = nil
155     end
156
157     token.update(:expiry => 1.week.from_now) if token && user
158
159     user
160   end
161
162   aasm :column => :status, :no_direct_assignment => true do
163     state :pending, :initial => true
164     state :active
165     state :confirmed
166     state :suspended
167     state :deleted
168
169     # A normal account is active
170     event :activate do
171       transitions :from => :pending, :to => :active
172     end
173
174     # Used in test suite, not something that we would normally need to do.
175     event :deactivate do
176       transitions :from => :active, :to => :pending
177     end
178
179     # To confirm an account is used to override the spam scoring
180     event :confirm do
181       transitions :from => [:pending, :active, :suspended], :to => :confirmed
182     end
183
184     event :suspend do
185       transitions :from => [:pending, :active], :to => :suspended
186     end
187
188     # Mark the account as deleted but keep all data intact
189     event :hide do
190       transitions :from => [:pending, :active, :confirmed, :suspended], :to => :deleted
191     end
192
193     event :unhide do
194       transitions :from => [:deleted], :to => :active
195     end
196
197     # Mark the account as deleted and remove personal data
198     event :soft_destroy do
199       before do
200         remove_personal_data
201       end
202
203       transitions :from => [:pending, :active, :confirmed, :suspended], :to => :deleted
204     end
205   end
206
207   def description
208     RichText.new(self[:description_format], self[:description])
209   end
210
211   def languages
212     attribute_present?(:languages) ? self[:languages].split(/ *[, ] */) : []
213   end
214
215   def languages=(languages)
216     self[:languages] = languages.join(",")
217   end
218
219   def preferred_language
220     languages.find { |l| Language.exists?(:code => l) }
221   end
222
223   def preferred_languages
224     @preferred_languages ||= Locale.list(languages)
225   end
226
227   def nearby(radius = Settings.nearby_radius, num = Settings.nearby_users)
228     if home_lon && home_lat
229       gc = OSM::GreatCircle.new(home_lat, home_lon)
230       sql_for_area = QuadTile.sql_for_area(gc.bounds(radius), "home_")
231       sql_for_distance = gc.sql_for_distance("home_lat", "home_lon")
232       nearby = User.active.identifiable
233                    .where.not(:id => id)
234                    .where(sql_for_area)
235                    .where("#{sql_for_distance} <= ?", radius)
236                    .order(Arel.sql(sql_for_distance))
237                    .limit(num)
238     else
239       nearby = []
240     end
241     nearby
242   end
243
244   def distance(nearby_user)
245     OSM::GreatCircle.new(home_lat, home_lon).distance(nearby_user.home_lat, nearby_user.home_lon)
246   end
247
248   def is_friends_with?(new_friend)
249     friendships.exists?(:befriendee => new_friend)
250   end
251
252   ##
253   # returns true if a user is visible
254   def visible?
255     %w[pending active confirmed].include? status
256   end
257
258   ##
259   # returns true if a user is active
260   def active?
261     %w[active confirmed].include? status
262   end
263
264   ##
265   # returns true if the user has the moderator role, false otherwise
266   def moderator?
267     has_role? "moderator"
268   end
269
270   ##
271   # returns true if the user has the administrator role, false otherwise
272   def administrator?
273     has_role? "administrator"
274   end
275
276   ##
277   # returns true if the user has the requested role
278   def has_role?(role)
279     roles.any? { |r| r.role == role }
280   end
281
282   ##
283   # returns the first active block which would require users to view
284   # a message, or nil if there are none.
285   def blocked_on_view
286     blocks.active.detect(&:needs_view?)
287   end
288
289   ##
290   # remove personal data - leave the account but purge most personal data
291   def remove_personal_data
292     avatar.purge_later
293
294     self.display_name = "user_#{id}"
295     self.description = ""
296     self.home_lat = nil
297     self.home_lon = nil
298     self.email_valid = false
299     self.new_email = nil
300     self.auth_provider = nil
301     self.auth_uid = nil
302
303     save
304   end
305
306   ##
307   # return a spam score for a user
308   def spam_score
309     changeset_score = changesets.size * 50
310     trace_score = traces.size * 50
311     diary_entry_score = diary_entries.visible.inject(0) { |acc, elem| acc + elem.body.spam_score }
312     diary_comment_score = diary_comments.visible.inject(0) { |acc, elem| acc + elem.body.spam_score }
313
314     score = description.spam_score / 4.0
315     score += diary_entries.where("created_at > ?", 1.day.ago).count * 10
316     score += diary_entry_score / diary_entries.length unless diary_entries.empty?
317     score += diary_comment_score / diary_comments.length unless diary_comments.empty?
318     score -= changeset_score
319     score -= trace_score
320
321     score.to_i
322   end
323
324   ##
325   # perform a spam check on a user
326   def spam_check
327     suspend! if may_suspend? && spam_score > Settings.spam_threshold
328   end
329
330   ##
331   # return an oauth 1 access token for a specified application
332   def access_token(application_key)
333     ClientApplication.find_by(:key => application_key).access_token_for_user(self)
334   end
335
336   ##
337   # return an oauth 2 access token for a specified application
338   def oauth_token(application_id)
339     application = Doorkeeper.config.application_model.find_by(:uid => application_id)
340
341     Doorkeeper.config.access_token_model.find_or_create_for(
342       :application => application,
343       :resource_owner => self,
344       :scopes => application.scopes
345     )
346   end
347
348   def fingerprint
349     digest = Digest::SHA256.new
350     digest.update(email)
351     digest.update(pass_crypt)
352     digest.hexdigest
353   end
354
355   def max_messages_per_hour
356     account_age_in_seconds = Time.now.utc - created_at
357     account_age_in_hours = account_age_in_seconds / 3600
358     recent_messages = messages.where("sent_on >= ?", Time.now.utc - 3600).count
359     active_reports = issues.with_status(:open).sum(:reports_count)
360     max_messages = account_age_in_hours.ceil + recent_messages - active_reports * 10
361     max_messages.clamp(0, Settings.max_messages_per_hour)
362   end
363
364   def max_friends_per_hour
365     account_age_in_seconds = Time.now.utc - created_at
366     account_age_in_hours = account_age_in_seconds / 3600
367     recent_friends = Friendship.where(:befriendee => self).where("created_at >= ?", Time.now.utc - 3600).count
368     active_reports = issues.with_status(:open).sum(:reports_count)
369     max_friends = account_age_in_hours.ceil + recent_friends - active_reports * 10
370     max_friends.clamp(0, Settings.max_friends_per_hour)
371   end
372
373   private
374
375   def encrypt_password
376     if pass_crypt_confirmation
377       self.pass_crypt, self.pass_salt = PasswordHash.create(pass_crypt)
378       self.pass_crypt_confirmation = nil
379     end
380   end
381
382   def update_tile
383     self.home_tile = QuadTile.tile_for_point(home_lat, home_lon) if home_lat && home_lon
384   end
385 end