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