rack-mogilefs 0.1.0

data/README.md

@@ -0,0 +1,63 @@
+# Rack middleware/endpoint for serving MogileFS files
+
+## Getting Started:
+
+First install the gem:
+
+    gem install rack-mogilefs
+
+There are a variety of ways to use it:
+
+### Rack middleware
+    # (config.ru)
+    use Rack::MogileFS, :path => %r{^/system/*}
+
+### Rails 3:
+
+    # (config/routes.rb)
+    match "/system/*" => Rack::MogileFS::Endpoint.new
+
+### Rails 2:
+
+    # (app/metal/mogilefs.rb)
+    class MogileFSMetal
+      def self.call(env)
+        if env["PATH_INFO"] =~ %r{^/system/*}
+          Rack::MogileFS::Endpoint.new.call(env)
+        else
+          [ 404, { "Content-Type" => "text/html" }, ["Not Found"] ]
+        end
+      end
+    end
+
+## File Lookup
+
+`Rack::MogileFS` uses the request path (`PATH_INFO`) for the MogileFS key. The
+content type of the file is detected by the extension, using the `mime-types`
+gem.
+
+## Configuration
+
+By default will look for a yaml config in `config/mogilefs.yml` that looks
+like this:
+
+    development:
+      connection:
+        domain: "development.myapp.com"
+        hosts:
+          - 127.0.0.1:7001
+          - 127.0.0.1:7001
+      class: "file"
+    staging:
+      ...
+
+
+and initialize a mogilefs client like this:
+
+    config = YAML.load_file( Rails.root.join("config/mogilefs.yml") )[Rails.env]
+    MogileFS::MogileFS.new( config["connection"].symbolize_keys )
+
+You can override this by passing in a MogileFS client of your own:
+
+    Rack::MogileFS::Endpoint.new :client => MyMogileFSClient.new
+    

data/Rakefile

@@ -0,0 +1,12 @@
+require 'rake/clean'
+require 'rake/testtask'
+
+task :default => [:test]
+
+task :test do
+  Rake::TestTask.new do |t|
+    t.libs << "test"
+    t.pattern = 'test/*_test.rb'
+    t.verbose = true
+  end
+end

data/lib/rack/mogilefs/endpoint.rb

@@ -0,0 +1,53 @@
+module Rack
+  class MogileFS
+    class Endpoint
+
+      def initialize(options={})
+        @options = {
+          :client => nil,
+          :default_content_type => "image/png"
+        }.merge(options)
+
+        @options[:client] ||= default_client
+      end
+
+      def call(env)
+        data = client.get_file_data(env['PATH_INFO'])
+        [ 200, { "Content-Type" => content_type(env['PATH_INFO']) }, [ data ] ]
+      rescue ::MogileFS::Backend::UnknownKeyError
+        [ 404, { "Content-Type" => "text/html" }, ["Not Found"] ]
+      rescue ::MogileFS::UnreachableBackendError => e
+        [ 503, { "Content-Type" => "text/html" }, [e.message] ]
+      rescue ::MogileFS::Error => e
+        [ 500, { "Content-Type" => "text/html" }, [e.message] ]
+      end
+
+      protected
+
+      def content_type(path_info)
+        ext = path_info.split(".").last
+        MIME::Types.type_for(ext).first.content_type
+      rescue
+        @options[:default_content_type]
+      end
+
+      def client
+        @options[:client]
+      end
+
+      def default_client
+        ::MogileFS::MogileFS.new(config)
+      end
+
+      def config
+        if defined?(Rails)
+          yml = YAML.load_file( Rails.root.join("config/mogilefs.yml") )
+          yml[Rails.env]["connection"].symbolize_keys
+        else
+          { :domain => "default", :hosts => ["127.0.0.1:7001"] }
+        end
+      end
+
+    end
+  end
+end

data/lib/rack/mogilefs.rb

@@ -0,0 +1,20 @@
+require 'mogilefs'
+require 'mime/types'
+
+module Rack
+  class MogileFS
+    autoload :Endpoint, "rack/mogilefs/endpoint"
+
+    def initialize(app, options={})
+      @app, @options = app, options
+    end
+
+    def call(env)
+      if env['PATH_INFO'] =~ @options[:path]
+        Endpoint.new(@options).call(env)
+      else
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/rack-mogilefs.rb

@@ -0,0 +1 @@
+require 'rack/mogilefs'

data/test/mogilefs_test.rb

@@ -0,0 +1,42 @@
+require 'test_helper'
+
+class TestMiddleWare < Test::Unit::TestCase
+  class MockMogileFsClient
+    def get_file_data(key)
+      "Mock Data"
+    end
+  end
+
+  def test_middleware_with_path
+    app = Rack::Builder.new do
+      use Rack::Lint
+      use Rack::MogileFS, :path => %r{^/system/*}
+      use Rack::Lint
+      run lambda { |env| [200, {'Content-Type' => 'text/plain'}, "Hello, World!"] }
+    end
+
+    response = Rack::MockRequest.new(app).get('/')
+    assert_equal "Hello, World!", response.body
+
+    response = Rack::MockRequest.new(app).get('/system/unreachable.txt')
+    assert_equal "couldn't connect to mogilefsd backend", response.body
+    assert_equal 503, response.status
+  end
+
+  def test_middleware_with_mock_mogile_client
+    app = Rack::Builder.new do
+      use Rack::Lint
+      use Rack::MogileFS, :path => %r{^/system/*}, :client => MockMogileFsClient.new
+      use Rack::Lint
+      run lambda { |env| [200, {'Content-Type' => 'text/plain'}, "Hello, World!"] }
+    end
+
+    response = Rack::MockRequest.new(app).get('/')
+    assert_equal "Hello, World!", response.body
+
+    response = Rack::MockRequest.new(app).get('/system/mocked.txt')
+    assert_equal "Mock Data", response.body
+    assert_equal 200, response.status
+    assert_equal "text/plain", response.headers['Content-Type']
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,6 @@
+require "rubygems"
+require "test/unit"
+require "rack"
+require "rack/mock"
+require "mocha"
+require "rack/mogilefs"

metadata

@@ -0,0 +1,116 @@
+--- !ruby/object:Gem::Specification 
+name: rack-mogilefs
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Ben Marini
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-09-08 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mime-types
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 47
+        segments: 
+        - 1
+        - 16
+        version: "1.16"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rack
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: mocha
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+description: A rack middleware and/or endpoint to serve up files from MogileFS
+email: bmarini@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/rack/mogilefs/endpoint.rb
+- lib/rack/mogilefs.rb
+- lib/rack-mogilefs.rb
+- README.md
+- Rakefile
+- test/mogilefs_test.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: http://github.com/bmarini/rack-mogilefs
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: A rack middleware and/or endpoint to serve up files from MogileFS
+test_files: 
+- test/mogilefs_test.rb
+- test/test_helper.rb