Add support for constrained windows on os x, based on Avi's GTMWindowSheetController.

chrome/browser/app_controller_mac.mm

 // Copyright (c) 2009 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #import "chrome/browser/app_controller_mac.h"
 
 #include "app/l10n_util.h"
 #include "base/command_line.h"
 #include "base/mac_util.h"
 #include "base/message_loop.h"
 #include "base/sys_string_conversions.h"
 #include "chrome/app/chrome_dll_resource.h"
 #include "chrome/browser/browser.h"
 #include "chrome/browser/browser_init.h"
 #include "chrome/browser/browser_list.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_shutdown.h"
 #include "chrome/browser/browser_window.h"
 #import "chrome/browser/cocoa/about_window_controller.h"
 #import "chrome/browser/cocoa/bookmark_menu_bridge.h"
+#import "chrome/browser/cocoa/browser_window_cocoa.h"
+#import "chrome/browser/cocoa/browser_window_controller.h"
 #import "chrome/browser/cocoa/history_menu_bridge.h"
 #import "chrome/browser/cocoa/clear_browsing_data_controller.h"
 #import "chrome/browser/cocoa/encoding_menu_controller_delegate_mac.h"
 #import "chrome/browser/cocoa/preferences_window_controller.h"
 #import "chrome/browser/cocoa/tab_strip_controller.h"
 #import "chrome/browser/cocoa/tab_window_controller.h"
 #import "chrome/browser/cocoa/ui_localizer.h"
 #include "chrome/browser/command_updater.h"
 #include "chrome/browser/download/download_manager.h"
+#include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/browser/sessions/tab_restore_service.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/pref_service.h"
 #include "chrome/browser/profile_manager.h"
 #include "chrome/common/temp_scaffolding_stubs.h"
 #include "grit/generated_resources.h"
 
 @interface AppController(PRIVATE)
 - (void)initMenuState;
 - (void)openURLs:(const std::vector<GURL>&)urls;
@@ skipping 61 matching lines @@
   [notificationCenter
       addObserver:self
          selector:@selector(tabsChanged:)
              name:kTabStripNumberOfTabsChanged
            object:nil];
 
   // Set up the command updater for when there are no windows open
   [self initMenuState];
 }
 
+// Checks if there are any tabs with sheets open, and if so, raises one of
+// the tabs with a sheet and returns NO.
+- (BOOL)shouldQuitWithOpenPerTabSheets {
+  BrowserList::const_iterator it = BrowserList::begin();
+  for (; it != BrowserList::end(); ++it) {
+    Browser* browser = *it;
+    BrowserWindowCocoa* window =
+        static_cast<BrowserWindowCocoa*>(browser->window());
+
+    // Could do this more nicely with a method e.g. on BWC. If I decide for
+    // keeping it this way, at least add a DCHECK().
+    BrowserWindowController* controller =
+        (BrowserWindowController*)[window->GetNativeHandle() windowController];
+
+    if (![controller shouldCloseWithOpenPerTabSheets])
+      return NO;
+  }
+
+  return YES;
+}
+
+// We do not use the normal application teardown process -- this function is
+// not called by the system but by us in |quit:|. |NSTerminateLater| is not a
+// return value that is supported by |quit:|.
+- (NSApplicationTerminateReply)applicationShouldTerminate:
+    (NSApplication *)sender {
+  // Do not quit if any per-tab sheets are open, as required by
+  // GTMWindowSheetController.
+  if (![self shouldQuitWithOpenPerTabSheets])
+      return NSTerminateCancel;
+
+  // Check for in-progress downloads, and prompt the user if they really want to
+  // quit (and thus cancel the downloads).
+  if (![self shouldQuitWithInProgressDownloads])
+    return NSTerminateCancel;
+
+  return NSTerminateNow;
+}
+
 // Called when the app is shutting down. Clean-up as appropriate.
 - (void)applicationWillTerminate:(NSNotification *)aNotification {
   DCHECK(!BrowserList::HasBrowserWithProfile([self defaultProfile]));
   if (!BrowserList::HasBrowserWithProfile([self defaultProfile])) {
     // As we're shutting down, we need to nuke the TabRestoreService, which will
     // start the shutdown of the NavigationControllers and allow for proper
     // shutdown. If we don't do this chrome won't shutdown cleanly, and may end
     // up crashing when some thread tries to use the IO thread (or another
     // thread) that is no longer valid.
     [self defaultProfile]->ResetTabRestoreService();
@@ skipping 210 matching lines @@
   }
 
   // No profiles or active downloads found, okay to exit.
   return YES;
 }
 
 // We can't use the standard terminate: method because it will abruptly exit
 // the app and leave things on the stack in an unfinalized state. We need to
 // post a quit message to our run loop so the stack can gracefully unwind.
 - (IBAction)quit:(id)sender {
+  if ([self applicationShouldTerminate:NSApp] == NSTerminateCancel)
+    return;
+
   // TODO(pinkerton):
   // since we have to roll it ourselves, ask the delegate (ourselves, really)
   // if we should terminate. For example, we might not want to if the user
   // has ongoing downloads or multiple windows/tabs open. However, this would
   // require posting UI and may require spinning up another run loop to
   // handle it. If it says to continue, post the quit message, otherwise
   // go back to normal.
 
-  // Check for in-progress downloads, and prompt the user if they really want to
-  // quit (and thus cancel the downloads).
-  if (![self shouldQuitWithInProgressDownloads])
-    return;
-
   NSAppleEventManager* em = [NSAppleEventManager sharedAppleEventManager];
   [em removeEventHandlerForEventClass:kInternetEventClass
                            andEventID:kAEGetURL];
   [em removeEventHandlerForEventClass:'WWW!'
                            andEventID:'OURL'];
   [em removeEventHandlerForEventClass:kCoreEventClass
                            andEventID:kAEOpenDocuments];
 
   // TODO(pinkerton): Not sure where this should live, including it here
   // causes all sorts of asserts from the open renderers. On Windows, it
@@ skipping 261 matching lines @@
                name:kUserClosedAboutNotification
              object:aboutController_.get()];
   }
   if (![[aboutController_ window] isVisible])
     [[aboutController_ window] center];
   [aboutController_ showWindow:self];
 #endif
 }
 
 @end