// SPDX-FileCopyrightText: 2026 Cesanta Software Limited
// SPDX-License-Identifier: GPL-2.0-only or commercial

#include "mongoose.h"

#if MG_ARCH == MG_ARCH_UNIX || MG_ARCH == MG_ARCH_WIN32
#define HTTP_ADDR "http://0.0.0.0:8000"
#else
#define HTTP_ADDR "http://0.0.0.0:80"
#endif

// Control panel
static bool s_led1, s_led2, s_led3;

static struct mg_field fields_leds[] = {
    {"led1", MG_VAL_BOOL, &s_led1, sizeof(s_led1)},
    {"led2", MG_VAL_BOOL, &s_led2, sizeof(s_led2)},
    {"led3", MG_VAL_BOOL, &s_led3, sizeof(s_led3)},
    {NULL, 0, NULL, 0},
};

static void sync_leds(bool is_write) {
  if (is_write) {
    // gpio_write(LED1_PIN, s_led1);
    // gpio_write(LED2_PIN, s_led2);
    // gpio_write(LED3_PIN, s_led3);
  } else {
    // s_led1 = gpio_read(LED1_PIN);
    // s_led2 = gpio_read(LED2_PIN);
    // s_led3 = gpio_read(LED3_PIN);
  }
}

// Read-only device Metrics
static int s_ram = 32, s_cpu = 7;
static double s_temperature = 24.8;

static struct mg_field fields_metrics[] = {
    {"ram", MG_VAL_INT, &s_ram, sizeof(s_ram)},
    {"cpu", MG_VAL_INT, &s_cpu, sizeof(s_cpu)},
    {"temperature", MG_VAL_DBL, &s_temperature, sizeof(s_temperature)},
    {NULL, 0, NULL, 0},
};

static void sync_metrics(bool is_write) {
  if (is_write) return;
  s_ram = 25 + (rand() % 16);  // Sumulate metric change
  s_cpu = 7 + (rand() % 21);
  s_temperature = 14.8 + ((double) rand() / RAND_MAX) * 20.0;
}

// Read-write device settings
static double s_volume = 17.2;
static int s_log_level = MG_LL_DEBUG;
static char s_name[20] = "Dublin";

static struct mg_field fields_settings[] = {
    {"volume", MG_VAL_DBL, &s_volume, sizeof(s_volume)},
    {"name", MG_VAL_STR, &s_name, sizeof(s_name)},
    {"log_level", MG_VAL_INT, &s_log_level, sizeof(s_log_level)},
    {NULL, 0, NULL, 0},
};

static void sync_settings(bool is_write) {
  if (is_write) {
    mg_log_level = s_log_level;
  } else {
    s_log_level = mg_log_level;
  }
}

#define NUM_POINTS 7  // How many data points to send

struct point {
  uint16_t x;   // X coordinate of the data point, usually this is time
  uint16_t y1;  // We can plot multiple values per graph,
  uint16_t y2;  // let's plot 2 values: y1 and y2.
};

// Serialise chart data into a string: [[x,y1,y2],[x,y1,y2],...]
// One uint16_t takes 5 character max, so one entry is 19 bytes max.
// We could be more efficient by storing hex or base64-encoded data, but
// let's keep it simple. However if we do, update data parsing logic in
// the web_root/main.js
#define POINT_SIZE 19
static char s_chart1_data[NUM_POINTS * POINT_SIZE + 2 + 1];

static struct mg_field fields_chart1[] = {
    {"data", MG_VAL_RAW, s_chart1_data, sizeof(s_chart1_data)},
    {NULL, 0, NULL, 0},
};

// Randomly generate graph points and serialise them into a string
void sync_chart1(bool is_write) {
  size_t i, len = 0;
  if (is_write) return;
  len += mg_snprintf(s_chart1_data + len, sizeof(s_chart1_data) - len, "[");
  for (i = 0; i < NUM_POINTS; i++) {
    struct point p;
    p.x = (uint16_t) i;
    p.y1 = 85 - (uint16_t) ((49 * i) / NUM_POINTS) + (rand() % 9);
    p.y2 = 20 + (uint16_t) ((69 * i) / NUM_POINTS) + (rand() % 5);
    len += mg_snprintf(s_chart1_data + len, sizeof(s_chart1_data) - len,
                       "%s[%hu,%hu,%hu]", i > 0 ? "," : "", p.x, p.y1, p.y2);
  }
  len += mg_snprintf(s_chart1_data + len, sizeof(s_chart1_data) - len, "]");
}

struct file {
  struct file *next;
  char *name;
  size_t size;
} *s_file_list;

static void file_add(struct mg_str name, size_t size) {
  struct file *f = mg_calloc(1, sizeof(*f));
  f->name = mg_strdup(name).buf;
  f->size = size;
  f->next = s_file_list;
  s_file_list = f;
}

static void file_del(struct mg_str name) {
  struct file **head, *f;
  for (head = &s_file_list, f = *head; f; head = &(*head)->next, f = *head) {
    if (mg_strcmp(mg_str(f->name), name) == 0) {
      MG_INFO(("Deleting %s", f->name));
      *head = f->next;
      mg_free(f->name);
      mg_free(f);
      return;
    }
  }
}

// Files array: [{"name": "foo.txt", "size": 1234}]
static char s_files[1024];
static struct mg_field fields_files[] = {
    {"data", MG_VAL_RAW, s_files, sizeof(s_files)},
    {NULL, 0, NULL, 0},
};
static void sync_files(bool is_write) {
  size_t len = 0;
  struct file *f;
  if (is_write) return;
  len += mg_snprintf(s_files + len, sizeof(s_files) - len, "[");
  for (f = s_file_list; f; f = f->next) {
    len += mg_snprintf(s_files + len, sizeof(s_files) - len, "%s{%m:%m,%m:%u}",
                       len > 1 ? "," : "", MG_ESC("name"), MG_ESC(f->name),
                       MG_ESC("size"), f->size);
  }
  len += mg_snprintf(s_files + len, sizeof(s_files) - len, "]");
}

static struct mg_field_set field_sets[] = {
    {"leds", fields_leds, sync_leds, 0, 0},
    {"metrics", fields_metrics, sync_metrics, 0, 0},
    {"settings", fields_settings, sync_settings, 0, 0},
    {"chart1", fields_chart1, sync_chart1, 0, 0},
    {"files", fields_files, sync_files, 0, 0},
    {0, 0, 0, 0, 0},
};
static struct mg_dash s_dashboard = {field_sets, NULL};

struct upload_state {
  char marker;      // Tells that we're a file upload connection
  size_t expected;  // POST data length, bytes
  size_t received;  // Already received bytes
  void *ctx;        // OTA context
};

static void file_ev_handler(struct mg_connection *c, int ev, void *ev_data) {
  struct upload_state *us = (struct upload_state *) c->data;
  if ((ev == MG_EV_HTTP_HDRS || ev == MG_EV_HTTP_MSG) && us->marker == 0) {
    struct mg_http_message *hm = (struct mg_http_message *) ev_data;
    if (hm->uri.len > 7) hm->uri.len -= 7, hm->uri.buf += 7;  // Strip /files/
    if (sizeof(*us) > sizeof(c->data)) {
      mg_error(c,
               "FAILURE: sizeof(c->data) == %lu, need %lu."
               " Set #define MG_DATA_SIZE XXX",
               sizeof(c->data), sizeof(*us));
    } else if (mg_strcmp(hm->query, mg_str("action=delete")) == 0) {
      // Query string ?action=delete - file deletion
      file_del(hm->uri);
      mg_dash_send_change(c->mgr, &s_dashboard, "files");
      mg_http_reply(c, 200, NULL, "true");
    } else if (mg_path_is_sane(hm->uri) == false) {
      // Bad file name, return error. Protect from traversal, etc
      mg_http_reply(c, 400, NULL, "Bad name");
    } else if (mg_match(hm->method, mg_str("POST"), NULL) ||
               mg_match(hm->method, mg_str("PUT"), NULL)) {
      // File upload
      char path[128];
      mg_snprintf(path, sizeof(path), "/tmp/%.*s", hm->uri.len, hm->uri.buf);
      us->ctx = fopen(path, "wb+");
      if (us->ctx == NULL) {
        mg_http_reply(c, 500, NULL, "File upload error %d\n", errno);
      } else {
        file_add(hm->uri, hm->body.len);
        MG_DEBUG(("Starting upload, [%.*s] %lu", hm->uri.len, hm->uri.buf, hm->body.len));
        us->marker = 'U';
        us->expected = hm->body.len;
        us->received = 0;
        mg_iobuf_del(&c->recv, 0, hm->head.len);
        c->pfn = NULL;
        c->fn = file_ev_handler;
        mg_call(c, MG_EV_READ, &c->recv.len);
      }
    } else {
      // Serve file
      char path[128];
      mg_snprintf(path, sizeof(path), "/tmp/%.*s", hm->uri.len, hm->uri.buf);
      mg_http_serve_file(c, ev_data, path, NULL);
    }
  } else if (ev == MG_EV_READ && us->marker == 'U') {
    // Write uploaded data
    MG_DEBUG(("Uploading.. recv.len=%u", c->recv.len));
    us->received += c->recv.len;
    fwrite(c->recv.buf, 1, c->recv.len, us->ctx);
    c->recv.len = 0;
    if (us->received >= us->expected) {
      fclose(us->ctx);
      MG_DEBUG(("Uploaded %lu", us->expected));
      mg_http_reply(c, 200, NULL, "%lu uploaded\n", us->expected);
      mg_dash_send_change(c->mgr, &s_dashboard, "files");
      memset(us, 0, sizeof(*us));
      c->is_draining = 1;
    }
  }
}

void mg_dash_init(struct mg_mgr *mgr) {
  file_add(mg_str("device-config.json"), 1234);
  file_add(mg_str("device-log-2026-04-25.txt"), 1327854);
  MG_DASH_REGISTER_CUSTOM_HANDLER(&s_dashboard, "/files/#", file_ev_handler);
  mg_http_listen(mgr, HTTP_ADDR, mg_dash_ev_handler, &s_dashboard);
}

void mg_dash_poll(struct mg_mgr *mgr) {
  // Send metrics change periodically
  static uint64_t timer = 0;
  if (mg_timer_expired(&timer, 30000, mg_now())) {
    mg_dash_send_change(mgr, &s_dashboard, "metrics");
  }
}