msurguy/ESPSockets.ino

## ESPSockets.ino
/*

When the sketch is uploaded, the ESP8266 will reboot and create its own access point called AutoConnect AP. Connecting to that AP will bring up the
list of available access points. Enter Username and PW for the access point and ESP module will save the credentials for you.

When the module is restarted, it will connect to the AP that you chose and should be available on the network. You can get its IP from the Serial monitor
or use mDNS library (it's uncommented by default, along with all usage of the library) to make the module discoverable.

To change colors on the module (Neopixels on Pin 2), simply go to the root URL of the web server.

*/

#include <Arduino.h>

#include <ESP8266WiFi.h>
#include <WiFiClient.h>
#include <WebSocketsServer.h>
#include <ESP8266WebServer.h>
//#include <ESP8266mDNS.h>
#include <FastLED.h>
#include <ESP8266HTTPUpdateServer.h>
#include <WiFiManager.h>

#include <Hash.h>

WebSocketsServer webSocket = WebSocketsServer(81);
ESP8266WebServer server = ESP8266WebServer(80);

ESP8266HTTPUpdateServer httpUpdater;

#define USE_SERIAL Serial
#define NUM_LEDS 16
CRGB leds[NUM_LEDS];

uint8_t gHue = 0;
int mode = 0;

void webSocketEvent(uint8_t num, WStype_t type, uint8_t * payload, size_t lenght) {

    switch(type) {
        case WStype_DISCONNECTED:
            USE_SERIAL.printf("[%u] Disconnected!\n", num);
            break;
        case WStype_CONNECTED:
            {
                IPAddress ip = webSocket.remoteIP(num);
                USE_SERIAL.printf("[%u] Connected from %d.%d.%d.%d url: %s\n", num, ip[0], ip[1], ip[2], ip[3], payload);
                mode = 0;
        				// send message to client
        				webSocket.sendTXT(num, "Connected");
            }
            break;
        case WStype_TEXT:
            USE_SERIAL.printf("[%u] get Text: %s\n", num, payload);
            //USE.SERIAL.printf("First character of the paylod %s\n", payload[0]);
            for(int i = 0; i < NUM_LEDS; i++) {
              //(int) strtol( &payload, NULL, 16);
              uint32_t rgb = (uint32_t) strtol((const char *) &payload[1], NULL, 16);
              leds[i] = rgb; //(long) strtol(&payload[1], NULL, 16);
            }

            // send message to client
            // webSocket.sendTXT(num, "message here");

            // send data to all connected clients
            // webSocket.broadcastTXT("message here");
            break;
        case WStype_BIN:
            USE_SERIAL.printf("[%u] get binary lenght: %u\n", num, lenght);
            hexdump(payload, lenght);

            // send message to client
            // webSocket.sendBIN(num, payload, lenght);
            break;
    }

}

void setup() {
    USE_SERIAL.begin(115200);
    USE_SERIAL.setDebugOutput(true);

    USE_SERIAL.println();

    WiFiManager wifiManager;
    wifiManager.autoConnect("AutoConnectAP");

    delay(2000);

    USE_SERIAL.println("Connection established!");
    FastLED.addLeds<NEOPIXEL, 2>(leds, NUM_LEDS);

    webSocket.begin();
    webSocket.onEvent(webSocketEvent);
    //if(MDNS.begin("esp8266")) {
    //    USE_SERIAL.println("MDNS responder started");
    //}

    // handle index
    server.on("/", []() {
        // send index.html
        server.send(200, "text/html", "<html><head><script>var connection = new WebSocket('ws://'+location.hostname+':81/', ['arduino']);connection.onopen = function () {  connection.send('Connect ' + new Date()); }; connection.onerror = function (error) {    console.log('WebSocket Error ', error);};connection.onmessage = function (e) {  console.log('Server: ', e.data);};function sendRGB() {  var r = parseInt(document.getElementById('r').value).toString(16);  var g = parseInt(document.getElementById('g').value).toString(16);  var b = parseInt(document.getElementById('b').value).toString(16);  if(r.length < 2) { r = '0' + r; }   if(g.length < 2) { g = '0' + g; }   if(b.length < 2) { b = '0' + b; }   var rgb = '#'+r+g+b;    console.log('RGB: ' + rgb); connection.send(rgb); }</script></head><body>LED Control:<br/><br/>R: <input id=\"r\" type=\"range\" min=\"0\" max=\"255\" step=\"1\" oninput=\"sendRGB();\" /><br/>G: <input id=\"g\" type=\"range\" min=\"0\" max=\"255\" step=\"1\" oninput=\"sendRGB();\" /><br/>B: <input id=\"b\" type=\"range\" min=\"0\" max=\"255\" step=\"1\" oninput=\"sendRGB();\" /><br/></body></html>");
    });

    httpUpdater.setup(&server);
    server.begin();

    // Add service to MDNS
    //MDNS.addService("http", "tcp", 80);
    //MDNS.addService("ws", "tcp", 81);
}

void loop() {
    webSocket.loop();
    server.handleClient();
    FastLED.show();
    FastLED.delay(1000/120);
}
	/*

	When the sketch is uploaded, the ESP8266 will reboot and create its own access point called AutoConnect AP. Connecting to that AP will bring up the
	list of available access points. Enter Username and PW for the access point and ESP module will save the credentials for you.

	When the module is restarted, it will connect to the AP that you chose and should be available on the network. You can get its IP from the Serial monitor
	or use mDNS library (it's uncommented by default, along with all usage of the library) to make the module discoverable.

	To change colors on the module (Neopixels on Pin 2), simply go to the root URL of the web server.

	*/

	#include <Arduino.h>

	#include <ESP8266WiFi.h>
	#include <WiFiClient.h>
	#include <WebSocketsServer.h>
	#include <ESP8266WebServer.h>
	//#include <ESP8266mDNS.h>
	#include <FastLED.h>
	#include <ESP8266HTTPUpdateServer.h>
	#include <WiFiManager.h>

	#include <Hash.h>

	WebSocketsServer webSocket = WebSocketsServer(81);
	ESP8266WebServer server = ESP8266WebServer(80);

	ESP8266HTTPUpdateServer httpUpdater;

	#define USE_SERIAL Serial
	#define NUM_LEDS 16
	CRGB leds[NUM_LEDS];

	uint8_t gHue = 0;
	int mode = 0;

	void webSocketEvent(uint8_t num, WStype_t type, uint8_t * payload, size_t lenght) {

	switch(type) {
	case WStype_DISCONNECTED:
	USE_SERIAL.printf("[%u] Disconnected!\n", num);
	break;
	case WStype_CONNECTED:
	{
	IPAddress ip = webSocket.remoteIP(num);
	USE_SERIAL.printf("[%u] Connected from %d.%d.%d.%d url: %s\n", num, ip[0], ip[1], ip[2], ip[3], payload);
	mode = 0;
	// send message to client
	webSocket.sendTXT(num, "Connected");
	}
	break;
	case WStype_TEXT:
	USE_SERIAL.printf("[%u] get Text: %s\n", num, payload);
	//USE.SERIAL.printf("First character of the paylod %s\n", payload[0]);
	for(int i = 0; i < NUM_LEDS; i++) {
	//(int) strtol( &payload, NULL, 16);
	uint32_t rgb = (uint32_t) strtol((const char *) &payload[1], NULL, 16);
	leds[i] = rgb; //(long) strtol(&payload[1], NULL, 16);
	}

	// send message to client
	// webSocket.sendTXT(num, "message here");

	// send data to all connected clients
	// webSocket.broadcastTXT("message here");
	break;
	case WStype_BIN:
	USE_SERIAL.printf("[%u] get binary lenght: %u\n", num, lenght);
	hexdump(payload, lenght);

	// send message to client
	// webSocket.sendBIN(num, payload, lenght);
	break;
	}

	}

	void setup() {
	USE_SERIAL.begin(115200);
	USE_SERIAL.setDebugOutput(true);

	USE_SERIAL.println();

	WiFiManager wifiManager;
	wifiManager.autoConnect("AutoConnectAP");

	delay(2000);

	USE_SERIAL.println("Connection established!");
	FastLED.addLeds<NEOPIXEL, 2>(leds, NUM_LEDS);

	webSocket.begin();
	webSocket.onEvent(webSocketEvent);
	//if(MDNS.begin("esp8266")) {
	// USE_SERIAL.println("MDNS responder started");
	//}

	// handle index
	server.on("/", []() {
	// send index.html
	server.send(200, "text/html", "<html><head><script>var connection = new WebSocket('ws://'+location.hostname+':81/', ['arduino']);connection.onopen = function () { connection.send('Connect ' + new Date()); }; connection.onerror = function (error) { console.log('WebSocket Error ', error);};connection.onmessage = function (e) { console.log('Server: ', e.data);};function sendRGB() { var r = parseInt(document.getElementById('r').value).toString(16); var g = parseInt(document.getElementById('g').value).toString(16); var b = parseInt(document.getElementById('b').value).toString(16); if(r.length < 2) { r = '0' + r; } if(g.length < 2) { g = '0' + g; } if(b.length < 2) { b = '0' + b; } var rgb = '#'+r+g+b; console.log('RGB: ' + rgb); connection.send(rgb); }</script></head><body>LED Control:<br/><br/>R: <input id=\"r\" type=\"range\" min=\"0\" max=\"255\" step=\"1\" oninput=\"sendRGB();\" /><br/>G: <input id=\"g\" type=\"range\" min=\"0\" max=\"255\" step=\"1\" oninput=\"sendRGB();\" /><br/>B: <input id=\"b\" type=\"range\" min=\"0\" max=\"255\" step=\"1\" oninput=\"sendRGB();\" /><br/></body></html>");
	});

	httpUpdater.setup(&server);
	server.begin();

	// Add service to MDNS
	//MDNS.addService("http", "tcp", 80);
	//MDNS.addService("ws", "tcp", 81);
	}

	void loop() {
	webSocket.loop();
	server.handleClient();
	FastLED.show();
	FastLED.delay(1000/120);
	}