#ifndef SESSION_ANDROID_USER_GROUPS_H
#define SESSION_ANDROID_USER_GROUPS_H

#include "jni.h"
#include "util.h"
#include "conversation.h"
#include "session/config/user_groups.hpp"

inline session::config::UserGroups* ptrToUserGroups(JNIEnv *env, jobject obj) {
    jclass configClass = env->FindClass("network/loki/messenger/libsession_util/UserGroupsConfig");
    jfieldID pointerField = env->GetFieldID(configClass, "pointer", "J");
    return (session::config::UserGroups*) env->GetLongField(obj, pointerField);
}

inline void deserialize_members_into(JNIEnv *env, jobject members_map, session::config::legacy_group_info *to_append_group) {
    jclass map_class = env->FindClass("java/util/Map");
    jclass map_entry_class = env->FindClass("java/util/Map$Entry");
    jclass set_class = env->FindClass("java/util/Set");
    jclass iterator_class = env->FindClass("java/util/Iterator");
    jclass boxed_bool = env->FindClass("java/lang/Boolean");

    jmethodID get_entry_set = env->GetMethodID(map_class, "entrySet", "()Ljava/util/Set;");
    jmethodID get_at = env->GetMethodID(set_class, "iterator", "()Ljava/util/Iterator;");
    jmethodID has_next = env->GetMethodID(iterator_class, "hasNext", "()Z");
    jmethodID next = env->GetMethodID(iterator_class, "next", "()Ljava/lang/Object;");
    jmethodID get_key = env->GetMethodID(map_entry_class, "getKey", "()Ljava/lang/Object;");
    jmethodID get_value = env->GetMethodID(map_entry_class, "getValue", "()Ljava/lang/Object;");
    jmethodID get_bool_value = env->GetMethodID(boxed_bool, "booleanValue", "()Z");

    jobject entry_set = env->CallObjectMethod(members_map, get_entry_set);
    jobject iterator = env->CallObjectMethod(entry_set, get_at);

    while (env->CallBooleanMethod(iterator, has_next)) {
        jobject entry = env->CallObjectMethod(iterator, next);
        jstring key = static_cast<jstring>(env->CallObjectMethod(entry, get_key));
        jobject boxed = env->CallObjectMethod(entry, get_value);
        bool is_admin = env->CallBooleanMethod(boxed, get_bool_value);
        auto member_string = env->GetStringUTFChars(key, nullptr);
        to_append_group->insert(member_string, is_admin);
        env->ReleaseStringUTFChars(key, member_string);
    }
}

inline session::config::legacy_group_info* deserialize_legacy_group_info(JNIEnv *env, jobject info) {
    auto clazz = env->FindClass("network/loki/messenger/libsession_util/util/GroupInfo$LegacyGroupInfo");
    auto id_field = env->GetFieldID(clazz, "sessionId", "Ljava/lang/String;");
    auto name_field = env->GetFieldID(clazz, "name", "Ljava/lang/String;");
    auto members_field = env->GetFieldID(clazz, "members", "Ljava/util/Map;");
    auto hidden_field = env->GetFieldID(clazz, "hidden", "Z");
    auto enc_pub_key_field = env->GetFieldID(clazz, "encPubKey", "[B");
    auto enc_sec_key_field = env->GetFieldID(clazz, "encSecKey", "[B");
    auto priority_field = env->GetFieldID(clazz, "priority", "I");
    jstring id = static_cast<jstring>(env->GetObjectField(info, id_field));
    jstring name = static_cast<jstring>(env->GetObjectField(info, name_field));
    jobject members_map = env->GetObjectField(info, members_field);
    bool hidden = env->GetBooleanField(info, hidden_field);
    jbyteArray enc_pub_key = static_cast<jbyteArray>(env->GetObjectField(info, enc_pub_key_field));
    jbyteArray enc_sec_key = static_cast<jbyteArray>(env->GetObjectField(info, enc_sec_key_field));
    int priority = env->GetIntField(info, priority_field);

    auto id_bytes = env->GetStringUTFChars(id, nullptr);
    auto name_bytes = env->GetStringUTFChars(name, nullptr);
    auto enc_pub_key_bytes = util::ustring_from_bytes(env, enc_pub_key);
    auto enc_sec_key_bytes = util::ustring_from_bytes(env, enc_sec_key);

    auto info_deserialized = new session::config::legacy_group_info(id_bytes);

    info_deserialized->priority = priority;
    deserialize_members_into(env, members_map, info_deserialized);
    info_deserialized->name = name_bytes;
    info_deserialized->hidden = hidden;
    info_deserialized->enc_pubkey = enc_pub_key_bytes;
    info_deserialized->enc_seckey = enc_sec_key_bytes;
    env->ReleaseStringUTFChars(id, id_bytes);
    env->ReleaseStringUTFChars(name, name_bytes);
    return info_deserialized;
}

inline jobject serialize_members(JNIEnv *env, std::map<std::string, bool> members_map) {
    jclass map_class = env->FindClass("java/util/HashMap");
    jclass boxed_bool = env->FindClass("java/lang/Boolean");
    jmethodID map_constructor = env->GetMethodID(map_class, "<init>", "()V");
    jmethodID insert = env->GetMethodID(map_class, "put", "(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;");
    jmethodID new_bool = env->GetMethodID(boxed_bool, "<init>", "(Z)V");

    jobject new_map = env->NewObject(map_class, map_constructor);
    for (auto it = members_map.begin(); it != members_map.end(); it++) {
        auto session_id = env->NewStringUTF(it->first.data());
        bool is_admin = it->second;
        auto jbool = env->NewObject(boxed_bool, new_bool, is_admin);
        env->CallObjectMethod(new_map, insert, session_id, jbool);
        ++it;
    }
    return new_map;
}

inline jobject serialize_legacy_group_info(JNIEnv *env, session::config::legacy_group_info info) {
    jstring session_id = env->NewStringUTF(info.session_id.data());
    jstring name = env->NewStringUTF(info.name.data());
    jobject members = serialize_members(env, info.members());
    jbyteArray enc_pubkey = util::bytes_from_ustring(env, info.enc_pubkey);
    jbyteArray enc_seckey = util::bytes_from_ustring(env, info.enc_seckey);
    int priority = info.priority;
    bool hidden = info.hidden;

    jclass legacy_group_class = env->FindClass("network/loki/messenger/libsession_util/util/GroupInfo$LegacyGroupInfo");
    jmethodID constructor = env->GetMethodID(legacy_group_class, "<init>", "(Ljava/lang/String;Ljava/lang/String;Ljava/util/Map;Z[B[BI)V");
    jobject serialized = env->NewObject(legacy_group_class, constructor, session_id, name, members, hidden, enc_pubkey, enc_seckey, priority);
    return serialized;
}

inline jobject serialize_community_info(JNIEnv *env, session::config::community_info info) {
    auto priority = info.priority;
    auto serialized_info = util::serialize_base_community(env, info);
    auto clazz = env->FindClass("network/loki/messenger/libsession_util/util/GroupInfo$CommunityGroupInfo");
    jmethodID constructor = env->GetMethodID(clazz, "<init>", "(Lnetwork/loki/messenger/libsession_util/util/BaseCommunityInfo;I)V");
    jobject serialized = env->NewObject(clazz, constructor, serialized_info, priority);
    return serialized;
}

#endif //SESSION_ANDROID_USER_GROUPS_H