// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/android/callback_android.h"
#include "base/functional/bind.h"
#include "chrome/browser/download/android/download_manager_service.h"
#include "chrome/browser/download/background_download_service_factory.h"
#include "chrome/browser/download/offline_item_utils.h"
#include "chrome/browser/profiles/profile_key.h"
#include "chrome/browser/profiles/profile_key_android.h"
#include "components/download/public/background_service/background_download_service.h"
#include "components/download/public/common/android/auto_resumption_handler.h"
#include "components/offline_items_collection/core/android/offline_item_bridge.h"
#include "content/public/browser/browser_context.h"

// Must come after all headers that specialize FromJniType() / ToJniType().
#include "chrome/android/chrome_jni_headers/DownloadBackgroundTaskCallback_jni.h"
#include "chrome/android/chrome_jni_headers/DownloadBackgroundTask_jni.h"

using base::android::JavaRef;

namespace download {
namespace android {

BackgroundDownloadService* GetDownloadService(const JavaRef<jobject>& jkey) {
  ProfileKey* key = ProfileKeyAndroid::FromProfileKeyAndroid(jkey);
  DCHECK(key);
  return BackgroundDownloadServiceFactory::GetForKey(key);
}

AutoResumptionHandler* GetAutoResumptionHandler() {
  if (!AutoResumptionHandler::Get())
    DownloadManagerService::CreateAutoResumptionHandler();
  return AutoResumptionHandler::Get();
}

void RunTaskFinishedCallback(
    const base::android::ScopedJavaGlobalRef<jobject>& j_callback_wrapper,
    bool should_reschedule) {
  JNIEnv* env = base::android::AttachCurrentThread();
  Java_DownloadBackgroundTaskCallback_finishTask(env, j_callback_wrapper,
                                                 should_reschedule);
}

void RunTaskNotifyCallback(
    const base::android::ScopedJavaGlobalRef<jobject>& j_callback_wrapper,
    DownloadItem* download) {
  JNIEnv* env = base::android::AttachCurrentThread();
  std::string name_space =
      OfflineItemUtils::GetDownloadNamespacePrefix(/*is_off_the_record=*/false);
  offline_items_collection::OfflineItem item =
      OfflineItemUtils::CreateOfflineItem(name_space, download);
  Java_DownloadBackgroundTaskCallback_notify(
      env, j_callback_wrapper,
      offline_items_collection::android::OfflineItemBridge::CreateOfflineItem(
          env, item));
}

// static
static void JNI_DownloadBackgroundTask_StartBackgroundTask(
    JNIEnv* env,
    const JavaRef<jobject>& jkey,
    int32_t task_type,
    const JavaRef<jobject>& j_callback_wrapper) {
  base::android::ScopedJavaGlobalRef<jobject> j_callback =
      base::android::ScopedJavaGlobalRef<jobject>(j_callback_wrapper);
  TaskFinishedCallback finish_callback =
      base::BindOnce(&RunTaskFinishedCallback, j_callback);

  auto type = static_cast<DownloadTaskType>(task_type);
  switch (type) {
    case DownloadTaskType::DOWNLOAD_AUTO_RESUMPTION_TASK:
    case DownloadTaskType::DOWNLOAD_AUTO_RESUMPTION_UNMETERED_TASK:
    case DownloadTaskType::DOWNLOAD_AUTO_RESUMPTION_ANY_NETWORK_TASK:
    case DownloadTaskType::DOWNLOAD_LATER_TASK:
      GetAutoResumptionHandler()->OnStartScheduledTask(
          type, std::move(finish_callback),
          base::BindOnce(&RunTaskNotifyCallback, j_callback));
      break;
    case DownloadTaskType::DOWNLOAD_TASK:
    case DownloadTaskType::CLEANUP_TASK:
      GetDownloadService(jkey)->OnStartScheduledTask(
          type, std::move(finish_callback));
      break;
  }
}

// static
static bool JNI_DownloadBackgroundTask_StopBackgroundTask(
    JNIEnv* env,
    const JavaRef<jobject>& jkey,
    int32_t task_type) {
  auto type = static_cast<DownloadTaskType>(task_type);
  switch (type) {
    case DownloadTaskType::DOWNLOAD_AUTO_RESUMPTION_TASK:
    case DownloadTaskType::DOWNLOAD_LATER_TASK:
    case DownloadTaskType::DOWNLOAD_AUTO_RESUMPTION_UNMETERED_TASK:
    case DownloadTaskType::DOWNLOAD_AUTO_RESUMPTION_ANY_NETWORK_TASK:
      GetAutoResumptionHandler()->OnStopScheduledTask(type);
      break;
    case DownloadTaskType::DOWNLOAD_TASK:
    case DownloadTaskType::CLEANUP_TASK:
      return GetDownloadService(jkey)->OnStopScheduledTask(type);
  }
  return false;
}

}  // namespace android
}  // namespace download

DEFINE_JNI(DownloadBackgroundTaskCallback)
DEFINE_JNI(DownloadBackgroundTask)