Ben / moon

#include "audio_player.h"
#include "lvgl.h"
#include "moon.h"
#include "playlist.h"
#include "screens/audiobooks_screen.h"
#include "screens/files_screen.h"
#include "screens/home_screen.h"
#include "screens/music_screen.h"
#include "screens/now_playing_screen.h"
#include "screens/queue_dialog_screen.h"
#include "screens/settings_screen.h"
#include "screens/usb_screen.h"
#include "storage.h"

#ifdef ESP_PLATFORM
#include "usb_msc.h"
#endif
#include "topbar.h"
#include "ui.h"

#include "fonts/fonts.h"

uint8_t button_state;

app_state_t app_state = {0};
ui_state_t ui_state = {0};

ui_state_t setup_screen(screen_type_t type) {
  switch (type) {
  case SCREEN_HOME:
    return setup_home_screen();
  case SCREEN_FILES:
    return setup_files_screen("/", "");
  case SCREEN_NOW_PLAYING:
    return setup_now_playing_screen();
  case SCREEN_QUEUE_DIALOG:
    return setup_queue_dialog_screen();
  case SCREEN_AUDIOBOOKS:
    return setup_audiobooks_screen(AUDIOBOOKS_ROOT, "");
  case SCREEN_MUSIC:
    return setup_music_screen(MUSIC_ROOT, "");
  case SCREEN_SETTINGS:
    return setup_settings_screen();
  case SCREEN_USB:
    return setup_usb_screen();
  default:
    return (ui_state_t){0};
  }
}

void free_screen(void) {
  switch (ui_state.type) {
  case SCREEN_HOME:
    free_home_screen(ui_state.home);
    break;
  case SCREEN_FILES:
    free_files_screen(ui_state.files);
    break;
  case SCREEN_NOW_PLAYING:
    free_now_playing_screen(ui_state.now_playing);
    break;
  case SCREEN_QUEUE_DIALOG:
    free_queue_dialog_screen(ui_state.queue_dialog);
    break;
  case SCREEN_AUDIOBOOKS:
    free_audiobooks_screen(ui_state.audiobooks);
    break;
  case SCREEN_MUSIC:
    free_music_screen(ui_state.music);
    break;
  case SCREEN_SETTINGS:
    free_settings_screen(ui_state.settings);
    break;
  case SCREEN_USB:
    free_usb_screen(ui_state.usb);
    break;
  default:
    break;
  }
  ui_state = (ui_state_t){0};
}

void update_screen(void) {
  switch (ui_state.type) {
  case SCREEN_HOME:
    update_home_screen(ui_state.home);
    break;
  case SCREEN_FILES:
    update_files_screen(ui_state.files);
    break;
  case SCREEN_NOW_PLAYING:
    update_now_playing_screen(ui_state.now_playing);
    break;
  case SCREEN_QUEUE_DIALOG:
    update_queue_dialog_screen(ui_state.queue_dialog);
    break;
  case SCREEN_AUDIOBOOKS:
    update_audiobooks_screen(ui_state.audiobooks);
    break;
  case SCREEN_MUSIC:
    update_music_screen(ui_state.music);
    break;
  case SCREEN_SETTINGS:
    update_settings_screen(ui_state.settings);
    break;
  case SCREEN_USB:
    update_usb_screen(ui_state.usb);
    break;
  default:
    break;
  }
}

void setup(void) {
  // Initialize storage subsystem
  storage_init();

  // Load playlist from SD card
  playlist_load();

  // Initialize audio player
  audio_player_init();

  // Set up dark theme with custom accent color
  lv_display_t *disp = lv_display_get_default();
  lv_theme_t *theme = lv_theme_default_init(
      disp,
      //lv_color_hex(0xE91E63), // Primary color (vibrant pink)
      //lv_color_hex(0x0288D1), // Secondary color (deep sky blue)
	  lv_palette_main(LV_PALETTE_PINK),
	  lv_palette_main(LV_PALETTE_LIGHT_BLUE),
      true,                   // Dark mode
      &roboto_condensed
      //&lv_font_source_han_sans_sc_14_cjk
  );
  lv_display_set_theme(disp, theme);

  topbar_create();
  ui_state = setup_screen(SCREEN_HOME);
}

void loop(void) {
#ifdef ESP_PLATFORM
  // Auto-switch to USB screen when host connects
  if (ui_state.type != SCREEN_USB && usb_msc_host_connected()) {
    navigate_to(SCREEN_USB);
    return;
  }
#endif
  audio_player_update();
  playlist_tick();
  topbar_update();
  update_screen();
}