HSend.cpp

#include "Plugin.h"
#include <winhttp.h>
#include <stdio.h>
#include <time.h>
#include <string.h>

#pragma comment(lib, "winhttp.lib")

struct SiteInterface gSite;

static bool SendHTTPSignal(int port, const char* jsonPayload) {
    HINTERNET hSession = WinHttpOpen(L"HSend/1.0", WINHTTP_ACCESS_TYPE_DEFAULT_PROXY, WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
    if (!hSession) return false;

    HINTERNET hConnect = WinHttpConnect(hSession, L"localhost", port, 0);
    if (!hConnect) {
        WinHttpCloseHandle(hSession);
        return false;
    }

    HINTERNET hRequest = WinHttpOpenRequest(hConnect, L"POST", L"/amibroker_signal", NULL, WINHTTP_NO_REFERER, WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
    if (!hRequest) {
        WinHttpCloseHandle(hConnect);
        WinHttpCloseHandle(hSession);
        return false;
    }

    const wchar_t* headers = L"Content-Type: application/json\r\n";
    int jsonLen = strlen(jsonPayload);
    
    bool success = WinHttpSendRequest(hRequest, headers, -1, (LPVOID)jsonPayload, jsonLen, jsonLen, 0);
    
    if (success) {
        success = WinHttpReceiveResponse(hRequest, NULL);
    }
    
    WinHttpCloseHandle(hRequest);
    WinHttpCloseHandle(hConnect);
    WinHttpCloseHandle(hSession);
    
    return success;
}

static AmiVar HSend(int NumArgs, AmiVar* ArgsTable) {
    AmiVar result = gSite.AllocArrayResult();
    
    if (NumArgs < 1) {
        float* resultArray = result.array;
        int arraySize = gSite.GetArraySize();
        for (int i = 0; i < arraySize; i++) {
            resultArray[i] = -1.0f;
        }
        return result;
    }

    const char* payload = (ArgsTable[0].type == VAR_STRING && ArgsTable[0].string) ? ArgsTable[0].string : "";
    
    AmiVar chartIdResult = gSite.CallFunction("GetChartID", 0, NULL);
    int chartId = (int)chartIdResult.val;
    
    SYSTEMTIME st;
    GetLocalTime(&st);
    char timeStr[15];
    sprintf(timeStr, "%02d%02d%02d.%03d", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds);
    
    char jsonPayload[2048];
    sprintf(jsonPayload, 
        "{\"timestamp\":\"%s\",\"chartID\":%d,\"payload\":\"%s\"}",
        timeStr, chartId, payload);

    bool httpSuccess = SendHTTPSignal(5000, jsonPayload);
    
    float* resultArray = result.array;
    int arraySize = gSite.GetArraySize();
    float resultValue = httpSuccess ? 1.0f : 0.0f;
    
    for (int i = 0; i < arraySize; i++) {
        resultArray[i] = resultValue;
    }
    
    return result;
}

static AmiVar VTest(int NumArgs, AmiVar* ArgsTable) {
    AmiVar result = gSite.AllocArrayResult();
    
    float* resultArray = result.array;
    int arraySize = gSite.GetArraySize();
    
    for (int i = 0; i < arraySize; i++) {
        resultArray[i] = 42.0f;
    }
    
    return result;
}

FunctionTag gFunctionTable[] = {
    "VTest", { VTest, 0, 0, 0, 0, NULL },
    "HSend", { HSend, 0, 1, 0, 0, NULL }
};

int gFunctionTableSize = sizeof(gFunctionTable) / sizeof(FunctionTag);

PLUGINAPI int GetPluginInfo(struct PluginInfo* pInfo) {
    pInfo->nStructSize = sizeof(struct PluginInfo);
    pInfo->nType = PLUGIN_TYPE_AFL;
    pInfo->nVersion = 10000;
    pInfo->nIDCode = PIDCODE('H','S','N','D');
    strcpy(pInfo->szName, "HSend Plugin");
    strcpy(pInfo->szVendor, "Test");
    return TRUE;
}

PLUGINAPI int SetSiteInterface(struct SiteInterface* pSite) {
    gSite = *pSite;
    return TRUE;
}

PLUGINAPI int GetFunctionTable(FunctionTag** ppFunctionTable) {
    *ppFunctionTable = gFunctionTable;
    return gFunctionTableSize;
}

PLUGINAPI int Init() {
    return TRUE;
}

PLUGINAPI int Release() {
    return TRUE;
}