Class: Rhcf::Utils::DownloadCache

Inherits:
Object
  • Object
show all
Includes:
FileUtils
Defined in:
lib/rhcf/utils/download_cache.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(cache_id = 'default', ttl = nil, root_path = nil) ⇒ DownloadCache 



10
11
12
13
14
 
# File 'lib/rhcf/utils/download_cache.rb', line 10

def initialize(cache_id= 'default', ttl = nil, root_path = nil)
  @cache_id = cache_id
  @ttl = ttl
  @root_path ||= "/tmp/#{self.class.name}"
end

Class Method Details

.hit_fname?(fname, ttl) ⇒ Boolean 



44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/rhcf/utils/download_cache.rb', line 44

def self.hit_fname?(fname, ttl)
  
  if File.exist?(fname) 
    if ttl
      File::Stat.new(fname).ctime > Time.now - ttl
    else 
      true
    end
  else
    false
  end
end

Instance Method Details

#download!(url, outfile) ⇒ Object 



29
30
31
32
33
34
35
36
37
 
# File 'lib/rhcf/utils/download_cache.rb', line 29

def download!(url, outfile)
  mkdir_p(File.dirname(outfile))
  File.open(outfile, 'wb') do |fd|
    open(url, "rb") do |down|
      fd.write(down.read)
    end
  end
  outfile
end

#filename_for(url) ⇒ Object 



57
58
59
60
61
62
 
# File 'lib/rhcf/utils/download_cache.rb', line 57

def filename_for(url)
  hash =  Digest::MD5.hexdigest(url)
  uri = URI(url)
  basename = File.basename(uri.path)
  File.join(@root_path, @cache_id, hash, basename) 
end

#get(url) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
 
# File 'lib/rhcf/utils/download_cache.rb', line 17

def get(url)
  outfile = filename_for(url)
  return outfile if self.class.hit_fname?(outfile, @ttl) # here goes the cache
  download!(url, outfile)
rescue 
  if File.exist?(outfile)
    File.unlink(outfile) 
  end
  raise
end

#hit?(url) ⇒ Boolean 



39
40
41
42
 
# File 'lib/rhcf/utils/download_cache.rb', line 39

def hit?(url)
  outfile = filename_for(url)
  self.class.hit_fname?(outfile, @ttl) # here goes the cache
end