#include <ArduinoJson.h>
#include "PubSubClient.h"
#include <WiFi.h>
#include <Keypad.h>
#include <Wire.h> //该头文件wokwi仿真时必不可少，否则编译不通过，但Arduino IDE编译时可以不写
#include <LiquidCrystal_I2C.h>
#include <ESP32Servo.h>

const uint8_t ROWS = 4;
const uint8_t COLS = 4;
char keys[ROWS][COLS] = {
  { '1', '2', '3', 'A' },
  { '4', '5', '6', 'B' },
  { '7', '8', '9', 'C' },
  { '*', '0', '#', 'D' }
};
uint8_t colPins[COLS] = { 19, 18, 5, 17 }; // Pins connected to C1, C2, C3, C4
uint8_t rowPins[ROWS] = { 32, 33, 25, 26}; // Pins connected to R1, R2, R3, R4

LiquidCrystal_I2C lcd(0x27,20,4);

Servo lockServo;
Keypad keypad = Keypad(makeKeymap(keys), rowPins, colPins, ROWS, COLS);
//要连接的WiFi热点名、WiFi热点密码 
#define WIFI_AP "Wokwi-GUEST"
#define WIFI_PASSWORD ""

//要修改以下三个变量的内容：根据自己在ThingsBoard服务器上创建的设备填写
//clientId: 添加设备时的客户端ID
//userName: 用户名
//password: 密码
const char* clientId="h6if6pks8t40dfzjd7ez"; 
const char* userName="cab4kox52c60df5k29nj"; 
const char* password="3vr9kfc28uw9y8tmy81j";

#define lightPin 2
#define SERVO_PIN 13
#define SERVO_LOCK_POS   20
#define SERVO_UNLOCK_POS 90
#define lightSensor A0
#define ECHO_PIN 0
#define TRIG_PIN 4
int counter =0;

//不要修改！这是CPS技术与创客实践课程的物联网云服务器网址、端口号
char thingsboardServer[] = "thingsboard.cloud";
int port = 1883;
String CODE_now="0000";//默认密码0000
bool isNearby;
WiFiClient wifiClient;


PubSubClient client(wifiClient);

int status = WL_IDLE_STATUS;
//开始开始开始开始开始
void setup() {
  pinMode(lightPin, OUTPUT);
  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);

  lockServo.attach(SERVO_PIN);

  lcd.init();
  
  lcd.setCursor(0,0);
  lcd.print("Hello !");
  Serial.begin(115200);


  InitWiFi();
  client.setServer( thingsboardServer, port );
  client.setCallback(on_message);
  
}

void loop() {
  if ( !client.connected() ) reconnect();
 
  client.loop();
  char key = keypad.getKey();
    if (key != NO_KEY) {
      Serial.println(key);
      if (key=='*'&&!isNearby){
        lcd.setCursor(0, 0);
        lcd.print("Enter new code:");
        String newCode = inputSecretCode();
        lcd.clear();
        lcd.setCursor(0, 0);
        lcd.print("Succuess!");
        lockServo.write(0);
        StaticJsonDocument<200> jsonBuffer;
        JsonObject data = jsonBuffer.to<JsonObject>();
        data["key"] = newCode;
        char payload[256];
        serializeJson(data,payload,sizeof(payload));
        String strPayload = String(payload);
        Serial.print("key sensor: ");
        Serial.println(strPayload);
        client.publish("v1/devices/me/telemetry", strPayload.c_str());
        CODE_now=newCode;
        delay(2000);
        lcd.setCursor(0, 0);
        lcd.print("please press #");
      }
      if(key=='#'&&isNearby){
        lcd.setCursor(0, 0);
        lcd.print("Enter the key:");
        String key = getCode();
        if(key==CODE_now){
          lcd.clear();
          lcd.setCursor(0, 0);
          lcd.print("Succuess!");
          lockServo.write(180);
          delay(2000);
          lcd.setCursor(0, 0);
          lcd.print("please press *");
        }
        else{
          lcd.clear();
          lcd.setCursor(0, 0);
          lcd.print("Key Erorr!");
          delay(2000);
          lcd.setCursor(0, 0);
          lcd.print("please press #");
        }
      }
    }
  if(counter>=20){
    counter=0;
    /*2秒钟上报1次传感器数据*/


    int value = analogRead(lightSensor);
    float distance = readDistanceCM();

    isNearby = distance < 100;
    digitalWrite(lightPin, isNearby);
    Serial.print("Measured distance: ");
    Serial.println(readDistanceCM());

    Serial.print("Light sensor:");
    Serial.println(value);
      // 准备JSON格式的数据payload
    StaticJsonDocument<200> jsonBuffer;
    JsonObject data = jsonBuffer.to<JsonObject>();
    data["distance"] = distance;
    char payload[256];
    serializeJson(data,payload,sizeof(payload));
    String strPayload = String(payload);
    Serial.print("Near sensor: ");
    Serial.println(strPayload);
    client.publish("v1/devices/me/telemetry", strPayload.c_str());
  } else {
    counter++;
    delay(100);
  }
}
void lock() {
  lockServo.write(SERVO_LOCK_POS);
}

void unlock() {
  lockServo.write(SERVO_UNLOCK_POS);
}

float readDistanceCM() {
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);
  int duration = pulseIn(ECHO_PIN, HIGH);
  return duration * 0.034 / 2;
}

void InitWiFi() {
  Serial.println("Connecting to AP ...");
  // 连接WiFi
  WiFi.mode(WIFI_STA);
  WiFi.begin(WIFI_AP, WIFI_PASSWORD);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("Connected to AP");
}

void reconnect() {
  // 循环直到重新连接成功
  while (!client.connected()) {
    status = WiFi.status();
    if ( status != WL_CONNECTED) {
      WiFi.begin(WIFI_AP, WIFI_PASSWORD);
      while (WiFi.status() != WL_CONNECTED) {
        delay(500);
        Serial.print(".");
      }
      Serial.println("Connected to AP");
    }
    Serial.print("Connecting to IoT Server ...");
    // 连接物联网云服务器，使用(clientId, username, password)
    if ( client.connect(clientId, userName, password) ) {
      Serial.println( "[DONE]" );
      client.subscribe("v1/devices/me/rpc/request/+"); 
    } else {
      Serial.print( "[FAILED] [ rc = " );
      Serial.print( client.state() );
      Serial.println( " : retrying in 5 seconds]" );
      // 重连前，等待5秒
      delay( 5000 );
    }
  }
}

// 回调函数：当从服务器收到PUBLISH消息时触发
void on_message(char* topic, byte* payload, unsigned int length) {
  Serial.println("On message");

  char json[length + 1];
  strncpy (json, (char*)payload, length);
  json[length] = '\0';

  Serial.print("Topic: ");
  Serial.println(topic);
  Serial.print("Message: ");
  Serial.println(json);

    // Decode JSON request
  JsonDocument data;
  DeserializationError error = deserializeJson(data, json);
  if (error)
  {
    Serial.println("parseObject() failed");
    return;
  }

  // 检查请求的方法method
  String methodName = String((const char*)data["method"]);
  
  if (methodName.equals("light")) {  
    // 应答Hi消息
    String responseTopic = String(topic);
    responseTopic.replace("request", "response");
        // 检查请求的参数params
    bool params = data["params"];
    Serial.println(params);
    if(params) {
       digitalWrite(lightPin, HIGH);
       Serial.println("light on");
      client.publish(responseTopic.c_str(), "{\"light\":\"on!\"}");
    }
    else {
       digitalWrite(lightPin, LOW);
       Serial.println("light off");
      client.publish(responseTopic.c_str(), "{\"light\":\"off!\"}");
    }
  } else  {
    // 应答"Unknow command"消息
    String responseTopic = String(topic);
    Serial.println("Error: Unknown command.");
    responseTopic.replace("request", "response");
    client.publish(responseTopic.c_str(), "{\"Error\":\"Unknown command\"}");
  }
}

String inputSecretCode() {
  lcd.setCursor(5, 1);
  lcd.print("[____]");
  lcd.setCursor(6, 1);
  String result = "";
  while (result.length() < 4) {
    char key = keypad.getKey();
    if (key >= '0' && key <= '9') {
      lcd.print('*');
      result += key;
    }
  }
  return result;
}

String getCode(){
  lcd.setCursor(5, 1);
  lcd.print("[____]");
  lcd.setCursor(6, 1);
  String result = "";
  while (result.length() < 4) {
    char key = keypad.getKey();
    if (key >= '0' && key <= '9') {
      lcd.print('*');
      result += key;
    }
  }
  return result;
}