rack-livereload/lib/rack/livereload.rb

108 lines
2.7 KiB
Ruby
Raw Normal View History

2011-11-17 20:44:48 +00:00
require 'erb'
2011-11-04 15:51:22 +00:00
module Rack
class LiveReload
LIVERELOAD_JS_PATH = '/__rack/livereload.js'
LIVERELOAD_LOCAL_URI = 'http://localhost:35729/livereload.js'
2011-11-04 15:51:22 +00:00
attr_reader :app
def initialize(app, options = {})
2011-11-04 15:51:22 +00:00
@app = app
@options = options
2011-11-04 15:51:22 +00:00
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
2011-11-04 15:51:22 +00:00
def call(env)
2011-11-17 20:44:48 +00:00
_, path, file = (env['PATH_INFO'] || '').split('/')
if path == '__rack' && ::File.file?(target = ::File.expand_path("../../../js/#{file}", __FILE__))
deliver_file(target)
2011-11-04 15:51:22 +00:00
else
status, headers, body = @app.call(env)
case headers['Content-Type']
when %r{text/html}
2011-11-07 13:49:42 +00:00
content_length = 0
2011-11-04 15:51:22 +00:00
body.each do |line|
2011-11-07 13:49:42 +00:00
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]
2011-11-04 15:51:22 +00:00
2011-11-17 20:44:48 +00:00
template = ERB.new(::File.read(::File.expand_path('../../../skel/livereload.html.erb', __FILE__)))
line.gsub!('</head>', %{#{template.result(binding)}</head>})
2011-11-07 13:41:51 +00:00
2011-11-07 13:49:42 +00:00
headers["X-Rack-LiveReload"] = '1'
2011-11-04 15:51:22 +00:00
end
2011-11-07 13:49:42 +00:00
content_length += line.length
2011-11-04 15:51:22 +00:00
end
2011-11-07 13:49:42 +00:00
headers['Content-Length'] = content_length.to_s
2011-11-04 15:51:22 +00:00
end
[ status, headers, body ]
end
end
private
def deliver_file(file)
2011-11-17 20:44:48 +00:00
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]
2011-11-04 15:51:22 +00:00
end
end
end