#include "LuaPB.h"
#include "ProtoImporter.h"

#include <google/protobuf/descriptor.h>
#include <google/protobuf/descriptor_database.h>
#include <google/protobuf/message.h>
#include <google/protobuf/repeated_field.h>
//#include <google/protobuf/dynamic_message.h>
#include <google/protobuf/stubs/common.h>
#include <google/protobuf/compiler/importer.h>
// #include "xxtea.h"
using namespace google::protobuf;
// const static string copyright = "fun101@copyright";
// const static string company = "_elex";
static int push_message(lua_State* L,
                        Message* message,
                        bool del)
{
    lua_pbmsg* tmp =
            static_cast<lua_pbmsg*>(lua_newuserdata(L, sizeof(lua_pbmsg)));
    if (tmp == NULL)
    {
        return 0;
    }

    tmp->msg = message;
    tmp->isDelete = del;

    luaL_getmetatable(L, PB_MESSAGE_META);
    lua_setmetatable(L, -2);
    return 1;
}

static int push_repeated_msg(lua_State* L,
                        Message* msg,
                        FieldDescriptor* field)
{

    lua_repeated_msg* repeated =
        static_cast<lua_repeated_msg*>(lua_newuserdata(L, sizeof(lua_repeated_msg)));

    if (!repeated) {
        return 0;
    }

    repeated->msg = msg;
    repeated->field = field;

    luaL_getmetatable(L, PB_REPEATED_MESSAGE_META);
    lua_setmetatable(L, -2);
    return 1;
}

static int pb_repeated_add(lua_State* L)
{
    lua_repeated_msg* repeated =
        (lua_repeated_msg*)luaL_checkudata(L, 1, PB_REPEATED_MESSAGE_META);

    Message* message = repeated->msg;
    if (!message)
    {
        luaL_argerror(L, 1, "pb_repeated_add, pb msg is nil");
        return 0;
    }

    FieldDescriptor* field = repeated->field;
    const Reflection* reflection = message->GetReflection();
    luaL_argcheck(L, field != NULL, 1, "pb_repeated_add, field is null");

    if(field->type() == google::protobuf::FieldDescriptor::TYPE_MESSAGE)
    {
        Message* msg = reflection->AddMessage(message, field);
        return push_message(L, msg, false);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_INT32)
    {
        int val = static_cast<int>(luaL_checkinteger(L, 2));
        reflection->AddInt32(message, field, val);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_INT64)
    {
        long long val = static_cast<long long>(luaL_checkinteger(L, 2));
        reflection->AddInt64(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_UINT32)
    {
        unsigned int val = static_cast<unsigned int>(luaL_checkinteger(L, 2));
        reflection->AddUInt32(message, field, val);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_UINT64)
    {
        unsigned long long val = static_cast<unsigned long long>(luaL_checkinteger(L, 2));
        reflection->AddUInt64(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_FLOAT)
    {
        float val = static_cast<float>(luaL_checknumber(L, 2));
        reflection->AddFloat(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_DOUBLE)
    {
        double val =  static_cast<double>(luaL_checknumber(L, 2));
        reflection->AddDouble(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_BOOL)
    {
        int val = static_cast<int>(luaL_checkinteger(L, 2));
        reflection->AddBool(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_STRING)
    {
        size_t strlen;
        const char* str = luaL_checklstring(L, 2, &strlen);
        reflection->AddString(message, field, str);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_BYTES)
    {
        size_t strlen;
        const char* str = luaL_checklstring(L, 2, &strlen);
        reflection->AddString(message, field, str);
    }
    else
    {
        luaL_argerror(L, (2), "pb_repeated_add field name type for add  is not support!!");
     }
    return 0;
}

static int pb_repeated_len(lua_State* L)
{
    lua_repeated_msg* repeated =
        (lua_repeated_msg*)luaL_checkudata(L, 1, PB_REPEATED_MESSAGE_META);

    Message* message = repeated->msg;
    if (!message)
    {
        luaL_argerror(L, 1, "pb_repeated_len, pb msg is nil");
        return 0;
    }

    const Reflection* reflection = message->GetReflection();
    FieldDescriptor* field = repeated->field;
    luaL_argcheck(L, field != NULL, 1, "pb_repeated_len field not exist");

    int fieldsize = reflection->FieldSize(*message, field);
    lua_pushinteger(L, fieldsize);
    return 1;
}

static int pb_repeated_get(lua_State* L)
{
    lua_repeated_msg* repeated =
        (lua_repeated_msg*)luaL_checkudata(L, 1, PB_REPEATED_MESSAGE_META);

    Message* message = repeated->msg;
    if (!message)
    {
        luaL_argerror(L, 1, "pb_repeated_get, pb msg is nil");
        return 0;
    }

    FieldDescriptor* field = repeated->field;
    const Reflection* reflection = message->GetReflection();
    luaL_argcheck(L, field != NULL, 1, "pb_repeated_get field not exist");

    // -1 为了和lua的下标从一开始保持一致
    int index = static_cast<int>(luaL_checkinteger(L, 2)) - 1;
    luaL_argcheck(L, index >= 0, 2, "pb_repeated_get index expected >= 1");

    int fieldsize = reflection->FieldSize(*message, field);
    if (index < 0 || index >= fieldsize)
    {
        luaL_argerror(L, 1, "pb_repeated_get, index is out of bounds");
        return 0;
    }

    if(field->type() == google::protobuf::FieldDescriptor::TYPE_INT32)
    {
        lua_pushinteger(L, reflection->GetRepeatedInt32(*message, field, index));
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_INT64) {
        lua_pushinteger(L, reflection->GetRepeatedInt64(*message, field, index));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_STRING)
    {
        lua_pushstring(L, reflection->GetRepeatedString(*message, field, index).data());
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_BYTES)
    {
        lua_pushstring(L, reflection->GetRepeatedString(*message, field, index).data());
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_UINT32)
    {
        lua_pushinteger(L, reflection->GetRepeatedUInt32(*message, field, index));
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_UINT64)
    {
        lua_pushinteger(L, reflection->GetRepeatedUInt64(*message, field, index));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_FLOAT)
    {
        lua_pushnumber(L, reflection->GetRepeatedFloat(*message, field, index));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_DOUBLE)
    {
        lua_pushnumber(L, reflection->GetRepeatedDouble(*message, field, index));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_BOOL)
    {
        lua_pushboolean(L, reflection->GetRepeatedBool(*message, field, index));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_MESSAGE)
    {
        Message* msg = reflection->MutableRepeatedMessage(message, field, index);
        return push_message(L, msg, false);
    }
    else
    {
        luaL_argerror(L, 0, "pb_repeated_get, field type for get not support!!!");
        return 0;
    }
    return 1;
}

static int pb_repeated_set(lua_State* L)
{
    lua_repeated_msg* repeated =
        (lua_repeated_msg*)luaL_checkudata(L, 1, PB_REPEATED_MESSAGE_META);

    Message* message = repeated->msg;
    if (!message)
    {
        luaL_argerror(L, 1, "pb_repeated_set, pb msg is nil");
        return 0;
    }

    const Reflection* reflection = message->GetReflection();
    FieldDescriptor* field = repeated->field;
    luaL_argcheck(L, field != NULL, 1, "pb_repeated_set field not exist");

    int index = static_cast<int>(luaL_checkinteger(L, 2)) - 1;
    luaL_argcheck(L, index >= 0, 2, "pb_repeated_set index expected >= 1");

    if(field->type() == google::protobuf::FieldDescriptor::TYPE_INT32)
    {
        int val = static_cast<int>(luaL_checkinteger(L, 3));
        reflection->SetRepeatedInt32(message, field, index, val);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_INT64)
    {
        long long val = static_cast<long long>(luaL_checkinteger(L, 3));
        reflection->SetRepeatedInt64(message, field,index, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_UINT32)
    {
        unsigned int val = static_cast<unsigned int>(luaL_checkinteger(L, 3));
        reflection->SetRepeatedUInt32(message, field, index, val);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_UINT64)
    {
        unsigned long long val = static_cast<unsigned long long>(luaL_checkinteger(L, 3));
        reflection->SetRepeatedUInt64(message, field, index, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_FLOAT)
    {
        float val = static_cast<float>(luaL_checknumber(L, 3));
        reflection->SetRepeatedFloat(message, field, index, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_DOUBLE)
    {
        double val = static_cast<double>(luaL_checknumber(L, 3));
        reflection->SetRepeatedDouble(message, field, index, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_BOOL)
    {
        int val = static_cast<int>(lua_toboolean(L, 3));
        reflection->SetRepeatedBool(message, field, index, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_STRING)
    {
        size_t strlen;
        const char *str = static_cast<const char *>(luaL_checklstring(L, 3, &strlen));
        reflection->SetRepeatedString(message, field, index, str);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_BYTES)
    {
        size_t strlen;
        const char *str = static_cast<const char *>(luaL_checklstring(L, 3, &strlen));
        reflection->SetRepeatedString(message, field, index, str);
    }
    else
    {
        luaL_argerror(L, (2), "pb_repeated_set type for set not support!!!");
    }
    return 0;
}

///////////////////////////////////////////////////////////
int pb_import(lua_State* L)
{
    int argc = lua_gettop(L);
    if(argc >= 1)
    {
        const char * filename = luaL_checkstring(L, 1);
        sProtoImporter.Import(filename);
    }else
    {
        luaL_argerror(L, 1, "pb_import,  params num at last 1");
    }
    return 0;
}
int pb_import_data(const char* filename, const void* data, int size) {
    sProtoImporter.Import(filename, data, size);
    return 0;
}

static int pb_get_description(lua_State* L)
{
    lua_pbmsg* luamsg = (lua_pbmsg*)luaL_checkudata(L, 1, PB_MESSAGE_META);

    Message* message = luamsg->msg;
    if (!message)
    {
        luaL_argerror(L, 1, "pb_get_description,  pb msg is nil");
        return 0;
    }

    const Descriptor* descriptor = message->GetDescriptor();
    int cont = descriptor->field_count();
    lua_createtable(L, cont, 0);
    for (int i = 0; i < cont; i++)
    {
        lua_pushstring(L, descriptor->field(i)->name().c_str());
        lua_rawseti(L, -2, i + 1);
    }

    return 1;
}

static int pb_map_path(lua_State* L)
{
    int argc = lua_gettop(L);
    if (argc == 2)
    {
        const char * virtualPath = luaL_checkstring(L, 1);
        const char * diskPath = luaL_checkstring(L, 2);
        sProtoImporter.MapPath(virtualPath, diskPath);
    }
    else
    {
        luaL_argerror(L, 1, "pb_mappath,  params num must 2");
    }
    return 0;
}
static int pb_clean_proto(lua_State* L)
{
    sProtoImporter.CleanCacheFile();
    sProtoImporter.CleanMapPath();
    return 0;
}
static int pb_new(lua_State* L)
{
    const char* type_name = luaL_checkstring(L, 1);
    Message* message = sProtoImporter.createDynamicMessage(type_name);
    if (!message)
    {
        fprintf(stderr, "pb_new error, result is typename(%s) not found!\n", type_name);
        return 0;
    }

    return push_message(L, message, true);
}

static int pb_delete(lua_State* L)
{
    lua_pbmsg* luamsg = (lua_pbmsg*)luaL_checkudata(L, 1, PB_MESSAGE_META);

    if (luamsg->isDelete && luamsg->msg)
    {
        Message* message = luamsg->msg;
        delete message;
        luamsg->msg = NULL;
    }
    return 0;
}

static int pb_tostring(lua_State* L)
{
    lua_pbmsg* luamsg = (lua_pbmsg*)luaL_checkudata(L, 1, PB_MESSAGE_META);
    google::protobuf::Message *message = luamsg->msg;
    if (!message)
    {
        luaL_argerror(L, 1, "pb_tostring,  pb msg is nil");
        return 0;
    }
    std::string msg(message->DebugString());
    lua_pushlstring(L, msg.c_str(), msg.length());
    return 1;
}

static int pb_get(lua_State* L)
{
    lua_pbmsg* luamsg = (lua_pbmsg*)luaL_checkudata(L, 1, PB_MESSAGE_META);
    const char* field_name = luaL_checkstring(L, 2);

    Message* message = luamsg->msg;
    if (!message)
    {
        luaL_argerror(L, 1, "pb_get,  pb msg is nil");
        return 0;
    }

    const Descriptor* descriptor = message->GetDescriptor();
    const Reflection* reflection = message->GetReflection();
    const FieldDescriptor* field = descriptor->FindFieldByName(field_name);
    luaL_argcheck(L, (field != NULL), 2, "pb_get, field_name error");

    if (field->is_repeated())
    {
        return push_repeated_msg(L, message, const_cast<FieldDescriptor*>(field));
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_INT32)
    {
        lua_pushinteger(L, reflection->GetInt32(*message, field));
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_INT64)
    {
        lua_pushinteger(L, reflection->GetInt64(*message, field));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_UINT32)
    {
        lua_pushinteger(L, reflection->GetUInt32(*message, field));
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_UINT64)
    {
        lua_pushinteger(L, reflection->GetUInt64(*message, field));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_FLOAT)
    {
         lua_pushnumber(L, reflection->GetFloat(*message, field));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_DOUBLE)
    {
         lua_pushnumber(L, reflection->GetDouble(*message, field));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_BOOL)
    {
        lua_pushboolean(L, reflection->GetBool(*message, field));
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_STRING)
    {
        std::string str(reflection->GetString(*message, field));
        lua_pushlstring(L, str.c_str(), str.length());
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_BYTES)
    {
        std::string str(reflection->GetString(*message, field));
        lua_pushlstring(L, str.c_str(), str.length());
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_MESSAGE)
    {
        Message* msg = reflection->MutableMessage(message, field);
        return push_message(L, msg, false);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_ENUM)
    {
        lua_pushinteger(L, reflection->GetEnumValue(*message, field));
    }
    return 1;
}

static int pb_set(lua_State* L)
{
    lua_pbmsg* luamsg = (lua_pbmsg*)luaL_checkudata(L, 1, PB_MESSAGE_META);
    const char* field_name = luaL_checkstring(L, 2);

    Message* message = luamsg->msg;
    if (!message)
    {
        luaL_argerror(L, 1, "pb_set, pb msg is nil");
        return 0;
    }

    const Descriptor* descriptor = message->GetDescriptor();
    const Reflection* reflection = message->GetReflection();
    const FieldDescriptor* field = descriptor->FindFieldByName(field_name);
    luaL_argcheck(L, field != NULL, 2, "LuaPB::set field_name error");
    luaL_argcheck(L, !field->is_repeated(), 2, "LuaPB::set field_name is repeated");

    if(field->type() == google::protobuf::FieldDescriptor::TYPE_STRING)
    {
        size_t strlen;
        const char *str = luaL_checklstring(L, 3, &strlen);
        reflection->SetString(message, field, str);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_BYTES)
    {
        size_t strlen;
        const char *str = luaL_checklstring(L, 3, &strlen);
        reflection->SetString(message, field, str);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_INT32)
    {
        int val = static_cast<int>(luaL_checkinteger(L, 3));
        reflection->SetInt32(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_INT64)
    {
        long long val = static_cast<long long>(luaL_checkinteger(L, 3));
        reflection->SetInt64(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_UINT32)
    {
        unsigned int val = static_cast<unsigned int>(luaL_checkinteger(L, 3));
        reflection->SetUInt32(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_UINT64)
    {
        unsigned long long val = static_cast<unsigned long long>(luaL_checkinteger(L, 3));
        reflection->SetUInt64(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_FLOAT)
    {
        float val = static_cast<float>(luaL_checknumber(L, 3));
        reflection->SetFloat(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_DOUBLE)
    {
        double val = static_cast<double>(luaL_checknumber(L, 3));
        reflection->SetDouble(message, field, val);
    }
    else if(field->type() == google::protobuf::FieldDescriptor::TYPE_BOOL)
    {
        int val = static_cast<int>(lua_toboolean(L, 3));
        reflection->SetBool(message, field, val);
    }
    else if (field->type() == google::protobuf::FieldDescriptor::TYPE_ENUM)
    {
        int val = static_cast<int>(luaL_checkinteger(L, 3));
        reflection->SetEnumValue(message, field, val);
    }
    else
    {
        luaL_argerror(L, 2, "pb_set field_name type error");
    }
    return 0;
}

static int pb_parseFromString(lua_State* L)
{
    lua_pbmsg* luamsg = (lua_pbmsg*)luaL_checkudata(L, 1, PB_MESSAGE_META);
    Message* message = luamsg->msg;

    luaL_checktype(L, 2, LUA_TSTRING);

    size_t bin_len;
    const char* bin = static_cast<const char*>(	luaL_checklstring(L, 2, &bin_len));
    message->ParseFromArray(bin, bin_len);
    return 0;
}

static int pb_serializeToString(lua_State* L)
{
    lua_pbmsg* luamsg = (lua_pbmsg*)luaL_checkudata(L, 1, PB_MESSAGE_META);
    Message* message = luamsg->msg;

    std::string msg;
    message->SerializeToString(&msg);
    lua_pushlstring(L, msg.c_str(), msg.length());
    return 1;
}
// void * lua_decode(const void *data, int* size)
// {
//     int company_length = strlen(company.c_str());
//     int copyright_length = strlen(copyright.c_str());
//     if(strncmp((const char*)data,company.c_str(),company_length) == 0)
//     {
//         fprintf(stderr, "the data need decode\n");
//         xxtea_long len = 0;
//         unsigned char * result;
//         result = xxtea_decrypt((unsigned char*)data + company_length, (xxtea_long)(*size) - (xxtea_long)company_length, (unsigned char*)copyright.c_str(), (xxtea_long)copyright_length,&len);
//         *size = len;
//         return result;
//     }
//     *size = 0;
//     fprintf(stderr, "can't decode ,because data is null\n");
//     return NULL;
// }

// void freeData(void* data)
// {
//     if(data != NULL)
//     {
//         fprintf(stderr, "free data success\n");
//         free(data);
//     }else
//     {
//         fprintf(stderr, "can't free ,because data is null\n");
//     }
// }
static const luaL_Reg lib[] =
{
    {"new", pb_new},
    {"mapPath",pb_map_path},
    {"tostring", pb_tostring},
    {"parseFromString", pb_parseFromString},
    {"serializeToString", pb_serializeToString},
    {"cleanProto",pb_clean_proto},
    {"getDescription", pb_get_description},
    {"import",pb_import},
    {NULL, NULL}
};

static const luaL_Reg libm[] =
{
    {"__index", pb_get},
    {"__newindex", pb_set},
    {"__gc", pb_delete},
    {NULL, NULL}
};

static const luaL_Reg repeatedlib[] = {
    {"add", pb_repeated_add},
    {"len", pb_repeated_len},
    {"get", pb_repeated_get},
    {"set", pb_repeated_set},
    {NULL, NULL},
};

void luaopen_luapb(lua_State* L)
{
    luaL_newmetatable(L, PB_REPEATED_MESSAGE_META);
    luaL_setfuncs(L,repeatedlib,0);

    lua_pushstring(L, "__index");
    lua_pushvalue(L, -2);
    lua_settable(L, -3);

    lua_pushstring(L, "__newindex");
    lua_pushcfunction(L, pb_repeated_set);
    lua_settable(L, -3);
    lua_pop(L,1);

    luaL_newmetatable(L, PB_MESSAGE_META);
    luaL_setfuncs(L,libm,0);
    lua_pop(L,1);
    luaL_newlib(L,lib);
    lua_setglobal(L,PB_MESSAGE);
}