109 lines
2.5 KiB
Ruby
109 lines
2.5 KiB
Ruby
class Hbc::CLI::Cleanup < Hbc::CLI::Base
|
|
OUTDATED_DAYS = 10
|
|
OUTDATED_TIMESTAMP = Time.now - (60 * 60 * 24 * OUTDATED_DAYS)
|
|
|
|
def self.help
|
|
"cleans up cached downloads and tracker symlinks"
|
|
end
|
|
|
|
def self.needs_init?
|
|
true
|
|
end
|
|
|
|
def self.run(*args)
|
|
if args.empty?
|
|
default.cleanup!
|
|
else
|
|
default.cleanup(args)
|
|
end
|
|
end
|
|
|
|
def self.default
|
|
@default ||= new(Hbc.cache, Hbc.cleanup_outdated)
|
|
end
|
|
|
|
attr_reader :cache_location, :outdated_only
|
|
def initialize(cache_location, outdated_only)
|
|
@cache_location = Pathname.new(cache_location)
|
|
@outdated_only = outdated_only
|
|
end
|
|
|
|
def cleanup!
|
|
remove_cache_files
|
|
end
|
|
|
|
def cleanup(tokens)
|
|
remove_cache_files(*tokens)
|
|
end
|
|
|
|
def cache_files
|
|
return [] unless cache_location.exist?
|
|
cache_location.children
|
|
.map(&method(:Pathname))
|
|
.reject(&method(:outdated?))
|
|
end
|
|
|
|
def outdated?(file)
|
|
outdated_only && file && file.stat.mtime > OUTDATED_TIMESTAMP
|
|
end
|
|
|
|
def incomplete?(file)
|
|
file.extname == ".incomplete"
|
|
end
|
|
|
|
def cache_incompletes
|
|
cache_files.select(&method(:incomplete?))
|
|
end
|
|
|
|
def cache_completes
|
|
cache_files.reject(&method(:incomplete?))
|
|
end
|
|
|
|
def disk_cleanup_size
|
|
Hbc::Utils.size_in_bytes(cache_files)
|
|
end
|
|
|
|
def remove_cache_files(*tokens)
|
|
message = "Removing cached downloads"
|
|
message.concat " for #{tokens.join(', ')}" unless tokens.empty?
|
|
message.concat " older than #{OUTDATED_DAYS} days old" if outdated_only
|
|
ohai message
|
|
|
|
deletable_cache_files = if tokens.empty?
|
|
cache_files
|
|
else
|
|
start_withs = tokens.map { |token| "#{token}--" }
|
|
|
|
cache_files.select { |path|
|
|
path.basename.to_s.start_with?(*start_withs)
|
|
}
|
|
end
|
|
|
|
delete_paths(deletable_cache_files)
|
|
end
|
|
|
|
def delete_paths(paths)
|
|
cleanup_size = 0
|
|
processed_files = 0
|
|
paths.each do |item|
|
|
next unless item.exist?
|
|
processed_files += 1
|
|
if Hbc::Utils.file_locked?(item)
|
|
puts "skipping: #{item} is locked"
|
|
next
|
|
end
|
|
puts item
|
|
item_size = File.size?(item)
|
|
cleanup_size += item_size unless item_size.nil?
|
|
item.unlink
|
|
end
|
|
|
|
if processed_files.zero?
|
|
puts "Nothing to do"
|
|
else
|
|
disk_space = disk_usage_readable(cleanup_size)
|
|
ohai "This operation has freed approximately #{disk_space} of disk space."
|
|
end
|
|
end
|
|
end
|