#include QMK_KEYBOARD_H

#include "ledmap.h"
#define MOON_LED_LEVEL LED_LEVEL

extern rgb_config_t rgb_matrix_config;

void keyboard_post_init_user(void) {
  rgb_matrix_enable();
}

void set_layer_color(int layer) {
  int n = sizeof(ledmap[layer]) / sizeof(*ledmap[layer]);

  for (int i = 0; i < n; i++) {
    /* The rgb index is counted from top to bottom, e.g.:
    *   0 4
    *   1 5
    *   2 6
    *   3 7 ..
    *
    *   But we have an index (`i`) which counts from left to right, e.g.:
    *   0 1 2 3
    *   4 5 6 7
    *   ..
    *
    *   Thus we use an translation table for the conversions (`i` -> `rgb_i`).
    */
    int rgb_i = translation[i];

    HSV hsv = {
        .h = pgm_read_byte(&ledmap[layer][i][0]),
        .s = pgm_read_byte(&ledmap[layer][i][1]),
        .v = pgm_read_byte(&ledmap[layer][i][2]),
    };

    if (!hsv.h && !hsv.s && !hsv.v) {
      rgb_matrix_set_color(rgb_i, 0, 0, 0);
    } else {
      RGB rgb = hsv_to_rgb(hsv);
      float f = (float)rgb_matrix_config.hsv.v / UINT8_MAX;
      rgb_matrix_set_color(rgb_i, f * rgb.r, f * rgb.g, f * rgb.b);
    }
  }
}

/* Callback. Is called once every time a new “frame” is rendered */
bool rgb_matrix_indicators_user(void) {
  if (keyboard_config.disable_layer_led) {
    return false;
  }

  int layer = biton32(layer_state);
  int defined_layer = sizeof(ledmap) / sizeof(*ledmap);

  if (layer <= (defined_layer - 1)) {
    set_layer_color(layer);
  } else {
    if (rgb_matrix_get_flags() == LED_FLAG_NONE) {
      rgb_matrix_set_color_all(0, 0, 0);
    }
  }

  return true;
}