rtomayko-rack-cache 0.2.0

data/lib/rack/cache/request.rb

@@ -0,0 +1,37 @@
+require 'rack/request'
+require 'rack/cache/headers'
+require 'rack/utils/environment_headers'
+
+module Rack::Cache
+  # Provides access to the HTTP request. The +request+ and +original_request+
+  # objects exposed by the Core caching engine are instances of this class.
+  #
+  # Request objects respond to a variety of convenience methods, including
+  # everything defined by Rack::Request as well as the Headers and
+  # RequestHeaders modules.
+
+  class Request < Rack::Request
+    include Rack::Cache::Headers
+    include Rack::Cache::RequestHeaders
+
+    # The HTTP request method. This is the standard implementation of this
+    # method but is respecified here due to libraries that attempt to modify
+    # the behavior to respect POST tunnel method specifiers. We always want
+    # the real request method.
+    def request_method
+      @env['REQUEST_METHOD']
+    end
+
+    # Determine if the request's method matches any of the values
+    # provided:
+    #   if request.request_method?('GET', 'POST')
+    #     ...
+    #   end
+    def request_method?(*methods)
+      method = request_method
+      methods.any? { |test| test.to_s.upcase == method }
+    end
+
+    alias_method :method?, :request_method?
+  end
+end

data/lib/rack/cache/response.rb

@@ -0,0 +1,76 @@
+require 'set'
+require 'rack/cache/headers'
+
+module Rack::Cache
+  # Provides access to the response generated by the downstream application. The
+  # +response+, +original_response+, and +entry+ objects exposed by the Core
+  # caching engine are instances of this class.
+  #
+  # Response objects respond to a variety of convenience methods, including
+  # those defined in Rack::Response::Helpers, Rack::Cache::Headers,
+  # and Rack::Cache::ResponseHeaders.
+  #
+  # Note that Rack::Cache::Response is not a subclass of Rack::Response and does
+  # not perform many of the same initialization and finalization tasks. For
+  # example, the body is not slurped during initialization and there are no
+  # facilities for generating response output.
+
+  class Response
+    include Rack::Response::Helpers
+    include Rack::Cache::Headers
+    include Rack::Cache::ResponseHeaders
+
+    # The response's status code (integer).
+    attr_accessor :status
+
+    # The response body. See the Rack spec for information on the behavior
+    # required by this object.
+    attr_accessor :body
+
+    # The response headers.
+    attr_reader :headers
+
+    # Create a Response instance given the response status code, header hash,
+    # and body.
+    def initialize(status, headers, body)
+      @status = status
+      @headers = headers
+      @body = body
+      @now = Time.now
+      @headers['Date'] ||= now.httpdate
+    end
+
+    def initialize_copy(other)
+      super
+      @headers = other.headers.dup
+    end
+
+    # Return the value of the named response header.
+    def [](header_name)
+      headers[header_name]
+    end
+
+    # Set a response header value.
+    def []=(header_name, header_value)
+      headers[header_name] = header_value
+    end
+
+    # Called immediately after an object is loaded from the cache.
+    def activate!
+      headers['Age'] = age.to_i.to_s
+    end
+
+    # Return the status, headers, and body in a three-tuple.
+    def to_a
+      [status, headers, body]
+    end
+
+    # Freezes
+    def freeze
+      @headers.freeze
+      super
+    end
+
+  end
+
+end

data/lib/rack/cache/storage.rb

@@ -0,0 +1,50 @@
+require 'uri'
+require 'rack/cache/metastore'
+require 'rack/cache/entitystore'
+
+module Rack::Cache
+  # Maintains a collection of MetaStore and EntityStore instances keyed by
+  # URI. A single instance of this class can be used across a single process
+  # to ensure that only a single instance of a backing store is created per
+  # unique storage URI.
+
+  class Storage
+    def initialize
+      @metastores = {}
+      @entitystores = {}
+    end
+
+    def resolve_metastore_uri(uri)
+      @metastores[uri.to_s] ||= create_store(MetaStore, uri)
+    end
+
+    def resolve_entitystore_uri(uri)
+      @entitystores[uri.to_s] ||= create_store(EntityStore, uri)
+    end
+
+    # Clear store instances.
+    def clear
+      @metastores.clear
+      @entitystores.clear
+      nil
+    end
+
+  private
+    def create_store(type, uri)
+      uri = URI.parse(uri) unless uri.respond_to?(:scheme)
+      if type.const_defined?(uri.scheme.upcase)
+        klass = type.const_get(uri.scheme.upcase)
+        klass.resolve(uri)
+      else
+        fail "Unknown storage provider: #{uri.to_s}"
+      end
+    end
+
+  public
+    @@singleton_instance = new
+    def self.instance
+      @@singleton_instance
+    end
+  end
+
+end

data/lib/rack/cache.rb

@@ -0,0 +1,51 @@
+require 'fileutils'
+require 'time'
+require 'rack'
+
+module Rack #:nodoc:
+end
+
+# = HTTP Caching For Rack
+#
+# Rack::Cache is suitable as a quick, drop-in component to enable HTTP caching
+# for Rack-enabled applications that produce freshness (+Expires+, +Cache-Control+)
+# and/or validation (+Last-Modified+, +ETag+) information.
+#
+# * Standards-based (RFC 2616 compliance)
+# * Freshness/expiration based caching and validation
+# * Supports HTTP Vary
+# * Portable: 100% Ruby / works with any Rack-enabled framework
+# * VCL-like configuration language for advanced caching policies
+# * Disk, memcached, and heap memory storage backends
+#
+# === Usage
+#
+# Create with default options:
+#   require 'rack/cache'
+#   Rack::Cache.new(app, :verbose => true, :entitystore => 'file:cache')
+#
+# Within a rackup file (or with Rack::Builder):
+#   require 'rack/cache'
+#   use Rack::Cache do
+#     set :verbose, true
+#     set :metastore, 'memcached://localhost:11211/meta'
+#     set :entitystore, 'file:/var/cache/rack'
+#   end
+#   run app
+#
+module Rack::Cache
+  require 'rack/cache/request'
+  require 'rack/cache/response'
+  require 'rack/cache/context'
+  require 'rack/cache/storage'
+
+  # Create a new Rack::Cache middleware component that fetches resources from
+  # the specified backend application. The +options+ Hash can be used to
+  # specify default configuration values (see attributes defined in
+  # Rack::Cache::Options for possible key/values). When a block is given, it
+  # is executed within the context of the newly create Rack::Cache::Context
+  # object.
+  def self.new(backend, options={}, &b)
+    Context.new(backend, options, &b)
+  end
+end

data/lib/rack/utils/environment_headers.rb

@@ -0,0 +1,78 @@
+require 'rack/utils'
+
+module Rack::Utils #:nodoc:
+  # A facade over a Rack Environment Hash that gives access to headers
+  # using their normal RFC 2616 names.
+
+  class EnvironmentHeaders
+    include Enumerable
+
+    # Create the facade over the given Rack Environment Hash.
+    def initialize(env)
+      @env = env
+    end
+
+    # Return the value of the specified header. The +header_name+ should
+    # be as specified by RFC 2616 (e.g., "Content-Type", "Accept", etc.)
+    def [](header_name)
+      @env[env_name(header_name)]
+    end
+
+    # Set the value of the specified header. The +header_name+ should
+    # be as specified by RFC 2616 (e.g., "Content-Type", "Accept", etc.)
+    def []=(header_name, value)
+      @env[env_name(header_name)] = value
+    end
+
+    # Determine if the underlying Rack Environment includes a header
+    # of the given name.
+    def include?(header_name)
+      @env.include?(env_name(header_name))
+    end
+
+    # Iterate over all headers yielding a (name, value) tuple to the
+    # block. Rack Environment keys that do not map to an header are not
+    # included.
+    def each
+      @env.each do |key,value|
+        next unless key =~ /^(HTTP_|CONTENT_)/
+        yield header_name(key), value
+      end
+    end
+
+    # Delete the entry in the underlying Rack Environment that corresponds
+    # to the given RFC 2616 header name.
+    def delete(header_name)
+      @env.delete(env_name(header_name))
+    end
+
+    # Return the underlying Rack Environment Hash.
+    def to_env
+      @env
+    end
+
+    alias_method :to_hash, :to_env
+
+  private
+
+    # Return the Rack Environment key for the given RFC 2616 header name.
+    def env_name(header_name)
+      case header_name = header_name.upcase
+      when 'CONTENT-TYPE'   then 'CONTENT_TYPE'
+      when 'CONTENT-LENGTH' then 'CONTENT_LENGTH'
+      else "HTTP_#{header_name.tr('-', '_')}"
+      end
+    end
+
+    # Return the RFC 2616 header name for the given Rack Environment key.
+    def header_name(env_name)
+      env_name.
+        sub(/^HTTP_/, '').
+        downcase.
+        capitalize.
+        gsub(/_(.)/) { '-' + $1.upcase }
+    end
+
+  end
+
+end

data/rack-cache.gemspec

@@ -0,0 +1,74 @@
+Gem::Specification.new do |s|
+  s.specification_version = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+
+  s.name = 'rack-cache'
+  s.version = '0.2.0'
+  s.date = '2008-10-24'
+
+  s.description = "HTTP Caching for Rack"
+  s.summary     = "HTTP Caching for Rack"
+
+  s.authors = ["Ryan Tomayko"]
+  s.email = "r@tomayko.com"
+
+  # = MANIFEST =
+  s.files = %w[
+    CHANGES
+    COPYING
+    README
+    Rakefile
+    TODO
+    doc/configuration.markdown
+    doc/events.dot
+    doc/faq.markdown
+    doc/index.markdown
+    doc/layout.html.erb
+    doc/license.markdown
+    doc/rack-cache.css
+    doc/storage.markdown
+    lib/rack/cache.rb
+    lib/rack/cache/config.rb
+    lib/rack/cache/config/busters.rb
+    lib/rack/cache/config/default.rb
+    lib/rack/cache/config/no-cache.rb
+    lib/rack/cache/context.rb
+    lib/rack/cache/core.rb
+    lib/rack/cache/entitystore.rb
+    lib/rack/cache/headers.rb
+    lib/rack/cache/metastore.rb
+    lib/rack/cache/options.rb
+    lib/rack/cache/request.rb
+    lib/rack/cache/response.rb
+    lib/rack/cache/storage.rb
+    lib/rack/utils/environment_headers.rb
+    rack-cache.gemspec
+    test/cache_test.rb
+    test/config_test.rb
+    test/context_test.rb
+    test/core_test.rb
+    test/entitystore_test.rb
+    test/environment_headers_test.rb
+    test/headers_test.rb
+    test/logging_test.rb
+    test/metastore_test.rb
+    test/options_test.rb
+    test/pony.jpg
+    test/response_test.rb
+    test/spec_setup.rb
+    test/storage_test.rb
+  ]
+  # = MANIFEST =
+
+  s.test_files = s.files.select {|path| path =~ /^test\/.*_test.rb/}
+
+  s.extra_rdoc_files = %w[README COPYING TODO CHANGES]
+  s.add_dependency 'rack', '~> 0.4'
+
+  s.has_rdoc = true
+  s.homepage = "http://tomayko.com/src/rack-cache/"
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Rack::Cache", "--main", "Rack::Cache"]
+  s.require_paths = %w[lib]
+  s.rubyforge_project = 'wink'
+  s.rubygems_version = '1.1.1'
+end

data/test/cache_test.rb

@@ -0,0 +1,35 @@
+require "#{File.dirname(__FILE__)}/spec_setup"
+
+def dumb_app(env)
+  body = block_given? ? [yield] : ['Hi']
+  [ 200, {'Content-Type' => 'text/plain'}, body ]
+end
+
+describe 'Rack::Cache::new' do
+  before { @app = method(:dumb_app) }
+
+  it 'takes a backend and returns a middleware component' do
+    Rack::Cache.new(@app).
+      should.respond_to :call
+  end
+  it 'takes an options Hash' do
+    lambda { Rack::Cache.new(@app, {}) }.
+      should.not.raise(ArgumentError)
+  end
+  it 'sets options provided in the options Hash' do
+    object = Rack::Cache.new(@app, :foo => 'bar', 'foo.bar' => 'bling')
+    object.options['foo.bar'].should.be == 'bling'
+    object.options['rack-cache.foo'].should.be == 'bar'
+  end
+  it 'takes a block; executes it during initialization' do
+    state, block_scope = 'not invoked', nil
+    object =
+      Rack::Cache.new @app do
+        block_scope = self
+        state = 'invoked'
+        should.respond_to :on
+      end
+    state.should.be == 'invoked'
+    object.should.be block_scope
+  end
+end

data/test/config_test.rb

@@ -0,0 +1,66 @@
+require "#{File.dirname(__FILE__)}/spec_setup"
+require 'rack/cache/config'
+
+class MockConfig
+  include Rack::Cache::Config
+  def configured!
+    @configured = true
+  end
+  def configured?
+    @configured
+  end
+end
+
+describe 'Rack::Cache::Config' do
+  before :each do
+    @config = MockConfig.new
+    @tempdir = create_temp_directory
+    $:.unshift @tempdir
+  end
+  after :each do
+    @config = nil
+    $:.shift if $:.first == @tempdir
+    remove_entry_secure @tempdir
+  end
+
+  def make_temp_file(filename, data='configured!')
+    create_temp_file @tempdir, filename, data
+  end
+
+  it 'loads config files from the load path when file is relative' do
+    make_temp_file 'foo/bar.rb'
+    @config.import 'foo/bar.rb'
+    @config.should.be.configured
+  end
+  it 'assumes a .rb file extension when no file extension exists' do
+    make_temp_file 'foo/bar.rb'
+    @config.import 'foo/bar'
+    @config.should.be.configured
+  end
+  it 'does not assume a .rb file extension when other file extension exists' do
+    make_temp_file 'foo/bar.conf'
+    @config.import 'foo/bar.conf'
+    @config.should.be.configured
+  end
+  it 'should locate files with absolute path names' do
+    make_temp_file 'foo/bar.rb'
+    @config.import File.join(@tempdir, 'foo/bar.rb')
+    @config.should.be.configured
+  end
+  it 'raises a LoadError when the file cannot be found' do
+    assert_raises(LoadError) {
+      @config.import('this/file/is/very-likely/not/to/exist.rb')
+    }
+  end
+  it 'executes within the context of the object instance' do
+    make_temp_file 'foo/bar.rb',
+      'self.should.be.kind_of Rack::Cache::Config ; configured!'
+    @config.import 'foo/bar'
+    @config.should.be.configured
+  end
+  it 'does not import files more than once' do
+    make_temp_file 'foo/bar.rb', "import 'foo/bar'"
+    @config.import('foo/bar').should.be true
+    @config.import('foo/bar').should.be false
+  end
+end