heroku-nav 0.1.0

data/.gitignore

@@ -0,0 +1 @@
+pkg/

data/Rakefile

@@ -0,0 +1,28 @@
+task :default => :spec
+
+desc 'Run specs (with story style output)'
+task 'spec' do
+  sh 'bacon -s spec/*_spec.rb'
+end
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gemspec|
+    gemspec.name = "heroku-nav"
+    gemspec.summary = ""
+    gemspec.description = ""
+    gemspec.email = "pedro@heroku.com"
+    gemspec.homepage = "http://heroku.com"
+    gemspec.authors = ["Todd Matthews", "Pedro Belo"]
+
+    gemspec.add_development_dependency(%q<baconmocha>, [">= 0"])
+    gemspec.add_development_dependency(%q<sinatra>, [">= 0"])
+    gemspec.add_development_dependency(%q<rack-test>, [">= 0"])
+    gemspec.add_dependency(%q<rest-client>, ["~> 1.2.0"])
+    gemspec.add_dependency(%q<json>, [">= 0"])
+
+    gemspec.version = '0.1.0'
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: gem install jeweler"
+end

data/heroku-nav.gemspec

@@ -0,0 +1,60 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{heroku-nav}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Todd Matthews", "Pedro Belo"]
+  s.date = %q{2010-02-24}
+  s.description = %q{}
+  s.email = %q{pedro@heroku.com}
+  s.files = [
+    ".gitignore",
+     "Rakefile",
+     "heroku-nav.gemspec",
+     "lib/heroku/nav.rb",
+     "spec/api_spec.rb",
+     "spec/base.rb",
+     "spec/nav_spec.rb"
+  ]
+  s.homepage = %q{http://heroku.com}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{}
+  s.test_files = [
+    "spec/api_spec.rb",
+     "spec/base.rb",
+     "spec/nav_spec.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<baconmocha>, [">= 0"])
+      s.add_development_dependency(%q<sinatra>, [">= 0"])
+      s.add_development_dependency(%q<rack-test>, [">= 0"])
+      s.add_runtime_dependency(%q<rest-client>, ["~> 1.2.0"])
+      s.add_runtime_dependency(%q<json>, [">= 0"])
+    else
+      s.add_dependency(%q<baconmocha>, [">= 0"])
+      s.add_dependency(%q<sinatra>, [">= 0"])
+      s.add_dependency(%q<rack-test>, [">= 0"])
+      s.add_dependency(%q<rest-client>, ["~> 1.2.0"])
+      s.add_dependency(%q<json>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<baconmocha>, [">= 0"])
+    s.add_dependency(%q<sinatra>, [">= 0"])
+    s.add_dependency(%q<rack-test>, [">= 0"])
+    s.add_dependency(%q<rest-client>, ["~> 1.2.0"])
+    s.add_dependency(%q<json>, [">= 0"])
+  end
+end
+

data/lib/heroku/nav.rb

@@ -0,0 +1,67 @@
+require 'rest_client'
+require 'json'
+
+module Heroku
+  module Nav
+    class Base
+      def initialize(app)
+        @app = app
+        refresh
+      end
+
+      def call(env)
+        @status, @headers, @body = @app.call(env)
+        insert! if can_insert?
+        [@status, @headers, @body]
+      end
+
+      def can_insert?
+        return unless @headers['Content-Type'] == 'text/html'
+        return unless @body.respond_to?(:[])
+        return unless @body[0] =~ /<body.*?>/i
+        true
+      end
+
+      def refresh
+        @html, @css = fetch
+      end
+
+      def fetch
+        raw   = RestClient.get(resource_url, :accept => :json)
+        attrs = JSON.parse(raw)
+        [attrs['html'], attrs['css']]
+      rescue => e
+        STDERR.puts "Failed to fetch the Heroku #{resource}: #{e.class.name} - #{e.message}"
+        nil
+      end
+
+      def resource
+        self.class.name.split('::').last.downcase
+      end
+
+      def resource_url
+        [api_url, resource].join
+      end
+
+      def api_url
+        ENV['API_URL'] || "http://nav.heroku.com/"
+      end
+    end
+
+    class Header < Base
+      def insert!
+        @body[0].gsub!(/(<head>)/i, "\\1<style type=\"text/css\">#{@css}</style>") if @css
+        @body[0].gsub!(/(<body.*?>\s*(<div .*?class=["'].*?container.*?["'].*?>)?)/i, "\\1#{@html}") if @html
+        @headers['Content-Length'] = @body[0].size.to_s
+      end
+    end
+
+    class Footer < Base
+      def insert!
+        @body[0].gsub!(/(<head>)/i, "\\1<style type=\"text/css\">#{@css}</style>") if @css
+        @body[0].gsub!(/(<\/body>)/i, "#{@html}\\1") if @html
+        @headers['Content-Length'] = @body[0].size.to_s
+      end
+    end
+  end
+end

data/spec/api_spec.rb

@@ -0,0 +1,25 @@
+require File.dirname(__FILE__) + '/base'
+
+describe "Api" do
+  before do
+    RestClient.stubs(:get).returns({ :html => '<!-- header -->', :css => '#header' }.to_json)
+    @header = Heroku::Nav::Header.new(:app)
+  end
+
+  it "has a resource based on the class name" do
+    @header.resource.should == 'header'
+  end
+
+  it "has a resource url based on the api url" do
+    @header.resource_url.should == 'http://nav.heroku.com/header'
+  end
+
+  it "doesn't raise" do
+    RestClient.stubs(:get).raises("error")
+    lambda { @header.fetch }.should.not.raise
+  end
+
+  it "parses the JSON response, returning the html and css" do
+    @header.fetch.should == ['<!-- header -->', '#header']
+  end
+end

data/spec/base.rb

@@ -0,0 +1,45 @@
+$: << File.dirname(__FILE__) + '/../lib'
+
+require 'rubygems'
+require 'heroku/nav'
+require 'sinatra/base'
+require 'baconmocha'
+require 'rack/test'
+
+class TestApp < Sinatra::Base
+  set :environment, :test
+
+  def self.body
+    @@body
+  end
+
+  def self.body=(body)
+    @@body = body
+  end
+
+  get '/' do
+    params[:body]
+  end
+
+  get '/alternate' do
+    params[:body]
+  end
+
+  get '/text' do
+    content_type 'text/plain'
+    params[:body]
+  end
+end
+
+# tiny factory to help making a Sinatra::Base application.
+# whatever is passed in the block will get eval'ed into the class
+def make_app(&blk)
+  handler = Class.new(TestApp)
+  handler.class_eval(&blk)
+  handler
+end
+
+# Make sure Rack::Test methods are available for all specs
+class Bacon::Context
+  include Rack::Test::Methods
+end

data/spec/nav_spec.rb

@@ -0,0 +1,60 @@
+require File.dirname(__FILE__) + '/base'
+
+class Heroku::Nav::Header
+  def fetch
+    ['<!-- header -->', '#header']
+  end
+end
+
+class Heroku::Nav::Footer
+  def fetch
+    ['<!-- footer -->', '#footer']
+  end
+end
+
+describe Heroku::Nav::Header do
+  def app
+    make_app { use Heroku::Nav::Header }
+  end
+
+  it "doesn't apply if content-type is not html" do
+    get '/text', :body => '<html><body>hi'
+    last_response.body.should.equal '<html><body>hi'
+  end
+
+  it "adds the html right after the body" do
+    get '/', :body => '<html><body>hi'
+    last_response.body.should.equal '<html><body><!-- header -->hi'
+  end
+
+  it "adds the html right after the body, even if it has properties" do
+    get '/', :body => '<html><body id="a" class="b">hi'
+    last_response.body.should.equal '<html><body id="a" class="b"><!-- header -->hi'
+  end
+
+  it "adds the html right after the first div if class is container" do
+    get '/', :body => '<html><body><div class="container">hi</div>'
+    last_response.body.should.equal '<html><body><div class="container"><!-- header -->hi</div>'
+  end
+
+  it "adds the css right after the head" do
+    get '/', :body => '<html><head>... <body>'
+    last_response.body.should.equal '<html><head><style type="text/css">#header</style>... <body><!-- header -->'
+  end
+end
+
+describe Heroku::Nav::Footer do
+  def app
+    make_app { use Heroku::Nav::Footer }
+  end
+
+  it "adds the html right before the body closing" do
+    get '/', :body => '<body>hi</body></html>'
+    last_response.body.should.equal '<body>hi<!-- footer --></body></html>'
+  end
+
+  it "adds the css right after the head" do
+    get '/', :body => '<html><head>... <body>'
+    last_response.body.should.equal '<html><head><style type="text/css">#footer</style>... <body>'
+  end
+end

metadata

@@ -0,0 +1,113 @@
+--- !ruby/object:Gem::Specification 
+name: heroku-nav
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Todd Matthews
+- Pedro Belo
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-02-24 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: baconmocha
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: sinatra
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: rack-test
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: rest-client
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.2.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: json
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: ""
+email: pedro@heroku.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Rakefile
+- heroku-nav.gemspec
+- lib/heroku/nav.rb
+- spec/api_spec.rb
+- spec/base.rb
+- spec/nav_spec.rb
+has_rdoc: true
+homepage: http://heroku.com
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: ""
+test_files: 
+- spec/api_spec.rb
+- spec/base.rb
+- spec/nav_spec.rb