pass test: Picture.store

[pettanr/pettanr.git] / app / models / picture.rb

diff --git a/app/models/picture.rb b/app/models/picture.rb
index f62ecbe..33d4bf4 100644 (file)
--- a/app/models/picture.rb
+++ b/app/models/picture.rb
@@ -1,2 +1,162 @@
 class Picture < ActiveRecord::Base
+  belongs_to :original_picture
+  belongs_to :license
+  belongs_to :artist
+  
+  validates :original_picture_id, :presence => true, :numericality => true, :existence => true
+  validates :revision, :presence => true, :numericality => true
+  validates :ext, :presence => true, :length => {:maximum => 4}, :inclusion => {:in => ['png', 'jpeg', 'gif']}\r
+  validates :width, :presence => true, :numericality => true, :natural_number => true
+  validates :height, :presence => true, :numericality => true, :natural_number => true
+  validates :filesize, :presence => true, :numericality => {:greater_than => 0, :less_than_or_equal_to => 2000000}, :natural_number => true
+  validates :license_id, :presence => true, :numericality => true, :existence => true
+  validates :artist_id, :presence => true, :numericality => true, :existence => true
+  validates :artist_name, :presence => true
+  
+  def dext
+    self.ext.downcase
+  end
+  
+  def filename
+    "#{self.id}.#{self.dext}"
+  end
+  
+  def mime_type
+    "image/#{self.dext}"
+  end
+  
+  def new_revision
+    Picture.maximum(:revision, :conditions => ['original_picture_id = ?', self.original_picture_id]).to_i + 1
+  end
+  
+  def v(rimg)
+    rimg.flip.to_blob
+  end
+  
+  def h(rimg)
+    rimg.flop.to_blob
+  end
+  
+  def vh(rimg)
+    rimg.flip.flop.to_blob
+  end
+  
+  def to_gif?
+    self.dext == 'png' and self.flag_gif_convert >= 0
+  end
+  
+  def self.png_to_gif(data)
+    res = nil
+    begin
+      mgk = Magick::Image.from_blob(data).shift
+      mgk.format = 'gif'
+      res = mgk
+    rescue
+      res = false
+    end
+    res
+  end
+  
+  def subdirs
+    flag_reverse < 0 ? [''] : ['', 'v', 'h', 'vh']
+  end
+  
+  def store(mgk)
+    res = false
+    self.revision = self.new_revision
+    if res = self.save
+      if res = self.store_picture(mgk)
+        if self.to_gif?
+          if gifmgk = Picture.png_to_gif(mgk.to_blob)
+            res = self.store_picture(gifmgk)
+          else
+            res = false
+          end
+        end
+      end
+    end
+    res
+  end
+  
+  def store_picture(mgk)
+    res = false
+    subdirs.each do |d|
+      picdata = d.empty? ? mgk.to_blob : self.__send__(d, mgk)
+      res = PictureIO.picture_io.put(picdata, "#{self.id}.#{mgk.format}", d)
+      break unless res
+    end
+    res
+  end
+  
+  def restore(subdir = nil)
+    PictureIO.resource_picture_io.get self.filename, subdir
+  end
+  
+  def flags
+    begin
+      @flags = JSON.parse(self.settings) unless @flags
+    rescue 
+    end
+    @flags
+  end
+  
+  def flags=(s)
+    @flags = s
+  end
+  
+  def flag_open
+    @flag_open = flags["open"] unless @flag_open
+    @flag_open
+  end
+  
+  def flag_commercial
+    @flag_commercial = flags["commercial"] unless @flag_commercial
+    @flag_commercial
+  end
+  
+  def flag_official
+    @flag_official = flags["official"] unless @flag_official
+    @flag_official
+  end
+  
+  def flag_attribution
+    @flag_attribution = flags["attribution"] unless @flag_attribution
+    @flag_attribution
+  end
+  
+  def flag_derive
+    @flag_derive = flags["derive"] unless @flag_derive
+    @flag_derive
+  end
+  
+  def flag_thumbnail
+    @flag_thumbnail = flags["thumbnail"] unless @flag_thumbnail
+    @flag_thumbnail
+  end
+  
+  def flag_gif_convert
+    @flag_gif_convert = flags["gif_convert"] unless @flag_gif_convert
+    @flag_gif_convert
+  end
+  
+  def flag_reverse
+    @flag_reverse = flags["reverse"] unless @flag_reverse
+    @flag_reverse
+  end
+  
+  def flag_resize
+    @flag_resize = flags["resize"] unless @flag_resize
+    @flag_resize
+  end
+  
+  def flag_sync_vh
+    @flag_sync_vh = flags["sync_vh"] unless @flag_sync_vh
+    @flag_sync_vh
+  end
+  
+  def flag_overlap
+    @flag_overlap = flags["overlap"] unless @flag_overlap
+    @flag_overlap
+  end
+  
 end