ESP32_ChinaDieselHeater_Control/Arduino/BTCDieselHeater/BTCWebServer.cpp

/*
 * This file is part of the "bluetoothheater" distribution 
 * (https://gitlab.com/mrjones.id.au/bluetoothheater) 
 *
 * Copyright (C) 2018  Ray Jones <ray@mrjones.id.au>
 * Copyright (C) 2018  James Clark
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 * 
 */


#include "BTCWebServer.h"
#include "DebugPort.h"
#include "TxManage.h"
#include "helpers.h"
#include "pins.h"
#include "Index.h"
#include <ArduinoJson.h>
#include <map>


WebServer server(80);
WebSocketsServer webSocket = WebSocketsServer(81);
bool bRxWebData = false;
bool bTxWebData = false;

DynamicJsonBuffer jsonBuffer(512);   // create a JSON buffer on the heap

class CModerator {
  std::map<const char*, float> fMemory;
  std::map<const char*, int> iMemory;
public:
  bool check(const char* name, float value);
  bool check(const char* name, int value);
	void reset();
};

void
CModerator::reset() 
{
  // install invalid values, retain maps (memory defrag reasons)
	for(auto it = fMemory.begin(); it != fMemory.end(); ++it)  it->second = -100;
	for(auto it = iMemory.begin(); it != iMemory.end(); ++it)  it->second = -100;
}

bool
CModerator::check(const char* name, float value) 
{
	bool retval = true;
	auto it = fMemory.find(name);
  if(it != fMemory.end()) {
		retval = it->second != value;
		it->second = value;
	}
	else {
		fMemory[name] = value;
	}
	return retval;
}

bool
CModerator::check(const char* name, int value) 
{
	bool retval = true;
	auto it = iMemory.find(name);
  if(it != iMemory.end()) {
		retval = it->second != value;
		it->second = value;
	}
	else {
		iMemory[name] = value;
	}
	return retval;
}

CModerator Moderator;

const int led = 13;

void handleRoot() {
	String s = MAIN_PAGE; //Read HTML contents
	server.send(200, "text/html", s); //Send web page
}

void handleNotFound() {
	digitalWrite(led, 1);
	String message = "File Not Found\n\n";
	message += "URI: ";
	message += server.uri();
	message += "\nMethod: ";
	message += (server.method() == HTTP_GET) ? "GET" : "POST";
	message += "\nArguments: ";
	message += server.args();
	message += "\n";
	for (uint8_t i = 0; i < server.args(); i++) {
		message += " " + server.argName(i) + ": " + server.arg(i) + "\n";
	}
	server.send(404, "text/plain", message);
	digitalWrite(led, 0);
}

void initWebServer(void) {

	
	if (MDNS.begin("BTCHeater")) {
		DebugPort.println("MDNS responder started");
	}
	
	server.on("/", handleRoot);
	server.onNotFound(handleNotFound);

	server.begin();
	webSocket.begin();
	webSocket.onEvent(webSocketEvent);
	DebugPort.println("HTTP server started");

}
unsigned char cVal;

bool doWebServer(void) {
	static unsigned long lastTx = 0;
	static int prevNumClients;
	webSocket.loop();
	server.handleClient();

	int numClients = webSocket.connectedClients();
	if(numClients != prevNumClients) {
		prevNumClients = numClients;
		Moderator.reset();   // force full update of params if number of clients change
		DebugPort.println("Changed number of web clients, resetting history");
	}

	if(numClients) {
		if(millis() > lastTx) {   // moderate the delivery of new messages - we simply cannot send every pass of the main loop!
			lastTx = millis() + 100;
			bool bSend = false;

			JsonObject& root = jsonBuffer.createObject();
			float tidyTemp = int(getActualTemperature() * 10) * 0.1f;  // round to 0.1 resolution (hopefully!)
			if(Moderator.check("CurrentTemp", tidyTemp)) {
				root.set("CurrentTemp", tidyTemp);
				bSend = true;
			}
			if(Moderator.check("RunState", getHeaterInfo().getRunState())) {
				root.set("RunState", getHeaterInfo().getRunState());
				bSend = true;
			}
			if(Moderator.check("DesiredTemp", getHeaterInfo().getTemperature_Desired())) {
				root.set("DesiredTemp", getHeaterInfo().getTemperature_Desired());
				bSend = true;
			}

			if(bSend) {
				bTxWebData = true;
      	String jsonToSend;
				root.printTo(jsonToSend);
      	webSocket.broadcastTXT(jsonToSend);
			}
		}
		return true;
	}
  return false;
}

void webSocketEvent(uint8_t num, WStype_t type, uint8_t * payload, size_t length) {
	if (type == WStype_TEXT) {
		bRxWebData = true;
		char cmd[256];
		memset(cmd, 0, 256);
		for (int i = 0; i < length && i < 256; i++) {
			cmd[i] = payload[i];
		}
//    DebugPort.println(cmd);
		interpretJsonCommand(cmd);  // send to the main heater controller decode routine
  }
}

bool hasWebClientSpoken(bool reset)
{
	bool retval = bRxWebData;
	if(reset)
  	bRxWebData = false;
	return retval;
}

bool hasWebServerSpoken(bool reset)
{
	bool retval = bTxWebData;
	if(reset)
		bTxWebData = false;
	return retval;
}


void interpretJsonCommand(char* pLine)
{
  if(strlen(pLine) == 0)
    return;

  DebugPort.print("JSON parse... "); Serial.print(pLine);

	JsonObject& obj = jsonBuffer.parseObject(pLine);
	if(!obj.success()) {
		DebugPort.println(" FAILED");
		return;
	}
	DebugPort.println(" OK"); 

	JsonObject::iterator it;
	for(it = obj.begin(); it != obj.end(); ++it) {

		if(strcmp("DesiredTemp", it->key) == 0) {
      reqTemp(it->value.as<unsigned char>());
		}
		else if(strcmp("RunState", it->key) == 0) {
			if(it->value.as<unsigned char>()) {
	      requestOn();
			}
			else {
	      requestOff();
			}
		}
		else if(strcmp("PumpMin", it->key) == 0) {
			setPumpMin(it->value.as<float>());
		}
		else if(strcmp("PumpMax", it->key) == 0) {
			setPumpMax(it->value.as<float>());
		}
		else if(strcmp("FanMin", it->key) == 0) {
			setFanMin(it->value.as<short>());
		}
		else if(strcmp("FanMax", it->key) == 0) {
			setFanMax(it->value.as<short>());
		}
		else if(strcmp("Thermostat", it->key) == 0) {
			setThermostatMode(it->value.as<unsigned char>());
		}
	}
}