attachment.rb
98 lines
| 2.9 KiB
| text/x-ruby
|
RubyLexer
Jean-Philippe Lang
|
r330 | # redMine - project management software | ||
| # Copyright (C) 2006-2007 Jean-Philippe Lang | ||||
| # | ||||
| # 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. | ||||
| # | ||||
| # 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. | ||||
| # | ||||
| # 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" | ||||
| validates_presence_of :container, :filename | ||||
|
Jean-Philippe Lang
|
r590 | validates_length_of :filename, :maximum => 255 | ||
| validates_length_of :disk_filename, :maximum => 255 | ||||
|
Jean-Philippe Lang
|
r330 | cattr_accessor :storage_path | ||
| @@storage_path = "#{RAILS_ROOT}/files" | ||||
| def validate | ||||
| errors.add_to_base :too_long if self.filesize > Setting.attachment_max_size.to_i.kilobytes | ||||
| end | ||||
| def file=(incomming_file) | ||||
| unless incomming_file.nil? | ||||
| @temp_file = incomming_file | ||||
| if @temp_file.size > 0 | ||||
| self.filename = sanitize_filename(@temp_file.original_filename) | ||||
| self.disk_filename = DateTime.now.strftime("%y%m%d%H%M%S") + "_" + self.filename | ||||
| self.content_type = @temp_file.content_type | ||||
| self.filesize = @temp_file.size | ||||
| end | ||||
| end | ||||
| end | ||||
| def file | ||||
| nil | ||||
| end | ||||
| # Copy temp file to its final location | ||||
| def before_save | ||||
| if @temp_file && (@temp_file.size > 0) | ||||
| logger.debug("saving '#{self.diskfile}'") | ||||
| File.open(diskfile, "wb") do |f| | ||||
| f.write(@temp_file.read) | ||||
| end | ||||
| self.digest = Digest::MD5.hexdigest(File.read(diskfile)) | ||||
| end | ||||
| end | ||||
| # Deletes file on the disk | ||||
| def after_destroy | ||||
| if self.filename? | ||||
| File.delete(diskfile) if File.exist?(diskfile) | ||||
| end | ||||
| end | ||||
| # Returns file's location on disk | ||||
| def diskfile | ||||
| "#{@@storage_path}/#{self.disk_filename}" | ||||
| end | ||||
| def increment_download | ||||
| increment!(:downloads) | ||||
| end | ||||
| # returns last created projects | ||||
| def self.most_downloaded | ||||
| find(:all, :limit => 5, :order => "downloads DESC") | ||||
| end | ||||
|
Jean-Philippe Lang
|
r538 | |||
| def project | ||||
| container.is_a?(Project) ? container : container.project | ||||
| end | ||||
|
Jean-Philippe Lang
|
r330 | private | ||
| def sanitize_filename(value) | ||||
| # get only the filename, not the whole path | ||||
| just_filename = value.gsub(/^.*(\\|\/)/, '') | ||||
| # NOTE: File.basename doesn't work right with Windows paths on Unix | ||||
| # INCORRECT: just_filename = File.basename(value.gsub('\\\\', '/')) | ||||
| # Finally, replace all non alphanumeric, underscore or periods with underscore | ||||
| @filename = just_filename.gsub(/[^\w\.\-]/,'_') | ||||
| end | ||||
|
Jean-Philippe Lang
|
r2 | |||
| end | ||||