sub_udp.cpp

#include "sub_udp.h"
#include "conf.h"
#include "sub_bmp.h"
#include "sub_enow.h"
#include "sub_json.h"
#include "sub_led.h"
#include "sub_wifi.h"

WiFiUDP Udp;
char udpBuf[255]; //buffer to hold incoming packet

void udp_begin()
{
	Udp.begin(8888);
}

int get_vcc(bool a)
{
	int ad = 0;
	for (int i = 0; i < 10; i++) ad += adc1_get_raw(conf.vccch);
	return (a ? 1 : conf.vcc) * ad / 10;
}

int get_vcc()
{
	return get_vcc(false);
}

int vcc2p(int gvcc)
{
	if (isnan(gvcc) || gvcc < 0 || gvcc > 4200) return 101;
	if (gvcc < 3300) return 0;
	const int svcc[12] = {0, 3300, 3400, 3500, 3600, 3700, 3800, 3900, 4000, 4100, 4150, 5000};
	const int pvcc[12] = {0,    1,   12,   23,   34,   45,   56,   67,   78,   89,  100,  101};
	int d = 0;
	for (int i = 0; i < 11; i++)
		if (gvcc >= svcc[i] && gvcc < svcc[i + 1])
		{
			d = i;
			break;
		}
	float ans = pvcc[d] + (gvcc - svcc[d]) / 9.00;
	return ans;
}

String get_answ(String san, String sav)
{
	if (san == "blink")
	{
		led_blinkudp();
		return "blink";
	}
	if (san == "bpm")
	{
		int savi = sav.toInt();
		if (savi > 10 && savi < 65500) state.bpm = savi;
		return String(state.bpm);
	}
	if (san == "brgn" || san == "brg")
	{
		String brgnwarn = "";
		if (sav == "m")
		{
			conf.brgn -= conf.brgn > 8 ? 4 : (conf.brgn > 1 ? 1 : 0);
		}
		if (sav == "p")
		{
			conf.brgn += conf.brgn < 128 ? (conf.brgn < 8 ? 1 : 4) : 0;
			if (conf.brgn == 128) brgnwarn = " max 128";
		}
		int savi = sav.toInt();
		if (savi >= 4 && savi <= 128)
		{
			if (savi == 8 && conf.brgn > 1)
				conf.brgn--;
			else
				conf.brgn = savi;
		}
		if (savi > 128) brgnwarn = " max 128 < " + String(savi);
		led_brgn();
		if (state.go == false) led_show();
		return String(conf.brgn, DEC) + brgnwarn;
	}
	if (san == "btmode")
	{
		if (sav == "one")
		{
			state.loop = false;
			return "mode one";
		}
		if (sav == "loop")
		{
			if (state.loop == false)
			{
				state.loop = true;
				state.next = true;
			}
			return "mode loop";
		}
	}
	if (san == "cmt")
	{
		json_save();
		return "save ok";
	}
	if (san == "conf")
	{
		String ans;
		ans += "prog:" + String(state.progname);
		ans += " macslen:" + String(conf.macslen) + " ";
		char bufm[14];
		for (int i = 0; i < conf.macslen; i++)
		{
			sprintf(bufm, "%02X%02X%02X%02X%02X%02X,", conf.macs[i][0],conf.macs[i][1],conf.macs[i][2],conf.macs[i][3],conf.macs[i][4],conf.macs[i][5]);
			ans += String(bufm);
		}
		return ans;
	}
	if (san == "cont")
	{
		int savi = sav.toInt();
		savi = savi > 128 ? 128 : savi;
		led_calccont(savi);
		conf.cont = savi;
		return "cont set ok " + sav;
	}
	if (san == "delconf")
	{
		json_del();
		conf.macslen = 0;
		return F("conf deleted");	
	}
	if (san == "drip")
	{
		IPAddress myip = WiFi.localIP();
		led_drawip(myip[3]);
		return "draw ip";
	}
	if (san == "enow")
	{
		if (sav == "2")
		{
			conf.bt = false;
			conf.skwf = true;
			conf.enow = true;
			conf.enowone = true;
			json_save();
			return "enow on";
		}
	}
	if (san == "format")
	{
		FILESYSTEM.end();
		bool f = FILESYSTEM.format();
		bool b = FILESYSTEM.begin();
		json_save();
		if (conf.mode == 13) bmp_savecost();
		return "format " + String(f ? "ok," : "fail,") + String(b ? " begin" : " not begin");
	}
	if (san == "free")
	{
		int tBytes = FILESYSTEM.totalBytes(); int uBytes = FILESYSTEM.usedBytes();
		return String(tBytes - uBytes, DEC);
	}
	if (san == "go")
	{
		if (state.go == false)
		{
			led_clear();
			if (state.calcmax)
			{
				bmp_max();
				state.calcmax = false;
			}
			if (state.maxbmp > 0) state.go = true;
		}
		if (state.maxbmp > 0)
		{
			//return ("runned " + String(conf.wait, DEC) + " " + String(conf.brgn, DEC) + " " + String(stat.whdr, DEC) + " " + String(/*whodr1*/0, DEC) + " " + String(/*whodr2*/0, DEC) + " " + String(/*whodr3*/0, DEC) + " " + String(stat.maxbmp, DEC) + " " + String(/*eNow*/0, DEC));
			return ("runned "   + String(conf.wait, DEC) + " " + String(conf.brgn, DEC) + " " + String(state.whdr, DEC) +                     " 0"                             " 0"                             " 0"        " " + String(state.maxbmp, DEC) +       " 0");
		}
		else
		{
			return "no pics";
		}
	}
	if (san == "heap")
	{
		return String(ESP.getFreeHeap(), DEC);
	}
	if (san == "info")
	{
		return (		"w " + String(conf.wait, DEC) +" b "+ String(conf.brgn, DEC) +" m "+ String(state.whdr, DEC) +" p "+ String(state.currprog + 1, DEC) + "/" + String(state.maxprog, DEC) + " b " + String(state.currbmp, DEC) + "/" + String(state.maxbmp, DEC));
	}
	if (san == "ip")
	{
		IPAddress myip = WiFi.localIP();
		return (String(myip[0], DEC) + "." + String(myip[1], DEC) + "." + String(myip[2], DEC) + "." + String(myip[3], DEC));
	}
	if (san == "leds")
	{
		if (sav == "1")
		{
			return String(conf.leds);
		}
		else
		{
			int savi = sav.toInt();
			conf.leds = savi;
			led_init();
			json_save();
			return "saved " + String(conf.leds);
		}
	}
	if (san == "mac")
	{
		return String(WiFi.macAddress());
	}
	if (san == "maca")
	{
		uint8_t mc[6];
		char mcb[24];
		WiFi.softAPmacAddress(mc);
		sprintf(mcb, "%02X%02X%02X%02X%02X%02X", mc[0], mc[1], mc[2], mc[3], mc[4], mc[5]);
		return String(mcb);
	}
	if (san == "macor")
	{
		int px = enow_getorder();
		for (int i = 0; i < 22; i++) led_setpxall(i, black);
		for (int i = 0; i < px; i++) led_setpxall(i, green);
		led_show();
		return "mac order";
	}
	if (san == "macs")
	{
		if (sav == "1")
		{
			conf.macslen = 0;
			memset(conf.macs, 255, 16 * 6);
			json_save();
			return "mac_ap removed";
		}
		else
		{
			uint8_t macb[6];
			mac_decode(sav, macb);
			char exs = 0;
			for (int i = 0; i < conf.macslen; i++)
			{
				if (strncmp((char *)macb, (char *)conf.macs[i], 6) == 0) {exs = 1; break;}
			}
			if (exs == 1)
			{
				Serial.print("mac exist ");
				for (int m = 0; m < 6; m++) Serial.printf("%02X", conf.macs[0][m]);
				Serial.println();
				return "mac_ap";
			}
			if (conf.macslen < 16)
			{
				for (int i = 0; i < 6; i++) conf.macs[conf.macslen][i] = macb[i];
				conf.macslen += 1;
				json_save();
				Serial.print("mac_ap saved ");
				for (int m = 0; m < 6; m++) Serial.printf("%02X", macb[m]);
				Serial.println();
				return "mac_ap";
			}
			else
			{
				Serial.println("mac list full\n");
				return "mac_ap";
			}
		}
	}
	if (san == "macun")
	{
		if (sav == "echo")
		{
			udp_sendmac();
			uint8_t uech[8] = "macun=1";
			udp_echo(uech, 8);
			return "mac sended all";
		}
		if (sav == "unsn")
		{
			conf.macslen = 0;
			memset(conf.macs, 255, 16 * 6);
			json_save();
			return "mac_ap removed";
		}
		else
		{
			udp_sendmac();
			return "mac sended";
		}
	}
	if (san == "mode")
	{
		if (sav == "1")
		{
			switch (conf.mode)
			{
				case 10: return "poi 6";
				case 11: return "mask1";
				case 12: return "mask2";
				case 13: return "свой";
				case 14: return "poi 4";
				default: return "err";
			}
		}
		else
		{
			int savi = sav.toInt();
			switch (savi)
			{
				case 10: 
					conf.mode = 10;
				#ifndef ARDUINO_ESP32C3_DEV
					conf.leds = 12;
				#else
					conf.leds = 32;
				#endif
					json_save();
					return "poi 6";
				case 11: conf.mode = 11; conf.leds = 930; json_save(); return "mask1";
				case 12: conf.mode = 12; conf.leds = 980; json_save(); return "mask2";
				case 13: conf.mode = 13; conf.leds = 850; json_save(); return "свой";
				case 14: conf.mode = 14; conf.leds = 20;  json_save(); return "poi 4";
				case 3 : state.whdr = 3; json_save(); return "files";
				case 4 : state.whdr = 4; json_save(); return "prog";
				default: return "err mode " + sav;
			}
		}
	}
	if (san == "modefile")
	{
		int savi = sav.toInt();
		if (savi > 0 && savi <= state.maxbmp)
		{
			state.setbmp = savi;
		}
		return "file " + sav;
	}
	if (san == "modeone")
	{
		int savi = sav.toInt();
		if (savi > 0 && savi <= state.maxbmp)
		{
			state.setbmp = savi;
			return "next " + sav;
		}
		else
		{
			state.next = true;
			return "next ?";
		}
	}
	if (san == "modp")
	{
		if (sav == "m")
		{
			state.loop = false;
			if (state.currbmp > 1) state.setbmp = state.currbmp - 1; else state.setbmp = state.maxbmp;
			return "prev " + String(state.setbmp, DEC);
		}
		if (sav == "p")
		{
			state.loop = false;
			if (state.currbmp < state.maxbmp) state.setbmp = state.currbmp + 1; else state.setbmp = 1;
			return "next " + String(state.setbmp, DEC);
		}
	}
	if (san == "pins")
	{
		char pinb[50];
		sav.toCharArray(pinb, 50);
		int pins[6];
		int pinc = sscanf(pinb, "%d%d%d%d%d%d", &pins[0], &pins[1], &pins[2], &pins[3], &pins[4], &pins[5]);
		if (pinc != 6) return "error numbers";
		for (int i = 0; i < 6; i++)
		{
			if (pins[i] > 34 || pins[i] == 1 || pins[i] == 2 || pins[i] == 3) return "error pin " + String(i)  + " cannot be " + String(pins[i]);
			for (int j = i + 1 ; j < 6; j++) if (pins[i] == pins[j]) return "error pin " + String(pins[i]) + " repeated";
		}
		Preferences preferences;
		preferences.begin("pins", false);
		preferences.putUInt("p0", pins[0]);
		preferences.putUInt("p1", pins[1]);
		preferences.putUInt("p2", pins[2]);
		preferences.putUInt("p3", pins[3]);
		preferences.putUInt("p4", pins[4]);
		preferences.putUInt("p5", pins[5]);
		preferences.end();
		return "saved " + sav;
	}
	if (san == "prg")
	{
		String prgwarn = "";
		if (sav == "m")
		{
			if (state.currprog > 0)
			{
				state.currprog--;
				state.progname = String(state.proglist[state.currprog]);
			}
		}
		if (sav == "p")
		{
			if (state.currprog < state.maxprog - 1)
			{
				state.currprog++;
				state.progname = String(state.proglist[state.currprog]);
			}
		}
		if (sav == "n")
		{
			state.currprog = 0;
			state.progname = "no_prog";
		}
		int savi = sav.toInt();
		if (savi > 0 && savi <= state.maxprog)
		{
			state.currprog = savi - 1;
			state.progname = String(state.proglist[state.currprog]);
		}
		return state.progname;
	}
	if (san == "prog")
	{
		return state.progname;
	}
	if (san == "restart")
	{
		state.go = false;
		return "restart";
	}
	if (san == "skwf")
	{
		if (sav == "1")
		{
			conf.skwf = true;
			return "skip wait wifi on";
		}
		else
		{
			conf.skwf = false;
			return "skip wait wifi off";
		}
	}
	if (san == "stp")
	{
		state.go = false;
		led_clear();
		led_show();
		return "stop";
	}
	if (san == "uptime")
	{
		if (sav == "1") state.uptime = 1;
		return String(state.uptime);
	}
	if (san == "vcc")
	{
		int savi = sav.toInt();
		if (savi == 1)
		{
			int gvcc = get_vcc();
			int pr = vcc2p(gvcc);
			if (state.go == false) led_drawvcc();
			return (String(pr, DEC) + "% (" + String(gvcc, DEC) + "mV) " + WiFi.RSSI());
		}
		else if (savi >= 100 && savi <= 5000)
		{
			conf.vcc = (float)savi / get_vcc(true);
			json_save();
			return String("saved " + String(conf.vcc, DEC));
		}
		else
		{
			return "vcc fail, use 100-5000 not " + sav;
		}
	}
	if (san == "ver")
	{
		uint8_t mc[6];
		WiFi.softAPmacAddress(mc);
		char buf[45] = {0};
		sprintf(buf, "%s %s_%02X%02X%02X", conf.ver, conf.wpref, mc[3], mc[4], mc[5]);
		return String(buf);
	}
	if (san == "wait" || san == "spd")
	{
		if (sav == "m")
		{
			conf.wait -= conf.wait > 0 ? 1 : 0;
		}
		if (sav == "p")
		{
			conf.wait += conf.wait < 200 ? 1 : 0;
		}
		int savi = sav.toInt();
		if (savi > 0 && savi < 200)
		{
			conf.wait = savi;
		}
		return String(conf.wait);
	}
	if (san == "wfaps")
	{
		int scn = WiFi.scanNetworks();
		String ans = "";
		if (scn > 0)
			for (int scni = 0; scni < scn; ++scni)
			{
				ans += "<label><input type='radio' name='ssid' value='";
				ans += WiFi.SSID(scni) + "'>" + WiFi.SSID(scni);
				ans += "</label> [";
				ans += String(WiFi.RSSI(scni), DEC) + "]<br>\n";
			}
			else
			{
				ans = "no ap found";
			}
		return ans;
	}
	if (san == "wpref")
	{
		if (sav == "0")
		{
			return "name=" + conf.wpref;
		}
		else if (sav == "1")
		{
			conf.wpref = "LedHDxx";
			json_save();
			return "prefix reseted";
			
		}
		else
		{
			conf.wpref = sav;
			json_save();
			return "prefix saved";
		}
	}
	return "no ans " + san + "=" + sav;
}

void udp_poll()
{
	int packetSize = Udp.parsePacket();
	if (packetSize)
	{
		int len = Udp.read(udpBuf, 255);
		if (len > 0)
		{
			udpBuf[len] = 0;
			String q = String(udpBuf);
			int dei = q.indexOf("=");
			if (dei != -1)
			{
				String sn = q.substring(0, dei);
				String sv = q.substring(dei + 1);
				String ans = get_answ(sn, sv);
				int len = ans.length() + 1;
				char ansBuf[len] = {0};
				ans.toCharArray(ansBuf, len);
				Udp.beginPacket(Udp.remoteIP(), 60201);
				uint8_t an[255] = {0};
				memcpy(an, ansBuf, 254);
				Udp.write(an, ans.length() > 255 ? 255 : ans.length());
				Udp.endPacket();
				if (ans == "restart")
				{
					led_clear();
					delay(500);
					ESP.restart();
				}
			}
		}
	}
}

void udp_sendip()
{
	IPAddress myip = WiFi.localIP();
	char ansBuf[21] = {0};
	sprintf(ansBuf, "myip %d.%d.%d.%d", myip[0], myip[1], myip[2], myip[3]);
	uint8_t uans[21] = {0};
	memcpy(uans, ansBuf, 21);
	myip[3] = 255;
	Udp.beginPacket(myip, 60201);
	Udp.write(uans, 21);
	Udp.endPacket();
}

void udp_sendmac()
{
	IPAddress myip = WiFi.localIP();
	myip[3] = 255;
	uint8_t mc[6];
	WiFi.softAPmacAddress(mc);
	char mcb[18] = {0};
	sprintf(mcb, "macs=%02X%02X%02X%02X%02X%02X", mc[0], mc[1], mc[2], mc[3], mc[4], mc[5]);
	uint8_t umcb[18] = {0};
	memcpy(umcb, mcb, 18);
	Udp.beginPacket(myip, 8888);
	Udp.write(umcb,18);
	Udp.endPacket();
}

void udp_echo(uint8_t *buf, size_t s)
{
	IPAddress myip = WiFi.localIP();
	myip[3] = 255;
	Udp.beginPacket(myip, 8888);
	Udp.write(buf, s);
	Udp.endPacket();
}