module WatirSpec class Server class App def response(path, data = nil) case path when '/' respond(self.class.name) when '/post_to_me' respond("You posted the following content:\n#{data}") when '/plain_text' respond('This is text/plain', 'Content-Type' => 'text/plain') when %r{/set_cookie} body = "C is for cookie, it's good enough for me" respond(body, 'Content-Type' => 'text/html', 'Set-Cookie' => 'monster=1') when static_file? respond_to_file(path) else respond('') end end private def respond_to_file(path) case path when css_file? respond(file_read(path), 'Content-Type' => 'text/css') when js_file? respond(file_read(path), 'Content-Type' => 'application/javascript') when png_file? respond(file_binread(path), 'Content-Type' => 'image/png') when gif_file? respond(file_read(path), 'Content-Type' => 'image/gif') else respond(file_read(path)) end end def respond(body, headers = {}, status = '200 OK') [status, headers, body] end def css_file? proc { |path| static_file(path) && path.end_with?('.css') } end def js_file? proc { |path| static_file(path) && path.end_with?('.js') } end def png_file? proc { |path| static_file(path) && path.end_with?('.png') } end def gif_file? proc { |path| static_file(path) && path.end_with?('.gif') } end def static_file? proc { |path| static_file(path) } end def static_files WatirSpec.htmls.flat_map do |html| Dir["#{html}/**/*"] end end def static_file(path) static_files.find do |file| file.end_with?(path) end end def file_read(path) File.read(static_file(path)) end def file_binread(path) File.binread(static_file(path)) end end end end