115 lines
3.1 KiB
Ruby
115 lines
3.1 KiB
Ruby
class Photo < ActiveRecord::Base
|
|
attr_accessor :file
|
|
|
|
validates_presence_of :author_id, :album_id
|
|
|
|
belongs_to :album
|
|
belongs_to :author
|
|
has_many :photo_tags, :dependent => :destroy
|
|
has_many :tags, :through => :photo_tags
|
|
|
|
before_validation_on_create :set_author_id
|
|
before_create :validate_image_sanity
|
|
after_create :create_directories
|
|
before_destroy :destroy_directories
|
|
|
|
attr_protected :author_id
|
|
|
|
##
|
|
# Determines the base directory for all files in this model.
|
|
#
|
|
def base_directory
|
|
"#{Merb.root}/public/photos/#{id}"
|
|
end
|
|
|
|
##
|
|
# Checks to see if the file is found on the filesystem.
|
|
#
|
|
def exist?
|
|
File.exist?(self.pathname)
|
|
end
|
|
|
|
##
|
|
# Returns the full path to the file on the filesystem.
|
|
#
|
|
def pathname
|
|
"#{self.base_directory}/#{self.filename}"
|
|
end
|
|
|
|
def self.popular_tags(limit = nil)
|
|
query = "SELECT tags.id, tags.name, count(*) AS count FROM photo_tags, tags, photos WHERE tags.id = tag_id AND photo_tags.photo_id = photos.id GROUP BY tags.id, tags.name ORDER BY count DESC"
|
|
query << " LIMIT #{limit}" unless limit.nil?
|
|
Tag.find_by_sql(query).sort { |a, b| a.name <=> b.name }
|
|
end
|
|
|
|
protected
|
|
|
|
##
|
|
# Sets the Author marker for ownership on creation.
|
|
#
|
|
def set_author_id
|
|
self.author_id ||= Application.current_author_id
|
|
end
|
|
|
|
##
|
|
# Checks to make sure that the file exists and is an image.
|
|
#
|
|
def validate_image_sanity
|
|
if self.file[:tempfile].nil?
|
|
self.errors.add(:file, 'File is not a file')
|
|
end
|
|
if self.file[:content_type] !~ /image\/\w+/
|
|
self.errors.add(:file, 'File is not a supported type')
|
|
end
|
|
if self.file[:size] > 3 * 1048576
|
|
self.errors.add(:file, 'File is too big (3MB max)')
|
|
end
|
|
if self.album.photos.count(:conditions => { :filename => self.file[:filename] }) > 0
|
|
self.errors.add(:file, 'already seems to exist in this album')
|
|
end
|
|
return false unless self.errors.empty?
|
|
|
|
begin
|
|
@fstr = self.file[:tempfile].read
|
|
iary = Magick::Image.from_blob(@fstr)
|
|
self.filename = File.basename(self.file[:filename]).gsub(/[^\w._-]/, '')
|
|
if iary.first.to_s =~ / (\d+)x(\d+) /
|
|
self.width = $1
|
|
self.height = $2
|
|
end
|
|
rescue
|
|
$stderr.puts("Caught an exception saving an image:")
|
|
$stderr.puts("* #{$!}")
|
|
self.errors.add(:file, 'File could not be read as an image')
|
|
return false
|
|
end
|
|
|
|
self.content_type = self.file[:content_type]
|
|
true
|
|
end
|
|
|
|
##
|
|
# Makes the directories and writes the file to disk.
|
|
#
|
|
def create_directories
|
|
File.umask(0022)
|
|
Dir.mkdir(base_directory) unless File.exist?(base_directory)
|
|
File.open("#{base_directory}/#{self.filename}", "w") do |f|
|
|
f.puts(@fstr)
|
|
end
|
|
File.chmod(0644, "#{base_directory}/#{self.filename}")
|
|
end
|
|
|
|
##
|
|
# Removes the directories and files associated with this model on destroy.
|
|
#
|
|
def destroy_directories
|
|
return unless File.exists?(base_directory)
|
|
Dir.foreach(base_directory) do |file|
|
|
next if file =~ /^\.\.?$/
|
|
File.delete(base_directory + '/' + file)
|
|
end
|
|
Dir.delete(base_directory)
|
|
end
|
|
end
|