140 lines
3.8 KiB
Ruby
140 lines
3.8 KiB
Ruby
require 'erb'
|
|
|
|
module Rack
|
|
class LiveReload
|
|
LIVERELOAD_JS_PATH = '/__rack/livereload.js'
|
|
HEAD_TAG_REGEX = /<head>|<head[^(er)][^<]*>/
|
|
|
|
BAD_USER_AGENTS = [ %r{MSIE} ]
|
|
|
|
def livereload_local_uri
|
|
"http://localhost:#{@port}/livereload.js"
|
|
end
|
|
|
|
attr_reader :app
|
|
|
|
def initialize(app, options = {})
|
|
@app, @options = app, options
|
|
@port = @options[:live_reload_port] || 35729
|
|
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, EOFError
|
|
@use_vendored = true
|
|
rescue => e
|
|
$stderr.puts e.inspect
|
|
raise e
|
|
end
|
|
end
|
|
|
|
@use_vendored
|
|
end
|
|
|
|
def call(env)
|
|
dup._call(env)
|
|
end
|
|
|
|
def _call(env)
|
|
_, path, file = (env['PATH_INFO'] || '').split('/')
|
|
|
|
if path == '__rack' && ::File.file?(target = ::File.expand_path("../../../js/#{file}", __FILE__))
|
|
deliver_file(target)
|
|
else
|
|
status, headers, body = result = @app.call(env)
|
|
|
|
if (headers['Content-Type'] and headers['Content-Type']['text/event-stream']) or
|
|
headers['Transfer-Encoding'] == 'chunked' or
|
|
headers['Content-Disposition'] =~ %r{^inline}
|
|
return result
|
|
end
|
|
|
|
body.close if body.respond_to?(:close)
|
|
|
|
new_body = [] ; body.each { |line| new_body << line.to_s }
|
|
|
|
if !ignored?(env['PATH_INFO']) and !bad_browser?(env['HTTP_USER_AGENT'])
|
|
if headers['Content-Type'] and status == 200 and headers['Content-Type'][%r{text/html}]
|
|
content_length = 0
|
|
|
|
new_body.each do |line|
|
|
if !headers['X-Rack-LiveReload'] && line['<head']
|
|
host_to_use = (@options[:host] || env['HTTP_HOST'] || 'localhost').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]
|
|
|
|
template = ERB.new(::File.read(::File.expand_path('../../../skel/livereload.html.erb', __FILE__)))
|
|
|
|
if line['<head']
|
|
line.gsub!(HEAD_TAG_REGEX) { |match| %{#{match}#{template.result(binding)}} }
|
|
end
|
|
|
|
headers["X-Rack-LiveReload"] = '1'
|
|
end
|
|
|
|
content_length += line.bytesize
|
|
end
|
|
|
|
headers['Content-Length'] = content_length.to_s
|
|
end
|
|
end
|
|
|
|
[ status, headers, new_body ]
|
|
end
|
|
end
|
|
|
|
def ignored?(path_info)
|
|
@options[:ignore] and @options[:ignore].any? { |filter| path_info[filter] }
|
|
end
|
|
|
|
def bad_browser?(user_agent)
|
|
BAD_USER_AGENTS.any? { |pattern| (user_agent || '')[pattern] }
|
|
end
|
|
|
|
private
|
|
def deliver_file(file)
|
|
type = case ::File.extname(file)
|
|
when '.js'
|
|
'text/javascript'
|
|
when '.swf'
|
|
'application/swf'
|
|
end
|
|
|
|
[ 200, { 'Content-Type' => type, 'Content-Length' => ::File.size(file).to_s }, [ ::File.read(file) ] ]
|
|
end
|
|
|
|
def force_swf?
|
|
@options[:force_swf]
|
|
end
|
|
|
|
def with_swf?
|
|
!@options[:no_swf]
|
|
end
|
|
end
|
|
end
|
|
|