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 BaseShellDialogImpl()18BaseShellDialogImpl::BaseShellDialogImpl() { 19 ++instance_count_; 20 } 21 ~BaseShellDialogImpl()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 BeginRun(HWND owner)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 // 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 EndRun(RunState run_state)46void 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 IsRunningDialogForOwner(HWND owner) const57bool BaseShellDialogImpl::IsRunningDialogForOwner(HWND owner) const { 58 return (owner && owners_.find(owner) != owners_.end()); 59 } 60 DisableOwner(HWND owner)61void BaseShellDialogImpl::DisableOwner(HWND owner) { 62 if (IsWindow(owner)) 63 EnableWindow(owner, FALSE); 64 } 65 66 // static CreateDialogThread()67base::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 EnableOwner(HWND owner)75void BaseShellDialogImpl::EnableOwner(HWND owner) { 76 if (IsWindow(owner)) 77 EnableWindow(owner, TRUE); 78 } 79 80 } // namespace ui 81