mirror of
https://github.com/codeninjasllc/discourse.git
synced 2024-11-25 00:29:30 -05:00
77 lines
2.3 KiB
Ruby
77 lines
2.3 KiB
Ruby
require_dependency 'letter_avatar'
|
|
|
|
class UserAvatar < ActiveRecord::Base
|
|
MAX_SIZE = 240
|
|
SYSTEM_AVATAR_VERSION = 1
|
|
|
|
belongs_to :user
|
|
belongs_to :system_upload, class_name: 'Upload', dependent: :destroy
|
|
belongs_to :gravatar_upload, class_name: 'Upload', dependent: :destroy
|
|
belongs_to :custom_upload, class_name: 'Upload', dependent: :destroy
|
|
|
|
def contains_upload?(id)
|
|
system_upload_id == id || gravatar_upload_id == id || custom_upload_id == id
|
|
end
|
|
|
|
# updates the letter based avatar
|
|
def update_system_avatar!
|
|
old_id = nil
|
|
if system_upload
|
|
old_id = system_upload_id
|
|
system_upload.destroy!
|
|
end
|
|
|
|
file = File.open(LetterAvatar.generate(user.username, MAX_SIZE, cache: false), "r")
|
|
self.system_upload = Upload.create_for(user_id, file, "avatar.png", file.size)
|
|
self.system_avatar_version = SYSTEM_AVATAR_VERSION
|
|
|
|
if old_id == user.uploaded_avatar_id
|
|
user.update_column(:uploaded_avatar_id, system_upload_id)
|
|
end
|
|
|
|
save!
|
|
end
|
|
|
|
def update_gravatar!
|
|
# special logic for our system user, we do not want the discourse email there
|
|
email_hash = user.id == -1 ? User.email_hash("info@discourse.org") : user.email_hash
|
|
|
|
self.last_gravatar_download_attempt = Time.new
|
|
gravatar_url = "http://www.gravatar.com/avatar/#{email_hash}.png?s=500&d=404"
|
|
tempfile = FileHelper.download(gravatar_url, 1.megabyte, "gravatar")
|
|
|
|
upload = Upload.create_for(user.id, tempfile, 'gravatar.png', File.size(tempfile.path))
|
|
|
|
if gravatar_upload_id != upload.id
|
|
gravatar_upload.try(:destroy!)
|
|
self.gravatar_upload = upload
|
|
save!
|
|
else
|
|
gravatar_upload
|
|
end
|
|
rescue OpenURI::HTTPError
|
|
save!
|
|
ensure
|
|
tempfile.unlink if tempfile
|
|
end
|
|
|
|
end
|
|
|
|
# == Schema Information
|
|
#
|
|
# Table name: user_avatars
|
|
#
|
|
# id :integer not null, primary key
|
|
# user_id :integer not null
|
|
# system_upload_id :integer
|
|
# custom_upload_id :integer
|
|
# gravatar_upload_id :integer
|
|
# last_gravatar_download_attempt :datetime
|
|
# created_at :datetime
|
|
# updated_at :datetime
|
|
# system_avatar_version :integer default(0)
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_user_avatars_on_user_id (user_id)
|
|
#
|