asciinema.org/app/models/asciicast.rb
2017-04-19 23:59:43 +02:00

167 lines
3.9 KiB
Ruby

class Asciicast < ActiveRecord::Base
ORDER_MODES = { date: 'created_at', popularity: 'views_count' }
mount_uploader :stdin_data, StdinDataUploader
mount_uploader :stdin_timing, StdinTimingUploader
mount_uploader :stdout_data, StdoutDataUploader
mount_uploader :stdout_timing, StdoutTimingUploader
mount_uploader :stdout_frames, StdoutFramesUploader
mount_uploader :file, AsciicastUploader
mount_uploader :image, ImageUploader
serialize :snapshot, ActiveSupportJsonProxy
belongs_to :user
has_many :comments, -> { order(:created_at) }, dependent: :destroy
has_many :likes, dependent: :destroy
validates :user, :duration, presence: true
validates :stdout_data, :stdout_timing, presence: true, unless: :file
validates :file, presence: true, unless: :stdout_data
validates :snapshot_at, numericality: { greater_than: 0, allow_blank: true }
validates :terminal_columns, presence: true, numericality: { greater_than: 0, less_than_or_equal_to: 1000 }
validates :terminal_lines, presence: true, numericality: { greater_than: 0, less_than_or_equal_to: 500 }
scope :featured, -> { where(featured: true) }
scope :by_date, -> { order("created_at DESC") }
scope :by_random, -> { order("RANDOM()") }
scope :non_private, -> { where(private: false) }
scope :homepage_latest, -> { non_private.by_date.limit(6).includes(:user) }
scope :homepage_featured, -> { non_private.featured.where("created_at > ?", 1.year.ago).by_random.limit(6).includes(:user) }
before_create :generate_secret_token
def self.find_by_id_or_secret_token!(thing)
if thing.size == 25
find_by_secret_token!(thing)
else
non_private.find(thing)
end
end
def self.cache_key
timestamps = scoped.select(:updated_at).map { |o| o.updated_at.to_i }
Digest::MD5.hexdigest timestamps.join('/')
end
def self.paginate(page, per_page)
page(page).per(per_page)
end
def self.for_category_ordered(category, order, page = nil, per_page = nil)
collection = self
if category == :featured
collection = collection.featured
end
collection = collection.order("#{ORDER_MODES.fetch(order)} DESC")
if page
collection = collection.paginate(page, per_page)
end
collection
end
def width
terminal_columns
end
def height
terminal_lines
end
def title=(value)
value ? super(value.strip[0...255]) : super
end
def command=(value)
value ? super(value.strip[0...255]) : super
end
def theme_name=(value)
if value == ""
value = nil
end
super(value)
end
def self.generate_secret_token
SecureRandom.hex.to_i(16).to_s(36).rjust(25, '0')
end
def to_param
if private?
secret_token
else
id.to_s
end
end
def data
if file.present?
file
else
stdout_frames
end
end
def data_url(options = {})
data.url(options)
end
def ready?
data_url && snapshot
end
def download_filename
"asciicast-#{id}.json"
end
def stdout
return @stdout if @stdout
@stdout = Stdout::Buffered.new(get_stdout)
end
def with_terminal
terminal = Terminal.new(terminal_columns, terminal_lines)
yield(terminal)
ensure
terminal.release if terminal
end
def theme
theme_name && Theme.for_name(theme_name)
end
def owner?(user)
user && self.user == user
end
private
def get_stdout
if version == 0
Stdout::MultiFile.new(stdout_data.decompressed_path,
stdout_timing.decompressed_path)
else
Stdout::SingleFile.new(file.absolute_url)
end
end
def image_hash
version = 2 # version of screenshot, increment to force regeneration
input = "#{version}/#{id}/#{snapshot_at}"
Digest::SHA1.hexdigest(input)
end
def generate_secret_token
begin
self.secret_token = self.class.generate_secret_token
end while self.class.exists?(secret_token: secret_token)
end
end