rack-rekon 0.1.0

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/.rvmrc

@@ -0,0 +1 @@
+rvm use 1.9.2@rack-rekon --create

data/Gemfile

@@ -0,0 +1,7 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in rack-rekon.gemspec
+gemspec
+
+gem 'ruby-debug19', :platform => :mri_19
+gem 'ruby-debug', :platform => :mri_18

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2011 SEOmoz
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,25 @@
+# Rack::Rekon
+
+[Riak](http://wiki.basho.com/) is awesome.
+[Rekon](https://github.com/adamhunter/rekon/) is an awesome to browse
+your riak data.
+
+Rack::Rekon provides a simple way to serve rekon from your
+rack/sinatra/rails app.
+
+## Usage
+
+Example `config.ru`:
+
+    require 'rack-rekon'
+
+    require 'rack/rekon'
+    use Rack::Lint
+    map '/rekon' do
+      run Rack::Rekon::App.new
+    end
+
+## Copyright
+
+Copyright 2011 SEOmoz. See LICENSE for details.
+

data/Rakefile

@@ -0,0 +1,7 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+desc "Start rack-rekon on port 3200"
+task :start do
+  sh "bundle exec shotgun config.ru -p 3200"
+end

data/config.ru

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+require 'bundler'
+Bundler.setup
+
+require 'rack/rekon'
+use Rack::Lint
+map '/rekon' do
+  run Rack::Rekon::App.new
+end

data/lib/rack-rekon.rb

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

data/lib/rack/rekon.rb

@@ -0,0 +1 @@
+require 'rack/rekon/app'

data/lib/rack/rekon/app.rb

@@ -0,0 +1,82 @@
+require 'rack/rekon/proxy'
+require 'rack/rekon/response_rewriters/html'
+require 'rack/rekon/response_rewriters/rekon_js'
+require 'rack/rekon/response_rewriters/sammy_template'
+require 'zlib'
+
+module Rack
+  module Rekon
+    class App < Proxy
+      attr_accessor :riak_host
+      attr_accessor :riak_port
+      attr_accessor :riak_protocol
+
+      RIAK_BUCKET = 'riak'
+
+      def initialize
+        yield self if block_given?
+
+        # set defaults
+        self.riak_host ||= 'localhost'
+        self.riak_port ||= 8098
+        self.riak_protocol ||= 'http'
+      end
+
+      def call(env)
+        request = Rack::Request.new(env)
+
+        # redirect to the rekon app root
+        if path_and_query_for(request) == ''
+          uri = URI(request.url)
+          uri.path = request.script_name + '/riak/rekon/go'
+          [301, { 'Location' => uri.to_s, 'Content-Type' => 'text/html' }, []]
+        else
+          super
+        end
+      end
+
+      def riak_root
+        @riak_root ||= "#{riak_protocol}://#{riak_host}:#{riak_port}"
+      end
+
+      def rekon_root
+        @rekon_root ||= riak_root + "/#{RIAK_BUCKET}/rekon"
+      end
+
+      def path_and_query_for(request)
+        start_index = request.url.index(request.script_name) + request.script_name.size
+        request.url[start_index..-1]
+      end
+
+      def uri_for(request)
+        riak_root + path_and_query_for(request)
+      end
+
+      def rewrite_response(request, status, headers, body)
+        rewriter = case headers['content-type']
+          when 'text/html'; ResponseRewriters::HTML
+          when 'application/x-sammy-template'; ResponseRewriters::SammyTemplate
+          when 'application/javascript'
+            case request.path.split('/').last
+              when 'rekon.js'; ResponseRewriters::RekonJS
+            end
+        end
+
+        if rewriter
+          rewriter = rewriter.new(request.env['SCRIPT_NAME'], plain_text_body(headers, body))
+          body = [rewriter.rewritten_body]
+        end
+
+        [status, headers, body]
+      end
+
+      def plain_text_body(headers, body)
+        body = body.first if body.is_a?(Array) && body.size == 1
+        return body unless headers['content-encoding'] == 'gzip'
+        headers.delete('content-encoding')
+        io = StringIO.new(body)
+        Zlib::GzipReader.new(io).read
+      end
+    end
+  end
+end

data/lib/rack/rekon/proxy.rb

@@ -0,0 +1,48 @@
+require 'net/http'
+require 'uri'
+
+module Rack
+  module Rekon
+    class Proxy
+      def call(env)
+        req = Rack::Request.new(env)
+        method = req.request_method.downcase
+        method[0..0] = method[0..0].upcase
+
+        uri = URI(uri_for(req))
+
+        sub_request = Net::HTTP.const_get(method).new("#{uri.path}#{"?" if uri.query}#{uri.query}")
+
+        if sub_request.request_body_permitted? and req.body
+          sub_request.body_stream = req.body
+          sub_request.content_length = req.content_length
+          sub_request.content_type = req.content_type
+        end
+
+        sub_request["X-Forwarded-For"] = (req.env["X-Forwarded-For"].to_s.split(/, +/) + [req.env['REMOTE_ADDR']]).join(", ")
+        sub_request["X-Requested-With"] = req.env['HTTP_X_REQUESTED_WITH'] if req.env['HTTP_X_REQUESTED_WITH']
+        sub_request["Accept-Encoding"] = req.accept_encoding
+        sub_request["Referer"] = req.referer
+        sub_request.basic_auth *uri.userinfo.split(':') if (uri.userinfo && uri.userinfo.index(':'))
+
+        sub_response = Net::HTTP.start(uri.host, uri.port) do |http|
+          http.request(sub_request)
+        end
+
+        headers = {}
+        sub_response.each_header do |k,v|
+          headers[k] = v unless k.to_s =~ /cookie|content-length|transfer-encoding/i
+        end
+
+        response_triplet = [sub_response.code.to_i, headers, [sub_response.read_body]]
+
+        if respond_to?(:rewrite_response)
+          rewrite_response(req, *response_triplet)
+        else
+          response_triplet
+        end
+      end
+    end
+  end
+end
+

data/lib/rack/rekon/response_rewriters/html.rb

@@ -0,0 +1,42 @@
+require 'nokogiri'
+
+module Rack
+  module Rekon
+    module ResponseRewriters
+      class HTML
+        attr_reader :doc, :app_root
+        def initialize(app_root, body)
+          @app_root, @doc = app_root, Nokogiri::HTML(body)
+        end
+
+        def rewritten_body
+          ensure_rewritten!
+          doc.to_html
+        end
+
+        private
+
+          def ensure_rewritten!
+            return if @rewritten
+            rewrite_javascripts!
+            rewrite_link_tags!
+            @rewritten = true
+          end
+
+          def rewrite_javascripts!
+            doc.css('script').each do |script|
+              src = script['src']
+              script['src'] = [app_root, 'riak', 'rekon', src].join('/')
+            end
+          end
+
+          def rewrite_link_tags!
+            doc.css('link').each do |link_tag|
+              href = link_tag['href']
+              link_tag['href'] = [app_root, 'riak', 'rekon', href].join('/')
+            end
+          end
+      end
+    end
+  end
+end

data/lib/rack/rekon/response_rewriters/rekon_js.rb

@@ -0,0 +1,41 @@
+module Rack
+  module Rekon
+    module ResponseRewriters
+      class RekonJS
+        attr_reader :js, :app_root
+        def initialize(app_root, js)
+          @app_root, @js = app_root, js
+        end
+
+        def rewritten_body
+          ensure_rewritten!
+          js
+        end
+
+        private
+
+          def ensure_rewritten!
+            return if @rewritten
+            rewrite_templates_paths!
+            rewrite_base_urls!
+            @rewritten = true
+          end
+
+          def rewrite_templates_paths!
+            js.gsub!(/([^\.\(\)\']+)\.html\.template/, [app_root, 'riak', 'rekon'].join("/") + '/\1.html.template')
+          end
+
+          def rewrite_base_urls!
+            js << <<-EOJS
+              Rekon.client = new RiakClient(
+                '#{app_root}/#{Rack::Rekon::App::RIAK_BUCKET}',
+                '#{app_root}/mapred'
+              );
+            EOJS
+
+            js.gsub!("'/stats'", "'#{app_root}/stats'")
+          end
+      end
+    end
+  end
+end

data/lib/rack/rekon/response_rewriters/sammy_template.rb

@@ -0,0 +1,44 @@
+require 'nokogiri'
+
+module Rack
+  module Rekon
+    module ResponseRewriters
+      class SammyTemplate
+        attr_reader :body, :app_root
+        def initialize(app_root, body)
+          @app_root, @body = app_root, body
+        end
+
+        def doc
+          @doc ||= Nokogiri::HTML::DocumentFragment.parse(body)
+        end
+
+        def rewritten_body
+          return body if ejs?
+
+          ensure_rewritten!
+          doc.to_html
+        end
+
+        private
+
+          def ejs?
+            body.include?('<%=')
+          end
+
+          def ensure_rewritten!
+            return if @rewritten
+            rewrite_images!
+            @rewritten = true
+          end
+
+          def rewrite_images!
+            doc.css('img').each do |img|
+              src = img['src']
+              img['src'] = [app_root, 'riak', 'recon', src].join('/')
+            end
+          end
+      end
+    end
+  end
+end

data/lib/rack/rekon/version.rb

@@ -0,0 +1,5 @@
+module Rack
+  module Rekon
+    VERSION = "0.1.0"
+  end
+end

data/rack-rekon.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "rack/rekon/version"
+
+Gem::Specification.new do |s|
+  s.name        = "rack-rekon"
+  s.version     = Rack::Rekon::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Myron Marston"]
+  s.email       = ["myron@seomoz.org"]
+  s.homepage    = ""
+  s.summary     = %q{Mount rekon as a rack app.}
+  s.description = %q{Proxies rekon as a rack app so you can include it in your rack-based riak application.}
+
+  s.rubyforge_project = "rack-rekon"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_runtime_dependency 'nokogiri', '~> 1.4'
+  s.add_development_dependency 'shotgun', '0.9'
+end

metadata

@@ -0,0 +1,112 @@
+--- !ruby/object:Gem::Specification 
+name: rack-rekon
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Myron Marston
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-04-19 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: nokogiri
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 1
+        - 4
+        version: "1.4"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: shotgun
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 25
+        segments: 
+        - 0
+        - 9
+        version: "0.9"
+  type: :development
+  version_requirements: *id002
+description: Proxies rekon as a rack app so you can include it in your rack-based riak application.
+email: 
+- myron@seomoz.org
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- .rvmrc
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- config.ru
+- lib/rack-rekon.rb
+- lib/rack/rekon.rb
+- lib/rack/rekon/app.rb
+- lib/rack/rekon/proxy.rb
+- lib/rack/rekon/response_rewriters/html.rb
+- lib/rack/rekon/response_rewriters/rekon_js.rb
+- lib/rack/rekon/response_rewriters/sammy_template.rb
+- lib/rack/rekon/version.rb
+- rack-rekon.gemspec
+has_rdoc: true
+homepage: ""
+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: rack-rekon
+rubygems_version: 1.5.2
+signing_key: 
+specification_version: 3
+summary: Mount rekon as a rack app.
+test_files: []
+