Implement OfflinePageDownloadBridge::startDownload

chrome/browser/android/offline_pages/downloads/offline_page_download_bridge.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 "chrome/browser/android/offline_pages/downloads/offline_page_download_bridge.h"
 
 #include <vector>
 
 #include "base/android/jni_string.h"
+#include "base/guid.h"
+#include "base/memory/ptr_util.h"
+#include "chrome/browser/android/offline_pages/offline_page_mhtml_archiver.h"
 #include "chrome/browser/android/offline_pages/offline_page_model_factory.h"
+#include "chrome/browser/android/tab_android.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_android.h"
+#include "components/offline_pages/client_namespace_constants.h"
 #include "components/offline_pages/downloads/download_ui_item.h"
 #include "components/offline_pages/offline_page_model.h"
+#include "content/public/browser/web_contents.h"
 #include "jni/OfflinePageDownloadBridge_jni.h"
 #include "net/base/filename_util.h"
 #include "url/gurl.h"
 
 using base::android::AttachCurrentThread;
 using base::android::ConvertJavaStringToUTF8;
 using base::android::ConvertUTF8ToJavaString;
 using base::android::JavaParamRef;
 using base::android::ScopedJavaGlobalRef;
 using base::android::ScopedJavaLocalRef;
@@ skipping 25 matching lines @@
       ConvertUTF8ToJavaString(env, item.target_path.value()).obj(),
       item.start_time.ToJavaTime(), item.total_bytes);
 }
 }  // namespace
 
 OfflinePageDownloadBridge::OfflinePageDownloadBridge(
     JNIEnv* env,
     const JavaParamRef<jobject>& obj,
     DownloadUIAdapter* download_ui_adapter)
     : weak_java_ref_(env, obj),
-      download_ui_adapter_(download_ui_adapter) {
+      download_ui_adapter_(download_ui_adapter),
+      weak_ptr_factory_(this) {
   DCHECK(download_ui_adapter_);
   download_ui_adapter_->AddObserver(this);
 }
 
 OfflinePageDownloadBridge::~OfflinePageDownloadBridge() {}
 
 // static
 bool OfflinePageDownloadBridge::Register(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
@@ skipping 36 matching lines @@
     JNIEnv* env,
     const JavaParamRef<jobject>& obj,
     const JavaParamRef<jstring>& j_guid) {
   std::string guid = ConvertJavaStringToUTF8(env, j_guid);
   GURL url = download_ui_adapter_->GetOfflineUrlByGuid(guid);
   if (!url.is_valid())
     return ScopedJavaLocalRef<jstring>();
   return ConvertUTF8ToJavaString(env, url.spec());
 }
 
+void OfflinePageDownloadBridge::StartDownload(
+    JNIEnv* env,
+    const JavaParamRef<jobject>& obj,
+    const JavaParamRef<jobject>& j_tab) {
+  TabAndroid* tab = TabAndroid::GetNativeTab(env, j_tab);
+  if (!tab)
+    return;
+
+  content::WebContents* web_contents = tab->web_contents();
+  if (!web_contents)
+    return;
+
+  GURL url = web_contents->GetLastCommittedURL();
+  auto archiver =
+      base::MakeUnique<offline_pages::OfflinePageMHTMLArchiver>(web_contents);
+
+  offline_pages::ClientId client_id;
+  client_id.name_space = offline_pages::kDownloadNamespace;
+  client_id.id = base::GenerateGUID();
+
+  offline_pages::OfflinePageModel* offline_page_model =

[fgorski, 2016/08/15 19:59:24]

nit: move the offline_page_model check up above url line.

+      OfflinePageModelFactory::GetForBrowserContext(
+      tab->GetProfile()->GetOriginalProfile());
+  if (!offline_page_model)
+    return;
+
+  offline_page_model->SavePage(
+      url, client_id, std::move(archiver),
+      base::Bind(&OfflinePageDownloadBridge::SavePageCallback,
+                 weak_ptr_factory_.GetWeakPtr()));
+}
+
 void OfflinePageDownloadBridge::ItemsLoaded() {
   JNIEnv* env = AttachCurrentThread();
   ScopedJavaLocalRef<jobject> obj = weak_java_ref_.get(env);
   if (obj.is_null())
     return;
   Java_OfflinePageDownloadBridge_downloadItemsLoaded(env, obj.obj());
 }
 
 void OfflinePageDownloadBridge::ItemAdded(const DownloadUIItem& item) {
   JNIEnv* env = AttachCurrentThread();
@@ skipping 15 matching lines @@
 
 void OfflinePageDownloadBridge::ItemUpdated(const DownloadUIItem& item) {
   JNIEnv* env = AttachCurrentThread();
   ScopedJavaLocalRef<jobject> obj = weak_java_ref_.get(env);
   if (obj.is_null())
     return;
   Java_OfflinePageDownloadBridge_downloadItemUpdated(
       env, obj.obj(), ToJavaOfflinePageDownloadItem(env, item).obj());
 }
 
+void OfflinePageDownloadBridge::SavePageCallback(
+    OfflinePageModel::SavePageResult result, int64_t offline_id) {
+  // TODO(dimich): Consider adding UMA here.
+}
+
 static jlong Init(JNIEnv* env,
                   const JavaParamRef<jobject>& obj,
                   const JavaParamRef<jobject>& j_profile) {
   Profile* profile = ProfileAndroid::FromProfileAndroid(j_profile);
   OfflinePageModel* offline_page_model =
       OfflinePageModelFactory::GetForBrowserContext(profile);
 
   DownloadUIAdapter* adapter =
       DownloadUIAdapter::FromOfflinePageModel(offline_page_model);
 
   return reinterpret_cast<jlong>(
       new OfflinePageDownloadBridge(env, obj, adapter));
 }
 
 }  // namespace android
 }  // namespace offline_pages