84 lines
2.2 KiB
Ruby
84 lines
2.2 KiB
Ruby
module Rack
|
|
class LiveReload
|
|
LIVERELOAD_JS_PATH = '/__rack/livereload.js'
|
|
LIVERELOAD_LOCAL_URI = 'http://localhost:35729/livereload.js'
|
|
|
|
attr_reader :app
|
|
|
|
def initialize(app, options = {})
|
|
@app = app
|
|
@options = options
|
|
end
|
|
|
|
def use_vendored?
|
|
return @use_vendored if @use_vendored
|
|
|
|
if @options[:source]
|
|
@use_vendored = (@options[:source] == :vendored)
|
|
else
|
|
require 'net/http'
|
|
require 'uri'
|
|
|
|
uri = URI.parse(LIVERELOAD_LOCAL_URI)
|
|
|
|
http = Net::HTTP.new(uri.host, uri.port)
|
|
http.read_timeout = 1
|
|
|
|
begin
|
|
http.send_request('GET', uri.path)
|
|
@use_vendored = false
|
|
rescue Timeout::Error, Errno::ECONNREFUSED
|
|
@use_vendored = true
|
|
end
|
|
end
|
|
|
|
@use_vendored
|
|
end
|
|
|
|
def call(env)
|
|
if env['PATH_INFO'] == LIVERELOAD_JS_PATH
|
|
deliver_file(::File.expand_path('../../../js/livereload.js', __FILE__))
|
|
else
|
|
status, headers, body = @app.call(env)
|
|
|
|
case headers['Content-Type']
|
|
when %r{text/html}
|
|
content_length = 0
|
|
|
|
body.each do |line|
|
|
if !headers['X-Rack-LiveReload'] && line['</head>']
|
|
host_to_use = @options[:host] || env['HTTP_HOST'].gsub(%r{:.*}, '')
|
|
|
|
if use_vendored?
|
|
src = LIVERELOAD_JS_PATH.dup + "?host=#{host_to_use}"
|
|
else
|
|
src = LIVERELOAD_LOCAL_URI.dup.gsub('localhost', host_to_use) + '?'
|
|
end
|
|
|
|
src << "&mindelay=#{@options[:min_delay]}" if @options[:min_delay]
|
|
src << "&maxdelay=#{@options[:max_delay]}" if @options[:max_delay]
|
|
src << "&port=#{@options[:port]}" if @options[:port]
|
|
|
|
line.gsub!('</head>', %{<script type="text/javascript" src="#{src}"></script></head>})
|
|
|
|
headers["X-Rack-LiveReload"] = '1'
|
|
end
|
|
|
|
content_length += line.length
|
|
end
|
|
|
|
headers['Content-Length'] = content_length.to_s
|
|
end
|
|
|
|
[ status, headers, body ]
|
|
end
|
|
end
|
|
|
|
private
|
|
def deliver_file(file)
|
|
[ 200, { 'Content-Type' => 'text/javascript', 'Content-Length' => ::File.size(file).to_s }, [ ::File.read(file) ] ]
|
|
end
|
|
end
|
|
end
|
|
|