// Copyright 2014 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/logo_bridge.h"

#include <jni.h>
#include <stdint.h>

#include "base/android/jni_android.h"
#include "base/android/jni_array.h"
#include "base/android/jni_string.h"
#include "base/android/scoped_java_ref.h"
#include "base/metrics/histogram_macros.h"
#include "chrome/browser/android/logo_service.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_android.h"
#include "components/search_provider_logos/logo_tracker.h"
#include "jni/LogoBridge_jni.h"
#include "net/url_request/url_fetcher.h"
#include "net/url_request/url_request_context_getter.h"
#include "net/url_request/url_request_status.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "ui/gfx/android/java_bitmap.h"

using base::android::ConvertJavaStringToUTF8;
using base::android::ConvertUTF8ToJavaString;
using base::android::ScopedJavaLocalRef;
using base::android::ToJavaByteArray;

namespace {

// Converts a C++ Logo to a Java Logo.
ScopedJavaLocalRef<jobject> ConvertLogoToJavaObject(
    JNIEnv* env,
    const search_provider_logos::Logo* logo) {
  if (!logo)
    return ScopedJavaLocalRef<jobject>();

  ScopedJavaLocalRef<jobject> j_bitmap = gfx::ConvertToJavaBitmap(&logo->image);

  ScopedJavaLocalRef<jstring> j_on_click_url;
  if (!logo->metadata.on_click_url.empty())
    j_on_click_url = ConvertUTF8ToJavaString(env, logo->metadata.on_click_url);

  ScopedJavaLocalRef<jstring> j_alt_text;
  if (!logo->metadata.alt_text.empty())
    j_alt_text = ConvertUTF8ToJavaString(env, logo->metadata.alt_text);

  ScopedJavaLocalRef<jstring> j_animated_url;
  if (!logo->metadata.animated_url.empty())
    j_animated_url = ConvertUTF8ToJavaString(env, logo->metadata.animated_url);

  return Java_LogoBridge_createLogo(env, j_bitmap.obj(), j_on_click_url.obj(),
                                    j_alt_text.obj(), j_animated_url.obj());
}

class LogoObserverAndroid : public search_provider_logos::LogoObserver {
 public:
  LogoObserverAndroid(base::WeakPtr<LogoBridge> logo_bridge,
                      JNIEnv* env,
                      jobject j_logo_observer)
      : logo_bridge_(logo_bridge) {
    j_logo_observer_.Reset(env, j_logo_observer);
  }

  ~LogoObserverAndroid() override {}

  // seach_provider_logos::LogoObserver:
  void OnLogoAvailable(const search_provider_logos::Logo* logo,
                       bool from_cache) override {
    if (!logo_bridge_)
      return;

    JNIEnv* env = base::android::AttachCurrentThread();
    ScopedJavaLocalRef<jobject> j_logo = ConvertLogoToJavaObject(env, logo);
    Java_LogoObserver_onLogoAvailable(
        env, j_logo_observer_.obj(), j_logo.obj(), from_cache);
  }

  void OnObserverRemoved() override { delete this; }

 private:
  // The associated LogoBridge. We won't call back to Java if the LogoBridge has
  // been destroyed.
  base::WeakPtr<LogoBridge> logo_bridge_;

  base::android::ScopedJavaGlobalRef<jobject> j_logo_observer_;
};

}  // namespace

static jlong Init(JNIEnv* env,
                  const JavaParamRef<jobject>& obj,
                  const JavaParamRef<jobject>& j_profile) {
  LogoBridge* logo_bridge = new LogoBridge(j_profile);
  return reinterpret_cast<intptr_t>(logo_bridge);
}

LogoBridge::LogoBridge(jobject j_profile) : weak_ptr_factory_(this) {
  Profile* profile = ProfileAndroid::FromProfileAndroid(j_profile);
  if (profile) {
    logo_service_ = LogoServiceFactory::GetForProfile(profile);
    request_context_getter_ = profile->GetRequestContext();
  }
}

LogoBridge::~LogoBridge() {
  ClearFetcher();
}

void LogoBridge::Destroy(JNIEnv* env, const JavaParamRef<jobject>& obj) {
  delete this;
}

void LogoBridge::GetCurrentLogo(JNIEnv* env,
                                const JavaParamRef<jobject>& obj,
                                const JavaParamRef<jobject>& j_logo_observer) {
  if (!logo_service_)
    return;

  // |observer| is deleted in LogoObserverAndroid::OnObserverRemoved().
  LogoObserverAndroid* observer = new LogoObserverAndroid(
      weak_ptr_factory_.GetWeakPtr(), env, j_logo_observer);
  logo_service_->GetLogo(observer);
}

void LogoBridge::GetAnimatedLogo(JNIEnv* env,
                                 const JavaParamRef<jobject>& obj,
                                 const JavaParamRef<jobject>& j_callback,
                                 const JavaParamRef<jstring>& j_url) {
  DCHECK(j_callback);
  if (!logo_service_)
    return;

  GURL url = GURL(ConvertJavaStringToUTF8(env, j_url));
  if (fetcher_ && fetcher_->GetOriginalURL() == url)
    return;

  j_callback_.Reset(env, j_callback);
  fetcher_ = net::URLFetcher::Create(url, net::URLFetcher::GET, this);
  fetcher_->SetRequestContext(request_context_getter_.get());
  fetcher_->Start();
  animated_logo_download_start_time_ = base::TimeTicks::Now();
}

void LogoBridge::OnURLFetchComplete(const net::URLFetcher* source) {
  if (!source->GetStatus().is_success() || (source->GetResponseCode() != 200) ||
      j_callback_.is_null()) {
    ClearFetcher();
    return;
  }

  UMA_HISTOGRAM_TIMES(
      "NewTabPage.AnimatedLogoDownloadTime",
      base::TimeTicks::Now() - animated_logo_download_start_time_);

  std::string response;
  source->GetResponseAsString(&response);
  JNIEnv* env = base::android::AttachCurrentThread();

  ScopedJavaLocalRef<jbyteArray> j_bytes =
      ToJavaByteArray(env, reinterpret_cast<const uint8_t*>(response.data()),
                      response.length());
  ScopedJavaLocalRef<jobject> j_gif_image =
      Java_LogoBridge_createGifImage(env, j_bytes.obj());
  Java_AnimatedLogoCallback_onAnimatedLogoAvailable(env, j_callback_.obj(),
                                                    j_gif_image.obj());
  ClearFetcher();
}

void LogoBridge::ClearFetcher() {
  fetcher_.reset();
  j_callback_.Reset();
}

// static
bool RegisterLogoBridge(JNIEnv* env) {
  return RegisterNativesImpl(env);
}