From 45d5868aa630cb05edf0c8ccddca79415ea7ba04 Mon Sep 17 00:00:00 2001
From: Adam McCrea <adam@edgecase.com>
Date: Fri, 8 Mar 2013 14:59:39 -0500
Subject: [PATCH] Handle Rails apps running on relative root

---
 lib/rack/livereload.rb       | 4 +++-
 skel/livereload.html.erb     | 4 ++--
 spec/rack/livereload_spec.rb | 9 +++++++++
 3 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/lib/rack/livereload.rb b/lib/rack/livereload.rb
index 0a7b613..349640d 100644
--- a/lib/rack/livereload.rb
+++ b/lib/rack/livereload.rb
@@ -77,8 +77,10 @@ module Rack
               if !headers['X-Rack-LiveReload'] && line['<head']
                 host_to_use = (@options[:host] || env['HTTP_HOST'] || 'localhost').gsub(%r{:.*}, '')
 
+                app_root = ENV['RAILS_RELATIVE_URL_ROOT'] || ''
+
                 if use_vendored?
-                  src = LIVERELOAD_JS_PATH.dup + "?host=#{host_to_use}"
+                  src = "#{app_root}#{LIVERELOAD_JS_PATH.dup}?host=#{host_to_use}"
                 else
                   src = livereload_local_uri.dup.gsub('localhost', host_to_use) + '?'
                 end
diff --git a/skel/livereload.html.erb b/skel/livereload.html.erb
index b953e93..db1cb61 100644
--- a/skel/livereload.html.erb
+++ b/skel/livereload.html.erb
@@ -5,8 +5,8 @@
       WEB_SOCKET_FORCE_FLASH = true;
     <% end %>
   </script>
-  <script type="text/javascript" src="/__rack/swfobject.js"></script>
-  <script type="text/javascript" src="/__rack/web_socket.js"></script>
+  <script type="text/javascript" src="<%= app_root %>/__rack/swfobject.js"></script>
+  <script type="text/javascript" src="<%= app_root %>/__rack/web_socket.js"></script>
 <% end %>
 <script type="text/javascript">
   RACK_LIVERELOAD_PORT = <%= @port %>;
diff --git a/spec/rack/livereload_spec.rb b/spec/rack/livereload_spec.rb
index 17c46a4..bdcdac9 100644
--- a/spec/rack/livereload_spec.rb
+++ b/spec/rack/livereload_spec.rb
@@ -161,6 +161,15 @@ describe Rack::LiveReload do
         body_dom.at_css("script:last-child")[:insert].should == "before"
       end
 
+      context 'when a relative URL root is specified' do
+        before do
+          ENV['RAILS_RELATIVE_URL_ROOT'] = '/a_relative_path'
+        end
+
+        it 'should prepend the relative path to the script src' do
+          body_dom.at_css("script:eq(5)")[:src].should match(%r{^/a_relative_path/})
+        end
+      end
     end
 
     describe "LIVERELOAD_PORT value" do