// Copyright 2019-2023 Tauri Programme within The Commons Conservancy
// SPDX-License-Identifier: Apache-2.0
// SPDX-License-Identifier: MIT

use crate::Runtime;
use jni::{
  errors::Error as JniError,
  objects::{JObject, JValueOwned},
  JNIEnv,
};
use serde_json::Value as JsonValue;
use tauri_runtime::RuntimeHandle;

fn json_to_java<'a, R: Runtime>(
  env: &mut JNIEnv<'a>,
  activity: &JObject<'_>,
  runtime_handle: &R::Handle,
  json: &JsonValue,
) -> Result<(&'static str, JValueOwned<'a>), JniError> {
  let (class, v) = match json {
    JsonValue::Null => ("Ljava/lang/Object;", JObject::null().into()),
    JsonValue::Bool(val) => ("Z", (*val).into()),
    JsonValue::Number(val) => {
      if let Some(v) = val.as_i64() {
        ("J", v.into())
      } else if let Some(v) = val.as_f64() {
        ("D", v.into())
      } else {
        ("Ljava/lang/Object;", JObject::null().into())
      }
    }
    JsonValue::String(val) => (
      "Ljava/lang/Object;",
      JObject::from(env.new_string(val)?).into(),
    ),
    JsonValue::Array(val) => {
      let js_array_class = runtime_handle.find_class(env, activity, "app/tauri/plugin/JSArray")?;
      let data = env.new_object(js_array_class, "()V", &[])?;

      for v in val {
        let (signature, val) = json_to_java::<R>(env, activity, runtime_handle, v)?;
        env.call_method(
          &data,
          "put",
          format!("({signature})Lorg/json/JSONArray;"),
          &[val.borrow()],
        )?;
      }

      ("Ljava/lang/Object;", data.into())
    }
    JsonValue::Object(val) => {
      let data = {
        let js_object_class =
          runtime_handle.find_class(env, activity, "app/tauri/plugin/JSObject")?;
        env.new_object(js_object_class, "()V", &[])?
      };

      for (key, value) in val {
        let (signature, val) = json_to_java::<R>(env, activity, runtime_handle, value)?;
        let key = env.new_string(key)?;
        env.call_method(
          &data,
          "put",
          format!("(Ljava/lang/String;{signature})Lapp/tauri/plugin/JSObject;"),
          &[(&key).into(), val.borrow()],
        )?;
      }

      ("Ljava/lang/Object;", data.into())
    }
  };
  Ok((class, v))
}

pub fn to_jsobject<'a, R: Runtime>(
  env: &mut JNIEnv<'a>,
  activity: &JObject<'_>,
  runtime_handle: &R::Handle,
  json: &JsonValue,
) -> Result<JValueOwned<'a>, JniError> {
  if let JsonValue::Object(_) = json {
    json_to_java::<R>(env, activity, runtime_handle, json).map(|(_class, data)| data)
  } else {
    Ok(empty_object::<R>(env, activity, runtime_handle)?.into())
  }
}

fn empty_object<'a, R: Runtime>(
  env: &mut JNIEnv<'a>,
  activity: &JObject<'_>,
  runtime_handle: &R::Handle,
) -> Result<JObject<'a>, JniError> {
  // currently the Kotlin lib cannot handle nulls or raw values, it must be an object
  let js_object_class = runtime_handle.find_class(env, activity, "app/tauri/plugin/JSObject")?;
  let data = env.new_object(js_object_class, "()V", &[])?;
  Ok(data)
}