Refactors building the system menu model into a common place

chrome/browser/ui/views/frame/browser_desktop_root_window_host_win.cc

 // Copyright (c) 2012 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.
 
 #include "chrome/browser/ui/views/frame/browser_desktop_root_window_host_win.h"
 
 #include <dwmapi.h>
 
 #include "chrome/browser/ui/views/frame/browser_frame.h"
 #include "chrome/browser/ui/views/frame/browser_view.h"
 #include "chrome/browser/ui/views/tabs/tab_strip.h"
 #include "ui/base/theme_provider.h"
+#include "ui/views/controls/menu/native_menu_win.h"
 
 #pragma comment(lib, "dwmapi.lib")
 
 namespace {
 
 const int kClientEdgeThickness = 3;
 // We need to offset the DWMFrame into the toolbar so that the blackness
 // doesn't show up on our rounded corners.
 const int kDWMFrameTopOffset = 3;
 
@@ skipping 11 matching lines @@
     : DesktopRootWindowHostWin(native_widget_delegate,
                                desktop_native_widget_aura,
                                initial_bounds),
       browser_view_(browser_view),
       browser_frame_(browser_frame) {
 }
 
 BrowserDesktopRootWindowHostWin::~BrowserDesktopRootWindowHostWin() {
 }
 
+views::NativeMenuWin* BrowserDesktopRootWindowHostWin::GetSystemMenu() {
+  if (!system_menu_.get()) {
+    system_menu_.reset(
+        new views::NativeMenuWin(browser_frame_->GetSystemMenuModel(),
+                                 GetHWND()));
+    system_menu_->Rebuild();
+  }
+  return system_menu_.get();
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // BrowserDesktopRootWindowHostWin, BrowserDesktopRootWindowHost implementation:
 
 views::DesktopRootWindowHost*
     BrowserDesktopRootWindowHostWin::AsDesktopRootWindowHost() {
   return this;
 }
 
 int BrowserDesktopRootWindowHostWin::GetMinimizeButtonOffset() const {
   return minimize_button_metrics_.GetMinimizeButtonOffsetX();
 }
 
+bool BrowserDesktopRootWindowHostWin::UsesNativeSystemMenu() const {
+  return true;
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // BrowserDesktopRootWindowHostWin, views::DesktopRootWindowHostWin overrides:
 
 int BrowserDesktopRootWindowHostWin::GetInitialShowState() const {
   STARTUPINFO si = {0};
   si.cb = sizeof(si);
   si.dwFlags = STARTF_USESHOWWINDOW;
   GetStartupInfo(&si);
   return si.wShowWindow;
 }
@@ skipping 23 matching lines @@
   // the window region.
   UpdateDWMFrame();
   DesktopRootWindowHostWin::HandleFrameChanged();
 }
 
 bool BrowserDesktopRootWindowHostWin::PreHandleMSG(UINT message,
                                                    WPARAM w_param,
                                                    LPARAM l_param,
                                                    LRESULT* result) {
   switch (message) {
-  case WM_ACTIVATE:
-    if (LOWORD(w_param) != WA_INACTIVE)
-      minimize_button_metrics_.OnHWNDActivated();
-    return false;
-  case WM_ENDSESSION:
-    browser::SessionEnding();
-    return true;
+    case WM_ACTIVATE:
+      if (LOWORD(w_param) != WA_INACTIVE)
+        minimize_button_metrics_.OnHWNDActivated();
+      return false;
+    case WM_ENDSESSION:
+      browser::SessionEnding();
+      return true;
+    case WM_INITMENUPOPUP:
+      GetSystemMenu()->UpdateStates();
+      return true;
   }
   return DesktopRootWindowHostWin::PreHandleMSG(
       message, w_param, l_param, result);
 }
 
 void BrowserDesktopRootWindowHostWin::PostHandleMSG(UINT message,
                                                     WPARAM w_param,
                                                     LPARAM l_param) {
   switch (message) {
   case WM_CREATE:
@@ skipping 92 matching lines @@
         views::DesktopNativeWidgetAura* desktop_native_widget_aura,
         const gfx::Rect& initial_bounds,
         BrowserView* browser_view,
         BrowserFrame* browser_frame) {
   return new BrowserDesktopRootWindowHostWin(native_widget_delegate,
                                              desktop_native_widget_aura,
                                              initial_bounds,
                                              browser_view,
                                              browser_frame);
 }