#include "spy.h"
#include "load_calls.h"
#include "log.h"
#include "rpc_server.h"
#include "util.h"

WxCalls_t g_WxCalls      = { 0 };
DWORD g_WeChatWinDllAddr = 0;

void InitSpy()
{
    wchar_t version[16] = { 0 };
    InitLogger();
    g_WeChatWinDllAddr = (DWORD)GetModuleHandle(L"WeChatWin.dll"); //获取wechatWin模块地址
    if (g_WeChatWinDllAddr == 0) {
        LOG_ERROR("获取wechatWin.dll模块地址失败");
        return;
    }

    if (!GetWeChatVersion(version)) { //获取微信版本
        LOG_ERROR("获取微信版本失败");
        return;
    }

    if (LoadCalls(version, &g_WxCalls) != 0) { //加载微信版本对应的Call地址
        LOG_ERROR("不支持当前版本");
        return;
    }

    RpcStartServer();
}

void CleanupSpy()
{
    RpcStopServer();
    // FreeLibraryAndExitThread(hModule, 0);
}

int IsLogin(void) { return (int)GET_DWORD(g_WeChatWinDllAddr + g_WxCalls.login); }

std::string GetSelfWxid() { return GET_STRING(g_WeChatWinDllAddr + g_WxCalls.ui.wxid); }