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 "ui/shell_dialogs/base_shell_dialog_win.h"
6
7#include <algorithm>
8
9#include "base/threading/thread.h"
10#include "base/win/scoped_com_initializer.h"
11
12namespace ui {
13
14// static
15BaseShellDialogImpl::Owners BaseShellDialogImpl::owners_;
16int BaseShellDialogImpl::instance_count_ = 0;
17
18BaseShellDialogImpl::BaseShellDialogImpl() {
19  ++instance_count_;
20}
21
22BaseShellDialogImpl::~BaseShellDialogImpl() {
23  // All runs should be complete by the time this is called!
24  if (--instance_count_ == 0)
25    DCHECK(owners_.empty());
26}
27
28BaseShellDialogImpl::RunState BaseShellDialogImpl::BeginRun(HWND owner) {
29  // Cannot run a modal shell dialog if one is already running for this owner.
30  DCHECK(!IsRunningDialogForOwner(owner));
31  // The owner must be a top level window, otherwise we could end up with two
32  // entries in our map for the same top level window.
33  DCHECK(!owner || owner == GetAncestor(owner, GA_ROOT));
34  RunState run_state;
35  run_state.dialog_thread = CreateDialogThread();
36  run_state.owner = owner;
37  if (owner) {
38    owners_.insert(owner);
39    DisableOwner(owner);
40  }
41  return run_state;
42}
43
44void BaseShellDialogImpl::EndRun(RunState run_state) {
45  if (run_state.owner) {
46    DCHECK(IsRunningDialogForOwner(run_state.owner));
47    EnableOwner(run_state.owner);
48    DCHECK(owners_.find(run_state.owner) != owners_.end());
49    owners_.erase(run_state.owner);
50  }
51  DCHECK(run_state.dialog_thread);
52  delete run_state.dialog_thread;
53}
54
55bool BaseShellDialogImpl::IsRunningDialogForOwner(HWND owner) const {
56  return (owner && owners_.find(owner) != owners_.end());
57}
58
59void BaseShellDialogImpl::DisableOwner(HWND owner) {
60  if (IsWindow(owner))
61    EnableWindow(owner, FALSE);
62}
63
64// static
65base::Thread* BaseShellDialogImpl::CreateDialogThread() {
66  base::Thread* thread = new base::Thread("Chrome_ShellDialogThread");
67  thread->init_com_with_mta(false);
68  bool started = thread->Start();
69  DCHECK(started);
70  return thread;
71}
72
73void BaseShellDialogImpl::EnableOwner(HWND owner) {
74  if (IsWindow(owner))
75    EnableWindow(owner, TRUE);
76}
77
78}  // namespace ui
79