#include <Arduino.h>            //build in
#include <CustomJWT.h>          //find from library
#include <ESPAsyncWebServer.h>  // Get from https://github.com/me-no-dev/ESPAsyncWebServer
#include <ArduinoJson.h>        //find from library
#include <WiFiClientSecure.h>   //build in
#include <WiFiUdp.h>            //build in
#include <time.h>               //build in
#include <HTTPClient.h>         //find from library

void WiFiConnection();
void initTime();
void splitApiKey(const char *apikey);

char header[50];
char payload[500];
char signature[100];
char out[500];
char jsonString[500];

char idCharArray[150];
char secretCharArray[100];

const char* ssid = "FireBat";
const char* password = "123456789x";

const char *getMessage;

unsigned long timestamp;
time_t now;
struct tm timeinfo;

bool Process = false;
bool checkIP = true;

String getReturnJson, responseMessage;
HTTPClient http, http_id;

char *role = "user";

char *version = "0.0.1";

char *setApiKey = "cde3b06db9072cc9ce0c0233939207d7.AVSBoJ2P9v0NrkSf";  //Get Api key from https://open.bigmodel.cn/

//char* web_hook = "https://open.bigmodel.cn/api/paas/v3/model-api/characterglm/async-invoke";  //ChatGLM Character
//char *web_hook = "https://open.bigmodel.cn/api/paas/v3/model-api/chatglm_pro/async-invoke";  //ChatGLM Pro
char* web_hook = "https://open.bigmodel.cn/api/paas/v3/model-api/chatglm_std/async-invoke";  //ChatGLM Std
//char *web_hook = "https://open.bigmodel.cn/api/paas/v3/model-api/chatglm_lite/async-invoke";  // ChatGLM lite

const char *ntpServer = "ntp.aliyun.com";  //You can change NTP from what you want, the default is aliyun
const long gmtOffset_sec = 28800;          //Change GMT, default is China's NTP +8
const int daylightOffset_sec = 0;

DynamicJsonDocument doc(2048);

void setup() {
  Serial.begin(115200);
  WiFiConnection();
  delay(100);

  initTime();
  if(getLocalTime(&timeinfo)){
      now = mktime(&timeinfo);
      splitApiKey(setApiKey);
  } else {
      Serial.println(F("Failed to obtain Beijing time"));   //if you change your ntp you had better to change it
  }
}

void loop() {
    String userMessage="";
    if (WiFi.status() == WL_CONNECTED) {
        while(Serial.available()<0);
        while(Serial.available()>0){ // 串口有输入
            userMessage += char(Serial.read());
            delay(10);      // 延时函数用于等待字符完全进入缓冲区，可以尝试没有延时，输出结果会是什么
        }
        // 检查是否接收到数据，如果接收到数据，则输出该数据
        if(userMessage!=""){
            Serial.print("You:");
            Serial.print(userMessage);
            // userMessage = "100字内回答：" + userMessage; // 需要节约tokens时，则添加这句
            Process = true;
        }

        if(Process){
            // 开始向AI的http提问请求，获取task_id
            http.begin(web_hook);
            http.addHeader("Accept", "application/json");   
            http.addHeader("Content-Type", "application/json; charset=UTF-8");
            http.addHeader("Authorization", getReturnJson);
            // String metaData = "\"meta\": {\"user_info\": \"" + String(user_info) + "\", \"bot_info\": \"" + String(bot_info) + "\", \"bot_name\": \"" + String(bot_name) + "\", \"user_name\": \"" + String(user_name) + "\"}";
            // String payloadMessage = "{\"prompt\": [{\"role\": \"" + String(role) + "\", \"content\": \"" + userMessage + "\"}], " + metaData + "}";
            String payloadMessage = "{\"prompt\": \"" + userMessage + "\"}";

            int httpResponseCode = http.POST(payloadMessage);
            if (httpResponseCode > 0) {
                String returnJsonMessage = http.getString();
                DeserializationError error = deserializeJson(doc, returnJsonMessage);
                // Serial.println(returnJsonMessage);
                if (error) {
                  Serial.print(F("JSON parsing failed: "));
                  Serial.println(F(error.c_str()));
                }
            }
            // 结束向AI的http提问请求
            userMessage.clear();
            http.end();
            
            // 开始向AI的http处理结果请求
            String task_id = doc["data"]["task_id"]; // 获取task_id
            String web_search_id = "https://open.bigmodel.cn/api/paas/v3/model-api/-/async-invoke/" + task_id;  //GET Method(+below)
            http_id.begin(web_search_id);
            http_id.addHeader("Accept", "application/json");
            http_id.addHeader("Content-Type", "application/json; charset=UTF-8");
            http_id.addHeader("Authorization", getReturnJson);
            Serial.println();
            while(true){
              int httpResponseIDCode = http_id.GET();
              if (httpResponseIDCode > 0) {
                responseMessage = http_id.getString();
                DeserializationError error_id = deserializeJson(doc, responseMessage);
                String task_status = doc["data"]["task_status"];
                if(task_status == "PROCESSING"){
                  Serial.print('.');
                  doc.clear();
                  delay(500);
                  continue;
                }else{
                  Serial.println();
                  DeserializationError error_id = deserializeJson(doc, responseMessage);
                  String AssistantMessage = responseMessage.substring(responseMessage.indexOf('[')+35,responseMessage.lastIndexOf(']')-4);
                  Serial.println("AI:" + AssistantMessage);
                  break;
                }
              } else {
                  responseMessage = "Error: External API request failed!";
              }
            }
            http_id.end();
            task_id.clear();
            responseMessage.clear(); // 清空响应内容的本地字符串
            Process = false;
        } 
    }
}

void WiFiConnection() {
    WiFi.begin(ssid,password);
    Serial.print("Connecting to WiFi");
    while(WiFi.status() != WL_CONNECTED){
        Serial.print('.');
        delay(500);
    }
    Serial.println();
    Serial.println("WiFi successfully connected!");
    Serial.print("IP address:");
    Serial.println(WiFi.localIP());
}

void initTime() { // 时间同步
  Serial.println("Initializing time synchronization...");
  configTime(gmtOffset_sec, daylightOffset_sec, ntpServer);
  for (int i = 0; i < 10; i++) {
    if (time(nullptr)) {
      Serial.println(F("Time synchronized"));
      return;
    }
    delay(1000);
    Serial.println(F("Waiting for time synchronization..."));
  }
  Serial.println(F("Time synchronization failed. Check NTP server and network."));
}

void splitApiKey(const char *apikey) {  //IMPORTANT
  const char *delimiter = strchr(apikey, '.');

  if (delimiter != NULL) {
    size_t idLength = delimiter - apikey;

    if (idLength < sizeof(idCharArray)) {
      strncpy(idCharArray, apikey, idLength);
      idCharArray[idLength] = '\0';
      strcpy(secretCharArray, delimiter + 1);
      snprintf(jsonString, sizeof(jsonString), "{\"api_key\":\"%s\",\"exp\":%lu,\"timestamp\":%lu}", idCharArray, now * 5, now);  //expire time is set 5 * now
      CustomJWT jwt(secretCharArray, header, sizeof(header), payload, sizeof(payload), signature, sizeof(signature), out, sizeof(out));
      jwt.encodeJWT(jsonString);
      getReturnJson = jwt.out;  //return jwt
      jwt.clear();
    } else {
      Serial.println("ID part of API key is too long.");
    }
  } else {
    Serial.println("Invalid API key format.");
  }
}

DeserializationError error_id = deserializeJson(doc, responseMessage);
String AssistantMessage = responseMessage.substring(responseMessage.indexOf('[')+35,responseMessage.lastIndexOf(']')-4);
Serial.println("AI:" + AssistantMessage);

while(true){
  // 循环GET获取服务器处理状态
  int httpResponseIDCode = http_id.GET();
  if (httpResponseIDCode > 0) {
    responseMessage = http_id.getString();
    DeserializationError error_id = deserializeJson(doc, responseMessage);
    String task_status = doc["data"]["task_status"];
    if(task_status == "PROCESSING"){
      // 等待
      delay(500);
      continue;
    }else{
      // 查询到状态是SUCCESS时打印content的内容
      break;
    }
  } else {
      responseMessage = "Error: External API request failed!";
  }
}