download: Add a GetMenuModel() that grabs the correct model.

chrome/browser/download/download_shelf_context_menu.h

 // Copyright (c) 2011 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.
 
 #ifndef CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SHELF_CONTEXT_MENU_H_
 #define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SHELF_CONTEXT_MENU_H_
 #pragma once
 
 #include "base/basictypes.h"
 #include "base/compiler_specific.h"
@@ skipping 18 matching lines @@
   };
 
   virtual ~DownloadShelfContextMenu();
 
   DownloadItem* download_item() const { return download_item_; }
   void set_download_item(DownloadItem* item) { download_item_ = item; }
 
  protected:
   explicit DownloadShelfContextMenu(BaseDownloadItemModel* download_model);
 
-  ui::SimpleMenuModel* GetInProgressMenuModel();
-  ui::SimpleMenuModel* GetFinishedMenuModel();
+  // Returns the correct menu model depending whether the download item is
+  // completed or not.
+  ui::SimpleMenuModel* GetMenuModel();
 
   // ui::SimpleMenuModel::Delegate:
   virtual bool IsCommandIdEnabled(int command_id) const OVERRIDE;
   virtual bool IsCommandIdChecked(int command_id) const OVERRIDE;
   virtual void ExecuteCommand(int command_id) OVERRIDE;
   virtual bool GetAcceleratorForCommandId(
       int command_id,
       ui::Accelerator* accelerator) OVERRIDE;
   virtual bool IsItemForCommandIdDynamic(int command_id) const OVERRIDE;
   virtual string16 GetLabelForCommandId(int command_id) const OVERRIDE;
 
  private:
+  ui::SimpleMenuModel* GetInProgressMenuModel();
+  ui::SimpleMenuModel* GetFinishedMenuModel();
+
   // We show slightly different menus if the download is in progress vs. if the
   // download has finished.
   scoped_ptr<ui::SimpleMenuModel> in_progress_download_menu_model_;
   scoped_ptr<ui::SimpleMenuModel> finished_download_menu_model_;
 
   // A model to control the cancel behavior.
   BaseDownloadItemModel* download_model_;
 
   // Information source.
   DownloadItem* download_item_;
 
   DISALLOW_COPY_AND_ASSIGN(DownloadShelfContextMenu);
 };
 
 #endif  // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SHELF_CONTEXT_MENU_H_