#include <Arduino.h>  //Arduino开发环境，包含Arduino编程所需的基本函数和定义
#include <DHT.h>      //用于读取DHT系列的温度和湿度传感器的数据
#include <Wire.h>     //I2C通信
#include <Adafruit_SSD1306.h>     //用于驱动SSD1306 OLED显示屏
#include "WiFi.h"     //用于ESP8266或ESP32等带有WiFi功能的Arduino板与WiFi网络进行通信
#include "PubSubClient.h"   //Arduino设备通过MQTT协议发布和订阅消息
#include "Ticker.h"   //提供了一个简单的定时器功能
#include "ArduinoJson.h"      //处理JSON数据的库

//OLED SSD1306
#define SCREEN_WIDTH 128  //OLED宽度 // OLED display width, in pixels 
#define SCREEN_HEIGHT 64  //OLED高度 // OLED display height, in pixels  
#define OLED_RESET     -1 //复位  // Reset pin # (or -1 if sharing Arduino reset pin)
#define SCREEN_ADDRESS 0x3C //OLED显示屏的I2C地址  //< See datasheet for Address; 0x3D for 128x64, 0x3C for 128x32
// 0x3D for 128x64, 0x3C for 128x32，但是这里用的128*64 0x3C
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET); //创建了一个Adafruit_SSD1306类的实例，命名为display
// 创建实例？

//DHT22配置
int DHTPin= 13; //表示DHT22传感器的数据引脚连接到了Arduino的13号引脚
int DHTType= DHT22; // 定义了一个整型变量DHTType，并将其设置为DHT22
// ?
DHT dht (DHTPin, DHTType); // 创建实例，调用dht.readTemperature()和dht.readHumidity()来获取当前的温度和湿度，来自<DHT.h> 
 
//uFire_SHT20 sht20;
 
const char* ssid = "Wokwi-GUEST";              //wifi名
const char* password = "";      //wifi密码
const char* mqtt_server = "mqtts.heclouds.com"; //onenet 的 IP地址 mqtts.heclouds.com 183.230.40.96
const int port = 1883;                    //端口号

#define mqtt_pubid "b1o48niXbq"    //产品ID
#define mqtt_devid "test"       //设备名称
//鉴权信息 token
#define mqtt_password "version=2018-10-31&res=products%2Fb1o48niXbq%2Fdevices%2Ftest&et=1742904952&method=md5&sign=wmdIiPZhJPAdinaMnkpMwg%3D%3D"       //鉴权信息

WiFiClient espClient;           //  ? //创建一个WIFI连接客户端
PubSubClient client(espClient); //  ? // 创建一个PubSub客户端, 传入创建的WIFI客户端
Ticker tim1;                    //定时器,用来循环上传数据
Ticker tim2;                    //定时器,用来循环上传数据
float temp;
float humi;
float dis;

//设备上传数据的post主题 //上传端
#define ONENET_TOPIC_PROP_POST "$sys/" mqtt_pubid "/" mqtt_devid "/dp/post/json" //"$sys/" mqtt_pubid "/" mqtt_devid "/thing/property/post"
//接收下发属性设置主题  // 订阅
#define ONENET_TOPIC_PROP_SET  "$sys/" mqtt_pubid "/" mqtt_devid "/dp/post/json/+" //"$sys/" mqtt_pubid "/" mqtt_devid "/thing/property/set"
// ??
//接收下发属性设置成功的回复主题
#define ONENET_TOPIC_PROP_SET_REPLY "$sys/" mqtt_pubid "/" mqtt_devid "/dp/post/accepted" //"$sys/" mqtt_pubid "/" mqtt_devid "/thing/property/set_reply"
//接收设备属性获取命令主题
#define ONENET_TOPIC_PROP_GET "$sys/" mqtt_pubid "/" mqtt_devid "/cmd/request/+" //"$sys/" mqtt_pubid "/" mqtt_devid "/thing/property/get"
//接收设备属性获取命令成功的回复主题
#define ONENET_TOPIC_PROP_GET_REPLY "$sys/" mqtt_pubid "/" mqtt_devid "/cmd/response/+/+" //"$sys/" mqtt_pubid "/" mqtt_devid "/thing/property/get_reply"
 
//这是post上传数据使用的模板
#define ONENET_POST_BODY_FORMAT "{\"id\":%d,\"dp\":%s}"

//#define ONENET_POST_BODY_FORMAT
int postMsgId = 0; //记录已经post了多少条

void setupDisplay() // 用于初始化，成功则start ...，失败则SSD1306 allocation failed
{
  if (!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) //初始化成功，不进入，初始化失败则死循环，输出SSD1306 allocation failed
  {
    Serial.println(F("SSD1306 allocation failed")); // F() 内的字符串存放在Flash中，节省RAM
    // Serial.println(data): 发送数据到串行端口，并在数据后添加一个换行符。
    for (;;); // Don't proceed, loop forever
  }

  display.clearDisplay();
  display.setTextSize(1);           
  display.setTextColor(SSD1306_WHITE);      
  display.setCursor(0,0); 
  display.println("start ...");
  display.display();
}

//连接WIFI相关函数
void setupWifi()
{
  delay(10);
  Serial.println("connect WIFI");
  WiFi.begin(ssid, password);

  while (!WiFi.isConnected()) //链接成功则会返回ture，退出循环
  {
    Serial.print(".");
    delay(500);
  }

  Serial.println("OK");
  Serial.println("Wifi connected!");
  Serial.println("IP address: ");
  Serial.println(WiFi.localIP());

}

//向主题发送模拟的温湿度数据
void sendTempAndHumi()
{  
  temp = dht.readTemperature();
  humi = dht.readHumidity();
  
  if (client.connected())
  {
    //先拼接出json字符串
    char param[82];
    char jsonBuf[178];
    char buf[60];
    sprintf(param, "{ \"temp\":[{\"v\":%.2f}],\"humi\":[{\"v\":%.2f}] }", temp, humi); //我们把要上传的数据写在param里
    //sprintf(param, "{ \"dis\":[{\"v\":%.2f}] }", dis); //我们把要上传的数据写在param里
    
    postMsgId += 1;
    
    sprintf(jsonBuf, ONENET_POST_BODY_FORMAT, postMsgId, param);  //上传数据到onenet
    /*显示在终端*/
    sprintf(buf,"temp:%.2fhumi:%.2f", temp, humi);
    display.clearDisplay();
    display.setTextSize(2);           
    display.setTextColor(SSD1306_WHITE);      
    display.setCursor(0,0); 
    display.println(buf);
    display.display();

    //再从mqtt客户端中发布post消息
    if (client.publish(ONENET_TOPIC_PROP_POST, jsonBuf))
    {
      Serial.print("Post message to cloud: ");
      Serial.println(jsonBuf);
    }
    else
    {
      Serial.println("Publish message to cloud failed!");
    }

  }

}

//重连函数, 如果客户端断线,可以通过此函数重连
void clientReconnect()
{
  while (!client.connected()) //再重连客户端
  {
    Serial.println("reconnect MQTT...");
    if ( client.connect(mqtt_devid, mqtt_pubid, mqtt_password) )
    {
      Serial.println("connected");
    }
    else
    {
      Serial.println("failed");
      Serial.println(client.state());
      Serial.println("try again in 5 sec");
      delay(5000);
    }
  }
}
/**/
void callback(char *topic, byte *payload, unsigned int length)
{
  Serial.println("message rev:");
  Serial.println(topic);

  for (size_t i = 0; i < length; i++)
  {
    Serial.print((char)payload[i]);
  }
  Serial.println();
 
  if (strstr(topic, ONENET_TOPIC_PROP_SET))
  {
    DynamicJsonDocument doc(100);               
    DeserializationError error = deserializeJson(doc, payload);               
    if (error)
    {
      Serial.println("parse json failed");
      return;
    }
    JsonObject setAlinkMsgObj = doc.as<JsonObject>();  //??
    serializeJsonPretty(setAlinkMsgObj, Serial);
    String str = setAlinkMsgObj["id"];
    Serial.println(str);
    char sendbuf[100];
    sprintf(sendbuf, "{\"id\": \"%s\",\"code\":200,\"msg\":\"success\"}", str.c_str());   //??
    Serial.println(sendbuf);
    client.publish(ONENET_TOPIC_PROP_SET_REPLY, sendbuf);
  }

  if (strstr(topic, ONENET_TOPIC_PROP_GET))
  {
    DynamicJsonDocument doc(100);
    DeserializationError error = deserializeJson(doc, payload);
    if (error)
    {
      Serial.println("parse json failed");
      return;
    }
    JsonObject setAlinkMsgObj = doc.as<JsonObject>();
    serializeJsonPretty(setAlinkMsgObj, Serial);
    String str = setAlinkMsgObj["id"];
    Serial.println(str);
    char sendbuf[100];
    //sprintf(sendbuf, "{\"id\": \"%s\",\"code\":200,\"msg\":\"success\",\"data\":{\"temp\":%.2f,\"humi\":%.2f}}", str.c_str(), temp, humi);
    sprintf(sendbuf, "{\"id\": \"%s\",\"code\":200,\"msg\":\"success\",\"data\":{\"dis\":%.2f }}", str.c_str(), temp);
    Serial.println(sendbuf);
    client.publish(ONENET_TOPIC_PROP_GET_REPLY, sendbuf);
  }
}

void update_date(void)
{
  temp++;
  humi++;
  dis++;
}


void setup() {
  // put your setup code here, to run once:
  Serial.begin(115200); //初始化串口
  //Wire.begin();
  //sht20.begin();
  delay(1000);
  

  setupWifi();                                           //调用函数连接WIFI
  setupDisplay();                                        //初始化显示屏
  client.setServer(mqtt_server, port);                   //设置客户端连接的服务器,连接Onenet服务器, 使用1883端口
  delay(2000);
  Serial.println("setServer Init!"); 
  //client.setCallback(callback);
  client.connect(mqtt_devid, mqtt_pubid, mqtt_password); //客户端连接到指定的产品的指定设备.同时输入鉴权信息
  delay(2000);
  Serial.println("connect Init!"); 
  
  if (client.connected())
  {
    Serial.println("OneNet is connected!"); //判断以下是不是连好了.
  }

  //DHT22工作
  dht.begin();

  client.subscribe(ONENET_TOPIC_PROP_SET);
  client.subscribe(ONENET_TOPIC_PROP_GET);
  client.setCallback(callback);
  tim1.attach(8, sendTempAndHumi); //定时每20秒调用一次发送数据函数sendTempAndHumi
  tim2.attach(4, update_date);
  temp =10;
  humi =20;
  dis = 100;
}

int count = 0;



void loop() {
  // put your main code here, to run repeatedly:

  if (!WiFi.isConnected()) //先看WIFI是否还在连接
  {
    setupWifi();
  }

  if (!client.connected()) //如果客户端没连接ONENET, 重新连接
  {
    clientReconnect();
    delay(100);
  }

  client.loop(); //客户端循环检测
 
}