| OLD | NEW | 
|---|
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be | 
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. | 
| 4 | 4 | 
|  | 5 #include "content/public/common/user_agent.h" | 
| 5 #include "headless/public/headless_browser.h" | 6 #include "headless/public/headless_browser.h" | 
|  | 7 #include "net/url_request/url_request_context_getter.h" | 
| 6 | 8 | 
| 7 using Options = headless::HeadlessBrowser::Options; | 9 using Options = headless::HeadlessBrowser::Options; | 
| 8 using Builder = headless::HeadlessBrowser::Options::Builder; | 10 using Builder = headless::HeadlessBrowser::Options::Builder; | 
| 9 | 11 | 
| 10 namespace headless { | 12 namespace headless { | 
| 11 | 13 | 
|  | 14 // Product name for building the default user agent string. | 
|  | 15 namespace { | 
|  | 16 const char kProductName[] = "HeadlessChrome"; | 
|  | 17 } | 
|  | 18 | 
| 12 Options::Options(int argc, const char** argv) | 19 Options::Options(int argc, const char** argv) | 
| 13     : argc(argc), argv(argv), devtools_http_port(kInvalidPort) {} | 20     : argc(argc), | 
|  | 21       argv(argv), | 
|  | 22       user_agent(content::BuildUserAgentFromProduct(kProductName)), | 
|  | 23       devtools_http_port(kInvalidPort), | 
|  | 24       message_pump(nullptr) {} | 
| 14 | 25 | 
| 15 Options::~Options() {} | 26 Options::~Options() {} | 
| 16 | 27 | 
| 17 Builder::Builder(int argc, const char** argv) : options_(argc, argv) {} | 28 Builder::Builder(int argc, const char** argv) : options_(argc, argv) {} | 
| 18 | 29 | 
| 19 Builder::~Builder() {} | 30 Builder::~Builder() {} | 
| 20 | 31 | 
| 21 Builder& Builder::SetUserAgent(const std::string& user_agent) { | 32 Builder& Builder::SetUserAgent(const std::string& user_agent) { | 
| 22   options_.user_agent = user_agent; | 33   options_.user_agent = user_agent; | 
| 23   return *this; | 34   return *this; | 
| 24 } | 35 } | 
| 25 | 36 | 
| 26 Builder& Builder::EnableDevToolsServer(int port) { | 37 Builder& Builder::EnableDevToolsServer(const std::string& address, int port) { | 
|  | 38   options_.devtools_http_address = address; | 
| 27   options_.devtools_http_port = port; | 39   options_.devtools_http_port = port; | 
| 28   return *this; | 40   return *this; | 
| 29 } | 41 } | 
| 30 | 42 | 
| 31 Builder& Builder::SetURLRequestContextGetter( | 43 Builder& Builder::SetMessagePump(base::MessagePump* message_pump) { | 
| 32     scoped_refptr<net::URLRequestContextGetter> url_request_context_getter) { | 44   options_.message_pump = message_pump; | 
| 33   options_.url_request_context_getter = url_request_context_getter; |  | 
| 34   return *this; | 45   return *this; | 
| 35 } | 46 } | 
| 36 | 47 | 
| 37 Options Builder::Build() { | 48 Options Builder::Build() { | 
| 38   return options_; | 49   return options_; | 
| 39 } | 50 } | 
| 40 | 51 | 
| 41 }  // namespace headless | 52 }  // namespace headless | 
| OLD | NEW | 
|---|