#include <raylib.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include "tiled.h"

int lastSelectedTile[2] = {0, 0};
int selectedTile[2] = {0, 0};

int mode = -1;

void drawOverlay(Tiled tiled) {
    Vector2 screenPos = translateTiledScreenPosition(tiled, (Vector2){selectedTile[0], selectedTile[1]});
    
    DrawRectangleLinesEx((Rectangle) {
                screenPos.x,
                screenPos.y,
                tiled.zoom,
                tiled.zoom
            }, tiled.zoom/16, GREEN);
}

void modifyTile(Tiled *tiled, int tile) {
    setTiledMapTile(tiled->tiledMap, selectedTile, tile);
    redrawTiledMap(*tiled);
}

void setDrawMode(Tiled *tiled, int tile) {
    mode = tile % tiled->tiledMap.tileCount;
    if (mode < 0) mode += tiled->tiledMap.tileCount;
    modifyTile(tiled, mode);
}

void handleInputs(Tiled *tiled) {
    if (IsMouseButtonPressed(MOUSE_BUTTON_LEFT))
        setDrawMode(tiled, getTiledMapTile(tiled->tiledMap, selectedTile) + 1);

    if (IsMouseButtonPressed(MOUSE_BUTTON_RIGHT))
        setDrawMode(tiled, getTiledMapTile(tiled->tiledMap, selectedTile) - 1);

    if (IsMouseButtonPressed(MOUSE_BUTTON_MIDDLE))
        setDrawMode(tiled, 0);

    if (IsMouseButtonReleased(MOUSE_BUTTON_LEFT) ||
        IsMouseButtonReleased(MOUSE_BUTTON_RIGHT) ||
        IsMouseButtonReleased(MOUSE_BUTTON_MIDDLE)) {
        mode = -1;
    }


    if (!(selectedTile[0] == lastSelectedTile[0] &&
          selectedTile[1] == lastSelectedTile[1])) {
        if (mode != -1) modifyTile(tiled, mode);
    }
}

void update(Tiled *tiled) {
    updateTiledCamera(tiled);
    Vector2 mousePos = GetMousePosition();
    Vector2 mapPos = translateTiledPosition(*tiled, mousePos);
    if (mapPos.x >= 0 && mapPos.x < tiled->tiledMap.width
        && mapPos.y >= 0 && mapPos.y < tiled->tiledMap.height) {
        lastSelectedTile[0] = selectedTile[0];
        lastSelectedTile[1] = selectedTile[1];
        selectedTile[0] = mapPos.x;
        selectedTile[1] = mapPos.y;
    }
    handleInputs(tiled);
}

TiledMap launchEditor(TiledMap tiledMap) {
    SetConfigFlags(FLAG_WINDOW_RESIZABLE);
    InitWindow(SCREEN_W, SCREEN_H, "tiled");

    Tiled tiled = initTiled(tiledMap);

    while (!WindowShouldClose()) {
        update(&tiled);

        BeginDrawing();

        ClearBackground(LIGHTGRAY);

        drawTiled(&tiled);
        drawOverlay(tiled);
        DrawFPS(16, 16);

        EndDrawing();
    }

    unloadTiled(&tiled);

    CloseWindow();
    return tiled.tiledMap;
}

void printUsage(char *progname) {
    fprintf(stderr, "%s [-h] [-t tile_size] [-s map_size] [-a atlas.png]\n", progname);
    exit(1);
}

int main(int argc, char *argv[]) {
    char * tiledFilePath;
    const char * atlasFilePath = NULL;
    int tileSize = 16;
    int mapSize = 16;

    int flags, opt;
    while ((opt = getopt(argc, argv, "s:a:h")) != -1) {
        switch (opt) {
            case 's':
                mapSize = atoi(optarg);
            case 't':
                tileSize = atoi(optarg);
            case 'a':
                atlasFilePath = optarg;
        }
    }
    
    if (optind >= argc)
        printUsage(argv[0]);

    tiledFilePath = argv[optind];

    TiledMap tiledMap;
    if (access(tiledFilePath, F_OK)) {
        if (atlasFilePath == NULL) {
            fprintf(stderr, "Atlas file must be specified!\n");
            printUsage(argv[0]);
        }

        Image atlasImage = LoadImage(atlasFilePath);
        tiledMap = newTiledMap(atlasImage, tileSize, mapSize, mapSize);
    } else {
        tiledMap = loadTiledMap(tiledFilePath);
    }

    TiledMap editedTiledMap = launchEditor(tiledMap);
    saveTiledMap(tiledFilePath, tiledMap);
}