Remove DistilledPageInfo and related structs.

components/dom_distiller/core/distiller.cc

 // Copyright 2013 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 "components/dom_distiller/core/distiller.h"
 
 #include <map>
 #include <vector>
 
 #include "base/auto_reset.h"
@@ skipping 117 matching lines @@
         base::Bind(&DistillerImpl::OnPageDistillationFinished,
                    weak_factory_.GetWeakPtr(),
                    page_num,
                    url));
   }
 }
 
 void DistillerImpl::OnPageDistillationFinished(
     int page_num,
     const GURL& page_url,
-    scoped_ptr<DistilledPageInfo> distilled_page,
+    scoped_ptr<proto::DomDistillerResult> distiller_result,
     bool distillation_successful) {
-  DCHECK(distilled_page.get());
+  DCHECK(distiller_result.get());
   DCHECK(started_pages_index_.find(page_num) != started_pages_index_.end());
   if (distillation_successful) {
     DistilledPageData* page_data =
         GetPageAtIndex(started_pages_index_[page_num]);
     page_data->distilled_page_proto =
         new base::RefCountedData<DistilledPageProto>();
     page_data->page_num = page_num;
-    page_data->distilled_page_proto->data.set_title(distilled_page->title);
+    if (distiller_result->has_title()) {
+      page_data->distilled_page_proto->data.set_title(
+          distiller_result->title());
+    }
     page_data->distilled_page_proto->data.set_url(page_url.spec());
-    page_data->distilled_page_proto->data.set_html(distilled_page->html);
-
-    GURL next_page_url(distilled_page->next_page_url);
-    if (next_page_url.is_valid()) {
-      // The pages should be in same origin.
-      DCHECK_EQ(next_page_url.GetOrigin(), page_url.GetOrigin());
-      AddToDistillationQueue(page_num + 1, next_page_url);
+    if (distiller_result->has_distilled_content() &&
+        distiller_result->distilled_content().has_html()) {
+      page_data->distilled_page_proto->data.set_html(
+          distiller_result->distilled_content().html());
     }
 
-    GURL prev_page_url(distilled_page->prev_page_url);
-    if (prev_page_url.is_valid()) {
-      DCHECK_EQ(prev_page_url.GetOrigin(), page_url.GetOrigin());
-      AddToDistillationQueue(page_num - 1, prev_page_url);
+    if (distiller_result->has_pagination_info()) {
+      proto::PaginationInfo pagination_info =
+          distiller_result->pagination_info();
+      if (pagination_info.has_next_page()) {
+        GURL next_page_url(pagination_info.next_page());
+        if (next_page_url.is_valid()) {
+          // The pages should be in same origin.
+          DCHECK_EQ(next_page_url.GetOrigin(), page_url.GetOrigin());
+          AddToDistillationQueue(page_num + 1, next_page_url);
+        }
+      }
+
+      if (pagination_info.has_prev_page()) {
+        GURL prev_page_url(pagination_info.prev_page());
+        if (prev_page_url.is_valid()) {
+          DCHECK_EQ(prev_page_url.GetOrigin(), page_url.GetOrigin());
+          AddToDistillationQueue(page_num - 1, prev_page_url);
+        }
+      }
     }
 
-    for (size_t img_num = 0; img_num < distilled_page->image_urls.size();
+    for (int img_num = 0; img_num < distiller_result->image_urls_size();
          ++img_num) {
       std::string image_id =
           base::IntToString(page_num + 1) + "_" + base::IntToString(img_num);
-      FetchImage(page_num, image_id, distilled_page->image_urls[img_num]);
+      FetchImage(page_num, image_id, distiller_result->image_urls(img_num));
     }
 
     AddPageIfDone(page_num);
     DistillNextPage();
   } else {
     started_pages_index_.erase(page_num);
     RunDistillerCallbackIfDone();
   }
 }
 
@@ skipping 105 matching lines @@
     DCHECK(finished_pages_index_.empty());
 
     base::AutoReset<bool> dont_delete_this_in_callback(&destruction_allowed_,
                                                        false);
     finished_cb_.Run(article_proto.Pass());
     finished_cb_.Reset();
   }
 }
 
 }  // namespace dom_distiller