Sha256: f269ab833c3011211733693907d6b7597e30f3f7f0df1f0e04f377cede92819d
Contents?: true
Size: 1.83 KB
Versions: 11
Compression:
Stored size: 1.83 KB
Contents
module FriendlyId
class TaskRunner
extend Forwardable
attr_accessor :days
attr_accessor :klass
attr_accessor :task_options
def_delegators :klass, :find, :friendly_id_config, :update_all
OLD_SLUG_DAYS = 45
def initialize(&block)
self.klass = ENV["MODEL"]
self.days = ENV["DAYS"]
end
def days=(days)
@days ||= days.blank? ? OLD_SLUG_DAYS : days.to_i
end
def klass=(klass)
@klass ||= klass.to_s.classify.constantize unless klass.blank?
end
def make_slugs
validate_uses_slugs
cond = "slugs.id IS NULL"
options = {:limit => 100, :include => :slugs, :conditions => cond, :order => "#{klass.table_name}.id ASC"}.merge(task_options || {})
while records = find(:all, options) do
break if records.size == 0
records.each do |record|
record.save(:validate => false)
yield(record) if block_given?
end
options[:conditions] = cond + " and #{klass.table_name}.id > #{records.last.id}"
end
end
def delete_slugs
validate_uses_slugs
Slug.destroy_all(["sluggable_type = ?", klass.to_s])
if column = friendly_id_config.cache_column
update_all("#{column} = NULL")
end
end
def delete_old_slugs
conditions = ["created_at < ?", DateTime.now - days]
if klass
conditions[0] << " AND sluggable_type = ?"
conditions << klass.to_s
end
Slug.all(:conditions => conditions).select(&:outdated?).map(&:destroy)
end
def validate_uses_slugs
(raise "You need to pass a MODEL=<model name> argument to rake") if klass.blank?
unless friendly_id_config.use_slug?
raise "Class '%s' doesn't use slugs" % klass.to_s
end
rescue NoMethodError
raise "Class '%s' doesn't use FriendlyId" % klass.to_s
end
end
end
Version data entries
11 entries across 11 versions & 1 rubygems