src/ui/shell_dialogs/base_shell_dialog_win.cc

   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
  12 namespace ui {
  13
  14 // static
  15 BaseShellDialogImpl::Owners BaseShellDialogImpl::owners_;
  16 int BaseShellDialogImpl::instance_count_ = 0;
  17
  18 BaseShellDialogImpl::BaseShellDialogImpl() {
  19   ++instance_count_;
  20 }
  21
  22 BaseShellDialogImpl::~BaseShellDialogImpl() {
  23   // All runs should be complete by the time this is called!
  24   if (--instance_count_ == 0)
  25     DCHECK(owners_.empty());
  26 }
  27
  28 BaseShellDialogImpl::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   // TODO(scottmg): This should be re-enabled when Chrome Frame is removed.
  34   // http://crbug.com/310264
  35   // DCHECK(!owner || owner == GetAncestor(owner, GA_ROOT));
  36   RunState run_state;
  37   run_state.dialog_thread = CreateDialogThread();
  38   run_state.owner = owner;
  39   if (owner) {
  40     owners_.insert(owner);
  41     DisableOwner(owner);
  42   }
  43   return run_state;
  44 }
  45
  46 void BaseShellDialogImpl::EndRun(RunState run_state) {
  47   if (run_state.owner) {
  48     DCHECK(IsRunningDialogForOwner(run_state.owner));
  49     EnableOwner(run_state.owner);
  50     DCHECK(owners_.find(run_state.owner) != owners_.end());
  51     owners_.erase(run_state.owner);
  52   }
  53   DCHECK(run_state.dialog_thread);
  54   delete run_state.dialog_thread;
  55 }
  56
  57 bool BaseShellDialogImpl::IsRunningDialogForOwner(HWND owner) const {
  58   return (owner && owners_.find(owner) != owners_.end());
  59 }
  60
  61 void BaseShellDialogImpl::DisableOwner(HWND owner) {
  62   if (IsWindow(owner))
  63     EnableWindow(owner, FALSE);
  64 }
  65
  66 // static
  67 base::Thread* BaseShellDialogImpl::CreateDialogThread() {
  68   base::Thread* thread = new base::Thread("Chrome_ShellDialogThread");
  69   thread->init_com_with_mta(false);
  70   bool started = thread->Start();
  71   DCHECK(started);
  72   return thread;
  73 }
  74
  75 void BaseShellDialogImpl::EnableOwner(HWND owner) {
  76   if (IsWindow(owner))
  77     EnableWindow(owner, TRUE);
  78 }
  79
  80 }  // namespace ui