Momentum-Apps/esp32_marauder/esp32_marauder.ino

/* FLASH SETTINGS
  Board: LOLIN D32
  Flash Frequency: 80MHz
  Partition Scheme: Minimal SPIFFS
  https://www.online-utility.org/image/convert/to/XBM
*/

#include "configs.h"

#ifndef HAS_SCREEN
#define MenuFunctions_h
#define Display_h
#endif
#include <WiFi.h>
#include <Wire.h>
#include "esp_wifi.h"
#include "esp_wifi_types.h"
#include <stdio.h>
#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include "esp_system.h"
#include <Arduino.h>
#ifdef ESP32_CAM

#include <Preferences.h>
Preferences preferences;

#include "FS.h"                // SD Card ESP32
#include "SD_MMC.h"            // SD Card ESP32
#include "esp_camera.h"
#include "soc/soc.h"           // Disable brownout problems
#include "soc/rtc_cntl_reg.h"  // Disable brownout problems
#include "driver/rtc_io.h"

// Pin definition for CAMERA_MODEL_AI_THINKER
#define PWDN_GPIO_NUM     32
#define RESET_GPIO_NUM    -1
#define XCLK_GPIO_NUM      0
#define SIOD_GPIO_NUM     26
#define SIOC_GPIO_NUM     27

#define Y9_GPIO_NUM       35
#define Y8_GPIO_NUM       34
#define Y7_GPIO_NUM       39
#define Y6_GPIO_NUM       36
#define Y5_GPIO_NUM       21
#define Y4_GPIO_NUM       19
#define Y3_GPIO_NUM       18
#define Y2_GPIO_NUM        5
#define VSYNC_GPIO_NUM    25
#define HREF_GPIO_NUM     23
#define PCLK_GPIO_NUM     22

bool camera_initialized = false;

#endif
#include "Assets.h"
#include "WiFiScan.h"
#ifdef USE_SD_MMC_ALTERNATIVE
#include "SD_MMCInterface.h"
#else
#include "SDInterface.h"
#endif
#include "Web.h"
#include "Buffer.h"
#include "BatteryInterface.h"
#include "TemperatureInterface.h"
#include "LedInterface.h"
#include "esp_interface.h"
#include "settings.h"
#include "CommandLine.h"
#include "lang_var.h"
#include "flipperLED.h"

#ifdef HAS_SCREEN
#include "Display.h"
#include "MenuFunctions.h"
#include "a32u4_interface.h"
#endif

#ifdef MARAUDER_MINI
#include <SwitchLib.h>
SwitchLib u_btn = SwitchLib(U_BTN, 1000, true);
SwitchLib d_btn = SwitchLib(D_BTN, 1000, true);
SwitchLib l_btn = SwitchLib(L_BTN, 1000, true);
SwitchLib r_btn = SwitchLib(R_BTN, 1000, true);
SwitchLib c_btn = SwitchLib(C_BTN, 1000, true);
#endif

WiFiScan wifi_scan_obj;
SDInterface sd_obj;
Web web_obj;
Buffer buffer_obj;
BatteryInterface battery_obj;
TemperatureInterface temp_obj;
LedInterface led_obj;
EspInterface esp_obj;
Settings settings_obj;
CommandLine cli_obj;
flipperLED flipper_led;

#ifdef HAS_SCREEN
Display display_obj;
MenuFunctions menu_function_obj;
A32u4Interface a32u4_obj;
#endif

const String PROGMEM version_number = MARAUDER_VERSION;

Adafruit_NeoPixel strip = Adafruit_NeoPixel(Pixels, PIN, NEO_GRB + NEO_KHZ800);

uint32_t currentTime  = 0;


void backlightOn() {
#ifdef HAS_SCREEN
#ifdef MARAUDER_MINI
  digitalWrite(TFT_BL, LOW);
#endif

#ifndef MARAUDER_MINI
  digitalWrite(TFT_BL, HIGH);
#endif
#endif
}

void backlightOff() {
#ifdef HAS_SCREEN
#ifdef MARAUDER_MINI
  digitalWrite(TFT_BL, HIGH);
#endif

#ifndef MARAUDER_MINI
  digitalWrite(TFT_BL, LOW);
#endif
#endif
}


void setup()
{
#ifndef ESP32_CAM
  pinMode(FLASH_BUTTON, INPUT);
#endif

#ifdef HAS_SCREEN
  pinMode(TFT_BL, OUTPUT);
#endif

  backlightOff();
#if BATTERY_ANALOG_ON == 1
  pinMode(BATTERY_PIN, OUTPUT);
  pinMode(CHARGING_PIN, INPUT);
#endif

  // Preset SPI CS pins to avoid bus conflicts
#ifdef HAS_SCREEN
  digitalWrite(TFT_CS, HIGH);
#endif

  /*pinMode(SD_CS, OUTPUT);

    delay(10);

    digitalWrite(SD_CS, HIGH);

    delay(10);*/

  Serial.begin(230400);
  preferences.begin("esp32cam", false);
  if (preferences.getBool("streaming", false))
  {
    if (preferences.getBool("return", false))
    {
      preferences.putBool("streaming", false);
      preferences.end();
      ESP.restart();
    }
    preferences.putBool("return", true);
    preferences.end();
    delay(2000);
    preferences.begin("esp32cam", false);
    preferences.putBool("return", false);
    preferences.end();
    cam_stream_setup();
    for (;;)
      cam_stream_loop();
  }
  //Serial.begin(115200);

  //Serial.println("\n\nHello, World!\n");

  Serial.println("ESP-IDF version is: " + String(esp_get_idf_version()));

  //#ifdef HAS_SCREEN
  //  Serial.println("Has Screen");
  //#else
  //  Serial.println("Does not have screen");
  //#endif

#ifdef HAS_SCREEN
  display_obj.RunSetup();
  display_obj.tft.setTextColor(TFT_WHITE, TFT_BLACK);
#endif

  backlightOff();

  // Draw the title screen
#ifdef HAS_SCREEN
  display_obj.drawJpeg("/marauder3L.jpg", 0 , 0);     // 240 x 320 image
#endif

#ifdef HAS_SCREEN
  //showCenterText(version_number, 250);
#ifndef MARAUDER_MINI
  display_obj.tft.drawCentreString(display_obj.version_number, 120, 250, 2);
#endif

#ifdef MARAUDER_MINI
  display_obj.tft.drawCentreString(display_obj.version_number, TFT_WIDTH / 2, TFT_HEIGHT, 1);
#endif
#endif

  backlightOn(); // Need this

#ifdef HAS_SCREEN
  delay(2000);

  display_obj.clearScreen();

  display_obj.tft.setTextColor(TFT_CYAN, TFT_BLACK);

  display_obj.tft.println(text_table0[0]);

  delay(2000);

  display_obj.tft.println("Marauder " + display_obj.version_number + "\n");

  display_obj.tft.println(text_table0[1]);
#endif

  //Serial.println("Internal Temp: " + (String)((temprature_sens_read() - 32) / 1.8));

  settings_obj.begin();

#ifdef MARAUDER_FLIPPER
  flipper_led.RunSetup();
#endif

  //Serial.println("This is a test Channel: " + (String)settings_obj.loadSetting<uint8_t>("Channel"));
  //if (settings_obj.loadSetting<bool>( "Force PMKID"))
  //  Serial.println("This is a test Force PMKID: true");
  //else
  //  Serial.println("This is a test Force PMKID: false");

  wifi_scan_obj.RunSetup();

  //Serial.println(wifi_scan_obj.freeRAM());

#ifdef HAS_SCREEN
  display_obj.tft.println(F(text_table0[2]));
#endif

  // Do some SD stuff
  if (sd_obj.initSD()) {
    //Serial.println(F("SD Card supported"));
#ifdef HAS_SCREEN
    display_obj.tft.println(F(text_table0[3]));
#endif
  }
  else {
    Serial.println(F("SD Card NOT Supported"));
#ifdef HAS_SCREEN
    display_obj.tft.setTextColor(TFT_RED, TFT_BLACK);
    display_obj.tft.println(F(text_table0[4]));
    display_obj.tft.setTextColor(TFT_CYAN, TFT_BLACK);
#endif
  }

  battery_obj.RunSetup();

#ifdef HAS_SCREEN
  display_obj.tft.println(F(text_table0[5]));
#endif

  // Temperature stuff
#ifndef MARAUDER_FLIPPER
  temp_obj.RunSetup();
#endif

#ifdef HAS_SCREEN
  display_obj.tft.println(F(text_table0[6]));
#endif

#ifndef MARAUDER_FLIPPER
  battery_obj.battery_level = battery_obj.getBatteryLevel();

  //    if (battery_obj.i2c_supported) {
  //      Serial.println(F("IP5306 I2C Supported: true"));
  //    }
  //    else
  //      Serial.println(F("IP5306 I2C Supported: false"));
#endif

  // Do some LED stuff
#ifndef MARAUDER_FLIPPER
  led_obj.RunSetup();
#endif

#ifdef HAS_SCREEN
  display_obj.tft.println(F(text_table0[7]));

  delay(500);
#endif

#ifdef HAS_SCREEN
  display_obj.tft.println(F(text_table0[8]));

  display_obj.tft.setTextColor(TFT_WHITE, TFT_BLACK);

  delay(2000);
#endif

#ifdef HAS_SCREEN
  menu_function_obj.RunSetup();
#endif

  //Serial.println(F("\n\n--------------------------------\n"));
  //Serial.println(F("         ESP32 Marauder      \n"));
  //Serial.println("            " + version_number + "\n");
  //Serial.println(F("       By: justcallmekoko\n"));
  //Serial.println(F("--------------------------------\n\n"));

  Serial.println("CLI Ready");
  cli_obj.RunSetup();
}


void loop()
{
  currentTime = millis();
  bool mini = false;

#ifdef MARAUDER_MINI
  mini = true;
#endif

  // Update all of our objects
#ifdef HAS_SCREEN
  bool do_draw = display_obj.draw_tft;
#else
  bool do_draw = false;
#endif

  if ((!do_draw) && (wifi_scan_obj.currentScanMode != ESP_UPDATE))
  {
    cli_obj.main(currentTime);
#ifdef HAS_SCREEN
    display_obj.main(wifi_scan_obj.currentScanMode);
#endif
    wifi_scan_obj.main(currentTime);
    sd_obj.main();
#ifndef MARAUDER_FLIPPER
    battery_obj.main(currentTime);
    temp_obj.main(currentTime);
#endif
    settings_obj.main(currentTime);
    if (((wifi_scan_obj.currentScanMode != WIFI_PACKET_MONITOR) && (wifi_scan_obj.currentScanMode != WIFI_SCAN_EAPOL)) ||
        (mini)) {
#ifdef HAS_SCREEN
      menu_function_obj.main(currentTime);
#endif
#ifndef MARAUDER_FLIPPER
      led_obj.main(currentTime);
#endif
      //cli_obj.main(currentTime);
    }
    if (wifi_scan_obj.currentScanMode == OTA_UPDATE)
      web_obj.main();
#ifdef HAS_SCREEN
    delay(1);
#else
    delay(50);
#endif
  }
#ifdef HAS_SCREEN
  else if ((display_obj.draw_tft) &&
           (wifi_scan_obj.currentScanMode != OTA_UPDATE))
  {
    display_obj.drawStylus();
  }
#endif
  else if (wifi_scan_obj.currentScanMode == ESP_UPDATE) {
#ifdef HAS_SCREEN
    display_obj.main(wifi_scan_obj.currentScanMode);
    menu_function_obj.main(currentTime);
#endif
#ifndef MARAUDER_FLIPPER
    led_obj.main(currentTime);
#endif
    //cli_obj.main(currentTime);
    delay(1);
  }
}