#include <ESP8266HTTPClient.h>
#include <ArduinoJson.h>
#include <ESP8266WiFi.h>
#include <vector>
#include <Ticker.h>

Ticker ticker;

struct Weather{
  String wea="";
  String temp="";
  String windpower="";  //风力,单位：级
  String humi="";
};

const char* ssid      = "xxxxx";     
const char* password  = "xxxxxx";   

const String timeUrl = "http://quan.suning.com/getSysTime.do";
const String gaoDeUrl = "http://restapi.amap.com/v3/weather/weatherInfo?key=xxxxxx&city=610100";
String timeMsg="";
String weatherMsg="";
String timeAfterParse="";
struct Weather weatherAfterParse;
String data2stm="";   //打包后发送给stm32的信息
String data2client="";  //发送给socket客户端的信息

String callHttp(String httpUrl);
String parseJsonTime(String tjson);
struct Weather parseJsonWeather(String wjson);
void led_on_off();
String tqzh2py(String zhtq);//中文天气转拼音
String windpowerFormat(String wp);

void connectWIFI(char* ssid, char* password);
void NewClient();
void setLedOnOff(WiFiClient serverClient);
void sendData2Client(WiFiClient serverClient, uint8_t sbuf[], int len);
void receiveDataFromClient(WiFiClient serverClient);
void getSerialDataAndProcess();


WiFiServer server(5045);  //端口号，随意修改，范围0-65535
WiFiClient serverClient;  //连接到服务器的客户端

//初始化
void setup() {
  Serial.begin(115200);
  server.begin();
  pinMode(2, OUTPUT);  //设置内置LED引脚为输出模式
  digitalWrite(2, LOW);
  
  connectWIFI(ssid, password);

//  ticker.attach(1, getTimeMsg);
//  ticker.attach(10, getWeatherMsg);
  
  timeMsg = callHttp(timeUrl);
  weatherMsg = callHttp(gaoDeUrl);
}

String clientMsg="";
String stm32Msg="";
int n = 0;

void loop() {
  n++;
  NewClient();
  setLedOnOff(serverClient);  //指示灯提示是否连接到客户端

  
  //get data from client, Internet
  timeMsg = callHttp(timeUrl);
  if(n >= 12){  //大概10s请求一次
      //timeMsg = callHttp(timeUrl);
      weatherMsg = callHttp(gaoDeUrl);
      n=0;
  }
  receiveDataFromClient(serverClient);

  //get data from stm32 and process
  getSerialDataAndProcess();

  //data process
  if(timeMsg!=""){
    timeAfterParse = parseJsonTime(timeMsg);
  }
  
  if(weatherMsg!=""){
    weatherAfterParse = parseJsonWeather(weatherMsg);
  }


  //打包数据发到stm32
  //data2stm += (clientMsg + ",");
  if(timeMsg!="" && weatherMsg!=""){
      data2stm = (timeAfterParse + "," + tqzh2py(weatherAfterParse.wea) + "," + weatherAfterParse.temp + "," +weatherAfterParse.humi +"," + windpowerFormat(weatherAfterParse.windpower) + "," + clientMsg + "," + "\r\n"); //日期,天气,温度,湿度,风力等级
      Serial.print(data2stm);
//      if(n%6 == 0){
//        Serial.print(data2stm);
//      }
  }


  //data2stm="";
  

//send data to stm32
//  if(n==5 && data2stm!=""){
//    Serial.print("data to stm: " + data2stm);
//  }
  

  // send data to client
//  if(stm32Msg!=""){
//    data2client = stm32Msg;
//    sendData2Client(serverClient, data2client);
//    stm32Msg="";
//  }
  

  //delay(100);
}


void led_on_off(){
  static bool LEDState=0;
  LEDState = !LEDState;
  digitalWrite(LED_BUILTIN, LEDState);
}


String callHttp(String httpUrl) { 
  HTTPClient hClient;
  String message;
  hClient.begin(httpUrl); 
  
  int httpCode = hClient.GET();
  if (httpCode > 0) {
    if (httpCode == HTTP_CODE_OK) {
      String payload = hClient.getString();
      message = payload;
    }
  } else {
    message = "[1-HTTP]failed, error:" + String(hClient.errorToString(httpCode).c_str());
  }
  hClient.end();
  return message;
}

void getTimeMsg(){
  timeMsg = callHttp(timeUrl);
}

void getWeatherMsg(){
  weatherMsg = callHttp(gaoDeUrl);  
}

String parseJsonTime(String tjson){

  const size_t capacity = 96;
  DynamicJsonDocument tdoc(capacity);
 
  // 反序列化数据
  deserializeJson(tdoc, tjson);
 
  // 获取解析后的数据信息
  String datetime = tdoc["sysTime2"].as<String>();

  return datetime;
}


//{"status":"1","count":"1","info":"OK","infocode":"10000","lives":[{"province":"陕西","city":"西安市","adcode":"610100","weather":"晴","temperature":"13","winddirection":"北","windpower":"≤3","humidity":"88","reporttime":"2021-10-29 21:03:10"}]}
struct Weather parseJsonWeather(String wjson){
  
  struct Weather weather;

  const size_t capacity = 512;
  DynamicJsonDocument wdoc(capacity);
 
  // 反序列化数据
  deserializeJson(wdoc, wjson);

  JsonObject lives_0 = wdoc["lives"][0];
  
  weather.wea = lives_0["weather"].as<String>(); // "晴"
  weather.temp = lives_0["temperature"].as<String>(); // "13"
  weather.humi = lives_0["humidity"].as<String>(); // "88"
  weather.windpower = lives_0["windpower"].as<String>(); // "≤3"
  return weather;
}

String tqzh2py(String zhtq){  

  String zh_cn[68] = {"晴","少云","晴间多云","多云","阴","有风","平静","微风","和风","清风","强风/劲风","疾风","大风","烈风","风暴","狂爆风","飓风","热带风暴","霾","中度霾","重度霾","严重霾","阵雨","雷阵雨","雷阵雨并伴有冰雹","小雨","中雨","大雨","暴雨","大暴雨","特大暴雨","强阵雨","强雷阵雨","极端降雨","毛毛雨/细雨","雨","小雨-中雨","中雨-大雨","大雨-暴雨","暴雨-大暴雨","大暴雨-特大暴雨","雨雪天气","雨夹雪","阵雨夹雪","冻雨","雪","阵雪","小雪","中雪","大雪","暴雪","小雪-中雪","中雪-大雪","大雪-暴雪","浮尘","扬沙","沙尘暴","强沙尘暴","龙卷风","雾","浓雾","强浓雾","轻雾","大雾","特强浓雾","热","冷","未知",};
  String py[68] = {"qing","shaoyun","qingjianduoyun","duoyun","yin","youfeng","pingjing","weifeng","hefeng","qingfeng","qiangfeng/jinfeng","jifeng","dafeng","liefeng","fengbao","kuangbaofeng","jufeng","redaifengbao","mai","zhongdumai","zhongdumai","yanzhongmai","zhenyu","leizhenyu","leizhenyubingbanyoubingbao","xiaoyu","zhongyu","dayu","baoyu","dabaoyu","tedabaoyu","qiangzhenyu","qiangleizhenyu","jiduanjiangyu","maomaoyu/xiyu","yu","xiaoyu-zhongyu","zhongyu-dayu","dayu-baoyu","baoyu-dabaoyu","dabaoyu-tedabaoyu","yuxuetianqi","yujiaxue","zhenyujiaxue","dongyu","xue","zhenxue","xiaoxue","zhongxue","daxue","baoxue","xiaoxue-zhongxue","zhongxue-daxue","daxue-baoxue","fuchen","yangsha","shachenbao","qiangshachenbao","longjuanfeng","wu","nongwu","qiangnongwu","qingwu","dawu","teqiangnongwu","re","leng","weizhi"};
  
  for(int i=0; i<68; i++){
    
   // if(strstr())
    if(zh_cn[i] == zhtq)
      return py[i];
    }
    
  return zhtq;  //没有就返回中文
}
 
String windpowerFormat(String wp){
  
  if(wp=="≤3"){
    return "<=3";
  }
  return wp;
}

void connectWIFI(const char* ssid, const char* password) {
  WiFi.mode(WIFI_STA);  //WIFI_STA、WIFI_AP、WIFI_AP_STA
  WiFi.begin(ssid, password);

  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
  }
}

//检测新的client
void NewClient() {
  uint8_t i=0;
  
  //检测是否有新的client请求进来
  if (server.hasClient()) {
      //释放旧无效或者断开的client
    if (!serverClient || !serverClient.connected()) {
      if (serverClient) {
        serverClient.stop();
      }
      //分配最新的client
      serverClient = server.available();
      //Serial.print("New client: ");
      //Serial.print(i);
      i=1;
    }
    if(i==1){
      //当达到最大连接数 无法释放无效的client，需要拒绝连接
      WiFiClient serverClient = server.available();
      serverClient.stop();
      //Serial.println("Connection rejected ");
    }
  }
}

//客户端连接时常亮，无连接时闪烁
void setLedOnOff(WiFiClient serverClient) { 
  //连接情况 用led灯的状态显示
  if(serverClient.connected()){    
    digitalWrite(2, LOW);
    delay(800);
  } else{
    digitalWrite(2, HIGH);
    delay(400);
    digitalWrite(2, LOW);
    delay(400);
  }
}

//接收客户端的数据
void receiveDataFromClient(WiFiClient serverClient) {
  uint8_t buffLen=0;
  char buff[100];

  if (serverClient && serverClient.connected()) {
//    while (serverClient.available()){
//        r[i] = serverClient.read();
//        i++;
//    }
//    if (serverClient.read() == -1){
//      Serial.printf("%s", r);
//    }

    if(serverClient.available()){
      buffLen = serverClient.available();
      serverClient.readBytes(buff, buffLen);
      //Serial.write(buff, buffLen);
      //Serial.printf("\r\nbufflen: %d\r\n", buffLen);
      serverClient.flush();
      
      clientMsg = "";
      for(uint8_t i=2; i<buffLen; i++){  //i=2表示舍弃客户端发送的第一个字符（安卓客户端在数据开头会发送一个特殊字符，由于是UTF-8编码方式，占用2Bytes）
        clientMsg += (buff[i]);
      }
      //Serial.println("msg: "+ clientMsg);
    }
    //else Serial.println("--------------------");  
  }
}


void sendData2Client(WiFiClient serverClient, char data[], uint8_t len) {  //len为字节数
  if (serverClient && serverClient.connected()) {
    serverClient.write(data, len);       //将数据发送给客户端
//    Serial.write(sbuf, len);           
    delay(200);
  }
}


//得到数据后处理方式,把它当作一个串口(伪)中断函数
void getSerialDataAndProcess(){
  if (Serial.available()) {                     //串口读取到的转发到wifi，因为串口是一位一位的发送所以在这里缓存完再发送
    size_t countBytes = Serial.available();
    char sbuf[countBytes];
    Serial.readBytes(sbuf, countBytes);

//    Serial.write(sbuf, countBytes);
    sendData2Client(serverClient, sbuf, countBytes);
    Serial.flush();
  }
}