1. /**
    2.  * ESP8266 自动配网 MQTT
    3.  */
    4. // https://github.com/tzapu/WiFiManager/
    5. #include <WiFiManager.h>
    6. /**
    7.  * MQTT
    8.  */
    9. //PubSubClient by Nick O`Leary
    10. #include <PubSubClient.h>
    11. #include <ArduinoJson.h>
    12. /**
    13.  * ws2812 led
    14.  */
    15. #include <Adafruit_NeoPixel.h>
    17. // mqtt 参数
    18. char* mqtt_server ="192.168.0.200";
    19. char* mqtt_port = "1883";
    20. WiFiClient espClient;
    21. PubSubClient mqtt_client(espClient);
    23. #define RING_LED_PIN    2
    24. #define NUMPIXELS 24
    25. #define BRIGHTNESS 50
    26. Adafruit_NeoPixel strip = Adafruit_NeoPixel(NUMPIXELS, RING_LED_PIN, NEO_GRB + NEO_KHZ800);
    28. StaticJsonDocument<256> jsonState;                      //设备状态
    29. /*============================================== web 配网 下===========================================================*/
    30. /**
    31.  * 初始化wifimanager
    32.  */
    33. void initWifi()
    34. {
    35.   WiFiManager wifiManager;                               // 创建 wifimanager 对象
    36. //  wifiManager.resetSettings();                           // 重置保存的修改
    37.   wifiManager.setConnectTimeout(10);                     // 配置连接超时
    38.   wifiManager.setDebugOutput(true);                      // 打印调试内容
    39.   wifiManager.setMinimumSignalQuality(30);               // 设置最小信号强度
    40.   //设置AP
    41.   IPAddress _ip = IPAddress(192, 168, 4, 1);             // 设置固定AP信息
    42.   IPAddress _gw = IPAddress(192, 168, 4, 1);             // 设置固定AP信息
    43.   IPAddress _sn = IPAddress(255, 255, 255, 0);           // 设置固定AP信息
    44.   wifiManager.setAPStaticIPConfig(_ip, _gw, _sn);        // 设置固定AP信息
    45.   // 回调
    46.   wifiManager.setAPCallback(configModeCallback);         // 设置进入AP模式的回调
    47.   wifiManager.setSaveConfigCallback(saveConfigCallback); // 设置点击保存的回调
    48.   wifiManager.setBreakAfterConfig(true);                 // 设置 如果配置错误的ssid或者密码 退出配置模式
    49.   // 自定义参数
    50.   WiFiManagerParameter custom_mqtt_server("server", "mqtt server", mqtt_server, 40);
    51.   WiFiManagerParameter custom_mqtt_port("port", "mqtt port", mqtt_port, 6);
    52.   wifiManager.addParameter(&custom_mqtt_server);
    53.   wifiManager.addParameter(&custom_mqtt_port);
    54.   // 尝试连接网络，失败去到配置页面
    55.   if (!wifiManager.autoConnect("ESP8266_AP", ""))    
    56.   {
    57.     Serial.println("failed to connect and hit timeout");
    58.     ESP.reset();                                          // 重置并重试
    59.     delay(1000);
    60.   }
    61.   // WiFi连接成功后将通过串口监视器输出连接成功信息
    62.   Serial.println("== WIFI CONNECT SUCCESS ! ==");
    63.   Serial.print("IP Address : ");
    64.   Serial.println(WiFi.localIP());
    65.   // 读取配置页面配置好的信息
    66.   strcpy(mqtt_server, custom_mqtt_server.getValue());
    67.   strcpy(mqtt_port, custom_mqtt_port.getValue());
    68. }
    69. // 配置进入AP模式通知回调
    70. void configModeCallback(WiFiManager *myWiFiManager)
    71. {
    72.   Serial.println("Entered config mode"); //进入配置模式
    73.   Serial.print("Wifi Name : ");
    74.   Serial.println("Wifi Name : " + myWiFiManager->getConfigPortalSSID());
    75.   Serial.print("Wifi IP : ");
    76.   Serial.println(WiFi.softAPIP());
    77.   // 附加参数
    78.   Serial.print("MQTT Server : ");
    79.   Serial.println(mqtt_server);
    80.   Serial.print("MQTT Port : ");
    81.   Serial.println(mqtt_port);
    82. }
    83. // 设置点击保存的回调
    84. void saveConfigCallback()
    85. {
    86.   Serial.println("Should save config");
    87. }
    88. /*============================================== MQTT 下===========================================================*/
    89. /* 初始化设备属性状态*/
    90. void initState()
    91. {
    92.   jsonState["name"] = "ESP8266";
    93.   jsonState["led"] = "OFF";
    94.   jsonState["mode"] = "0";
    95.   jsonState["update_time"] = random(100);
    96. }
    97. void initMQTT(){
    98.   mqtt_client.setServer(mqtt_server, atoi(mqtt_port));
    99.   mqtt_client.setCallback(mqtt_callback);
    100. }
    101. void mqtt_reconnect() {
    102.     String clientId = "ESP-01A-" + String(random(0xffff), HEX);
    103.     if (mqtt_client.connect(clientId.c_str())) {         //连接成功
    104.       Serial.println("=== MQTT CONNECT SUCCESS ! ===");
    105.       String msg;
    106.       serializeJson(jsonState,msg);
    107.       Serial.println(msg);
    108.       mqtt_client.publish("hass/sensor/esp8266_1/led/state",msg.c_str());     // 连接成功发送初始状态信息
    109.       mqtt_client.subscribe("hass/sensor/esp8266_1/led/set");
    110.     } else {
    111.       delay(5000);                                  // Wait 5 seconds before retrying
    112.     }
    113. }
    114. /**
    115.  * MQTT 接收数据
    116.  */
    117. void mqtt_callback(char* topic, byte* payload, unsigned int length) {
    118.   Serial.print("接收消息：");
    119.   DynamicJsonDocument doc(200);
    120.   deserializeJson(doc, payload);
    121.   serializeJson(doc,Serial);
    122.   Serial.println("");
    123.   // 业务处理
    124.   String key = doc["key"];
    125.   String value = doc["value"];
    126.   // 更新状态
    127.   jsonState[key] = value;
    128.   jsonState["update_time"] = random(100);
    129.   String msg;
    130.   serializeJson(jsonState,msg);
    131.   Serial.print("更新状态：");
    132.   Serial.println(msg);
    133.   mqtt_client.publish("hass/sensor/esp8266_1/led/state",msg.c_str());     // 状态信息
    134. }
    135. /*============================================== 环形灯===========================================================*/
    136. void initRingLED(){
    137.   strip.setBrightness(BRIGHTNESS);
    138.   // 初始化Adafruit_NeoPixel库
    139.   strip.begin();
    140.   // 初始化时关闭所有LED      
    141.   strip.show();
    142. }
    143. /**
    144.  * 模式: 颜色擦除
    145.  */
    146. void colorWipe(uint32_t c, uint8_t wait) {
    147.   strip.clear();
    148.   for (uint16_t i = 0; i < strip.numPixels(); i++) {
    149.     strip.setPixelColor(i, c);
    150.     strip.show();
    151.     delay(wait);
    152.   }
    153. }
    155. /**
    156.  * 彩虹呼吸
    157.  */
    158. void rainbow(uint8_t wait) {
    159.   uint16_t i, j;
    160.   for (j = 0; j < 256; j++) {
    161.     for (i = 0; i < strip.numPixels(); i++) {
    162.       strip.setPixelColor(i, Wheel((i + j) & 255 ));
    163.     }
    164.     strip.show();
    165.     delay(wait);
    166.   }
    167. }
    169. /**
    170.  * 彩虹呼吸-旋转
    171.  */
    172. void rainbowCycle(uint8_t wait) {
    173.   uint16_t i, j;
    174.   for (j = 0; j < 256 * 5; j++) { // 5 cycles of all colors on wheel
    175.     for (i = 0; i < strip.numPixels(); i++) {
    176.       strip.setPixelColor(i, Wheel(((i * 256 / strip.numPixels()) + j) & 255));
    177.     }
    178.     strip.show();
    179.     delay(wait);
    180.   }
    181. }
    182. /**
    183.  * 走马灯
    184.  */
    185. void theaterChase(uint32_t c, uint8_t wait) {
    186.   for (int j=0; j<10; j++) {  //do 10 cycles of chasing
    187.     for (int q=0; q < 3; q++) {
    188.       for (uint16_t i=0; i < strip.numPixels(); i=i+3) {
    189.         strip.setPixelColor(i+q, c);    //turn every third pixel on
    190.       }
    191.       strip.show();
    193.       delay(wait);
    195.       for (uint16_t i=0; i < strip.numPixels(); i=i+3) {
    196.         strip.setPixelColor(i+q, 0);        //turn every third pixel off
    197.       }
    198.     }
    199.   }
    200. }
    201. /**
    202.  * 走马灯-彩虹
    203.  */
    204. void theaterChaseRainbow(uint8_t wait) {
    205.   for (int j=0; j < 256; j++) {     // cycle all 256 colors in the wheel
    206.     for (int q=0; q < 3; q++) {
    207.       for (uint16_t i=0; i < strip.numPixels(); i=i+3) {
    208.         strip.setPixelColor(i+q, Wheel( (i+j) % 255));    //turn every third pixel on
    209.       }
    210.       strip.show();
    212.       delay(wait);
    214.       for (uint16_t i=0; i < strip.numPixels(); i=i+3) {
    215.         strip.setPixelColor(i+q, 0);        //turn every third pixel off
    216.       }
    217.     }
    218.   }
    219. }
    220. uint32_t Wheel(byte WheelPos) {
    221.   if (WheelPos < 85) {
    222.     return strip.Color(WheelPos * 3, 255 - WheelPos * 3, 0);
    223.   } else if (WheelPos < 170) {
    224.     WheelPos -= 85;
    225.     return strip.Color(255 - WheelPos * 3, 0, WheelPos * 3);
    226.   } else {
    227.     WheelPos -= 170;
    228.     return strip.Color(0, WheelPos * 3, 255 - WheelPos * 3);
    229.   }
    230. }
    232. //----------------------------------------------- 程序入口 ---------------------------------------------------
    233. void setup()
    234. {
    235.   Serial.begin(9600);                             // 初始化串口设置
    236.   pinMode(LED_BUILTIN, OUTPUT);                   // 设置 LED_BUILTIN（GPIO-16） 为输出功能
    237.   digitalWrite(LED_BUILTIN, HIGH);                // 默认关闭
    238.   // 配置状态信息
    239.   initState();
    240.   // 自动配网
    241.   initWifi();
    242.   // 配置MQTT
    243.   initMQTT();
    244.   // 配置环形灯
    245.   initRingLED();
    246. }
    247. void loop()
    248. {
    249.   // mqtt
    250.   if (!mqtt_client.connected()) {
    251.     mqtt_reconnect();
    252.   }
    253.    mqtt_client.loop();
    254.    // LED
    255.    if (jsonState["led"] == "OFF") {          // 通过接收的信息 '1' 来控制开或关
    256.      digitalWrite(LED_BUILTIN, HIGH); 
    257.    } else {
    258.      digitalWrite(LED_BUILTIN, LOW); 
    259.    }
    260.   // ring led
    261.    if(jsonState["mode"]== "0"){ strip.clear(); }
    262.    else if(jsonState["mode"]== "1"){ colorWipe(strip.Color(150, 150, 150), 20); }// BlueWite
    263.    else if(jsonState["mode"]== "2"){ rainbow(5);}
    264.    else if(jsonState["mode"]== "3"){ rainbowCycle(1);}
    265.    else if(jsonState["mode"]== "4"){ theaterChase(strip.Color(127, 127, 127), 50);}
    266.    else if(jsonState["mode"]== "5"){ theaterChaseRainbow(5);}
    268. }