//
// Created by ps on 18-11-29.
//

#ifndef SYNCDBTOOL_ERLANGDBTOOL_H
#define SYNCDBTOOL_ERLANGDBTOOL_H

#include <iostream>
#include <cstring>
#include <unistd.h>

#include <sys/types.h>
#include <zconf.h>
#include <thread>

#include <map>
#include <vector>
#include <mutex>

#include <stdio.h>
#include <stdarg.h>
#include <jsoncpp/json/json.h>
#include <arpa/inet.h>
#include <SqliteToolkit.hpp>

#include "erl_interface.h"
#include "ei.h"

#include "SyncDB.hpp"

#define TIMEOUT (5000)
#define BUFSIZE 1000
#define PORT 13001


//typedef void (*CBF)(std::string);

extern void erlangCallBackFunc(std::string);

//#TODO 多线程优化
namespace ErlangTool {

    static int m_loop = 0;
    //map<uuid，人脸特征信息>
    typedef std::map<std::string, SyncDB::Feature_Info> map_FaceDataCache;
    //map<uuid，设备信息>
    typedef std::map<std::string, SyncDB::Device_Info> map_DevDataCache;
    //map<表名，底库表信息>
    typedef std::map<std::string, SyncDB::Table_Info> map_TabDataCache;
    //map<表名，黑白名单类型>
    typedef std::map<std::string, SyncDB::Bw_Info> map_BwDataCache;

    typedef std::vector<SyncDB::AddPersRet> vec_AddDataCache;
    typedef std::vector<SyncDB::PerExistRet> vec_PerExistRet;
    typedef std::vector<SyncDB::UpdPersRet> vec_UpdPersRet;
    //c++11 互斥锁
    // LockG(Parma);
    typedef std::lock_guard<std::mutex> LockG;

    /***
     * 判断文件夹路径是否存在，不存在则创建
     * @param dir 文件路径
     * @param mkdir_flag 是否创建创建文件夹
     * @return 是否成功
     */
    static bool dir_file_exists(std::string dir, bool mkdir_flag = true);

    /***
     * 测试节点是否启动
     * @param nodeName 节点名称
     * @param cookie   节点cookie
     * @return 启动状态
     */
    static bool pingNode(const std::string &nodeName, const std::string &cookie);

    /***
     * 检查ETERM的数据类型
     * @param elemen
     * @return 返回数据类型
     */
    static int checkETERMType(ETERM *elemen);

    static void erlangFreeEterm(int count, ...);

    /***
     * Erlang交互工具
     * #todo map保存到内存？
     */
    class ErlangDbTool {
    public:
        explicit ErlangDbTool();

        /***
         * 初始化节点(zhizuo chaxun shiyong)
         * @param nodeName 节点名称
         * @param cookie 集群cookie
         */
        ErlangDbTool(const std::string &nodeName, const std::string &cookie, const std::string &pName);

        /***
         * 初始化节点
         * @param path 节点数据库保存路径
         * @param nodeName 节点名称
         * @param cookie 集群cookie
         */
        ErlangDbTool(const std::string &path, const std::string &nodeName, const std::string &cookie,
                     const std::string &pName);

        //clusterID clusterName
        /***
         * 初始化节点
         * @param path 节点数据库保存路径
         * @param nodeName 节点名称
         * @param cookie 集群cookie
         * @param clusterID 集群id
         * @param clusterName 集群名称
         */
        ErlangDbTool(const std::string &path, const std::string &nodeName, const std::string &cookie,
                     const std::string &clusterID, const std::string &clusterName);

        //clusterID clusterName
        /***
         * 初始化节点
         * @param path 节点数据库保存路径
         * @param nodeName 节点名称
         * @param cookie 集群cookie
         * @param clusterID 集群id
         * @param clusterName 集群名称
         * @param pName c节点标识名
         */
        ErlangDbTool(const std::string &path, const std::string &nodeName, const std::string &cookie,
                     const std::string &clusterID, const std::string &clusterName, const std::string &pName);

        virtual ~ErlangDbTool();

    private:
        void initCNode();

        bool getExecSqlResult(const std::string &strSql);

    public:
        //#todo setCNodeName

        bool setCNodeName();

        bool sendMessage();

        //#todo sendMessage

        void test(pthread_cond_t *, pthread_mutex_t *);

        /***
         * 启动erlang节点
         * @return 启动状态
         */
        int initErlang();

        std::string getConfigJsonString();

        /***
         * 启动节点数据库
         * 新节点自动创建数据库
         * @param FatherNodeName 第一个节点无需传入此参数，其他节点需要传入引导节点的名称
         * @param DeviceName 设备名称
         * @param DevAdd 设备地址
         * @return
         */
        bool startNodeDb(std::string FatherNodeName, std::string DeviceId = "", std::string DevAdd = "DevAddress");


        /***
         * 删除节点数据库，同时删除本地文件
         * @return 返回状态
         */
        bool removeNode();


        bool modifyCluName(std::string CluId, std::string CluName);


        /***
         * 查找在线节点信息
         * #TODO 有一个小bug
         * device_info.create_by会被重置
         * @return 节点集合信息
         */
        map_DevDataCache findAllNode();

        /***
         * 创建底库表
         * @param TableType car||person
         * @param TableName
         * @param SyncType true 1||false 0
         * @param BwType   "Black"1||"White"0
         * @param StartTime
         * @param EndTime
         * @return 创建状态
         */
        bool createDatabase(std::string TableType, std::string TableName, bool SyncType, std::string BwType,
                            std::string StartTime = "", std::string EndTime = "");

        /***
         * 创建底库表
         * @param TableType TableType car||person
         * @param TableName
         * @param SyncType  true 1||false 0
         * @param BwType    "Black"1||"White"0
         * @param UploadFlag 是否上传
         * @param CmpThreshold 对比阈值
         * @param Enabled 是否启用
         * @param StartTime
         * @param EndTime
         * @return
         */
        bool createDatabase(std::string t_tableName, FieldValues &fieldValues);

        /***
         * 更新底库表
         * @param TableType car||person
         * @param TableName
         * @param SyncType true 1||false 0
         * @param BwType   "white"0||"black"1
         * @return 创建状态
         */
        bool updateDatabase(std::string UUID, std::string TableType, std::string TableName, bool SyncType,
                            std::string BwType, std::string StartTime = "", std::string EndTime = "");

        bool updateDatabase(std::string t_tableName, FieldValues &fieldValues);

        /***
         * 删除数据表
         * @param TableType car||person
         * @param TableName biaoming
         * @param SyncType true 1||false 0
         * @return
         */
        bool deleteDatabase(std::string TableType, std::string TableName, bool SyncType);

        bool deleteDatabase(std::string t_tableName);

        /***
         * 查询所有数据表
         * @return
         */
        map_TabDataCache findAllDatabase();

        /***
         * 查询所有表的黑白名单类型
         * @return
         */
        map_BwDataCache findAllTypeInfo();

        /***
         * 增加人员
         * @param UUID  out
         * @param TableName
         * @param Feature
         * @param PersonName
         * @param Age
         * @param Sex
         * @param IdCard
         * @param PhoneNum
         * @return
         */
        bool addPerson(std::string &UUID, std::string TableName, std::string Feature, std::string ImgUrl = " ",
                       std::string IdCard = " ", std::string PersonName = " ", std::string Age = " ",
                       std::string Sex = " ", std::string PhoneNum = " ");

        bool
        ErlangTool::ErlangDbTool::addPerson(std::string t_tableName, AddFaceData &faceData, FieldValues &fieldValues);

        bool updatePerson(std::string t_tableName, AddFaceData &faceData, FieldValues &fieldValues);

        vec_AddDataCache addPersons(std::string TableUuid, std::vector<SyncDB::AddPersonInfo> &tmpPer);

        /***
         * 删除人员
         * @param UUID
         * @param TableName
         * @return
         */
        bool delPerson(std::string UUID, std::string TableName);


        /***
         * 根据表名加载人脸特征
         * @param TableName
         * @return
         */
        map_FaceDataCache loadFaceFeaData(std::string TableName);


        /***
         * 根据表名加载人脸特征2
         * @param TableName
         * @return
         */
        map_FaceDataCache loadFaceFeaData2(std::string TableName);

        vec_PerExistRet personIsExistOnClu(std::string personUid);

        vec_UpdPersRet singlePersonUpdate(std::string NewPerId, std::string OldId, std::string Idcard,
                                          std::string PerPicUrl, std::string PerFea);
//    private:
        /***
         * 重新连接节点
         * @return 连接状态
         */
        bool resetConn();

        /***
         * 等待节点启动，阻塞
         * @return
         */
        bool waitNode();

    public:

    private:
        //存储路径
        std::string m_path;
        std::string m_nodeName;
        std::string m_cNodeName;
        std::string m_pName;
        std::string m_cookie;

        std::string m_clusterID;
        std::string m_clusterName;

        bool m_ret;
        int m_fd;

        std::mutex m_mutex;
    };

    static int my_listen(int port);

}


#endif //SYNCDBTOOL_ERLANGDBTOOL_H