chrome/browser/android/download/download_manager_bridge.cc

// Copyright 2019 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/download/download_manager_bridge.h"

#include <memory>

#include "base/android/jni_android.h"
#include "base/android/jni_string.h"
#include "base/android/scoped_java_ref.h"
#include "base/feature_list.h"
#include "components/download/public/common/download_features.h"
#include "content/public/browser/browser_thread.h"
#include "jni/DownloadManagerBridge_jni.h"
#include "url/gurl.h"

using base::android::ConvertUTF8ToJavaString;
using base::android::JavaParamRef;
using base::android::ScopedJavaLocalRef;
using content::BrowserThread;
using download::DownloadItem;

static void JNI_DownloadManagerBridge_OnAddCompletedDownloadDone(
    JNIEnv* env,
    jlong callback_id,
    jlong download_id,
    jboolean can_resolve) {
  DCHECK_CURRENTLY_ON(BrowserThread::UI);
  DCHECK(callback_id);

  // Convert java long long int to c++ pointer, take ownership.
  std::unique_ptr<AddCompletedDownloadCallback> cb(
      reinterpret_cast<AddCompletedDownloadCallback*>(callback_id));
  std::move(*cb).Run(download_id, can_resolve);
}

void DownloadManagerBridge::AddCompletedDownload(
    download::DownloadItem* download,
    AddCompletedDownloadCallback callback) {
  if (!base::FeatureList::IsEnabled(
          download::features::kUseDownloadOfflineContentProvider)) {
    return;
  }

  JNIEnv* env = base::android::AttachCurrentThread();
  ScopedJavaLocalRef<jstring> jfile_name =
      ConvertUTF8ToJavaString(env, download->GetFileNameToReportUser().value());
  ScopedJavaLocalRef<jstring> jmime_type =
      ConvertUTF8ToJavaString(env, download->GetMimeType());
  ScopedJavaLocalRef<jstring> jfile_path =
      ConvertUTF8ToJavaString(env, download->GetTargetFilePath().value());
  int64_t file_size = download->GetReceivedBytes();
  ScopedJavaLocalRef<jstring> joriginal_url =
      ConvertUTF8ToJavaString(env, download->GetOriginalUrl().spec());
  ScopedJavaLocalRef<jstring> jreferer = base::android::ConvertUTF8ToJavaString(
      env, download->GetReferrerUrl().spec());
  ScopedJavaLocalRef<jstring> jdownload_guid =
      base::android::ConvertUTF8ToJavaString(env, download->GetGuid());

  // Make copy on the heap so we can pass the pointer through JNI.
  intptr_t callback_id = reinterpret_cast<intptr_t>(
      new AddCompletedDownloadCallback(std::move(callback)));

  Java_DownloadManagerBridge_addCompletedDownload(
      env, jfile_name, jfile_name, jmime_type, jfile_path, file_size,
      joriginal_url, jreferer, jdownload_guid, callback_id);
}

void DownloadManagerBridge::RemoveCompletedDownload(
    download::DownloadItem* download) {
  if (!base::FeatureList::IsEnabled(
          download::features::kUseDownloadOfflineContentProvider)) {
    return;
  }

  JNIEnv* env = base::android::AttachCurrentThread();
  ScopedJavaLocalRef<jstring> jdownload_guid =
      base::android::ConvertUTF8ToJavaString(env, download->GetGuid());
  Java_DownloadManagerBridge_removeCompletedDownload(
      env, jdownload_guid, download->GetFileExternallyRemoved());
}