engine/app/models/theme_asset.rb

class ThemeAsset
  
  include Locomotive::Mongoid::Document
  
  ## Extensions ##  
  include Models::Extensions::Asset::Vignette
  
  ## fields ##
  field :slug
  field :content_type
  field :width, :type => Integer
  field :height, :type => Integer
  field :size, :type => Integer
  mount_uploader :source, ThemeAssetUploader
  
  ## associations ##
  belongs_to_related :site
  
  ## callbacks ##
  before_validate :sanitize_slug
  before_validate :store_plain_text
  before_save     :set_slug
  
  ## validations ##
  validate :extname_can_not_be_changed
  validates_presence_of :site, :source
  validates_presence_of :slug, :if => Proc.new { |a| a.new_record? && a.performing_plain_text? }
  validates_uniqueness_of :slug, :scope => [:site_id, :content_type]
  validates_integrity_of :source
    
  ## accessors ##
  attr_accessor :performing_plain_text
  
  ## methods ##
  
  %w{movie image stylesheet javascript}.each do |type|
    define_method("#{type}?") do
      self.content_type == type
    end  
  end
  
  def plain_text
    @plain_text ||= (if self.stylesheet? || self.javascript?
      self.source.read
    else
      nil
    end)
  end
  
  def plain_text=(source)
    self.performing_plain_text = true if self.performing_plain_text.nil?
    @plain_text = source
  end
  
  def performing_plain_text?
    return true if !self.new_record? && !self.image? && !self.movie? && self.errors.empty?
    
    !(self.performing_plain_text.blank? || self.performing_plain_text == 'false' || self.performing_plain_text == false)
  end
  
  def store_plain_text
    return if self.plain_text.blank?
    
    self.source = CarrierWave::SanitizedFile.new({ 
      :tempfile => StringIO.new(self.plain_text),
      :filename => "#{self.slug}.#{self.stylesheet? ? 'css' : 'js'}"
    })
  end
  
  def to_liquid
    { :url => self.source.url }.merge(self.attributes)
  end
    
  protected
  
  def sanitize_slug
    self.slug.slugify!(:underscore => true) if self.slug.present?
  end
  
  def set_slug
    if self.slug.blank?
      self.slug = File.basename(self.source_filename, File.extname(self.source_filename))
      self.sanitize_slug
    end
  end
  
  def extname_can_not_be_changed
    return if self.new_record?
        
    if File.extname(self.source.file.original_filename) != File.extname(self.source_filename)
      self.errors.add(:source, :extname_changed)
    end
  end
end