Chatter2-Firmware.ino

#include <Arduino.h>
#include <CircuitOS.h>
#include <Chatter2.h>
#include <Loop/LoopManager.h>
#include <lvgl.h>
#include "src/InputChatter.h"
#include "src/FSLVGL.h"
#include "src/ChatterTheme.h"
#include <SPIFFS.h>
#include "src/Screens/MainMenu.h"
#include "src/Storage/Storage.h"
#include "src/Services/LoRaService.h"
#include "src/Services/MessageService.h"
#include "src/Elements/IntroScreen.h"
#include "src/Interface/Pics.h"
#include "src/Services/ProfileService.h"
#include "src/Screens/UserHWTest.h"
#include <Settings.h>
#include "src/Services/SleepService.h"
#include "src/Services/ShutdownService.h"
#include "src/Services/BuzzerService.h"
#include "src/JigHWTest/JigHWTest.h"
#include "src/Games/GameEngine/Game.h"

lv_disp_draw_buf_t drawBuffer;
Display* display;

void lvglFlush(lv_disp_drv_t* disp, const lv_area_t* area, lv_color_t* color_p){
	uint32_t w = (area->x2 - area->x1 + 1);
	uint32_t h = (area->y2 - area->y1 + 1);

	TFT_eSPI &tft = *display->getTft();
	tft.startWrite();
	tft.setAddrWindow(area->x1, area->y1, w, h);
	tft.pushColors(&color_p->full, w * h, true);
	tft.endWrite();
	lv_disp_flush_ready(disp);
}

struct {
	UID_t uid;
	const char* nickname;
	uint8_t avatar;
	uint16_t hue;
} const Chatters[] = {
		{ 0xdcba4bf23a08, "Grle v0.5", 1, 0 },
		{ 0x48275612cfa4, "Choki v0.6", 2, 80 },
		{ 0x88a61312cfa4, "Zeus v0.6", 5, 90 },
		{ 0x48e4e0e6e2e0, "Johnny v0.4", 3, 160 },
		{ 0xe4c14bf23a08, "Rus v0.5", 4, 240 },
		{ 0x38c24bf23a08, "Crni v0.5", 6, 40 },
};

void loadMock(bool clear = false){
	if(clear){
		for(UID_t uid : Storage.Messages.all()) Storage.Messages.remove(uid);
		for(UID_t uid : Storage.Friends.all()) Storage.Friends.remove(uid);
		for(UID_t uid : Storage.Convos.all()) Storage.Convos.remove(uid);
	}

	for(const auto& chatter : Chatters){
		Friend fren;
		fren.uid = chatter.uid;
		fren.profile.avatar = chatter.avatar;
		fren.profile.hue = (uint8_t) (chatter.hue / 2);
		strncpy(fren.profile.nickname, chatter.nickname, 15);
		Storage.Friends.add(fren);

		if(chatter.uid == ESP.getEfuseMac()) continue;

		Convo convo;
		convo.uid = fren.uid;

		const int count = LoRa.rand() % 5;
		for(int i = 0; i < count; i++){
			Message message;
			if(LoRa.rand() % 2){
				message.setText("Hello!");
			}else{
				message.setPic(LoRa.rand() % NUM_PICS);
			}

			message.uid = LoRa.randUID();
			message.received = LoRa.rand() % 2;
			message.outgoing = LoRa.rand() % 2;
			convo.messages.push_back(message.uid);
			Storage.Messages.add(message);
		}

		Storage.Convos.add(convo);
	}

	// Reinit my profile
	Profiles.begin();
}

void printData(){
	for(UID_t uid : Storage.Friends.all()){
		Friend fren = Storage.Friends.get(uid);
		printf("%llx | Fren: %s | Hue: %d | Avatar: %d\n", fren.uid, fren.profile.nickname, fren.profile.hue, fren.profile.avatar);

		Convo convo = Storage.Convos.get(uid);
		printf("%llx | Convo: %d messages\n", convo.uid, convo.messages.size());

		for(UID_t uid : convo.messages){
			Message message = Storage.Messages.get(uid);
			printf("%llx | Message: %s\n", message.uid, message.getText().c_str());
		}
	}
}

void boot(){
	lv_timer_handler();

	display->getBaseSprite()->drawIcon(SPIFFS.open("/splash.raw"), 0, 0, 160, 128);
	display->commit();

	Chatter.fadeIn();

	FSLVGL::loadCache();

	Storage.begin();
	Messages.begin();

	LoRa.begin();
	Profiles.begin();

	//loadMock(true);
	//printData();

	Sleep.begin();


	auto intro = new IntroScreen([](){
		Shutdown.begin();
		Buzz.begin();
	});

	lv_timer_handler();

	intro->start();
}

bool jigManual = false;
bool checkJig(){
	char buf[7];
	int wp = 0;

	uint32_t start = millis();
	while(millis() - start < 1000){
		while(Serial.available()){
			buf[wp] = Serial.read();
			wp = (wp + 1) % 7;

			for(int i = 0; i < 7; i++){
				int match = 0;
				jigManual = false;
				static const char* target = "JIGTEST";

				for(int j = 0; j < 7; j++){
					match += buf[(i+j) % 7] == target[j];
					if(match == 6 && j == 6 && buf[(i+j) % 7] == 'M'){
						jigManual = true;
						match++;
					}
				}

				if(match == 7) return true;
			}
		}
	}

	return false;
}

void initLog(){
	esp_log_level_set("*", ESP_LOG_NONE);
	return;

	const static auto tags = { "*" };

	for(const char* tag : tags){
		esp_log_level_set(tag, ESP_LOG_VERBOSE);
	}
}

void setup(){
	Serial.begin(115200);

	randomSeed(analogRead(BATTERY_PIN) * 13 + analogRead(BATTERY_PIN) * 7 + 2);

	LoopManager::reserve(24);

	Chatter.begin(false);
	display = Chatter.getDisplay();

	initLog();

	if(checkJig()){
		auto test = new JigHWTest(display, jigManual);
		test->start();
		for(;;);
	}

	if(Battery.getPercentage() == 0){
		LoRa.initStateless();
		Sleep.turnOff();
		for(;;);
	}

	Piezo.setMute(!Settings.get().sound);

	lv_init();
	lv_disp_draw_buf_init(&drawBuffer, display->getBaseSprite()->getBuffer(), NULL, 160 * 128);

	new FSLVGL(SPIFFS, 'S');

	static lv_disp_drv_t displayDriver;
	lv_disp_drv_init(&displayDriver);
	displayDriver.hor_res = 160;
	displayDriver.ver_res = 128;
	displayDriver.flush_cb = lvglFlush;
	displayDriver.draw_buf = &drawBuffer;
	lv_disp_t * disp = lv_disp_drv_register(&displayDriver);
	chatterThemeInit(disp);

	Chatter.getInput()->addListener(new InputChatter());

	printf("UID: 0x%llx\n", ESP.getEfuseMac());

	if(!Settings.get().tested){
		FSLVGL::loadCache();

		auto test = new UserHWTest([](){
			Settings.get().tested = true;
			Settings.store();
			Chatter.fadeOut();
			boot();
		});

		test->start();
		lv_timer_handler();
		Chatter.fadeIn();
		return;
	}

	boot();
}

bool gameStarted = false;
Game* startedGame = nullptr;

void loop(){
	if(!gameStarted){
		lv_timer_handler();
	}
	LoopManager::loop();
}