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