#pragma once

#ifndef MenuFunctions_h
#define MenuFunctions_h

#include "configs.h"

#ifdef MARAUDER_CARDPUTER
  #include "Keyboard.h"
#endif

#ifdef HAS_TOUCH
  #include "TouchKeyboard.h"
#endif

#ifdef HAS_SCREEN

#define BATTERY_ANALOG_ON 0

#include "WiFiScan.h"
#include "BatteryInterface.h"
#include "SDInterface.h"
#include "settings.h"

#ifdef HAS_BUTTONS
  #include "Switches.h"
  #if (U_BTN >= 0)
    extern Switches u_btn;
  #endif
  #if (D_BTN >= 0)
    extern Switches d_btn;
  #endif
  #if (L_BTN >= 0)
    extern Switches l_btn;
  #endif
  #if (R_BTN >= 0)
    extern Switches r_btn;
  #endif
  #if (C_BTN >= 0)
    extern Switches c_btn;
  #endif
#endif

extern WiFiScan wifi_scan_obj;
extern SDInterface sd_obj;
extern BatteryInterface battery_obj;
extern Settings settings_obj;

#define FLASH_BUTTON 0

#if BATTERY_ANALOG_ON == 1
#define BATTERY_PIN 13
#define ANALOG_PIN 34
#define CHARGING_PIN 27
#endif

// Icon definitions
#define ATTACKS 0
#define BEACON_SNIFF 1
#define BLUETOOTH 2
#define BLUETOOTH_SNIFF 3
#define DEAUTH_SNIFF 4
#define DRAW 5
#define PACKET_MONITOR 6
#define PROBE_SNIFF 7
#define SCANNERS 8
#define CC_SKIMMERS 9
#define SNIFFERS 10
#define WIFI 11
#define BEACON_SPAM 12
#define RICK_ROLL 13
#define REBOOT 14
#define GENERAL_APPS 15
#define UPDATE 16
#define DEVICE 17
#define DEVICE_INFO 18
#define SD_UPDATE 19
#define WEB_UPDATE 20
#define EAPOL 21
#define STATUS_BAT 22
#define STATUS_SD 23
#define PWNAGOTCHI 24
#define SHUTDOWN 25
#define BEACON_LIST 26
#define GENERATE 27
#define CLEAR_ICO 28
#define KEYBOARD_ICO 29
#define JOIN_WIFI 30
#define LANGUAGE 31
#define STATUS_GPS 32
#define GPS_MENU 33
#define DISABLE_TOUCH 34
#define FLIPPER 35
#define BLANK 36
#define PINESCAN_SNIFF 37 // Use blanks icon
#define MULTISSID_SNIFF 37 // Use blanks icon
#define JOINED 38
#define FORCE 39
#define FUNNY_BEACON 40
#define FLOCK 41

struct Menu;

// Individual Nodes of a menu

struct MenuNode {
  String name;
  bool command;
  uint8_t color;
  uint8_t icon;
  TFT_eSPI_Button* button;
  bool selected;
  std::function<void()> callable;
};

// Full Menus
struct Menu {
  String name;
  LinkedList<MenuNode>* list;
  Menu                * parentMenu;
  uint16_t               selected = 0;
};


class MenuFunctions
{
  private:

    String u_result = "";


    float _graph_scale = 1.0;
    uint32_t initTime = 0;
    int menu_start_index = 0;
    uint8_t mini_kb_index = 0;
    uint8_t old_gps_sat_count = 0;
    uint8_t max_graph_value = 0;

    // Main menu stuff
    Menu mainMenu;

    Menu wifiMenu;
    Menu bluetoothMenu;
    #ifdef HAS_GPS
      Menu gpsMenu;   // H4W9 Added GPS Menu option to Main Menu
    #endif
    Menu badusbMenu;
    Menu deviceMenu;

    // Device menu stuff
    //Menu whichUpdateMenu;
    Menu failedUpdateMenu;
    Menu confirmMenu;
    Menu updateMenu;
    Menu settingsMenu;
    Menu specSettingMenu;
    //Menu languageMenu;
    Menu sdDeleteMenu;

    // WiFi menu stuff
    Menu wifiSnifferMenu;
    Menu wifiScannerMenu;
    Menu wifiAttackMenu;
    #ifdef HAS_GPS
      Menu wardrivingMenu;
    #endif
    Menu wifiGeneralMenu;
    Menu wifiAPMenu;
    Menu wifiIPMenu;
    Menu ssidsMenu;
    //#ifdef HAS_BT
    //  Menu airtagMenu;
    //#endif
    //#ifndef HAS_ILI9341
      Menu wifiStationMenu;
    //#endif

    // WiFi General Menu
    Menu htmlMenu;
    Menu miniKbMenu;
    Menu saveFileMenu;
    Menu genAPMacMenu;
    Menu cloneAPMacMenu;
    Menu setMacMenu;
    Menu selectProbeSSIDsMenu;

    // Bluetooth menu stuff
    Menu bluetoothSnifferMenu;
    Menu bluetoothAttackMenu;

    // Settings things menus
    Menu generateSSIDsMenu;

    Menu evilPortalMenu;

    static void lv_tick_handler();

    // Menu icons

    void setupSDFileList(bool update = false);
    void buildSDFileMenu(bool update = false);
    void displayMenuButtons();
    uint16_t getColor(uint16_t color);
    void drawAvgLine(int16_t value);
    void drawMaxLine(int16_t value, uint16_t color);
    void drawMaxLine(uint8_t value, uint16_t color);
    float calculateGraphScale(int16_t value);
    float calculateGraphScale(uint8_t value);
    float graphScaleCheck(const int16_t array[TFT_WIDTH]);
    #ifndef HAS_DUAL_BAND
      float graphScaleCheckSmall(const uint8_t array[MAX_CHANNEL]);
    #else
      float graphScaleCheckSmall(const uint8_t array[DUAL_BAND_CHANNELS]);
    #endif
    void drawGraph(int16_t *values);
    void drawGraphSmall(uint8_t *values);
    void renderGraphUI(uint8_t scan_mode = 0);
    //void addNodes(Menu* menu, String name, uint16_t color, Menu* child, int place, std::function<void()> callable, bool selected = false, String command = "");
    void addNodes(Menu* menu, String name, uint8_t color, Menu* child, int place, std::function<void()> callable, bool selected = false, String command = "");
    void battery(bool initial = false);
    void battery2(bool initial = false);
    void showMenuList(Menu* menu, int layer);
    String callSetting(String key);
    void runBoolSetting(String ley);
    void displaySetting(String key, Menu* menu, int index);
    void buttonSelected(int b, int x = -1);
    void buttonNotSelected(int b, int x = -1);
    //#if (!defined(HAS_ILI9341) && defined(HAS_BUTTONS))
    #ifdef HAS_MINI_KB
      String miniKeyboard(Menu * targetMenu, bool do_pass = false);
    #endif
    //#endif

    #ifdef MARAUDER_CARDPUTER
      Keyboard_Class M5CardputerKeyboard = Keyboard_Class();
      bool isKeyPressed(char c);
    #endif

  public:
    MenuFunctions();

    Menu* current_menu;
    Menu clearSSIDsMenu;
    Menu clearAPsMenu;
    
    // Save Files Menu
    Menu saveSSIDsMenu;
    Menu loadSSIDsMenu;
    Menu saveAPsMenu;
    Menu loadAPsMenu;
    Menu saveATsMenu;
    Menu loadATsMenu;

    #ifdef HAS_GPS
      // GPS Menu
      Menu gpsInfoMenu;
      Menu gpsPOIMenu;
    #endif

    Menu infoMenu;
    Menu apInfoMenu;

    Ticker tick;

    uint16_t x = -1, y = -1;
    boolean pressed = false;

    bool disable_touch;

    String loaded_file = "";

    void joinWiFiGFX(String essid, bool start_ap = false);
    void setGraphScale(float scale);
    void updateStatusBar();
    void addSSIDGFX();
    void buildButtons(Menu* menu, int starting_index = 0, String button_name = "");
    void changeMenu(Menu* menu, bool simple_change = false);
    void drawStatusBar();
    void displayCurrentMenu(int start_index = 0);
    void main(uint32_t currentTime);
    void RunSetup();
    void orientDisplay();
};


#endif
#endif