attachment.rb
250 lines
| 8.3 KiB
| text/x-ruby
|
RubyLexer
Toshi MARUYAMA
|
r5673 | # Redmine - project management software | ||
|
Toshi MARUYAMA
|
r8892 | # Copyright (C) 2006-2012 Jean-Philippe Lang | ||
Jean-Philippe Lang
|
r330 | # | ||
| # This program is free software; you can redistribute it and/or | ||||
| # modify it under the terms of the GNU General Public License | ||||
| # as published by the Free Software Foundation; either version 2 | ||||
| # of the License, or (at your option) any later version. | ||||
|
Toshi MARUYAMA
|
r5673 | # | ||
|
Jean-Philippe Lang
|
r330 | # This program is distributed in the hope that it will be useful, | ||
| # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
| # GNU General Public License for more details. | ||||
|
Toshi MARUYAMA
|
r5673 | # | ||
|
Jean-Philippe Lang
|
r330 | # You should have received a copy of the GNU General Public License | ||
| # along with this program; if not, write to the Free Software | ||||
| # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. | ||||
| require "digest/md5" | ||||
| class Attachment < ActiveRecord::Base | ||||
| belongs_to :container, :polymorphic => true | ||||
| belongs_to :author, :class_name => "User", :foreign_key => "author_id" | ||||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r8771 | validates_presence_of :filename, :author | ||
|
Jean-Philippe Lang
|
r590 | validates_length_of :filename, :maximum => 255 | ||
| validates_length_of :disk_filename, :maximum => 255 | ||||
|
Jean-Philippe Lang
|
r9817 | validates_length_of :description, :maximum => 255, :allow_blank => true | ||
|
Toshi MARUYAMA
|
r6594 | validate :validate_max_file_size | ||
|
Jean-Philippe Lang
|
r663 | |||
| acts_as_event :title => :filename, | ||||
|
Jean-Philippe Lang
|
r1669 | :url => Proc.new {|o| {:controller => 'attachments', :action => 'download', :id => o.id, :filename => o.filename}} | ||
|
Jean-Philippe Lang
|
r663 | |||
|
Jean-Philippe Lang
|
r1692 | acts_as_activity_provider :type => 'files', | ||
| :permission => :view_files, | ||||
|
Jean-Philippe Lang
|
r2064 | :author_key => :author_id, | ||
|
Toshi MARUYAMA
|
r5673 | :find_options => {:select => "#{Attachment.table_name}.*", | ||
|
Jean-Philippe Lang
|
r1692 | :joins => "LEFT JOIN #{Version.table_name} ON #{Attachment.table_name}.container_type='Version' AND #{Version.table_name}.id = #{Attachment.table_name}.container_id " + | ||
|
Jean-Philippe Lang
|
r2501 | "LEFT JOIN #{Project.table_name} ON #{Version.table_name}.project_id = #{Project.table_name}.id OR ( #{Attachment.table_name}.container_type='Project' AND #{Attachment.table_name}.container_id = #{Project.table_name}.id )"} | ||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r1692 | acts_as_activity_provider :type => 'documents', | ||
| :permission => :view_documents, | ||||
|
Jean-Philippe Lang
|
r2064 | :author_key => :author_id, | ||
|
Toshi MARUYAMA
|
r5673 | :find_options => {:select => "#{Attachment.table_name}.*", | ||
|
Jean-Philippe Lang
|
r1692 | :joins => "LEFT JOIN #{Document.table_name} ON #{Attachment.table_name}.container_type='Document' AND #{Document.table_name}.id = #{Attachment.table_name}.container_id " + | ||
| "LEFT JOIN #{Project.table_name} ON #{Document.table_name}.project_id = #{Project.table_name}.id"} | ||||
|
Jean-Philippe Lang
|
r330 | cattr_accessor :storage_path | ||
|
Toshi MARUYAMA
|
r5954 | @@storage_path = Redmine::Configuration['attachments_storage_path'] || "#{Rails.root}/files" | ||
|
Toshi MARUYAMA
|
r5673 | |||
|
Toshi MARUYAMA
|
r6636 | before_save :files_to_final_location | ||
|
Toshi MARUYAMA
|
r6643 | after_destroy :delete_from_disk | ||
|
Toshi MARUYAMA
|
r6636 | |||
|
Jean-Philippe Lang
|
r8771 | def container_with_blank_type_check | ||
| if container_type.blank? | ||||
| nil | ||||
| else | ||||
| container_without_blank_type_check | ||||
| end | ||||
| end | ||||
| alias_method_chain :container, :blank_type_check unless method_defined?(:container_without_blank_type_check) | ||||
|
Jean-Philippe Lang
|
r8556 | # Returns an unsaved copy of the attachment | ||
| def copy(attributes=nil) | ||||
| copy = self.class.new | ||||
| copy.attributes = self.attributes.dup.except("id", "downloads") | ||||
| copy.attributes = attributes if attributes | ||||
| copy | ||||
| end | ||||
|
Toshi MARUYAMA
|
r6594 | def validate_max_file_size | ||
|
Jean-Philippe Lang
|
r8556 | if @temp_file && self.filesize > Setting.attachment_max_size.to_i.kilobytes | ||
|
Jean-Philippe Lang
|
r8817 | errors.add(:base, l(:error_attachment_too_big, :max_size => Setting.attachment_max_size.to_i.kilobytes)) | ||
|
Jean-Philippe Lang
|
r2575 | end | ||
|
Jean-Philippe Lang
|
r330 | end | ||
|
Jean-Philippe Lang
|
r1306 | def file=(incoming_file) | ||
| unless incoming_file.nil? | ||||
| @temp_file = incoming_file | ||||
| if @temp_file.size > 0 | ||||
|
Jean-Philippe Lang
|
r8808 | if @temp_file.respond_to?(:original_filename) | ||
| self.filename = @temp_file.original_filename | ||||
|
Jean-Philippe Lang
|
r9200 | self.filename.force_encoding("UTF-8") if filename.respond_to?(:force_encoding) | ||
|
Jean-Philippe Lang
|
r8808 | end | ||
| if @temp_file.respond_to?(:content_type) | ||||
| self.content_type = @temp_file.content_type.to_s.chomp | ||||
| end | ||||
| if content_type.blank? && filename.present? | ||||
|
Jean-Philippe Lang
|
r3144 | self.content_type = Redmine::MimeType.of(filename) | ||
| end | ||||
|
Jean-Philippe Lang
|
r1306 | self.filesize = @temp_file.size | ||
| end | ||||
| end | ||||
| end | ||||
|
Toshi MARUYAMA
|
r8892 | |||
|
Jean-Philippe Lang
|
r1306 | def file | ||
| nil | ||||
| end | ||||
|
Jean-Philippe Lang
|
r8808 | def filename=(arg) | ||
| write_attribute :filename, sanitize_filename(arg.to_s) | ||||
| if new_record? && disk_filename.blank? | ||||
| self.disk_filename = Attachment.disk_filename(filename) | ||||
| end | ||||
| filename | ||||
| end | ||||
|
Jean-Philippe Lang
|
r2578 | # Copies the temporary file to its final location | ||
| # and computes its MD5 hash | ||||
|
Toshi MARUYAMA
|
r6636 | def files_to_final_location | ||
|
Jean-Philippe Lang
|
r1306 | if @temp_file && (@temp_file.size > 0) | ||
|
Jean-Philippe Lang
|
r6200 | logger.info("Saving attachment '#{self.diskfile}' (#{@temp_file.size} bytes)") | ||
|
Jean-Philippe Lang
|
r2578 | md5 = Digest::MD5.new | ||
|
Toshi MARUYAMA
|
r5673 | File.open(diskfile, "wb") do |f| | ||
|
Jean-Philippe Lang
|
r9470 | if @temp_file.respond_to?(:read) | ||
| buffer = "" | ||||
| while (buffer = @temp_file.read(8192)) | ||||
| f.write(buffer) | ||||
| md5.update(buffer) | ||||
| end | ||||
| else | ||||
| f.write(@temp_file) | ||||
| md5.update(@temp_file) | ||||
|
Jean-Philippe Lang
|
r2578 | end | ||
|
Jean-Philippe Lang
|
r1306 | end | ||
|
Jean-Philippe Lang
|
r2578 | self.digest = md5.hexdigest | ||
|
Jean-Philippe Lang
|
r1306 | end | ||
|
Jean-Philippe Lang
|
r6200 | @temp_file = nil | ||
|
Jean-Philippe Lang
|
r1306 | # Don't save the content type if it's longer than the authorized length | ||
| if self.content_type && self.content_type.length > 255 | ||||
| self.content_type = nil | ||||
| end | ||||
| end | ||||
|
Jean-Philippe Lang
|
r8556 | # Deletes the file from the file system if it's not referenced by other attachments | ||
|
Toshi MARUYAMA
|
r6643 | def delete_from_disk | ||
|
Jean-Philippe Lang
|
r8556 | if Attachment.first(:conditions => ["disk_filename = ? AND id <> ?", disk_filename, id]).nil? | ||
| delete_from_disk! | ||||
| end | ||||
|
Jean-Philippe Lang
|
r1306 | end | ||
| # Returns file's location on disk | ||||
| def diskfile | ||||
| "#{@@storage_path}/#{self.disk_filename}" | ||||
| end | ||||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r330 | def increment_download | ||
| increment!(:downloads) | ||||
| end | ||||
|
Jean-Philippe Lang
|
r538 | |||
| def project | ||||
|
Jean-Philippe Lang
|
r8771 | container.try(:project) | ||
|
Jean-Philippe Lang
|
r538 | end | ||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r2114 | def visible?(user=User.current) | ||
|
Jean-Philippe Lang
|
r8771 | container && container.attachments_visible?(user) | ||
|
Jean-Philippe Lang
|
r2114 | end | ||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r2114 | def deletable?(user=User.current) | ||
|
Jean-Philippe Lang
|
r8771 | container && container.attachments_deletable?(user) | ||
|
Jean-Philippe Lang
|
r2114 | end | ||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r636 | def image? | ||
|
Toshi MARUYAMA
|
r7771 | self.filename =~ /\.(bmp|gif|jpg|jpe|jpeg|png)$/i | ||
|
Jean-Philippe Lang
|
r636 | end | ||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r1506 | def is_text? | ||
| Redmine::MimeType.is_type?('text', filename) | ||||
| end | ||||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r1502 | def is_diff? | ||
| self.filename =~ /\.(patch|diff)$/i | ||||
| end | ||||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r2600 | # Returns true if the file is readable | ||
| def readable? | ||||
| File.readable?(diskfile) | ||||
| end | ||||
Eric Davis
|
r3409 | |||
|
Jean-Philippe Lang
|
r8771 | # Returns the attachment token | ||
| def token | ||||
| "#{id}.#{digest}" | ||||
| end | ||||
| # Finds an attachment that matches the given token and that has no container | ||||
| def self.find_by_token(token) | ||||
| if token.to_s =~ /^(\d+)\.([0-9a-f]+)$/ | ||||
| attachment_id, attachment_digest = $1, $2 | ||||
| attachment = Attachment.first(:conditions => {:id => attachment_id, :digest => attachment_digest}) | ||||
| if attachment && attachment.container.nil? | ||||
| attachment | ||||
| end | ||||
| end | ||||
| end | ||||
|
Eric Davis
|
r3409 | # Bulk attaches a set of files to an object | ||
| # | ||||
| # Returns a Hash of the results: | ||||
| # :files => array of the attached files | ||||
| # :unsaved => array of the files that could not be attached | ||||
| def self.attach_files(obj, attachments) | ||||
|
Jean-Philippe Lang
|
r8771 | result = obj.save_attachments(attachments, User.current) | ||
| obj.attach_saved_attachments | ||||
| result | ||||
|
Eric Davis
|
r3409 | end | ||
|
Toshi MARUYAMA
|
r5673 | |||
|
Toshi MARUYAMA
|
r7788 | def self.latest_attach(attachments, filename) | ||
|
Toshi MARUYAMA
|
r8892 | attachments.sort_by(&:created_on).reverse.detect { | ||
|
Toshi MARUYAMA
|
r7788 | |att| att.filename.downcase == filename.downcase | ||
| } | ||||
| end | ||||
|
Jean-Philippe Lang
|
r8771 | def self.prune(age=1.day) | ||
|
Jean-Philippe Lang
|
r8773 | attachments = Attachment.all(:conditions => ["created_on < ? AND (container_type IS NULL OR container_type = '')", Time.now - age]) | ||
|
Jean-Philippe Lang
|
r8771 | attachments.each(&:destroy) | ||
| end | ||||
|
Jean-Philippe Lang
|
r8556 | private | ||
| # Physically deletes the file from the file system | ||||
| def delete_from_disk! | ||||
| if disk_filename.present? && File.exist?(diskfile) | ||||
| File.delete(diskfile) | ||||
| end | ||||
| end | ||||
|
Jean-Philippe Lang
|
r330 | def sanitize_filename(value) | ||
|
Jean-Philippe Lang
|
r1306 | # get only the filename, not the whole path | ||
| just_filename = value.gsub(/^.*(\\|\/)/, '') | ||||
|
Jean-Philippe Lang
|
r330 | |||
|
Jean-Philippe Lang
|
r7797 | # Finally, replace invalid characters with underscore | ||
| @filename = just_filename.gsub(/[\/\?\%\*\:\|\"\'<>]+/, '_') | ||||
|
Jean-Philippe Lang
|
r330 | end | ||
|
Toshi MARUYAMA
|
r5673 | |||
|
Jean-Philippe Lang
|
r1418 | # Returns an ASCII or hashed filename | ||
| def self.disk_filename(filename) | ||||
|
Jean-Philippe Lang
|
r3397 | timestamp = DateTime.now.strftime("%y%m%d%H%M%S") | ||
| ascii = '' | ||||
|
Jean-Philippe Lang
|
r1418 | if filename =~ %r{^[a-zA-Z0-9_\.\-]*$} | ||
|
Jean-Philippe Lang
|
r3397 | ascii = filename | ||
|
Jean-Philippe Lang
|
r1418 | else | ||
|
Jean-Philippe Lang
|
r3397 | ascii = Digest::MD5.hexdigest(filename) | ||
|
Jean-Philippe Lang
|
r1418 | # keep the extension if any | ||
|
Jean-Philippe Lang
|
r3397 | ascii << $1 if filename =~ %r{(\.[a-zA-Z0-9]+)$} | ||
|
Jean-Philippe Lang
|
r1418 | end | ||
|
Jean-Philippe Lang
|
r3397 | while File.exist?(File.join(@@storage_path, "#{timestamp}_#{ascii}")) | ||
| timestamp.succ! | ||||
| end | ||||
| "#{timestamp}_#{ascii}" | ||||
|
Jean-Philippe Lang
|
r1418 | end | ||
|
Jean-Philippe Lang
|
r2 | end | ||