adbanner-test/app/models/ad_image.rb

125 lines
4.0 KiB
Ruby

require 'uri'
class AdImage
include Mongoid::Document
include Mongoid::Timestamps
mount_uploader :file, BannerImageUploader
mount_uploader :video_snapshot, BannerImageUploader
mount_uploader :video_file, BannerUploader
mount_uploader :video_file_webm, BannerUploader
field :default_video_snapshot, type: Boolean, default: false
field :auto_convert_video, type: Boolean, default: false
field :title, type: String, localize: true
field :context, type: String, localize: true
field :weight, type: Integer, default: 1
field :out_link, type: String
field :link_open, type: String
field :postdate , :type => DateTime, :default => Time.now
field :deadline , :type => DateTime
field :youtube , :type => String
field :sort_number, :type => Integer
field :language_enabled, :type => Array, :default => ["en","zh_tw"]
field :exchange_item, :default => "1"
LINK_OPEN_TYPES = ["local", "new_window"]
belongs_to :banner
before_validation :add_http
# validates :file, presence: true
validates :out_link, format: {:with=> /\A(http:\/\/|https:\/\/|\/)/i}, allow_blank: true
# validates :title, presence: true
scope :can_display, ->{self.and(AdImage.unscoped.or({:postdate.lte=>Time.now},{:postdate=>nil}).selector,AdImage.unscoped.or({:deadline.gte=>Time.now},{:deadline=>nil}).selector)}
scope :is_expired, ->{self.and(AdImage.unscoped.or({:deadline.lte=>Time.now}).selector)}
scope :not_expired, ->{self.and(AdImage.unscoped.or({:deadline.gte=>Time.now},{:deadline=>nil}).selector)}
before_save do
unless @skip_callback
if self.video_snapshot_changed?
self.default_video_snapshot = false
end
end
end
after_save do
unless @skip_callback
if (self.video_file_changed? || self.auto_convert_video_changed?) && self.auto_convert_video
Thread.new do
self.generate_webm
end
end
if self.video_file? && self.video_file_changed? && (!(self.video_snapshot?) || self.default_video_snapshot)
Thread.new do
self.generate_video_snapshot
end
end
end
end
def generate_video_snapshot
if self.video_file?
self.default_video_snapshot = true
self[:video_snapshot] = File.basename(self.video_file.file.path).split(/\.[^.]+$/)[0] + ".jpg"
FileUtils.mkdir_p(File.dirname(self.video_snapshot.file.path))
system("tmp/ffmpeg/ffmpeg -i #{self.video_file.file.path} -vframes 1 #{self.video_snapshot.file.path}")
@skip_callback = true
self.save
@skip_callback = false
true
end
end
def generate_webm
video_path = self.video_file.path rescue nil
if !video_path.blank?
video_webm = video_path.split('.')[0...-1].join('.')+".webm"
core_num = [`cat /proc/cpuinfo | grep processor | wc -l`.to_i/2,1].max
flag = system("tmp/ffmpeg/ffmpeg -i #{video_path} -c:v libvpx-vp9 -crf 35 -b:v 0 -b:a 96k -c:a libopus -filter:v fps=20 #{video_webm} -cpu-used #{core_num}")
if flag
self.video_file_webm = File.open(video_webm)
@skip_callback = true
self.save
@skip_callback = false
true
else
puts "generate webm failed"
end
end
end
def expired?
self.deadline<Time.now rescue false
end
def embed_url
uri = URI.parse(self.youtube) rescue nil
if !uri.nil? && uri.host == "www.youtube.com"
params = CGI.parse(uri.query)
v = params['v'].first
return "//www.youtube.com/embed/#{v}"
else
return ""
end
end
def youtube_thumb
uri = URI.parse(self.youtube) rescue nil
if !uri.nil? && uri.host == "www.youtube.com"
params = CGI.parse(uri.query) rescue {}
if params.has_key?('v')
v = params['v'].first
return "http://img.youtube.com/vi/#{v}/0.jpg"
else
return ""
end
else
return ""
end
end
protected
def add_http
unless self.out_link.blank? || self.out_link[/^http:\/\//] || self.out_link[/^https:\/\//] || self.out_link[/^\//]
self.out_link = 'http://' + self.out_link
end
end
end