// Copyright (c) 2014 The Chromium Embedded Framework 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 "resource_handler.h"

#include "jni_util.h"
#include "util.h"

ResourceHandler::ResourceHandler(JNIEnv* env, jobject handler)
    : handle_(env, handler) {}

bool ResourceHandler::ProcessRequest(CefRefPtr<CefRequest> request,
                                     CefRefPtr<CefCallback> callback) {
  ScopedJNIEnv env;
  if (!env)
    return false;

  ScopedJNIRequest jrequest(env, request);
  jrequest.SetTemporary();
  ScopedJNICallback jcallback(env, callback);
  jboolean jresult = JNI_FALSE;

  JNI_CALL_METHOD(
      env, handle_, "processRequest",
      "(Lorg/cef/network/CefRequest;Lorg/cef/callback/CefCallback;)Z", Boolean,
      jresult, jrequest.get(), jcallback.get());

  if (jresult == JNI_FALSE) {
    // If the Java method returns "false" the callback won't be used and
    // the reference can therefore be removed.
    jcallback.SetTemporary();
  }

  return (jresult != JNI_FALSE);
}

bool ResourceHandler::Open(CefRefPtr<CefRequest> request,
                           bool& handle_request,
                           CefRefPtr<CefCallback> callback) {
  ScopedJNIEnv env;
  if (!env)
    return false;

  ScopedJNIRequest jrequest(env, request);
  jrequest.SetTemporary();
  ScopedJNIBoolRef jhandleRequest(env, handle_request);
  ScopedJNICallback jcallback(env, callback);
  jboolean jresult = JNI_FALSE;

  JNI_CALL_METHOD(env, handle_, "open",
                  "(Lorg/cef/network/CefRequest;Lorg/cef/misc/BoolRef;Lorg/cef/"
                  "callback/CefCallback;)Z",
                  Boolean, jresult, jrequest.get(), jhandleRequest.get(),
                  jcallback.get());

  if (jresult == JNI_FALSE) {
    // If the Java method returns "false" the callback won't be used and
    // the reference can therefore be removed.
    jcallback.SetTemporary();
  }

  handle_request = jhandleRequest;

  return (jresult != JNI_FALSE);
}

void ResourceHandler::GetResponseHeaders(CefRefPtr<CefResponse> response,
                                         int64_t& response_length,
                                         CefString& redirectUrl) {
  ScopedJNIEnv env;
  if (!env)
    return;

  ScopedJNIResponse jresponse(env, response);
  jresponse.SetTemporary();
  ScopedJNIIntRef jresponseLength(env, (int)response_length);
  ScopedJNIStringRef jredirectUrl(env, redirectUrl);

  JNI_CALL_VOID_METHOD(env, handle_, "getResponseHeaders",
                       "(Lorg/cef/network/CefResponse;Lorg/cef/misc/"
                       "IntRef;Lorg/cef/misc/StringRef;)V",
                       jresponse.get(), jresponseLength.get(),
                       jredirectUrl.get());

  response_length = jresponseLength;
  redirectUrl = jredirectUrl;
}

bool ResourceHandler::ReadResponse(void* data_out,
                                   int bytes_to_read,
                                   int& bytes_read,
                                   CefRefPtr<CefCallback> callback) {
  ScopedJNIEnv env;
  if (!env)
    return false;

  ScopedJNIIntRef jbytesRead(env, bytes_read);
  jbyteArray jbytes = env->NewByteArray(bytes_to_read);
  ScopedJNICallback jcallback(env, callback);
  jboolean jresult = JNI_FALSE;

  JNI_CALL_METHOD(env, handle_, "readResponse",
                  "([BILorg/cef/misc/IntRef;Lorg/cef/callback/CefCallback;)Z",
                  Boolean, jresult, jbytes, bytes_to_read, jbytesRead.get(),
                  jcallback.get());

  bytes_read = jbytesRead;

  bool result = (jresult != JNI_FALSE);
  if (!result || bytes_read > 0) {
    // The callback won't be used and the reference can therefore be removed.
    jcallback.SetTemporary();
  }

  jbyte* jbyte = env->GetByteArrayElements(jbytes, nullptr);
  if (jbyte) {
    memmove(data_out, jbyte,
            (bytes_read < bytes_to_read ? bytes_read : bytes_to_read));
    env->ReleaseByteArrayElements(jbytes, jbyte, JNI_ABORT);
  }
  env->DeleteLocalRef(jbytes);

  return result;
}

bool ResourceHandler::Read(void* data_out,
                           int bytes_to_read,
                           int& bytes_read,
                           CefRefPtr<CefResourceReadCallback> callback) {
  ScopedJNIEnv env;
  if (!env)
    return false;

  ScopedJNIIntRef jbytesRead(env, bytes_read);
  jbyteArray jbytes = env->NewByteArray(bytes_to_read);
  ScopedJNIResourceReadCallback jcallback(env, callback);

  // This callback must retain a reference to the data_out buffer
  JNI_CALL_VOID_METHOD(env, jcallback.get(), "setBufferRefs", "(J[B)V",
                       (jlong)data_out, jbytes);

  jboolean jresult = JNI_FALSE;

  JNI_CALL_METHOD(
      env, handle_, "read",
      "([BILorg/cef/misc/IntRef;Lorg/cef/callback/CefResourceReadCallback;)Z",
      Boolean, jresult, jbytes, bytes_to_read, jbytesRead.get(),
      jcallback.get());

  bytes_read = jbytesRead;

  bool result = (jresult != JNI_FALSE);
  if (!result || bytes_read > 0) {
    // The callback won't be used and the reference can therefore be removed.
    jcallback.SetTemporary();
  }

  if (bytes_read > 0) {
    jbyte* jbyte = env->GetByteArrayElements(jbytes, nullptr);
    if (jbyte) {
      memmove(data_out, jbyte,
              (bytes_read < bytes_to_read ? bytes_read : bytes_to_read));
      env->ReleaseByteArrayElements(jbytes, jbyte, JNI_ABORT);
    }
  }
  env->DeleteLocalRef(jbytes);

  return result;
}

bool ResourceHandler::Skip(int64_t bytes_to_skip,
                           int64_t& bytes_skipped,
                           CefRefPtr<CefResourceSkipCallback> callback) {
  ScopedJNIEnv env;
  if (!env)
    return false;

  ScopedJNILongRef jbytesSkipped(env, bytes_skipped);
  ScopedJNIResourceSkipCallback jcallback(env, callback);
  jboolean jresult = JNI_FALSE;

  JNI_CALL_METHOD(
      env, handle_, "skip",
      "(JLorg/cef/misc/LongRef;Lorg/cef/callback/CefResourceSkipCallback;)Z",
      Boolean, jresult, bytes_to_skip, jbytesSkipped.get(), jcallback.get());

  bytes_skipped = jbytesSkipped;

  bool result = (jresult != JNI_FALSE);
  if (!result || bytes_skipped > 0) {
    // The callback won't be used and the reference can therefore be removed.
    jcallback.SetTemporary();
  }

  return result;
}

void ResourceHandler::Cancel() {
  ScopedJNIEnv env;
  if (!env)
    return;
  JNI_CALL_VOID_METHOD(env, handle_, "cancel", "()V");
}