Merge branch 'arduinoota' of Lomanic/PiedThon into master

Add OTA and wifi setup
This commit is contained in:
jeanjack 2019-11-23 11:26:50 +00:00 committed by Gitea
commit a9174a59b8
2 changed files with 122 additions and 84 deletions

View File

@ -1,3 +1,5 @@
# PiedThon # PiedThon
Feux PiedThon ESP8266 Feux PiedThon ESP8266
https://wiki.fuz.re/doku.php?id=projets:fuz:piedthon

View File

@ -1,12 +1,15 @@
/*** /***
* Fuz Piedthon Fuz Piedthon
* Pedestrian traffic lights, green and red, controlled over MQTT Pedestrian traffic lights, green and red, controlled over MQTT
* Board: ESP8266 Board: ESP8266
*
*/ */
#include <Arduino.h> #include <Arduino.h>
#include <ESP8266mDNS.h>
#include <WiFiUdp.h>
#include <ArduinoOTA.h>
/// Pins configuration /// Pins configuration
@ -18,14 +21,14 @@
#define MQTT_SERVER "sonic.fuz.re" #define MQTT_SERVER "sonic.fuz.re"
#define MQTT_SERVERPORT 1883 #define MQTT_SERVERPORT 1883
#define ESP_HOSTNAME "piedthon"
#include "WifiCredFuz.cpp" // Wifi and MQTT Credentials #include "WifiCredFuz.cpp" // Wifi and MQTT Credentials
/* with /* with
#define WLAN_SSID "" #define WLAN_SSID ""
#define WLAN_PASS "" #define WLAN_PASS ""
#define MQTT_USERNAME "" //omg no user #define MQTT_USERNAME "" //omg no user
#define MQTT_KEY "" //omg no key #define MQTT_KEY "" //omg no key
*/ */
/// End of user config /// End of user config
@ -38,20 +41,20 @@ WiFiClient client;
void setupWifi() { void setupWifi() {
// Connect to WiFi access point. // Connect to WiFi access point.
Serial.print("Connecting to "); Serial.print("Connecting to ");
Serial.println(WLAN_SSID); Serial.println(WLAN_SSID);
WiFi.begin(WLAN_SSID, WLAN_PASS); WiFi.begin(WLAN_SSID, WLAN_PASS);
while (WiFi.status() != WL_CONNECTED) { while (WiFi.status() != WL_CONNECTED) {
delay(500); delay(500);
Serial.print("."); Serial.print(".");
} }
Serial.println(); Serial.println();
Serial.print("WiFi connected"); Serial.print("WiFi connected");
Serial.print(" IP address: "); Serial.print(" IP address: ");
Serial.println(WiFi.localIP()); Serial.println(WiFi.localIP());
} }
/// MQTT /// MQTT
@ -67,8 +70,8 @@ Adafruit_MQTT_Client mqtt(&client, MQTT_SERVER, MQTT_SERVERPORT, MQTT_USERNAME,
/****************************** Feeds ***************************************/ /****************************** Feeds ***************************************/
// Setup a feed called 'pub_photocell' for publishing. // Setup a feed called 'pub_piedthon' for publishing.
Adafruit_MQTT_Publish pub_photocell = Adafruit_MQTT_Publish(&mqtt, "piedthon/debug"); Adafruit_MQTT_Publish pub_piedthon = Adafruit_MQTT_Publish(&mqtt, "piedthon/debug");
// Setup a feed called 'sub_piedthon' for subscribing to changes. // Setup a feed called 'sub_piedthon' for subscribing to changes.
Adafruit_MQTT_Subscribe sub_piedthon = Adafruit_MQTT_Subscribe(&mqtt, "piedthon/input"); Adafruit_MQTT_Subscribe sub_piedthon = Adafruit_MQTT_Subscribe(&mqtt, "piedthon/input");
@ -78,55 +81,55 @@ char *mqtt_message;
// Function to connect and reconnect as necessary to the MQTT server. // Function to connect and reconnect as necessary to the MQTT server.
void MQTT_connect() { void MQTT_connect() {
int8_t ret; int8_t ret;
// Stop if already connected. // Stop if already connected.
if (mqtt.connected()) { if (mqtt.connected()) {
return; return;
}
Serial.print("Connecting to MQTT... ");
uint8_t retries = 3;
while ((ret = mqtt.connect()) != 0) { // connect will return 0 for connected
Serial.println(mqtt.connectErrorString(ret));
Serial.println("Retrying MQTT connection in 5 seconds...");
mqtt.disconnect();
delay(5000); // wait 5 seconds
retries--;
if (retries == 0) {
// basically die and wait for WDT to reset me
Serial.println("Fail to connect to MQTT after 3 tries ... ");
return; // will recall the function
} }
}
Serial.print("Connecting to MQTT... "); Serial.println("MQTT Connected!");
uint8_t retries = 3;
while ((ret = mqtt.connect()) != 0) { // connect will return 0 for connected
Serial.println(mqtt.connectErrorString(ret));
Serial.println("Retrying MQTT connection in 5 seconds...");
mqtt.disconnect();
delay(5000); // wait 5 seconds
retries--;
if (retries == 0) {
// basically die and wait for WDT to reset me
Serial.println("Fail to connect to MQTT after 3 tries ... ");
return; // will recall the function
}
}
Serial.println("MQTT Connected!");
} }
/// PiedThon /// PiedThon
void messageToPiedthon(char *msg) { void messageToPiedthon(char *msg) {
// Clignotement ? deux en même temps ? // Clignotement ? deux en même temps ?
switch (msg[0]) { switch (msg[0]) {
// LOW active le relay // LOW active le relay
case '0' : // Rouge et Vert case '0' : // Rouge et Vert
digitalWrite(GPIO_ROUGE, LOW); digitalWrite(GPIO_ROUGE, LOW);
digitalWrite(GPIO_VERT, LOW); digitalWrite(GPIO_VERT, LOW);
break; break;
case '1' : // ROUGE case '1' : // ROUGE
digitalWrite(GPIO_VERT, HIGH); digitalWrite(GPIO_VERT, HIGH);
digitalWrite(GPIO_ROUGE, LOW); digitalWrite(GPIO_ROUGE, LOW);
break; break;
case '2' : // VERT case '2' : // VERT
digitalWrite(GPIO_VERT, LOW); digitalWrite(GPIO_VERT, LOW);
digitalWrite(GPIO_ROUGE, HIGH); digitalWrite(GPIO_ROUGE, HIGH);
break; break;
case '3' : // Rien case '3' : // Rien
digitalWrite(GPIO_ROUGE, HIGH); digitalWrite(GPIO_ROUGE, HIGH);
digitalWrite(GPIO_VERT, HIGH); digitalWrite(GPIO_VERT, HIGH);
break; break;
} }
} }
//// Bug workaround for Arduino 1.6.6, it seems to need a function declaration //// Bug workaround for Arduino 1.6.6, it seems to need a function declaration
@ -135,32 +138,65 @@ void messageToPiedthon(char *msg) {
void setup() { void setup() {
Serial.begin(115200);
delay(10);
Serial.println(F("Adafruit MQTT PiedThon remix"));
Serial.begin(115200); pinMode(GPIO_VERT, OUTPUT);
delay(10); pinMode(GPIO_ROUGE, OUTPUT);
Serial.println(F("Adafruit MQTT PiedThon remix"));
pinMode(GPIO_VERT, OUTPUT); randomSeed(analogRead(0));
pinMode(GPIO_ROUGE, OUTPUT);
setupWifi(); setupWifi();
// Setup MQTT subscription for onoff feed. // Setup MQTT subscription for onoff feed.
mqtt.subscribe(&sub_piedthon); mqtt.subscribe(&sub_piedthon);
ArduinoOTA.setHostname(ESP_HOSTNAME);
ArduinoOTA.onStart([]() { // switch off all the GPIOs during upgrade
digitalWrite(GPIO_ROUGE, HIGH);
digitalWrite(GPIO_VERT, HIGH);
});
ArduinoOTA.onProgress([](unsigned int progress, unsigned int total) {
/*if (progress % (total / 4) == 0) {
digitalWrite(GPIO_ROUGE, random(2) ? LOW : HIGH);
digitalWrite(GPIO_VERT, random(2) ? LOW : HIGH);
}*/
});
ArduinoOTA.onEnd([]() { // do a fancy thing with our board led at end
for (int i = 0; i < 10; i++) {
int RANDOM_GPIO = GPIO_ROUGE;
if (random(2)) { // random(2) returns 0 or 1 randomly
RANDOM_GPIO = GPIO_VERT;
}
digitalWrite(RANDOM_GPIO, !digitalRead(RANDOM_GPIO));
delay(400);
}
digitalWrite(GPIO_ROUGE, LOW);
digitalWrite(GPIO_VERT, LOW);
});
ArduinoOTA.onError([](ota_error_t error) {
(void)error;
ESP.restart();
});
ArduinoOTA.begin();
} }
void loop() { void loop() {
MQTT_connect();
ArduinoOTA.handle();
MQTT_connect(); Adafruit_MQTT_Subscribe *subscription;
while ((subscription = mqtt.readSubscription(5000))) {
Adafruit_MQTT_Subscribe *subscription; if (subscription == &sub_piedthon) {
while ((subscription = mqtt.readSubscription(5000))) { mqtt_message = (char *) sub_piedthon.lastread;
if (subscription == &sub_piedthon) { Serial.println("Got: " + String(mqtt_message));
mqtt_message = (char *) sub_piedthon.lastread; messageToPiedthon(mqtt_message);
Serial.println("Got: "+String(mqtt_message));
messageToPiedthon(mqtt_message);
}
} }
}
} }