t#31715:speech_balloon boost

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

diff --git a/app/models/resource_picture.rb b/app/models/resource_picture.rb
index c8bd3f7..30eb7e6 100644 (file)
--- a/app/models/resource_picture.rb
+++ b/app/models/resource_picture.rb
@@ -18,6 +18,15 @@ class ResourcePicture < ActiveRecord::Base
   validates :classname, :presence => true, :length => {:maximum => 50}
   validates :picture_id, :presence => true, :numericality => true, :existence => {:both => false}
   
+  before_validation :valid_encode
+  
+  def valid_encode
+    ['artist_name', 'classname', 'credit', 'settings'].each do |a|
+      next if attributes[a] == nil
+      raise Pettanr::BadRequest unless attributes[a].valid_encoding?
+    end
+  end
+  
   def supply_default
   end
   
@@ -28,18 +37,18 @@ class ResourcePicture < ActiveRecord::Base
     self.attributes = attr
   end
   
-  def own? ar
-    return false unless ar.is_a?(Artist)
-    ar.id == self.artist_id
+  def own? roles
+    roles = [roles] unless roles.respond_to?(:each)
+    ar = ResourcePicture.get_artist_from_roles roles
+    return false unless ar
+    self.artist_id == ar.id
   end
   
-  def visible? au
-    if au == nil
-      return false if MagicNumber['run_mode'] == 1
-    elsif au.is_a?(Author)
-      return true
+  def visible? roles
+    if MagicNumber['run_mode'] == 0
+      return false unless guest_role_check(roles)
     else
-      return false
+      return false unless resource_reader_role_check(roles)
     end
     true
   end
@@ -99,27 +108,51 @@ class ResourcePicture < ActiveRecord::Base
     page_size
   end
   
+  def self.mylist_where ar
+    ['resource_pictures.artist_id = ?', ar.id]
+  end
+  
+  def self.himlist_where ar
+    ['resource_pictures.artist_id = ?', ar.id]
+  end
+  
   def self.list page = 1, page_size = self.default_page_size
-    opt = {}
-    opt.merge!(self.list_opt)
-    opt.merge!({:limit => page_size, :offset => (page -1) * page_size}) if page_size > 0
-    opt.merge!({:order => 'updated_at desc'})
-    ResourcePicture.find(:all, opt)
+    ResourcePicture.includes(ResourcePicture.list_opt).order('resource_pictures.updated_at desc').offset((page -1) * page_size).limit(page_size)
+  end
+  
+  def self.mylist ar, page = 1, page_size = Author.default_resource_picture_page_size
+    ResourcePicture.where(self.mylist_where(ar)).includes(ResourcePicture.list_opt).order('resource_pictures.updated_at desc').offset((page -1) * page_size).limit(page_size)
+  end
+  
+  def self.himlist ar, page = 1, page_size = Author.default_resource_picture_page_size
+    ResourcePicture.where(self.himlist_where(ar)).includes(ResourcePicture.list_opt).order('resource_pictures.updated_at desc').offset((page -1) * page_size).limit(page_size)
+  end
+  
+  def self.list_paginate page = 1, page_size = self.default_page_size
+    Kaminari.paginate_array(Array.new(ResourcePicture.count, nil)).page(page).per(page_size)
+  end
+  
+  def self.mylist_paginate ar, page = 1, page_size = Author.default_resource_picture_page_size
+    Kaminari.paginate_array(Array.new(ResourcePicture.where(self.mylist_where(ar)).count, nil)).page(page).per(page_size)
+  end
+  
+  def self.himlist_paginate ar, page = 1, page_size = Author.default_resource_picture_page_size
+    Kaminari.paginate_array(Array.new(ResourcePicture.where(self.himlist_where(ar)).count, nil)).page(page).per(page_size)
   end
   
   def self.list_opt
-    {:include => {:license => {}, :artist => {}, :picture => {}} }
+    {:license => {}, :artist => {}, :picture => {} }
   end
   
   def self.list_json_opt
     {:include => {:license => {}, :artist => {}, :picture => {}} }
   end
   
-  def self.show rid, au
+  def self.show rid, roles
     opt = {}
     opt.merge!(self.show_opt)
     r = ResourcePicture.find(rid, opt)
-    raise ActiveRecord::Forbidden unless r.visible?(au)
+    raise ActiveRecord::Forbidden unless r.visible?(roles)
     r
   end
   
@@ -154,6 +187,9 @@ class ResourcePicture < ActiveRecord::Base
     res = false
     self.overwrite self.original_picture
     ResourcePicture.transaction do
+      self.original_picture.published_at = Time.now
+      self.original_picture.stopped_at = nil
+      raise ActiveRecord::Rollback unless self.original_picture.save
       pc = self.new_picture imager
       if pc
         self.picture_id = pc.id
@@ -205,19 +241,32 @@ class ResourcePicture < ActiveRecord::Base
   end
   
   def unpublish
-    r = true
+    res = false
     ResourcePicture.transaction do
-      self.destroy
-      PictureIO.resource_picture_io.delete(self.filename)
-      PictureIO.resource_picture_io.delete(self.filename, 'full')
+      self.original_picture.published_at = nil
+      self.original_picture.stopped_at = Time.now
+      raise ActiveRecord::Rollback unless self.original_picture.save
+      begin
+        PictureIO.resource_picture_io.delete(self.filename) if PictureIO.resource_picture_io.exist?(self.filename)
+        PictureIO.resource_picture_io.delete(self.filename, 'full') if PictureIO.resource_picture_io.exist?(self.filename, 'full')
+      rescue PictureIO::Error
+        res = false
+        raise ActiveRecord::Rollback
+      end
+      res = self.destroy
+      raise ActiveRecord::Rollback unless res
     end
-    r
+    res
   end
   
   def self.visible_count
     ResourcePicture.count
   end
   
+  def picture_data
+    Base64.encode64(self.restore 'full')
+  end
+  
   def credit_template
     "#{self.classname.tableize}/attributes/credit"
   end