Make Me Flliper Zero

#include <Arduino.h>
#include <Wire.h>
#include <SPI.h>
#include <SD.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <Adafruit_PN532.h>
#include <IRremote.hpp>
#include <OneWire.h>
#include <ELECHOUSE_CC1101_SRC_DRV.h>


struct StoredIr {
  uint32_t magic;
  uint8_t protocol;
  uint16_t address;
  uint16_t command;
  uint32_t raw;
};


// Forward declarations
void setStatus(const char *line1, const char *line2);
void drawScreen();
void loadSavedIr();
void saveIr(const IRData &data);
void sendSavedIr();
void pollIr();
void pollNfc();
void pollRfid();
void pollIButton();
void pollRadio();
void handleControls();

constexpr uint8_t OLED_SDA_PIN = 21;
constexpr uint8_t OLED_SCL_PIN = 22;
constexpr uint8_t IR_RECEIVE_PIN = 27;
constexpr uint8_t IR_SEND_PIN = 25;
constexpr uint8_t JOYSTICK_X_PIN = 32;
constexpr uint8_t JOYSTICK_Y_PIN = 33;
constexpr uint8_t JOYSTICK_SW_PIN = 14;
constexpr uint8_t CLEAR_BUTTON_PIN = 26;
constexpr uint8_t RFID_RX_PIN = 16;
constexpr uint8_t IBUTTON_PIN = 17;
constexpr uint8_t SPI_SCK_PIN = 18;
constexpr uint8_t SPI_MISO_PIN = 19;
constexpr uint8_t SPI_MOSI_PIN = 23;
constexpr uint8_t CC1101_CS_PIN = 4;
constexpr uint8_t SD_CS_PIN = 13;
constexpr uint8_t SCREEN_WIDTH = 128;
constexpr uint8_t SCREEN_HEIGHT = 64;
constexpr uint8_t OLED_RESET = 255;
constexpr uint32_t IR_FILE_MAGIC = 0x49525231UL;

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);
Adafruit_PN532 nfc(-1, -1, &Wire);
OneWire iButton(IBUTTON_PIN);
HardwareSerial rfidSerial(2);

const char *modeNames[] = {"IR remote", "NFC", "125k RFID", "iButton", "Sub-GHz RX"};
constexpr uint8_t MODE_COUNT = sizeof(modeNames) / sizeof(modeNames[0]);
uint8_t mode = 0;
char statusLine[42] = "Starting...";
char detailLine[42] = "";
bool sdReady = false;
bool nfcReady = false;
bool radioReady = false;
bool haveSavedIr = false;
uint32_t lastInputMs = 0;
uint32_t lastPollMs = 0;


StoredIr savedIr{};

void setStatus(const char *line1, const char *line2 = "") {
  strncpy(statusLine, line1, sizeof(statusLine) - 1);
  statusLine[sizeof(statusLine) - 1] = '\0';
  strncpy(detailLine, line2, sizeof(detailLine) - 1);
  detailLine[sizeof(detailLine) - 1] = '\0';
}

void drawScreen() {
  display.clearDisplay();
  display.setTextColor(SSD1306_WHITE);
  display.setTextSize(1);
  display.setCursor(0, 0);
  display.print(F("Pocket tool: "));
  display.println(modeNames[mode]);
  display.drawLine(0, 10, 127, 10, SSD1306_WHITE);
  display.setCursor(0, 18);
  display.println(statusLine);
  display.setCursor(0, 32);
  display.println(detailLine);
  display.setCursor(0, 49);
  display.println(F("L/R mode  press action"));
  display.println(F("Button clears screen"));
  display.display();
}

void loadSavedIr() {
  if (!sdReady) return;
  File file = SD.open("/lastir.bin", FILE_READ);
  if (file && file.read(reinterpret_cast<uint8_t *>(&savedIr), sizeof(savedIr)) == sizeof(savedIr) && savedIr.magic == IR_FILE_MAGIC) {
    haveSavedIr = true;
  }
  if (file) file.close();
}

void saveIr(const IRData &data) {
  savedIr.magic = IR_FILE_MAGIC;
  savedIr.protocol = static_cast<uint8_t>(data.protocol);
  savedIr.address = data.address;
  savedIr.command = data.command;
  savedIr.raw = data.decodedRawData;
  haveSavedIr = true;
  if (sdReady) {
    SD.remove("/lastir.bin");
    File file = SD.open("/lastir.bin", FILE_WRITE);
    if (file) {
      file.write(reinterpret_cast<const uint8_t *>(&savedIr), sizeof(savedIr));
      file.close();
    }
  }
}

void sendSavedIr() {
  if (!haveSavedIr) {
    setStatus("No saved IR command", "Aim remote at receiver");
    drawScreen();
    return;
  }
  IRData data{};
  data.protocol = static_cast<decode_type_t>(savedIr.protocol);
  data.address = savedIr.address;
  data.command = savedIr.command;
  data.decodedRawData = savedIr.raw;
  IrSender.write(&data, NO_REPEATS);
  setStatus("Sent saved IR command", "Only use on your equipment");
  drawScreen();
}

void pollIr() {
  if (!IrReceiver.decode()) return;
  if (!(IrReceiver.decodedIRData.flags & IRDATA_FLAGS_IS_REPEAT)) {
    saveIr(IrReceiver.decodedIRData);
    snprintf(statusLine, sizeof(statusLine), "Saved protocol %u", static_cast<unsigned>(IrReceiver.decodedIRData.protocol));
    snprintf(detailLine, sizeof(detailLine), "Addr %04X Cmd %04X", IrReceiver.decodedIRData.address, IrReceiver.decodedIRData.command);
    drawScreen();
  }
  IrReceiver.resume();
}

void pollNfc() {
  if (!nfcReady || millis() - lastPollMs < 400) return;
  lastPollMs = millis();
  uint8_t uid[7];
  uint8_t uidLength = 0;
  if (nfc.readPassiveTargetID(PN532_MIFARE_ISO14443A, uid, &uidLength, 30)) {
    char uidText[25] = "UID ";
    for (uint8_t i = 0; i < uidLength && strlen(uidText) < 20; i++) {
      char hex[4];
      snprintf(hex, sizeof(hex), "%02X", uid[i]);
      strcat(uidText, hex);
    }
    setStatus("NFC tag read", uidText);
    drawScreen();
  }
}

void pollRfid() {
  static char tag[15] = "";
  static uint8_t index = 0;
  while (rfidSerial.available()) {
    char c = static_cast<char>(rfidSerial.read());
    if (c == '\n' || c == '\r') {
      if (index > 0) {
        tag[index] = '\0';
        setStatus("125 kHz tag read", tag);
        drawScreen();
        index = 0;
      }
    } else if (isxdigit(c) && index < sizeof(tag) - 1) {
      tag[index++] = c;
    }
  }
}

void pollIButton() {
  if (millis() - lastPollMs < 600) return;
  lastPollMs = millis();
  uint8_t address[8];
  iButton.reset_search();
  if (iButton.search(address)) {
    char serial[25] = "ROM ";
    for (uint8_t i = 0; i < 8; i++) {
      char hex[4];
      snprintf(hex, sizeof(hex), "%02X", address[i]);
      strcat(serial, hex);
    }
    setStatus("iButton read", serial);
    drawScreen();
  }
}

void pollRadio() {
  if (!radioReady || millis() - lastPollMs < 250) return;
  lastPollMs = millis();
  if (ELECHOUSE_cc1101.CheckRxFifo(0)) {
    byte packet[64];
    byte length = ELECHOUSE_cc1101.ReceiveData(packet);
    snprintf(statusLine, sizeof(statusLine), "RF packet received");
    snprintf(detailLine, sizeof(detailLine), "%u bytes at 433.92 MHz", static_cast<unsigned>(length));
    drawScreen();
  }
}

void handleControls() {
  if (millis() - lastInputMs < 250) return;
  int x = analogRead(JOYSTICK_X_PIN);
  if (x < 800 || x > 3300) {
    mode = x < 800 ? (mode + MODE_COUNT - 1) % MODE_COUNT : (mode + 1) % MODE_COUNT;
    lastInputMs = millis();
    setStatus("Select a function", "Bring your own tag/device");
    drawScreen();
  }
  if (digitalRead(CLEAR_BUTTON_PIN) == LOW) {
    lastInputMs = millis();
    setStatus("Ready", mode == 0 ? "Press stick to send saved IR" : "Waiting for a nearby tag");
    drawScreen();
  }
  if (digitalRead(JOYSTICK_SW_PIN) == LOW) {
    lastInputMs = millis();
    if (mode == 0) sendSavedIr();
  }
}

void setup() {
  pinMode(CLEAR_BUTTON_PIN, INPUT_PULLUP);
  pinMode(JOYSTICK_SW_PIN, INPUT_PULLUP);
  Wire.begin(OLED_SDA_PIN, OLED_SCL_PIN);
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  SPI.begin(SPI_SCK_PIN, SPI_MISO_PIN, SPI_MOSI_PIN);
  sdReady = SD.begin(SD_CS_PIN);
  loadSavedIr();

  IrReceiver.begin(IR_RECEIVE_PIN, DISABLE_LED_FEEDBACK);
  IrSender.begin(IR_SEND_PIN, DISABLE_LED_FEEDBACK);
  rfidSerial.begin(9600, SERIAL_8N1, RFID_RX_PIN, -1);

  nfc.begin();
  uint32_t version = nfc.getFirmwareVersion();
  nfcReady = version != 0;
  if (nfcReady) nfc.SAMConfig();

  ELECHOUSE_cc1101.setSpiPin(SPI_SCK_PIN, SPI_MISO_PIN, SPI_MOSI_PIN, CC1101_CS_PIN);
  ELECHOUSE_cc1101.Init();
  radioReady = true;
  ELECHOUSE_cc1101.setMHZ(433.92);
  ELECHOUSE_cc1101.SetRx();
  setStatus("Ready", "L/R chooses a function");
  drawScreen();
}

void loop() {
  handleControls();
  if (mode == 0) pollIr();
  else if (mode == 1) pollNfc();
  else if (mode == 2) pollRfid();
  else if (mode == 3) pollIButton();
  else pollRadio();
}