Preliminary work on resuming downloads whose connections have expired.

chrome/browser/download/download_shelf.cc

 // 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.
 
 #include "chrome/browser/download/download_shelf.h"
 
 #include "app/l10n_util.h"
+#include "base/command_line.h"
 #include "base/file_util.h"
 #include "chrome/browser/browser.h"
 #include "chrome/browser/dom_ui/downloads_ui.h"
 #include "chrome/browser/download/download_item.h"
 #include "chrome/browser/download/download_item_model.h"
 #include "chrome/browser/download/download_manager.h"
 #include "chrome/browser/download/download_util.h"
 #include "chrome/browser/metrics/user_metrics.h"
 #include "chrome/common/url_constants.h"
+#include "chrome/common/chrome_switches.h"
 #include "grit/generated_resources.h"
 
 // DownloadShelfContextMenu ----------------------------------------------------
 
 DownloadShelfContextMenu::DownloadShelfContextMenu(
     BaseDownloadItemModel* download_model)
     : download_(download_model->download()),
       model_(download_model) {
 }
 
@@ skipping 16 matching lines @@
       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_SHOW);
     case OPEN_WHEN_COMPLETE:
       if (download_->state() == DownloadItem::IN_PROGRESS)
         return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_OPEN_WHEN_COMPLETE);
       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_OPEN);
     case ALWAYS_OPEN_TYPE:
       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_ALWAYS_OPEN_TYPE);
     case CANCEL:
       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_CANCEL);
     case TOGGLE_PAUSE: {
-      if (download_->is_paused())
+      if (download_->is_paused() || download_->IsInterruptedDownload())
         return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_RESUME_ITEM);
       else
         return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_PAUSE_ITEM);
     }
     default:
       NOTREACHED();
   }
   return string16();
 }
 
 bool DownloadShelfContextMenu::IsCommandIdEnabled(int command_id) const {
+  static bool can_restart = CommandLine::ForCurrentProcess()->HasSwitch(
+      switches::kEnableDownloadRestarts);
   switch (command_id) {
     case SHOW_IN_FOLDER:
     case OPEN_WHEN_COMPLETE:
       return download_->state() != DownloadItem::CANCELLED;
     case ALWAYS_OPEN_TYPE:
       return download_->CanOpenDownload();
     case CANCEL:
-      return download_->state() == DownloadItem::IN_PROGRESS;
+      return download_->IsPartialDownload();
     case TOGGLE_PAUSE:
-      return download_->state() == DownloadItem::IN_PROGRESS;
+      return ((download_->state() == DownloadItem::IN_PROGRESS) ||
+              (can_restart &&
+                   (download_->IsInterruptedDownload())));
     default:
       return command_id > 0 && command_id < MENU_LAST;
   }
 }
 
 void DownloadShelfContextMenu::ExecuteCommand(int command_id) {
   switch (command_id) {
     case SHOW_IN_FOLDER:
       download_->ShowDownloadInShell();
       break;
     case OPEN_WHEN_COMPLETE:
       download_->OpenDownload();
       break;
     case ALWAYS_OPEN_TYPE: {
       download_->OpenFilesBasedOnExtension(
           !IsCommandIdChecked(ALWAYS_OPEN_TYPE));
       break;
     }
     case CANCEL:
       model_->CancelTask();
       break;
     case TOGGLE_PAUSE:
       // It is possible for the download to complete before the user clicks the
       // menu item, recheck if the download is in progress state before toggling
       // pause.
-      if (download_->state() == DownloadItem::IN_PROGRESS)
+      if (download_->IsPartialDownload())
         download_->TogglePause();
       break;
     default:
       NOTREACHED();
   }
 }
 
 bool DownloadShelfContextMenu::GetAcceleratorForCommandId(
     int command_id, menus::Accelerator* accelerator) {
   return false;
@@ skipping 38 matching lines @@
       ALWAYS_OPEN_TYPE, IDS_DOWNLOAD_MENU_ALWAYS_OPEN_TYPE);
   finished_download_menu_model_->AddSeparator();
   finished_download_menu_model_->AddItemWithStringId(
       SHOW_IN_FOLDER, IDS_DOWNLOAD_MENU_SHOW);
   finished_download_menu_model_->AddSeparator();
   finished_download_menu_model_->AddItemWithStringId(
       CANCEL, IDS_DOWNLOAD_MENU_CANCEL);
 
   return finished_download_menu_model_.get();
 }