1 files changed, 118 insertions, 2 deletions

diff --git a/src/state.c b/src/state.c
index baf67c6..bfc4416 100644
--- a/src/state.c
+++ b/src/state.c
@@ -1,6 +1,9 @@
+#include <string.h>
+
 #include <engine/state.h>
 #include <engine/logging.h>
 #include <engine/dltools.h>
+#include <engine/input.h>
 
 typedef StateType state_update_t(void*);
 
@@ -132,8 +135,117 @@ StateType State_update(StateType type, memory *mem) {
   return next_state;
 }
 
+bool state_refresh_input_ctx(void *lib, i_ctx **ctx, usize ctx_len) {
+	if (ctx == NULL) return true;
+	if (ctx_len > 0 && ctx[0] == NULL) return false;
+	if (lib == NULL) return false;
+
+	for (usize c = 0; c < ctx_len; c++) {
+		LOG("ctx[%d]->len = %d", c, ctx[c]->len);
+		for (isize b = 0; b < ctx[c]->len; b++) {
+			switch (ctx[c]->bindings[b].action.type) {
+				case InputType_error:
+					break;
+				case InputType_action:
+					if (strcmp("NULL", ctx[c]->bindings[b].action.action.callback_str) != 0) {
+
+						ctx[c]->bindings[b].action.action.callback =
+							(input_callback_t*)dynamic_library_get_symbol(lib, ctx[c]->bindings[b].action.action.callback_str);
+
+							if (ctx[c]->bindings[b].action.action.callback == NULL) {
+								ERROR("Failed to get binding for %s: %s",
+										ctx[c]->bindings[b].action.action.callback_str,
+										dynamic_library_get_error());
+								return false;
+							}
+					}
+					break;
+				case InputType_state:
+					if (strcmp("NULL", ctx[c]->bindings[b].action.state.activate_str) != 0) {
+
+						ctx[c]->bindings[b].action.state.activate =
+							(input_callback_t*)dynamic_library_get_symbol(lib, ctx[c]->bindings[b].action.state.activate_str);
+
+							if (ctx[c]->bindings[b].action.state.activate == NULL) {
+								ERROR("Failed to get binding for %s: %s",
+										ctx[c]->bindings[b].action.state.activate_str,
+										dynamic_library_get_error());
+								return false;
+							}
+					}
+
+					if (strcmp("NULL", ctx[c]->bindings[b].action.state.deactivate_str) != 0) {
+
+						ctx[c]->bindings[b].action.state.deactivate =
+							(input_callback_t*)dynamic_library_get_symbol(lib, ctx[c]->bindings[b].action.state.deactivate_str);
+
+							if (ctx[c]->bindings[b].action.state.deactivate == NULL) {
+								ERROR("Failed to get binding for %s: %s",
+										ctx[c]->bindings[b].action.state.deactivate_str,
+										dynamic_library_get_error());
+								return false;
+							}
+					}
+					break;
+				case InputType_range:
+				default:
+					break;
+			}
+		}
+	}
+
+	return true;
+}
+
+/* IMPLEMENT CPY(dst,src) FOR THE SAME TYPE, SO WE CAN OVERRIDE THE OLD ONE AND
+ * FREE UP `ret->bindings` AND `ret` */
+i_ctx* i_ctx_dup(i_ctx **ctx, usize ctx_len) {
+	usize num_binds = 0;
+	for (usize c = 0; c < ctx_len; c++) {
+		num_binds += ctx[c]->len;
+	}
+
+	binding_t *bb = calloc(num_binds, sizeof(binding_t));
+	i_ctx *ret = calloc(ctx_len, sizeof(i_ctx));
+
+	usize cumsum = 0;
+	for (usize c = 0; c < ctx_len; c++) {
+			binding_t *b = ctx[c]->bindings;
+			ret[c].len = ctx[c]->len;
+			ret[c].bindings = &bb[cumsum];
+
+			for (isize i = 0; i < ctx[c]->len; i++) {
+				switch (b[i].action.type) {
+				case InputType_error:
+					break;
+				case InputType_action:
+					bb[cumsum] = BindActionLazy(
+							b[i].scancode,
+							b[i].scancode_alt,
+							b[i].action.action.callback_str);
+					break;
+				case InputType_state:
+					bb[cumsum] = BindStateLazy(
+							b[i].scancode,
+							b[i].scancode_alt,
+							b[i].action.state.activate_str,
+							b[i].action.state.deactivate_str);
+					break;
+				case InputType_range:
+				default:
+					break;
+				}
+				cumsum++;
+			}
+	}
+	return ret;
+}
+
 #ifdef DAW_BUILD_HOTRELOAD
-bool State_reload(StateType type) {
+bool State_reload(StateType type, i_ctx **ctx, usize ctx_len) {
+	void* libptr = NULL;
+	//i_ctx* ctx_cpy = i_ctx_dup(ctx, ctx_len);
+
   switch (type) {
 #define State(name)                                                             \
     case (STATE_##name): {                                                      \
@@ -153,6 +265,7 @@ bool State_reload(StateType type) {
       name##_init = (state_##name##_init_t*)dynamic_library_get_symbol(libstate_##name, STR( name##_init ) ); \
       name##_free = (state_##name##_free_t*)dynamic_library_get_symbol(libstate_##name, STR( name##_free ) ); \
       name##_update = (state_##name##_update_t*)dynamic_library_get_symbol(libstate_##name, STR( name##_update ) ); \
+      libptr = libstate_##name;                                                 \
       break;                                                                    \
     }
 #include <states/list_of_states.h>
@@ -161,10 +274,13 @@ bool State_reload(StateType type) {
     case STATE_quit:
       ERROR("Invalid state");
       DEBUG("Got %s state.\n", StateTypeStr[type]);
-      break;
+      return false;
     default:
       exit(EXIT_FAILURE);
   }
+	state_refresh_input_ctx(libptr, ctx, ctx_len);
+	//state_refresh_input_ctx(libptr, &ctx_cpy, ctx_len);
+	//ctx = &ctx_cpy;
   return true;
 }
 #endif