//
// Created by Marc Rousavy on 21.02.24.
//

#include "RNFJFilamentRecorder.h"
#include "RNFJNISharedPtr.h"
#include "java-bindings/RNFJDispatcher.h"
#include <android/native_window_jni.h>

namespace margelo {

JFilamentRecorder::JFilamentRecorder(const jni::alias_ref<jhybridobject>& javaThis, std::shared_ptr<Dispatcher> rendererDispatcher,
                                     int width, int height, int fps, double bitRate)
    : FilamentRecorder(rendererDispatcher, width, height, fps, bitRate), _javaPart(make_global(javaThis)) {}

JFilamentRecorder::~JFilamentRecorder() {
  __android_log_write(ANDROID_LOG_INFO, TAG, "Destroying JFilamentRecorder...");
  if (_nativeWindow != nullptr) {
    ANativeWindow_release(_nativeWindow);
  }
}

void JFilamentRecorder::registerNatives() {
  registerHybrid({
      makeNativeMethod("initHybrid", JFilamentRecorder::initHybrid),
      makeNativeMethod("onReadyForMoreData", JFilamentRecorder::onReadyForMoreDataJava),
  });
}

void JFilamentRecorder::renderFrame(double timestamp) {
  // TODO: Do anything here??
}

jni::local_ref<JDispatcher::javaobject> JFilamentRecorder::getRecorderDispatcher() {
  static const auto method = javaClassLocal()->getMethod<JDispatcher::javaobject()>("getRecorderDispatcher");
  return method(_javaPart);
}

void* JFilamentRecorder::getNativeWindow() {
  if (_nativeWindow == nullptr) {
    static const auto method = javaClassLocal()->getMethod<jobject()>("getNativeWindow");
    jni::local_ref<jobject> nativeWindow = method(_javaPart);
    _nativeWindow = ANativeWindow_fromSurface(jni::Environment::current(), nativeWindow.get());
  }
  return _nativeWindow;
}

std::future<void> JFilamentRecorder::startRecording() {
  auto self = shared<JFilamentRecorder>();
  auto dispatcher = getRecorderDispatcher();
  return dispatcher->cthis()->runAsyncAwaitable<void>([self]() {
    static const auto method = self->javaClassLocal()->getMethod<void()>("startRecording");
    method(self->_javaPart);
  });
}

std::future<std::string> JFilamentRecorder::stopRecording() {
  auto self = shared<JFilamentRecorder>();
  auto dispatcher = getRecorderDispatcher();
  return dispatcher->cthis()->runAsyncAwaitable<std::string>([self]() {
    static const auto method = self->javaClassLocal()->getMethod<void()>("stopRecording");
    method(self->_javaPart);
    return self->getOutputFile();
  });
}

std::string JFilamentRecorder::getOutputFile() {
  static const auto method = javaClassLocal()->getMethod<std::string()>("getOutputFile");
  jni::local_ref<jstring> path = method(_javaPart);
  return path->toString();
}

bool JFilamentRecorder::getIsRecording() {
  static const auto method = javaClassLocal()->getMethod<jboolean()>("getIsRecording");
  return method(_javaPart);
}

jni::local_ref<JFilamentRecorder::jhybriddata> JFilamentRecorder::initHybrid(jni::alias_ref<JFilamentRecorder::jhybridobject> jThis,
                                                                             jni::alias_ref<JDispatcher::javaobject> rendererDispatcher,
                                                                             int width, int height, int fps, double bitRate) {
  __android_log_write(ANDROID_LOG_INFO, TAG, "Initializing JFilamentRecorder...");
  jni::global_ref<JDispatcher::javaobject> globalRef = jni::make_global(rendererDispatcher);
  std::shared_ptr<Dispatcher> dispatcher = JNISharedPtr::make_shared_from_jni<JDispatcher>(globalRef);
  return makeCxxInstance(jThis, dispatcher, width, height, fps, bitRate);
}

} // namespace margelo