WIP debug console for testing playback

11 files changed, 278 insertions, 78 deletions

diff --git a/src/dev_console/CMakeLists.txt b/src/dev_console/CMakeLists.txt
new file mode 100644
index 00000000..b7e6357e
--- /dev/null
+++ b/src/dev_console/CMakeLists.txt
@@ -0,0 +1,5 @@
+idf_component_register(
+  SRCS "console.cpp"
+  INCLUDE_DIRS "include"
+  REQUIRES "console")
+target_compile_options(${COMPONENT_LIB} PRIVATE ${EXTRA_WARNINGS})
diff --git a/src/dev_console/console.cpp b/src/dev_console/console.cpp
new file mode 100644
index 00000000..9b51caaf
--- /dev/null
+++ b/src/dev_console/console.cpp
@@ -0,0 +1,87 @@
+#include "console.hpp"
+#include <stdio.h>
+#include <string.h>
+
+#include <algorithm>
+#include <iostream>
+#include <string>
+
+#include "esp_console.h"
+#include "esp_log.h"
+#include "esp_system.h"
+
+namespace console {
+
+int CmdLogLevel(int argc, char** argv) {
+  static const std::string usage =
+      "usage: loglevel [VERBOSE,DEBUG,INFO,WARN,ERROR,NONE]";
+  if (argc != 2) {
+    std::cout << usage << std::endl;
+    return 1;
+  }
+  std::string level_str = argv[1];
+  std::transform(level_str.begin(), level_str.end(), level_str.begin(),
+                 [](unsigned char c) { return std::toupper(c); });
+
+  esp_log_level_t level;
+  if (level_str == "VERBOSE") {
+    level = ESP_LOG_VERBOSE;
+  } else if (level_str == "DEBUG") {
+    level = ESP_LOG_DEBUG;
+  } else if (level_str == "INFO") {
+    level = ESP_LOG_INFO;
+  } else if (level_str == "WARN") {
+    level = ESP_LOG_WARN;
+  } else if (level_str == "ERROR") {
+    level = ESP_LOG_ERROR;
+  } else if (level_str == "NONE") {
+    level = ESP_LOG_NONE;
+  } else {
+    std::cout << usage << std::endl;
+    return 1;
+  }
+
+  esp_log_level_set("*", level);
+
+  return 0;
+}
+
+void RegisterLogLevel() {
+  esp_console_cmd_t cmd{
+      .command = "loglevel",
+      .help =
+          "Sets the log level to one of \"VERBOSE\", \"DEBUG\", \"INFO\", "
+          "\"WARN\", \"ERROR\", \"NONE\"",
+      .hint = "level",
+      .func = &CmdLogLevel,
+      .argtable = NULL};
+  esp_console_cmd_register(&cmd);
+}
+
+Console::Console() {}
+Console::~Console() {}
+
+auto Console::RegisterCommonComponents() -> void {
+  esp_console_register_help_command();
+  RegisterLogLevel();
+}
+
+auto Console::Launch() -> void {
+  esp_console_repl_t* repl = nullptr;
+  esp_console_repl_config_t repl_config = ESP_CONSOLE_REPL_CONFIG_DEFAULT();
+  repl_config.max_history_len = 16;
+  repl_config.prompt = " →";
+  repl_config.max_cmdline_length = 256;
+  repl_config.task_stack_size = 1024 * GetStackSizeKiB();
+
+  esp_console_dev_uart_config_t hw_config =
+      ESP_CONSOLE_DEV_UART_CONFIG_DEFAULT();
+  ESP_ERROR_CHECK(esp_console_new_repl_uart(&hw_config, &repl_config, &repl));
+
+  RegisterCommonComponents();
+  RegisterExtraComponents();
+
+  ESP_ERROR_CHECK(esp_console_start_repl(repl));
+}
+
+}  // namespace console
diff --git a/src/dev_console/include/console.hpp b/src/dev_console/include/console.hpp
new file mode 100644
index 00000000..cf5180dd
--- /dev/null
+++ b/src/dev_console/include/console.hpp
@@ -0,0 +1,22 @@
+#pragma once
+
+#include <cstdint>
+
+namespace console {
+
+class Console {
+ public:
+  Console();
+  virtual ~Console();
+
+  auto Launch() -> void;
+
+ protected:
+  virtual auto GetStackSizeKiB() -> uint16_t { return 0; }
+  virtual auto RegisterExtraComponents() -> void {}
+
+ private:
+  auto RegisterCommonComponents() -> void;
+};
+
+}  // namespace console
diff --git a/src/drivers/audio_playback.cpp b/src/drivers/audio_playback.cpp
index 55da2bfb..1bc5cb3b 100644
--- a/src/drivers/audio_playback.cpp
+++ b/src/drivers/audio_playback.cpp
@@ -124,6 +124,14 @@ void AudioPlayback::Play(const std::string& filename) {
   output_->SetVolume(volume_);
 }
 
+void AudioPlayback::Toggle() {
+  if (GetPlaybackState() == PLAYING) {
+    Pause();
+  } else if (GetPlaybackState() == PAUSED) {
+    Resume();
+  }
+}
+
 void AudioPlayback::Resume() {
   if (GetPlaybackState() == PAUSED) {
     current_state_ = PLAYING;
diff --git a/src/main/CMakeLists.txt b/src/main/CMakeLists.txt
index 210abed1..9bfefb1d 100644
--- a/src/main/CMakeLists.txt
+++ b/src/main/CMakeLists.txt
@@ -1,4 +1,5 @@
 idf_component_register(
-  SRCS "main.cpp"
-  REQUIRES "drivers")
+  SRCS "main.cpp" "app_console.cpp"
+  INCLUDE_DIRS "."
+  REQUIRES "drivers" "dev_console")
 target_compile_options(${COMPONENT_LIB} PRIVATE ${EXTRA_WARNINGS})
diff --git a/src/main/app_console.cpp b/src/main/app_console.cpp
new file mode 100644
index 00000000..613d5a9a
--- /dev/null
+++ b/src/main/app_console.cpp
@@ -0,0 +1,104 @@
+#include "app_console.hpp"
+
+#include <dirent.h>
+#include <cstdio>
+#include <iostream>
+#include <string>
+#include "esp_console.h"
+
+namespace console {
+
+static AppConsole* sInstance = nullptr;
+
+int CmdListDir(int argc, char** argv) {
+  static const std::string usage = "usage: ls [directory]";
+  if (argc > 2) {
+    std::cout << usage << std::endl;
+    return 1;
+  }
+  std::string path = drivers::kStoragePath;
+  if (argc == 2) {
+    path += "/";
+    path += argv[1];
+  }
+
+  DIR* dir;
+  struct dirent* ent;
+  dir = opendir(path.c_str());
+  while ((ent = readdir(dir))) {
+    std::cout << ent->d_name << std::endl;
+  }
+  closedir(dir);
+
+  return 0;
+}
+
+void RegisterListDir() {
+  esp_console_cmd_t cmd{.command = "ls",
+                        .help = "Lists SD contents",
+                        .hint = NULL,
+                        .func = &CmdListDir,
+                        .argtable = NULL};
+  esp_console_cmd_register(&cmd);
+}
+
+int CmdPlayFile(int argc, char** argv) {
+  static const std::string usage = "usage: play [file]";
+  if (argc != 2) {
+    std::cout << usage << std::endl;
+    return 1;
+  }
+  std::string path = drivers::kStoragePath;
+  path += "/";
+  path += argv[1];
+
+  sInstance->playback_->Play(path.c_str());
+
+  return 0;
+}
+
+void RegisterPlayFile() {
+  esp_console_cmd_t cmd{.command = "play",
+                        .help = "Begins playback of the file at the given path",
+                        .hint = "filepath",
+                        .func = &CmdPlayFile,
+                        .argtable = NULL};
+  esp_console_cmd_register(&cmd);
+}
+
+int CmdToggle(int argc, char** argv) {
+  static const std::string usage = "usage: toggle";
+  if (argc != 1) {
+    std::cout << usage << std::endl;
+    return 1;
+  }
+
+  sInstance->playback_->Toggle();
+
+  return 0;
+}
+
+void RegisterToggle() {
+  esp_console_cmd_t cmd{.command = "toggle",
+                        .help = "Toggles between play and pause",
+                        .hint = NULL,
+                        .func = &CmdToggle,
+                        .argtable = NULL};
+  esp_console_cmd_register(&cmd);
+}
+
+AppConsole::AppConsole(std::unique_ptr<drivers::AudioPlayback> playback)
+    : playback_(std::move(playback)) {
+  sInstance = this;
+}
+AppConsole::~AppConsole() {
+  sInstance = nullptr;
+}
+
+auto AppConsole::RegisterExtraComponents() -> void {
+  RegisterListDir();
+  RegisterPlayFile();
+  RegisterToggle();
+}
+
+}  // namespace console
diff --git a/src/main/app_console.hpp b/src/main/app_console.hpp
new file mode 100644
index 00000000..fb051bd1
--- /dev/null
+++ b/src/main/app_console.hpp
@@ -0,0 +1,21 @@
+#pragma once
+
+#include "audio_playback.hpp"
+#include "console.hpp"
+
+#include <memory>
+
+namespace console {
+
+class AppConsole : public Console {
+ public:
+  AppConsole(std::unique_ptr<drivers::AudioPlayback> playback);
+  virtual ~AppConsole();
+
+  std::unique_ptr<drivers::AudioPlayback> playback_;
+
+ protected:
+  virtual auto RegisterExtraComponents() -> void;
+};
+
+}  // namespace console
diff --git a/src/main/main.cpp b/src/main/main.cpp
index 1bcf14ae..1f0f0db7 100644
--- a/src/main/main.cpp
+++ b/src/main/main.cpp
@@ -1,3 +1,5 @@
+#include "app_console.hpp"
+#include "audio_playback.hpp"
 #include "battery.hpp"
 #include "core/lv_disp.h"
 #include "core/lv_obj_pos.h"
@@ -11,8 +13,6 @@
 #include "i2s_audio_output.hpp"
 #include "misc/lv_color.h"
 #include "misc/lv_timer.h"
-#include "audio_playback.hpp"
-#include "i2s_audio_output.hpp"
 #include "spi.hpp"
 #include "storage.hpp"
 
@@ -114,7 +114,7 @@ extern "C" void app_main(void) {
                                 (void*)lvglArgs, 1, sLvglStack,
                                 &sLvglTaskBuffer, 1);
 
-  ESP_LOGI(TAG, "Init Audio Output (I2S)");
+  ESP_LOGI(TAG, "Init audio output (I2S)");
   auto sink_res = drivers::I2SAudioOutput::create(expander);
   if (sink_res.has_error()) {
     ESP_LOGE(TAG, "Failed: %d", sink_res.error());
@@ -122,7 +122,7 @@ extern "C" void app_main(void) {
   }
   std::unique_ptr<drivers::IAudioOutput> sink = std::move(sink_res.value());
 
-  ESP_LOGI(TAG, "Init Audio Pipeline");
+  ESP_LOGI(TAG, "Init audio pipeline");
   auto playback_res = drivers::AudioPlayback::create(std::move(sink));
   if (playback_res.has_error()) {
     ESP_LOGE(TAG, "Failed: %d", playback_res.error());
@@ -131,8 +131,9 @@ extern "C" void app_main(void) {
   std::unique_ptr<drivers::AudioPlayback> playback =
       std::move(playback_res.value());
 
-  ESP_LOGI(TAG, "Everything looks good! Waiting a mo for debugger.");
-  vTaskDelay(pdMS_TO_TICKS(1500));
+  ESP_LOGI(TAG, "Launch console");
+  console::AppConsole console(std::move(playback));
+  console.Launch();
 
   while (1) {
     playback->ProcessEvents(5);
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 1b1ae818..0834a2b8 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -9,7 +9,10 @@ idf_build_set_property(
 # Treat warnings as errors for test purposes.
 list(APPEND EXTRA_WARNINGS "-Werror")
 
-list(APPEND EXTRA_COMPONENT_DIRS "$ENV{PROJ_PATH}/src/drivers")
+list(APPEND EXTRA_COMPONENT_DIRS
+  "$ENV{PROJ_PATH}/src/drivers"
+  "$ENV{PROJ_PATH}/src/dev_console"
+  )
 
 # List all components that include tests here.
 set(TEST_COMPONENTS "drivers")
diff --git a/test/main/CMakeLists.txt b/test/main/CMakeLists.txt
index 650dcc24..f0d35421 100644
--- a/test/main/CMakeLists.txt
+++ b/test/main/CMakeLists.txt
@@ -1,4 +1,4 @@
 idf_component_register(
   SRCS "main.cpp"
   INCLUDE_DIRS "."
-  REQUIRES "catch2")
+  REQUIRES "catch2 dev_console")
diff --git a/test/main/main.cpp b/test/main/main.cpp
index 9e01dbc0..13369cb4 100644
--- a/test/main/main.cpp
+++ b/test/main/main.cpp
@@ -1,63 +1,13 @@
 #include <stdio.h>
-#include <string.h>
-
-#include <algorithm>
-#include <iostream>
-#include <string>
+#include <cstdint>
 
 #include "esp_console.h"
 #include "esp_log.h"
-#include "esp_system.h"
 
 #include "catch_runner.hpp"
+#include "console.hpp"
 
-int exec_loglevel(int argc, char** argv) {
-  static const std::string usage =
-      "usage: loglevel [VERBOSE,DEBUG,INFO,WARN,ERROR,NONE]";
-  if (argc != 2) {
-    std::cout << usage << std::endl;
-    return 1;
-  }
-  std::string level_str = argv[1];
-  std::transform(level_str.begin(), level_str.end(), level_str.begin(),
-                 [](unsigned char c) { return std::toupper(c); });
-
-  esp_log_level_t level;
-  if (level_str == "VERBOSE") {
-    level = ESP_LOG_VERBOSE;
-  } else if (level_str == "DEBUG") {
-    level = ESP_LOG_DEBUG;
-  } else if (level_str == "INFO") {
-    level = ESP_LOG_INFO;
-  } else if (level_str == "WARN") {
-    level = ESP_LOG_WARN;
-  } else if (level_str == "ERROR") {
-    level = ESP_LOG_ERROR;
-  } else if (level_str == "NONE") {
-    level = ESP_LOG_NONE;
-  } else {
-    std::cout << usage << std::endl;
-    return 1;
-  }
-
-  esp_log_level_set("*", level);
-
-  return 0;
-}
-
-void register_loglevel() {
-  esp_console_cmd_t cmd{
-      .command = "loglevel",
-      .help =
-          "Sets the log level to one of \"VERBOSE\", \"DEBUG\", \"INFO\", "
-          "\"WARN\", \"ERROR\", \"NONE\"",
-      .hint = "level",
-      .func = &exec_loglevel,
-      .argtable = NULL};
-  esp_console_cmd_register(&cmd);
-}
-
-void register_catch2() {
+void RegisterCatch2() {
   esp_console_cmd_t cmd{
       .command = "catch",
       .help = "Execute the catch2 test runner. Use -? for options.",
@@ -67,23 +17,21 @@ void register_catch2() {
   esp_console_cmd_register(&cmd);
 }
 
-extern "C" void app_main(void) {
-  esp_console_repl_t* repl = nullptr;
-  esp_console_repl_config_t repl_config = ESP_CONSOLE_REPL_CONFIG_DEFAULT();
-  repl_config.max_history_len = 16;
-  repl_config.prompt = " →";
-  repl_config.max_cmdline_length = 256;
-  // Catch2 needs a huge stack, since it does a lot of pretty string formatting.
-  repl_config.task_stack_size = 1024 * 24;
+namespace console {
 
-  esp_console_dev_uart_config_t hw_config =
-      ESP_CONSOLE_DEV_UART_CONFIG_DEFAULT();
-  ESP_ERROR_CHECK(esp_console_new_repl_uart(&hw_config, &repl_config, &repl));
+class TestConsole : public Console {
+ protected:
+  virtual auto RegisterExtraComponents() -> void { RegisterCatch2(); }
+  virtual auto GetStackSizeKiB() -> uint16_t {
+    // Catch2 requires a particularly large stack.
+    return 24;
+  }
+};
 
-  esp_console_register_help_command();
-  register_loglevel();
-  register_catch2();
+}  // namespace console
 
+extern "C" void app_main(void) {
   esp_log_level_set("*", ESP_LOG_WARN);
-  ESP_ERROR_CHECK(esp_console_start_repl(repl));
+  console::Console* c = new console::TestConsole();
+  c->Launch();
 }