Update reading list entry on download

ios/chrome/browser/reading_list/reading_list_download_service.cc

 // Copyright 2016 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 "ios/chrome/browser/reading_list/reading_list_download_service.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "ios/chrome/browser/reading_list/reading_list_entry.h"
 #include "ios/chrome/browser/reading_list/reading_list_model.h"
-#include "ios/chrome/browser/reading_list/url_downloader.h"
 
 ReadingListDownloadService::ReadingListDownloadService(
     ReadingListModel* reading_list_model,
     dom_distiller::DomDistillerService* distiller_service,
     PrefService* prefs,
     base::FilePath chrome_profile_path)
     : reading_list_model_(reading_list_model) {
   DCHECK(reading_list_model);
   url_downloader_ = std::unique_ptr<URLDownloader>(
       new URLDownloader(distiller_service, prefs, chrome_profile_path,
@@ skipping 56 matching lines @@
   }
   size = reading_list_model_->read_size();
   for (size_t i = 0; i < size; i++) {
     ReadingListEntry entry = reading_list_model_->GetReadEntryAtIndex(i);
     this->DownloadEntry(entry);
   }
 }
 
 void ReadingListDownloadService::DownloadEntry(const ReadingListEntry& entry) {
   DCHECK(reading_list_model_->loaded());
-  url_downloader_->DownloadOfflineURL(entry.URL());
+  if (entry.DistilledState() != ReadingListEntry::ERROR) {
+    reading_list_model_->SetEntryDistilledState(entry.URL(),
+                                                ReadingListEntry::PROCESSING);
+    url_downloader_->DownloadOfflineURL(entry.URL());
+  }
 }
 
 void ReadingListDownloadService::RemoveDownloadedEntry(
     const ReadingListEntry& entry) {
   DCHECK(reading_list_model_->loaded());
   url_downloader_->RemoveOfflineURL(entry.URL());
 }
 
-void ReadingListDownloadService::OnDownloadEnd(const GURL& url, bool success) {
-  // TODO(crbug.com/616747) update entry with offline info
+void ReadingListDownloadService::OnDownloadEnd(
+    const GURL& url,
+    URLDownloader::SuccessState success,
+    const GURL& distilledURL,

[sdefresne, 2016/09/09 09:58:57]

style: s/distilledURL/distilled_url/

[lody, 2016/09/09 12:17:53]

Done.

+    const std::string& title) {
+  DCHECK(reading_list_model_->loaded());
+  if ((success == URLDownloader::DOWNLOAD_SUCCESS ||
+       success == URLDownloader::DOWNLOAD_EXISTS) &&
+      distilledURL.is_valid()) {
+    reading_list_model_->SetEntryDistilledURL(url, distilledURL);
+  } else if (success == URLDownloader::ERROR_RETRY) {
+    reading_list_model_->SetEntryDistilledState(url,
+                                                ReadingListEntry::WILL_RETRY);
+  } else if (success == URLDownloader::ERROR_PERMANENT) {
+    reading_list_model_->SetEntryDistilledState(url, ReadingListEntry::ERROR);
+  }
 }
 
 void ReadingListDownloadService::OnDeleteEnd(const GURL& url, bool success) {
-  // TODO(crbug.com/616747) update entry with offline info
+  // Nothing to update as this is only called when deleting reading list entries
 }