1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "remoting/host/setup/win/start_host_window.h"
6
7#include <atlbase.h>
8#include <atlwin.h>
9#include <windows.h>
10
11#include "base/memory/scoped_ptr.h"
12#include "base/strings/string16.h"
13#include "base/strings/utf_string_conversions.h"
14#include "google_apis/gaia/gaia_urls.h"
15#include "remoting/host/service_urls.h"
16#include "remoting/host/setup/oauth_helper.h"
17#include "remoting/host/setup/pin_validator.h"
18#include "remoting/host/setup/win/load_string_from_resource.h"
19
20namespace remoting {
21
22StartHostWindow::StartHostWindow(
23    scoped_refptr<net::URLRequestContextGetter> url_request_context_getter)
24    : host_starter_(remoting::HostStarter::Create(
25          remoting::ServiceUrls::GetInstance()->directory_hosts_url(),
26          url_request_context_getter)),
27      consent_to_collect_data_(true),
28      mem_mgr_(GetProcessHeap()),
29      string_mgr_(&mem_mgr_),
30      weak_ptr_factory_(this),
31      weak_ptr_(weak_ptr_factory_.GetWeakPtr()) {
32}
33
34void StartHostWindow::OnCancel(UINT code, int id, HWND control) {
35  EndDialog(IDCANCEL);
36}
37
38void StartHostWindow::OnClose() {
39  EndDialog(IDCANCEL);
40}
41
42LRESULT StartHostWindow::OnInitDialog(HWND wparam, LPARAM lparam) {
43  // Failure of any of these calls is acceptable.
44  SetWindowText(LoadStringFromResource(IDS_TITLE));
45  GetDlgItem(IDC_CONSENT).SetWindowText(LoadStringFromResource(IDS_CONSENT));
46  CheckDlgButton(IDC_CONSENT, BST_CHECKED);
47  return TRUE;
48}
49
50void StartHostWindow::OnConsent(UINT code, int id, HWND control) {
51  bool checked = (IsDlgButtonChecked(IDC_CONSENT) == BST_CHECKED);
52  checked = !checked;
53  CheckDlgButton(IDC_CONSENT, checked ? BST_CHECKED : BST_UNCHECKED);
54}
55
56void StartHostWindow::OnOk(UINT code, int id, HWND control) {
57  host_name_ = GetDlgItemString(IDC_HOST_NAME);
58  pin_ = GetDlgItemString(IDC_PIN);
59  std::string confirm_pin = GetDlgItemString(IDC_CONFIRM_PIN);
60  consent_to_collect_data_ = (IsDlgButtonChecked(IDC_CONSENT) == BST_CHECKED);
61  if (pin_ != confirm_pin) {
62    MessageBox(LoadStringFromResource(IDS_SAME_PIN),
63               LoadStringFromResource(IDS_TITLE),
64               MB_ICONEXCLAMATION | MB_OK);
65    return;
66  }
67  if (!IsPinValid(pin_)) {
68    MessageBox(LoadStringFromResource(IDS_INVALID_PIN),
69               LoadStringFromResource(IDS_TITLE), MB_ICONEXCLAMATION | MB_OK);
70    return;
71  }
72  MessageBox(LoadStringFromResource(IDS_USE_BROWSER),
73             LoadStringFromResource(IDS_TITLE), MB_OK);
74  auth_code_getter_.GetAuthCode(
75      base::Bind(&StartHostWindow::OnAuthCode, weak_ptr_));
76}
77
78void StartHostWindow::OnAuthCode(const std::string& auth_code) {
79  if (auth_code.empty())
80    return;
81
82  host_starter_->StartHost(
83      host_name_, pin_, consent_to_collect_data_, auth_code,
84      GetDefaultOauthRedirectUrl(),
85      base::Bind(&StartHostWindow::OnHostStarted, weak_ptr_));
86}
87
88void StartHostWindow::OnHostStarted(remoting::HostStarter::Result result) {
89  if (result == remoting::HostStarter::START_COMPLETE) {
90    MessageBox(LoadStringFromResource(IDS_HOST_START_SUCCEEDED),
91               LoadStringFromResource(IDS_TITLE), MB_OK);
92    EndDialog(IDOK);
93  } else {
94    MessageBox(LoadStringFromResource(IDS_HOST_START_FAILED),
95               LoadStringFromResource(IDS_TITLE), MB_ICONEXCLAMATION | MB_OK);
96  }
97}
98
99std::string StartHostWindow::GetDlgItemString(int id) {
100  CSimpleString str(L"", &string_mgr_);
101  GetDlgItemText(id, str);
102  return UTF16ToUTF8(str.GetString());
103}
104
105}  // namespace remoting
106