fstick.ino

/*
 * Copyright 2023-2024 IMXNOOBX
 *
 * This software is licensed under the Attribution-NonCommercial-NoDerivatives 4.0 International License.
 * For more details, see: https://creativecommons.org/licenses/by-nc-nd/4.0/
 *
 * If you use or modify this code, please include an attribution to the original author and repository.
 * This notice applies to all files within this repository
 */

#define PLUS
// #define PLUS2
#define DEV // Mostly to disable battery saver and some debug messages
#include "classes/globals.h"

/**
 * @brief initialize classes and variables
 */
Led led;
Logger l;
Battery battery;

BLE b;
Info inf;
Settings cfg;
IrBlaster ir; // const uint16_t kIrSendPin = 9;  // IR Emitter Pin - M5 IR Unit
WifiManager wi;

// if (M5.Lcd.width() > 160)
extern const unsigned char logo[];
int last_update = millis() + 1000;
int is_ready = false;

/**
 * @brief Menu options
 */
MenuAction ma_sub_infrared[] = {
    {"Back", nullptr },
    {"Spam Sig", []() { ir.sendAllPowerCodes(); }, ActionType::LOOP, true, []() { ir.sendAllPowerCodesRender(); }, &ir.send_codes },
};

MenuAction ma_sub_wifi[] = {
    {"Back", nullptr },
    {"Scan AP", []() { wi.scanNetworks(); }, ActionType::ACTION, true, []() { wi.scanNetworksRender(); } },
    {"Select AP", ActionType::ACTION_MENU, &wi.select_status, []() { wi.scanNetworksRenderSelect(); }, []() { wi.next(); }, []() { wi.prev(); }, []() { wi.select(); } },
	{"Spam AP", []() { wi.accessPointLoop(); }, ActionType::LOOP, false, nullptr, &wi.loop_spam_ap },
	{"Clone AP", []() { wi.cloneAPLoop(); }, ActionType::LOOP, false, nullptr, &wi.loop_clone_spam_ap },
	{"Rogue AP", []() { wi.rogueAPloop(); }, ActionType::LOOP, false, nullptr, &wi.loop_rogue_ap },
	{"Probe AP", []() { wi.probeAPloop(); }, ActionType::LOOP, false, nullptr, &wi.loop_probe_ap },
	{"Deauth", []() { wi.deauthLoop(); }, ActionType::LOOP, false, nullptr, &wi.loop_deauth_ap },
};

MenuItem mi_sub_wifi[] = {
    {"Back", nullptr },
};

MenuAction ma_sub_ble[] = {
    {"Back", nullptr },
    {"Apple Spm", []() { b.advertiseApple(); } },
    {"Android Sp", []() { b.advertiseAndroid(); } },
	{"Windows S", []() { b.advertiseWindows(); } },
	{"@everyone", []() { b.toggleAdvertiseEveryone(); }, ActionType::LOOP, true, []() { b.advertiseEveryoneRender(); }, &b.advertise_everyone },
};

MenuAction ma_sub_settings[] = {
    {"Back", nullptr },
    {"Bat Saver", []() { cfg.toggleBattSaver(); }, ActionType::TOGGLE, false, nullptr, &cfg.battery_saver },
    {"Sounds", []() { cfg.toggleSound(); }, ActionType::TOGGLE, false, nullptr, &cfg.sound_enable },
    {"Led", []() { cfg.toggleLed(); }, ActionType::TOGGLE, false, nullptr, &cfg.led_enable },
    {"Rotate Sc", []() { cfg.switchRotation(); }, ActionType::TOGGLE, false, nullptr },
    {"Restart", []() { M5.Axp.DeepSleep(5); } },
    {"Shutdown", []() { M5.Axp.PowerOff(); } },
};

MenuAction ma_sub_info[] = {
    {"Back", nullptr },
    {"Donate", true, []() { inf.renderDonate(); } },
    {"Supporters", true, []() { inf.renderDonate(); } },
    {"Repository", true, []() { inf.renderRepository(); } },
    {"Credits", true, []() { inf.renderCredits(); } },
};

MenuItem mi_menu[] = {
    {"Infra Red", nullptr, 2, ma_sub_infrared },
    {"WiFi", nullptr, 8, ma_sub_wifi },
	{"Select WiFi", mi_sub_wifi, 11, nullptr },
    {"BLE", nullptr, 5, ma_sub_ble },
    {"Settings", nullptr, 7, ma_sub_settings },
    {"Info", nullptr, 4, ma_sub_info },
};

MenuRenderer mainMenu(NAME, mi_menu, sizeof(mi_menu) / sizeof(mi_menu[0]));

/**
 * @brief Display startup menu
 */
void setup() {
	M5.begin();
	
	#ifdef DEV
		l.log(Logger::WARNING, "This is a DEVELOPMENT build made in " + String(__DATE__) + ", its not recommended for every day use.");

		int attempts = 0;
		while (!Serial.available()) { // Wait for the serial connection to be establised.
			l.log(Logger::WARNING, "(" + String(attempts) + ") Waiting for serial to be available...");

			if (attempts > 3)
				break;

			attempts++;

			delay(300);
		}
	#endif 
	l.log(Logger::INFO, "Starting " + String(NAME) + "...");

	M5.Lcd.setRotation(1);
	M5.Lcd.fillScreen(BLACK);
	M5.Lcd.setTextColor(WHITE, BLACK);

	mainMenu.topBar();

	// Logo
	M5.Lcd.drawBitmap(10, 20, 105, 105, (uint16_t*)logo);

	M5.Lcd.setCursor(120, 40);
	M5.Lcd.setTextSize(2);
	M5.Lcd.print(NAME);
	M5.Lcd.setCursor(120, 60);
	M5.Lcd.setTextSize(1);
	M5.Lcd.println(VERSION);
	M5.Lcd.setCursor(120, 70);
	M5.Lcd.println(DEVICE);
	/**
	 * @brief initialize classes
	 */
	l.setShouldDisplayLog(true); // Set log output to screen

	if(cfg.init())
		l.log(Logger::INFO, "Settings has been initialized successfully!");
	else
		l.log(Logger::ERROR, "Failed to initialize BLE");

	delay(500); // I know its not clean, but looks good for the end user

	if(ir.init())
		l.log(Logger::INFO, "IrBlaster has been initialized successfully!");
	else
		l.log(Logger::ERROR, "Failed to initialize IrBlaster");

	delay(500);

	if(wi.init())
		l.log(Logger::INFO, "WifiManager has been initialized successfully!");
	else
		l.log(Logger::ERROR, "Failed to initialize WifiManager");

	delay(500);

	if(b.init())
		l.log(Logger::INFO, "BLE has been initialized successfully!");
	else
		l.log(Logger::ERROR, "Failed to initialize BLE");

	delay(500);

	l.setShouldDisplayLog(false); // Remove log output from screen
		
	led.flash();
	M5.Lcd.setTextSize(1);
	M5.Lcd.setCursor(120, SCREEN_HEIGHT - 20);
	M5.Lcd.print("Click to continue");
	l.log(Logger::INFO, "Menu is ready to use!");

	/**
	 * @brief Wait until user interacts with the device
	 * @details This is to prevent the user from missing the startup screen but i have to find a better way to do this
	 */
	while (!M5.BtnA.wasReleased()) {
		M5.update();
		battery.loop();
		delay(10);
	}

	mainMenu.render(true);
}

/**
 * @brief buttons
 * BtnA: M5 button, M5.BtnA.wasReleased()
 * BtnB: The other button, M5.BtnB.wasReleased()
 * Axp: Power button, M5.Axp.GetBtnPress()
 */
void loop() {
	M5.update(); // Its necesary to update the button states

	if (cfg.getSecretMode()) 
		return;

    if (M5.Axp.GetBtnPress()) {
		#ifdef DEV
		{
			l.log(Logger::INFO, "Pressed Axp button to navigate to the next option");
		}
		#endif

		battery.setLI(millis());
        mainMenu.nextOption();
		battery.restoreBrightness();
    }

	if (M5.BtnB.wasReleased()) {
		#ifdef DEV
		{
			l.log(Logger::INFO, "Pressed BtnB button to navigate to the previous option");
		}
		#endif
		
		battery.setLI(millis());
        mainMenu.previousOption();
		battery.restoreBrightness();
    }

    if (M5.BtnA.wasReleased()) {
		#ifdef DEV
		{
			l.log(Logger::INFO, "Pressed BtnA button to select option");
		}
		#endif

		battery.setLI(millis());
        mainMenu.select();
		// mainMenu.render(true);
		// mainMenu.render_feature(); // Not clean but should do its job
		battery.restoreBrightness();
    }

	if (M5.BtnA.wasReleasefor(3000)) {
		cfg.toggleScretMode();
		l.log(Logger::WARNING, "Secret mode has been activated, restart the device to exit it.");
	}

	if (cfg.getSecretMode()) 
		mainMenu.render_hww();
	
	if (last_update < millis()) { // refresh the screen every x seconds
		mainMenu.render_feature(true);
		mainMenu.topBar(); // Should be the last to be on top of everything
		last_update = millis() + 1000; // Refresh every 1s
	}

	/**
	 * @brief Loops
	 * Run the loop() function of each class every tick
	 */
	ir.loop();
	wi.loop();
	b.loop();

	battery.loop();
}