Class: Superbot::Web

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

Instance Method Summary collapse

Constructor Details

#initialize(webdriver_endpoint: nil, auth_token: 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
71
 
# File 'lib/superbot/web.rb', line 12

def initialize(webdriver_endpoint: nil, auth_token: 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)
  @auth_token = auth_token
  @request_settings = {
    userinfo: @auth_token,
    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 



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

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 



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

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

#incomming_headers(request) ⇒ Object 



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

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 



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

def quit!
  @sinatra&.quit!
end

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



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

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(*@auth_token.split(':')) if @auth_token
  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 



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

def run!
  @sinatra.run_async!
end

#run_async_after_running! ⇒ Object 



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

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

  loop do
    break if @sinatra.running?

    sleep 0.001
  end
end