Momentum-Apps/scenes/evil_portal_scene_console_output.c

#include "../evil_portal_app_i.h"
#include "../helpers/evil_portal_storage.h"

void evil_portal_console_output_handle_rx_data_cb(uint8_t *buf, size_t len,
                                                  void *context) {
  furi_assert(context);
  Evil_PortalApp *app = context;

  // If text box store gets too big, then truncate it
  app->text_box_store_strlen += len;
  if (app->text_box_store_strlen >= EVIL_PORTAL_TEXT_BOX_STORE_SIZE - 1) {
    furi_string_right(app->text_box_store, app->text_box_store_strlen / 2);
    app->text_box_store_strlen = furi_string_size(app->text_box_store) + len;
  }

  // Null-terminate buf and append to text box store
  buf[len] = '\0';
  furi_string_cat_printf(app->text_box_store, "%s", buf);

  view_dispatcher_send_custom_event(app->view_dispatcher,
                                    Evil_PortalEventRefreshConsoleOutput);
}

void evil_portal_scene_console_output_on_enter(void *context) {
  Evil_PortalApp *app = context;

  TextBox *text_box = app->text_box;
  text_box_reset(app->text_box);
  text_box_set_font(text_box, TextBoxFontText);
  if (app->focus_console_start) {
    text_box_set_focus(text_box, TextBoxFocusStart);
  } else {
    text_box_set_focus(text_box, TextBoxFocusEnd);
  }

  if (app->is_command) {
    furi_string_reset(app->text_box_store);
    app->text_box_store_strlen = 0;

    if (0 == strncmp("help", app->selected_tx_string, strlen("help"))) {
      const char *help_msg =
          "BLUE = Waiting\nGREEN = Good\nRED = Bad\n\nThis project is a "
          "WIP.\ngithub.com/bigbrodude6119/flipper-zero-evil-portal\n\n";
      furi_string_cat_str(app->text_box_store, help_msg);
      app->text_box_store_strlen += strlen(help_msg);
    }

    if (0 == strncmp("savelogs", app->selected_tx_string, strlen("savelogs"))) {
      const char *help_msg = "Logs saved.\n\n";
      furi_string_cat_str(app->text_box_store, help_msg);
      app->text_box_store_strlen += strlen(help_msg);
      write_logs(app->portal_logs);
    }

    if (0 == strncmp("sethtml", app->selected_tx_string, strlen("sethtml"))) {
      // create a command queue
      // add sethtml
      // add setap
      // add start



      // if (!app->sent_html) {
      //   const char *help_msg = "Set HTML before\nstarting portal.\n\n";
      //   furi_string_cat_str(app->text_box_store, help_msg);
      //   app->text_box_store_strlen += strlen(help_msg);
      // } else if (!app->sent_ap) {
      //   const char *help_msg = "Set ap name before\nstarting portal.\n\n";
      //   furi_string_cat_str(app->text_box_store, help_msg);
      //   app->text_box_store_strlen += strlen(help_msg);
      // }

      // app->command_queue[0] = "sethtml";
      app->command_queue[0] = "setap";
      // app->command_queue[1] = "start";
      app->has_command_queue = true;
      app->command_index = 0;
    }

    if (app->show_stopscan_tip) {
      const char *help_msg = "Press BACK to return\n";
      furi_string_cat_str(app->text_box_store, help_msg);
      app->text_box_store_strlen += strlen(help_msg);
    }
  }

  // Set starting text - for "View Log", this will just be what was already in
  // the text box store
  text_box_set_text(app->text_box, furi_string_get_cstr(app->text_box_store));

  scene_manager_set_scene_state(app->scene_manager,
                                Evil_PortalSceneConsoleOutput, 0);
  view_dispatcher_switch_to_view(app->view_dispatcher,
                                 Evil_PortalAppViewConsoleOutput);

  // Register callback to receive data
  evil_portal_uart_set_handle_rx_data_cb(
      app->uart,
      evil_portal_console_output_handle_rx_data_cb); // setup callback for rx
                                                     // thread

  // Send command with CR+LF or newline '\n'
  // it is sent here
  if (app->is_command && app->selected_tx_string) {
    if (app->TERMINAL_MODE == 1) {
      evil_portal_uart_tx((uint8_t *)(app->selected_tx_string),
                          strlen(app->selected_tx_string));
      evil_portal_uart_tx((uint8_t *)("\r\n"), 2);
    } else {
      // handle special commands here
      if (0 == strncmp("sethtml", app->selected_tx_string, strlen("sethtml"))) {
        evil_portal_read_index_html(context);

        char *data = malloc(
            (size_t)(strlen((char *)app->index_html) + strlen("sethtml=")));
        strcat(data, "sethtml=");
        strcat(data, (char *)app->index_html);

        evil_portal_uart_tx((uint8_t *)(data), strlen(data));
        evil_portal_uart_tx((uint8_t *)("\n"), 1);

        app->sent_html = true;

        free(data);
        free(app->index_html);

        evil_portal_read_ap_name(context);
      } else if (0 ==
                 strncmp("setap", app->selected_tx_string, strlen("setap"))) {

        evil_portal_read_ap_name(context);

        char *data =
            malloc((size_t)(strlen((char *)app->ap_name) + strlen("setap=")));
        strcat(data, "setap=");
        strcat(data, (char *)app->ap_name);

        evil_portal_uart_tx((uint8_t *)(data), strlen(data));
        evil_portal_uart_tx((uint8_t *)("\n"), 1);

        app->sent_ap = true;

        free(data);
        free(app->ap_name);
      } else if (0 ==
                 strncmp("reset", app->selected_tx_string, strlen("reset"))) {
        app->sent_html = false;
        app->sent_ap = false;
        evil_portal_uart_tx((uint8_t *)(app->selected_tx_string),
                            strlen(app->selected_tx_string));
        evil_portal_uart_tx((uint8_t *)("\n"), 1);
      } else if (0 ==
                 strncmp("help", app->selected_tx_string, strlen("help"))) {
        // do nothing?
      } else {
        evil_portal_uart_tx((uint8_t *)(app->selected_tx_string),
                            strlen(app->selected_tx_string));
        evil_portal_uart_tx((uint8_t *)("\n"), 1);
      }
    }
  }   
}

bool evil_portal_scene_console_output_on_event(void *context,
                                               SceneManagerEvent event) {
  Evil_PortalApp *app = context;

  bool consumed = false;

  if (event.type == SceneManagerEventTypeCustom) {
    text_box_set_text(app->text_box, furi_string_get_cstr(app->text_box_store));
    consumed = true;
  } else if (event.type == SceneManagerEventTypeTick) {
    consumed = true;
  }

  return consumed;
}

void evil_portal_scene_console_output_on_exit(void *context) {
  Evil_PortalApp *app = context;

  // Unregister rx callback
  evil_portal_uart_set_handle_rx_data_cb(app->uart, NULL);
}