Contents

module Puppet::Module::Tool

  # = Cache
  #
  # Provides methods for reading files from local cache, filesystem or network.
  class Cache
    include Puppet::Module::Tool::Utils::URI

    # Instantiate new cahe for the +repositry+ instance.
    def initialize(repository)
      @repository = repository
    end

    # Return filename retrieved from +uri+ instance. Will download this file and
    # cache it if needed.
    #
    # TODO: Add checksum support.
    # TODO: Add error checking.
    def retrieve(url)
      returning(path + File.basename(url.to_s)) do |cached_file|
        uri = normalize(url)
        unless cached_file.file?
          if uri.scheme == 'file'
            FileUtils.cp(uri.path, cached_file)
          else
            # TODO: Handle HTTPS; probably should use repository.contact
            data = read_retrieve(uri)
            cached_file.open('wb') { |f| f.write data }
          end
        end
      end
    end

    # Return contents of file at the given URI's +uri+.
    def read_retrieve(uri)
      return uri.read
    end

    # Return Pathname for repository's cache directory, create it if needed.
    def path
      return @path ||= (self.class.base_path + @repository.cache_key).tap{ |o| o.mkpath }
    end

    # Return the base Pathname for all the caches.
    def self.base_path
      return(Puppet::Module::Tool.working_dir + 'cache')
    end

    # Clean out all the caches.
    def self.clean
      base_path.rmtree if base_path.exist?
    end

  end

end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
puppet-module-0.3.4	lib/puppet/module/tool/cache.rb
puppet-module-0.3.3	lib/puppet/module/tool/cache.rb
puppet-module-0.3.2	lib/puppet/module/tool/cache.rb
puppet-module-0.3.0	lib/puppet/module/tool/cache.rb