Class: Superbot::Web

Inherits:
Object
  • Object
show all
Defined in:
lib/superbot/web.rb

Instance Method Summary collapse

Constructor Details

#initialize(webdriver_endpoint: nil) ⇒ Web

Returns a new instance of Web.



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
# File 'lib/superbot/web.rb', line 12

def initialize(webdriver_endpoint: nil)
  @sinatra = Sinatra.new
  @sinatra.set :bind, "127.0.0.1"
  @sinatra.set :silent_sinatra, true
  @sinatra.set :silent_webrick, true
  @sinatra.set :silent_access_log, false
  instance = self

  @sinatra.before do
    headers['Access-Control-Allow-Methods'] = 'GET, POST, PUT, DELETE, OPTIONS'
    headers['Access-Control-Allow-Origin'] = '*'
    headers['Access-Control-Allow-Headers'] = 'accept, authorization, origin'
  end

  @sinatra.options '*' do
    response.headers['Allow'] = 'HEAD,GET,PUT,DELETE,OPTIONS,POST'
    response.headers['Access-Control-Allow-Headers'] = 'X-Requested-With, X-HTTP-Method-Override, Content-Type, Cache-Control, Accept'
  end

  @sinatra.get "/__superbot/v1/ping" do
    "PONG"
  end

  @sinatra.post "/__superbot/v1/convert" do
    converted_script = Superbot::Capybara::Convert.call(request.body.read)
    instance.capybara_runner.run(converted_script)
  end

  return unless webdriver_endpoint

  webdriver_uri = URI.parse(webdriver_endpoint)
  @request_settings = {
    userinfo: webdriver_uri.userinfo,
    host:     webdriver_uri.host,
    port:     webdriver_uri.port,
    path:     webdriver_uri.path
  }

  %w(get post put patch delete).each do |verb|
    @sinatra.send(verb, "/wd/hub/*") do
      begin
        content_type 'application/json'
        response = instance.remote_webdriver_request(
          verb.capitalize,
          request.path_info,
          request.query_string,
          request.body,
          instance.incomming_headers(request)
        )
        status response.code
        headers instance.all_headers(response)
        response.body
      rescue => e
        puts e.message
        halt 500, { message: e.message }.to_json
      end
    end
  end
end

Instance Method Details

#all_headers(response) ⇒ Object 



92
93
94
95
96
97
98
 
# File 'lib/superbot/web.rb', line 92

def all_headers(response)
  header_list = {}
  response.header.each_capitalized do |k, v|
    header_list[k] = v unless k == "Transfer-Encoding"
  end
  header_list
end

#capybara_runner ⇒ Object 



72
73
74
 
# File 'lib/superbot/web.rb', line 72

def capybara_runner
  @capybara_runner ||= Superbot::Capybara::Runner.new
end

#incomming_headers(request) ⇒ Object 



100
101
102
 
# File 'lib/superbot/web.rb', line 100

def incomming_headers(request)
  request.env.map { |header, value| [header[5..-1].split("_").map(&:capitalize).join('-'), value] if header.start_with?("HTTP_") }.compact.to_h
end

#quit! ⇒ Object 



120
121
122
 
# File 'lib/superbot/web.rb', line 120

def quit!
  @sinatra&.quit!
end

#remote_webdriver_request(type, path, query_string, body, new_headers) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
 
# File 'lib/superbot/web.rb', line 76

def remote_webdriver_request(type, path, query_string, body, new_headers)
  uri = URI::HTTP.build(
    @request_settings.merge(
      path: [@request_settings[:path], path.gsub('wd/hub/', '')].join,
      query: query_string.empty? ? nil : query_string
    )
  )
  req = Net::HTTP.const_get(type).new(uri, new_headers.merge('Content-Type' => 'application/json'))
  req.basic_auth(*@request_settings[:userinfo].split(':')) if @request_settings[:userinfo]
  req.body = body.read
  Net::HTTP.new(uri.hostname, uri.port).start do |http|
    http.read_timeout = Superbot.cloud_timeout
    http.request(req)
  end
end

#run! ⇒ Object 



104
105
106
 
# File 'lib/superbot/web.rb', line 104

def run!
  @sinatra.run_async!
end

#run_async_after_running! ⇒ Object 



108
109
110
111
112
113
114
115
116
117
118
 
# File 'lib/superbot/web.rb', line 108

def run_async_after_running!
  Thread.new do
    @sinatra.run!
  end

  loop do
    break if @sinatra.running?

    sleep 0.001
  end
end