app/models/original_picture.rb

   1 class OriginalPicture < ActiveRecord::Base
   2   belongs_to :artist
   3   belongs_to :license
   4   has_one :resource_picture
   5
   6   validates :ext, :presence => true, :length => {:maximum => 4}, :inclusion => {:in => ['png', 'jpeg', 'gif']}\r
   7   validates :width, :presence => true, :numericality => true, :natural_number => true
   8   validates :height, :presence => true, :numericality => true, :natural_number => true
   9   validates :filesize, :presence => true, :numericality => {:greater_than => 0, :less_than_or_equal_to => 2000000}, :natural_number => true
  10   validates :artist_id, :presence => true, :numericality => true, :existence => true
  11   validates :license_id, :presence => true, :numericality => true, :existence => true
  12
  13   before_destroy :destroy_with_file
  14
  15   def validate
  16     errors.add(:filesize, 'size over(1MB)') if self.filesize > 1000000
  17   end
  18
  19   def supply_default art
  20     self.license_id = art.default_license_id if self.license_id.blank?
  21   end
  22
  23   def self.default_page_size
  24     25
  25   end
  26
  27   def self.max_page_size
  28     100
  29   end
  30
  31   def self.page prm = nil
  32     page = prm.to_i
  33     page = 1 if page < 1
  34     page
  35   end
  36
  37   def self.page_size prm = self.default_page_size
  38     page_size = prm.to_i
  39     page_size = self.max_page_size if page_size > self.max_page_size
  40     page_size = self.default_page_size if page_size < 1
  41     page_size
  42   end
  43
  44   def self.list artist_id, opt = {}, page = 1, page_size = self.default_page_size
  45     opt.merge!(self.list_opt) unless opt[:include]
  46     opt.merge!({:conditions => ['artist_id = ?', artist_id],
  47       :order => 'updated_at desc', :limit => page_size, :offset => (page -1) * page_size}
  48     )
  49     OriginalPicture.find(:all, opt)
  50   end
  51
  52   def self.list_opt
  53     {:include => [:license, :resource_picture]}
  54   end
  55
  56   def self.list_json_opt
  57     {:include => [:license, :resource_picture]}
  58   end
  59
  60   def self.show cid, artist, opt = {}
  61     pic = OriginalPicture.find(cid, :include => self.show_include_opt(opt))
  62     raise ActiveRecord::Forbidden unless pic.own?(artist)
  63     pic
  64   end
  65
  66   def self.show_include_opt opt = {}
  67     res = [:license, :resource_picture]
  68     res.push(opt[:include]) if opt[:include]
  69     res
  70   end
  71
  72   def self.show_json_include_opt
  73     {:include => [:license, :resource_picture]}
  74   end
  75
  76   def destroy_with_file
  77     PictureIO.original_picture_io.delete self.filename
  78     self.resource_picture.destroy
  79   end
  80
  81   def dext
  82     self.ext.downcase
  83   end
  84
  85   def filename
  86     "#{self.id}.#{self.dext}"
  87   end
  88
  89   def mime_type
  90     "image/#{self.dext}"
  91   end
  92
  93   def url
  94     '/original_pictures/' + filename
  95   end
  96
  97   def data_to_mgk picture_data
  98     begin
  99       mgk = Magick::Image.from_blob(picture_data).shift
 100     rescue
 101       self.errors.add :base, 'magick failed'
 102       return false
 103     end
 104     mgk
 105   end
 106
 107   def store(picture_data, art, lid = nil)
 108     mgk = data_to_mgk picture_data
 109     return false unless mgk
 110     res = false
 111     self.attributes = {:ext => mgk.format.downcase, :width => mgk.columns, :height => mgk.rows,
 112       :filesize => mgk.filesize, :artist_id => art.id,
 113       :license_id => lid.blank? ? art.default_license_id : lid.to_i
 114     }
 115     OriginalPicture.transaction do
 116       if res = self.save
 117         if res = PictureIO.original_picture_io.put(picture_data, self.filename)
 118           rp = ResourcePicture.update_picture(self)
 119           unless rp.store(mgk)
 120             res = false
 121             PictureIO.original_picture_io.delete(self.filename)
 122             self.errors.add :base, 'resource picture copying error'
 123             raise ActiveRecord::Rollback
 124           end
 125           res = true
 126         else
 127           self.errors.add :base, 'original picture io does not work'
 128           raise ActiveRecord::Rollback
 129         end
 130       end
 131     end
 132     res
 133   end
 134
 135   def restore(subdir = nil)
 136     PictureIO.original_picture_io.get self.filename, subdir
 137   end
 138
 139   def own? art
 140     return false unless art
 141     self.artist_id == art.id
 142   end
 143
 144 end