From efd241d9504090bbae049e9efcca79773afe6513 Mon Sep 17 00:00:00 2001
From: Klemen Plestenjak <plestenjak.klemen@gmail.com>
Date: Tue, 23 Jan 2024 18:55:56 +0100
Subject: [PATCH] Add gold mining

---
 game/components.h      |  2 --
 game/input.h           |  1 +
 game/systems/s_input.c | 54 ++++++++++++++++++++++--------------------
 3 files changed, 29 insertions(+), 28 deletions(-)

diff --git a/game/components.h b/game/components.h
index 5fadc05..49269e1 100644
--- a/game/components.h
+++ b/game/components.h
@@ -11,7 +11,6 @@
 extern ECS_TAG_DECLARE(GameEntity);
 
 typedef enum ResourceType {
-    RES_IRON,
     RES_WOOD,
     RES_GOLD,
     RES_FOOD,
@@ -20,7 +19,6 @@ typedef enum ResourceType {
 
 static const char *getResourceTypePrettyName(ResourceType type) {
     switch (type) {
-        case RES_IRON: return "Iron";
         case RES_WOOD: return "Wood";
         case RES_GOLD: return "Gold";
         case RES_FOOD: return "Food";
diff --git a/game/input.h b/game/input.h
index 740dd79..8d68f52 100644
--- a/game/input.h
+++ b/game/input.h
@@ -45,6 +45,7 @@ typedef struct InputState {
     enum {
         CURSOR_NONE,
         CURSOR_COLLECT_WOOD,
+        CURSOR_COLLECT_GOLD,
     } cursor;
 
     // INPUT_BUILDING
diff --git a/game/systems/s_input.c b/game/systems/s_input.c
index 661f093..6e17fdf 100644
--- a/game/systems/s_input.c
+++ b/game/systems/s_input.c
@@ -82,41 +82,37 @@ void inputUnitAction(Game *game, InputState *input) {
     input->cursor = CURSOR_NONE;
     ecs_entity_t taskEntity;
     if ((taskEntity = queryEntity(game->entityGrid, input->mouseWorld, Harvestable))) {
-        input->cursor = CURSOR_COLLECT_WOOD;
+        Resource resource = *ecs_get(ECS, taskEntity, Resource);
+        switch (resource.type) {
+            case RES_WOOD:
+                input->cursor = CURSOR_COLLECT_WOOD;
+                break;
+            case RES_GOLD:
+                input->cursor = CURSOR_COLLECT_GOLD;
+                break;
+            case RES_FOOD:
+                //input->cursor = CURSOR_COLLECT_FOOD;
+                break;
+            default:;
+        }
         if (isInputBtnJustUp(input, actionBtn)) {
             ecs_defer_begin(ECS);
             ecs_iter_t it = ecs_query_iter(ECS, query);
             while (ecs_query_next(&it)) {
                 for (i32 i = 0; i < it.count; i++) {
-                    const ecs_entity_t entity = it.entities[i];
-                    const Position target = *ecs_get(ECS, taskEntity, Position);
+                    ecs_entity_t entity = it.entities[i];
+                    Position target = *ecs_get(ECS, taskEntity, Position);
+
                     setAIBehaviour(entity, game->BTs.workerHarvest, &(AIBlackboard) {
-                        .as.worker = {
-                                .harvestType = RES_WOOD,
-                                .harvestTarget = taskEntity,
-                                .harvestPos = target,
-                        },
-                        .proximity = 6.0f,
+                            .as.worker = {
+                                    .harvestType = resource.type,
+                                    .harvestTarget = taskEntity,
+                                    .harvestPos = target,
+                            },
+                            .proximity = 6.0f,
                     });
-                    /*
-                    setUnitAI(entity, game, &(const UnitAI) {
-                        .type = AI_WORKER_HARVEST,
-                        .as.workerHarvest.resource = RES_WOOD,
-                        .as.workerHarvest.target = taskEntity,
-                        .as.workerHarvest.targetPosition = target
-                    });
-                     */
-                    //addAction(entity, game, &(const Action) {
-                    //    .type = ACTION_MOVE_TO,
-                    //    .as.moveTo.target = target,
-                    //    .as.moveTo.proximityThreshold = 10.0f,
-                    //});
-                    //ecs_set(ECS, entity, HarvestTask, {taskEntity});
-                    goto while_break;
                 }
             }
-            while_break:
-            ecs_iter_fini(&it);
             ecs_defer_end(ECS);
         }
         return;
@@ -348,6 +344,12 @@ void drawPlayerInputUI() {
             DrawText("Collect wood", point.x, point.y, 10.0f, RED);
             break;
         }
+        case CURSOR_COLLECT_GOLD: {
+            const Vector2 point = input->mouseWorld;
+            DrawCircle(point.x, point.y, 2.0f, RED);
+            DrawText("Collect gold", point.x, point.y, 10.0f, RED);
+            break;
+        }
         default: break;
     }