Contents

require "fileutils"

module Chuckle
  class Cache
    attr_accessor :hits, :misses

    def initialize(client)
      @client = client

      self.hits = self.misses = 0
    end

    def get(request)
      if !exists?(request) || expired?(request)
        self.misses += 1
        return
      end
      self.hits += 1
      Response.new(request)
    end

    def set(request, curl)
      # mkdirs
      FileUtils.mkdir_p([File.dirname(request.headers_path), File.dirname(request.body_path)])

      # now mv
      FileUtils.mv(curl.headers_path, request.headers_path)
      FileUtils.mv(curl.body_path, request.body_path)

      Response.new(request)
    end

    def clear(request)
      Util.rm_if_necessary(request.headers_path)
      Util.rm_if_necessary(request.body_path)
    end

    def exists?(request)
      File.exists?(request.body_path)
    end

    def expired?(request)
      return false if @client.expires_in == :never
      return false if !exists?(request)
      if File.stat(request.body_path).mtime + @client.expires_in < Time.now
        clear(request)
        true
      end
    end

    def body_path(request)
      uri = request.uri

      # calculate body_path
      s = @client.cache_dir
      s = "#{s}/#{pathify(uri.host)}"
      s = "#{s}/#{pathify(uri.path)}"
      if uri.query
        q = "?#{uri.query}"
        s = "#{s}#{pathify(q)}"
      end
      if body = request.body
        s = "#{s},#{pathify(body)}"
      end

      # shorten long paths to md5 checksum
      if s.length > 250
        dir, base = File.dirname(s), File.basename(s)
        s = "#{dir}/#{Util.md5(base)}"
      end

      s
    end

    protected

    # turn s into a string that can be a path
    def pathify(s)
      s = s.gsub(/^\//, "")
      s = s.gsub("..", ",")
      s = s.gsub(/[?\/&]/, ",")
      s = s.gsub(/[^A-Za-z0-9_.,=%-]/) do |i|
        hex = i.unpack("H2").first
        "%#{hex}"
      end
      s = s.downcase
      s = "_root_" if s.empty?
      s
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
chuckle-1.0.3	lib/chuckle/cache.rb
chuckle-1.0.2	lib/chuckle/cache.rb
chuckle-1.0.1	lib/chuckle/cache.rb