#include "pd_api.h"

#include "../../clay.h"

struct Clay_Playdate_Rect {
  float x;
  float y;
  float w;
  float h;
};

// Playdate drawText function expects the number of codepoints to draw, not byte length
static size_t Clay_Playdate_CountUtf8Codepoints(const char *str, size_t byte_len) {
  size_t count = 0;
  size_t i = 0;
  while (i < byte_len) {
    uint8_t c = (uint8_t)str[i];

    // Skip continuation bytes (10xxxxxx)
    if ((c & 0xC0) != 0x80) {
      count++;
    }

    i++;
  }
  return count;
}

// As the playdate can only display black and white, we need to resolve Clay_color to either black or white
// for both color and draw mode.
// TODO: Convert to grayscale and then map the grayscale value to different dithering patterns
static LCDColor clayColorToLCDColor(Clay_Color color) {
  if (color.r > 0 || color.g > 0 || color.b > 0) {
    return kColorWhite;
  }
  return kColorBlack;
}

static LCDBitmapDrawMode clayColorToDrawMode(Clay_Color color) {
  if (color.r > 0 || color.g > 0 || color.b > 0) {
    return kDrawModeFillWhite;
  }
  return kDrawModeCopy;
}

static void Clay_Playdate_Render(PlaydateAPI *pd, Clay_RenderCommandArray renderCommands, LCDFont **fonts) {
  for (uint32_t i = 0; i < renderCommands.length; i++) {
    Clay_RenderCommand *renderCommand = Clay_RenderCommandArray_Get(&renderCommands, i);
    Clay_BoundingBox boundingBox = renderCommand->boundingBox;
    switch (renderCommand->commandType) {
    case CLAY_RENDER_COMMAND_TYPE_RECTANGLE: {
      Clay_RectangleRenderData *config = &renderCommand->renderData.rectangle;
      struct Clay_Playdate_Rect rect = (struct Clay_Playdate_Rect){
        .x = boundingBox.x,
        .y = boundingBox.y,
        .w = boundingBox.width,
        .h = boundingBox.height,
      };
      // TODO: support different radius for each corner like clay API allows
      if (config->cornerRadius.topLeft > 0) {
        pd->graphics->fillRoundRect(
          rect.x, rect.y, rect.w, rect.h,
          config->cornerRadius.topLeft,
          clayColorToLCDColor(config->backgroundColor)
        );
      } else {
        pd->graphics->fillRect(
          rect.x, rect.y, rect.w, rect.h,
          clayColorToLCDColor(config->backgroundColor)
        );
      }
      break;
    }
    case CLAY_RENDER_COMMAND_TYPE_TEXT: {
      Clay_TextRenderData *config = &renderCommand->renderData.text;
      LCDFont *font = fonts[config->fontId];
      pd->graphics->setFont(font);
      struct Clay_Playdate_Rect destination = (struct Clay_Playdate_Rect){
        .x = boundingBox.x,
        .y = boundingBox.y,
        .w = boundingBox.width,
        .h = boundingBox.height,
      };
      pd->graphics->setDrawMode(clayColorToDrawMode(config->textColor));
      pd->graphics->drawText(
        renderCommand->renderData.text.stringContents.chars,
        Clay_Playdate_CountUtf8Codepoints(
          renderCommand->renderData.text.stringContents.chars,
          renderCommand->renderData.text.stringContents.length
        ),
        kUTF8Encoding,
        destination.x,
        destination.y
      );
      pd->graphics->setDrawMode(kDrawModeCopy);
      break;
    }
    case CLAY_RENDER_COMMAND_TYPE_SCISSOR_START: {
      struct Clay_Playdate_Rect currentClippingRectangle = (struct Clay_Playdate_Rect){
        .x = boundingBox.x,
        .y = boundingBox.y,
        .w = boundingBox.width,
        .h = boundingBox.height,
      };
      pd->graphics->setClipRect(
        currentClippingRectangle.x, currentClippingRectangle.y,
        currentClippingRectangle.w, currentClippingRectangle.h
      );
      break;
    }
    case CLAY_RENDER_COMMAND_TYPE_SCISSOR_END: {
      pd->graphics->clearClipRect();
      break;
    }
    case CLAY_RENDER_COMMAND_TYPE_IMAGE: {
      Clay_ImageRenderData *config = &renderCommand->renderData.image;
      LCDBitmap *texture = config->imageData;
      struct Clay_Playdate_Rect destination = (struct Clay_Playdate_Rect){
        .x = boundingBox.x,
        .y = boundingBox.y,
        .w = boundingBox.width,
        .h = boundingBox.height,
      };
      pd->graphics->drawBitmap(texture, destination.x, destination.y, kBitmapUnflipped);
      break;
    }
    case CLAY_RENDER_COMMAND_TYPE_BORDER: {
      Clay_BorderRenderData *config = &renderCommand->renderData.border;
      struct Clay_Playdate_Rect rect = (struct Clay_Playdate_Rect){
        .x = boundingBox.x,
        .y = boundingBox.y,
        .w = boundingBox.width,
        .h = boundingBox.height,
      };

      // TODO: properly support the different border thickness and radius
      // instead of just using topLeft corner /top thickness as a global setting
      if (config->cornerRadius.topLeft > 0) {
        pd->graphics->drawRoundRect(
          rect.x, rect.y, rect.w, rect.h,
          config->cornerRadius.topLeft,
          config->width.top,
          clayColorToLCDColor(config->color)
        );
      } else {
        pd->graphics->drawRect(
          rect.x, rect.y, rect.w, rect.h,
          clayColorToLCDColor(config->color)
        );
      }
      break;
    }
    default: {
      pd->system->logToConsole("Error: unhandled render command: %d\n", renderCommand->commandType);
      return;
    }
    }
  }
}