1 files changed, 6 insertions, 90 deletions
diff --git a/main/main.c b/main/main.c
index 4caa26a..691043d 100644
--- a/main/main.c
+++ b/main/main.c
@@ -4,9 +4,10 @@
 #include "esp_system.h"
 #include "freertos/FreeRTOS.h"
 #include "freertos/task.h"
-#include "lwip/sockets.h"
 #include "sdkconfig.h"
-#include "station.h"
+
+#include "ws2812b_writer.h"
+#include "tcp_server.h"
 
 #ifdef CONFIG_IDF_TARGET_ESP32
 #define LCD_HOST HSPI_HOST
@@ -43,92 +44,7 @@
 #define PIN_NUM_BCKL 5
 #endif
 
-struct {
-  char color[32];
-  ws2812b_t* drv;
-} ws_params;
-
-#define SIZE 300
-void calculate_colors(ws2812b_buffer_t* buffer)
-{
-  /* RACE Conditions for sure. Need to fix that. */
-  ws2812b_rgb_t color = {0};
-  if (!strcmp(ws_params.color, "black\n")) {
-  } else if (!strcmp(ws_params.color, "green\n")) {
-    color.g = 255;
-  } else if (!strcmp(ws_params.color, "red\n")) {
-    color.r = 255;
-  } else if (!strcmp(ws_params.color, "blue\n")) {
-    color.b = 255;
-  } else if (!strcmp(ws_params.color, "yellow\n")) {
-    color.r = 255;
-    color.g = 255;
-  } else if (!strcmp(ws_params.color, "magenta\n")) {
-    color.r = 255;
-    color.b = 255;
-  } else if (!strcmp(ws_params.color, "teal\n")) {
-    color.g = 255;
-    color.b = 255;
-  } else {
-    color.r = 128;
-    color.g = 128;
-    color.b = 128;
-  }
-
-  int i;
-  for (i = 0; i < SIZE; ++i) {
-    ws2812b_buffer_set_color(buffer, i, &color);
-  }
-}
-
-portTASK_FUNCTION(ws2812b_write_task, params)
-{
-  strcpy(ws_params.color, "red\n");
-
-  ws2812b_buffer_t* buffer = ws2812b_new_buffer(SIZE);
-  for (;;) {
-    calculate_colors(buffer);
-    ws2812b_write(ws_params.drv, buffer);
-    vTaskDelay(10 / portTICK_PERIOD_MS);
-  }
-}
-
-portTASK_FUNCTION(tcp_server, params)
-{
-  wifi_init_station("Wort", "JoshIsBau5");
-
-  int s = socket(AF_INET, SOCK_STREAM, 0);
-  struct sockaddr_in addr;
-  memset(&addr, 0, sizeof(addr));
-  addr.sin_port = htons(1234);
-  addr.sin_family = AF_INET;
-  addr.sin_addr.s_addr = INADDR_ANY;
-  int err = bind(s, (struct sockaddr*)&addr, sizeof(addr));
-  if (err) {
-    printf("Error binding to sockaddr: %d\n", err);
-    return;
-  }
-
-  printf("Listening ...\n");
-  listen(s, 1);
-
-  while (true) {
-    printf("Waiting for connection... \n");
-
-    struct sockaddr_in client_addr;
-    socklen_t size;
-    int sock = accept(s, (struct sockaddr*)&client_addr, &size);
-
-    printf("Accepted connection\n");
-    ssize_t len;
-
-    while ((len = read(sock, ws_params.color, sizeof(ws_params.color) - 1)) >
-           0) {
-      ws_params.color[len] = 0;
-      printf("Read %s\n", ws_params.color);
-    }
-  }
-}
+static ws_params_t ws_params;
 
 void app_main(void)
 {
@@ -168,6 +84,6 @@ void app_main(void)
 
   printf("Configuration complete!!\n");
 
-  xTaskCreate(ws2812b_write_task, "ws2812b_writer", 4096, NULL, 1, NULL);
-  xTaskCreate(tcp_server, "tcp_server", 4096, NULL, 2, NULL);
+  xTaskCreate(ws2812b_write_task, "ws2812b_writer", 4096, &ws_params, 1, NULL);
+  xTaskCreate(tcp_server, "tcp_server", 4096, &ws_params, 2, NULL);
 }