//1029新增控制燈號顏色

#include <WiFi.h>
#include <PubSubClient.h> //請先安裝PubSubClient程式庫

// ------ws2812b ------

#include <Adafruit_NeoPixel.h>
#ifdef __AVR__
  #include <avr/power.h>
#endif
#define PIN        2   //接腳
#define NUMPIXELS 16   //燈數量
#define ttb223_pin 12
bool ws2812b_mainled=false;
// ------ 以下修改成你自己的WiFi帳號密碼 ------


char* ssid = "LankuoAP"; //請修改為您連線的網路名稱
char* password = "abcd1234"; //請修改為您連線的網路密碼
// ------ 以下修改成你MQTT設定 ------
char* MQTTServer = "mqttgo.io";//免註冊MQTT伺服器
int MQTTPort = 1883;//MQTT Port 除非加密，否則不用更改port
char* MQTTUser = "";//不須帳密
char* MQTTPassword = "";//不須帳密

//訂閱主題1:改變LED燈號(記得改Topic)
char* MQTTSubTopic1 = "lankuo/ws2812/led/power";
char* MQTTSubTopic2 = "lankuo/ws2812/led/color";
//建立MQTT與聯網物件
WiFiClient WifiClient;
PubSubClient MQTTClient(WifiClient);

Adafruit_NeoPixel pixels(NUMPIXELS, PIN, NEO_GRB + NEO_KHZ800);
#define DELAYVAL 500
int r=150,g=150,b=150;


void setup() {
  Serial.begin(115200);
  //開始WiFi連線
  WifiConnecte();
  //開始MQTT連線
  MQTTConnecte();

  #if defined(__AVR_ATtiny85__) && (F_CPU == 16000000)
  clock_prescale_set(clock_div_1);
  #endif
  pixels.begin();

  pinMode(ttb223_pin, INPUT);

}

void loop() {
  //如果WiFi連線中斷，則重啟WiFi連線
  if (WiFi.status() != WL_CONNECTED) { WifiConnecte(); }

  //如果MQTT連線中斷，則重啟MQTT連線
  if (!MQTTClient.connected()) {  MQTTConnecte(); }

  MQTTClient.loop();//更新訂閱狀態
  delay(50);

  if(digitalRead(ttb223_pin)){
    ws2812b_mainled=!ws2812b_mainled;
    Serial.print("當下狀態：");
    Serial.println(ws2812b_mainled);
    while(digitalRead(ttb223_pin)){
      delay(50);
    }
    if(ws2812b_mainled){
      Serial.print("開啟燈號");
      //pixels.clear();
      for (int i=0;i<NUMPIXELS;i++){
        pixels.setPixelColor(i, pixels.Color(r, g, b));}
      pixels.show();
    }
    else{
      Serial.print("關閉燈號");
      //pixels.clear();
      for (int i=0;i<NUMPIXELS;i++){
        pixels.setPixelColor(i, pixels.Color(0, 0, 0));}
      pixels.show();
    }
  }

}

//開始WiFi連線
void WifiConnecte() {
  //開始WiFi連線
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("WiFi連線成功");
  Serial.print("IP Address:");
  Serial.println(WiFi.localIP());
}

//開始MQTT連線
void MQTTConnecte() {
  MQTTClient.setServer(MQTTServer, MQTTPort);
  //以下開始處理訂閱主題
  MQTTClient.setCallback(MQTTCallback); //收到訂閱資料時，要處理MQTTCALLBACK
  while (!MQTTClient.connected()) {
    //以亂數為ClietID
    String  MQTTClientid = "esp32-" + String(random(1000000, 9999999));//建立一個郵差，為了要不一樣，所以使用亂數
    if (MQTTClient.connect(MQTTClientid.c_str(), MQTTUser, MQTTPassword)) {
      //連結成功，顯示「已連線」。
      Serial.println("MQTT已連線");
      //訂閱SubTopic1主題
      MQTTClient.subscribe(MQTTSubTopic1);
      MQTTClient.subscribe(MQTTSubTopic2);
    } else {
      //若連線不成功，則顯示錯誤訊息，並重新連線
      Serial.print("MQTT連線失敗,狀態碼=");
      Serial.println(MQTTClient.state());
      Serial.println("五秒後重新連線");
      delay(5000);
    }
  }
}

//接收到訂閱時
void MQTTCallback(char* topic, byte* payload, unsigned int length) {
  Serial.print(topic); Serial.print("訂閱通知:");
  String payloadString;//將接收的payload轉成字串
  //顯示訂閱內容
  for (int i = 0; i < length; i++) {
    payloadString = payloadString + (char)payload[i];
  }
  Serial.println(payloadString);
  //比對主題是否為訂閱主題1，若相同則執行動作的部分，需要注意payloadString內容的資料是否與app內的訊息相同。
  if (strcmp(topic, MQTTSubTopic1) == 0) {
    Serial.println("改變燈號：" + payloadString);
    if (payloadString == "on") {
      Serial.print("開啟燈號");
      ws2812b_mainled=true;
      //pixels.clear();
      for (int i=0;i<NUMPIXELS;i++){
        pixels.setPixelColor(i, pixels.Color(r, g, b));}
      pixels.show();
    }
    if (payloadString == "off") {
      Serial.print("關閉燈號");
      ws2812b_mainled=false;
      //pixels.clear();
      for (int i=0;i<NUMPIXELS;i++){
        pixels.setPixelColor(i, pixels.Color(0, 0, 0));}
      pixels.show();
    }
  }
  if (strcmp(topic, MQTTSubTopic2) == 0) {
    r = strtol(payloadString.substring(1, 3).c_str(), NULL, 16);
    g = strtol(payloadString.substring(3, 5).c_str(), NULL, 16);
    b = strtol(payloadString.substring(5, 7).c_str(), NULL, 16);
    for (int i = 0; i < NUMPIXELS; i++) {
    pixels.setPixelColor(i, pixels.Color(r, g, b));
    }
    pixels.show();
  }
}