1 files changed, 102 insertions, 5 deletions

diff --git a/src/menu.cpp b/src/menu.cpp
index 269174d..56f8c95 100644
--- a/src/menu.cpp
+++ b/src/menu.cpp
@@ -30,7 +30,7 @@ void init_conmsg() {
 // menu tree structure
 // ---------------------------------------------------------------------------
 
-enum NodeType { NODE_ROOT, NODE_FEATURE, NODE_TOGGLE, NODE_BIND };
+enum NodeType { NODE_ROOT, NODE_FEATURE, NODE_TOGGLE, NODE_BIND, NODE_VALUE };
 
 struct MenuNode {
   NodeType type;
@@ -43,6 +43,10 @@ struct MenuNode {
   bool *active_ptr;  // NODE_TOGGLE: pointer to the active flag
   int *bind_ptr;     // NODE_BIND: pointer to the key binding (SDL scancode)
   const char *bind_name; // NODE_BIND: current bind display name
+  int *value_ptr = nullptr; // NODE_VALUE: pointer to the value
+  int value_min = 0;
+  int value_max = 0;
+  int value_step = 0;
 };
 
 // feature data
@@ -58,6 +62,20 @@ static FeatureState g_abh{true, 0};
 static FeatureState g_esp{true, 0};
 static FeatureState g_aimbot{true, 604}; // default Mouse5, rebindable in menu
 
+// Aimbot bone checklist. Head defaults on (v1 behaviour).
+enum {
+  AIM_BONE_HEAD = 0,
+  AIM_BONE_TORSO,
+  AIM_BONE_LARM,
+  AIM_BONE_RARM,
+  AIM_BONE_LLEG,
+  AIM_BONE_RLEG,
+  AIM_BONE_COUNT
+};
+static bool g_aim_bones[AIM_BONE_COUNT] = {true, false, false,
+                                           false, false, false};
+static int g_aim_smooth = 5; // degrees per tick
+
 // ESP sub-options (menu "ESP" children). Defaults match the pre-option
 // behaviour: everything on.
 static bool g_esp_opt[ESP_OPT_COUNT] = {true, true, true, true, true};
@@ -81,6 +99,17 @@ static MenuNode g_esp_opt_nodes[ESP_OPT_COUNT] = {
 static MenuNode g_aimbot_feat{NODE_FEATURE, "Aimbot", false, 1, {}, nullptr, nullptr, nullptr};
 static MenuNode g_aimbot_active{NODE_TOGGLE, "Active", false, 0, {}, &g_aimbot.active, nullptr, nullptr};
 static MenuNode g_aimbot_bind{NODE_BIND, "Bind", false, 0, {}, nullptr, &g_aimbot.bind, nullptr};
+static MenuNode g_aimbot_bones_feat{NODE_FEATURE, "Bones", false, 0, {}, nullptr, nullptr, nullptr};
+static MenuNode g_aimbot_bone_nodes[AIM_BONE_COUNT] = {
+    {NODE_TOGGLE, "Head", false, 0, {}, &g_aim_bones[AIM_BONE_HEAD], nullptr, nullptr},
+    {NODE_TOGGLE, "Torso", false, 0, {}, &g_aim_bones[AIM_BONE_TORSO], nullptr, nullptr},
+    {NODE_TOGGLE, "Left arm", false, 0, {}, &g_aim_bones[AIM_BONE_LARM], nullptr, nullptr},
+    {NODE_TOGGLE, "Right arm", false, 0, {}, &g_aim_bones[AIM_BONE_RARM], nullptr, nullptr},
+    {NODE_TOGGLE, "Left leg", false, 0, {}, &g_aim_bones[AIM_BONE_LLEG], nullptr, nullptr},
+    {NODE_TOGGLE, "Right leg", false, 0, {}, &g_aim_bones[AIM_BONE_RLEG], nullptr, nullptr},
+};
+static MenuNode g_aimbot_smooth_node{NODE_VALUE, "Smoothness", false, 0, {}, nullptr,
+                                     nullptr, nullptr, &g_aim_smooth, 1, 45, 1};
 
 static MenuNode *g_visible[32]; // flattened visible nodes for arrow navigation
 static int g_visible_count = 0;
@@ -96,7 +125,12 @@ static void build_tree() {
   g_esp_feat.children[0] = &g_esp_active;
   g_aimbot_feat.children[0] = &g_aimbot_active;
   g_aimbot_feat.children[1] = &g_aimbot_bind;
-  g_aimbot_feat.child_count = 2;
+  g_aimbot_feat.children[2] = &g_aimbot_bones_feat;
+  g_aimbot_feat.children[3] = &g_aimbot_smooth_node;
+  g_aimbot_feat.child_count = 4;
+  for (int i = 0; i < AIM_BONE_COUNT; i++)
+    g_aimbot_bones_feat.children[i] = &g_aimbot_bone_nodes[i];
+  g_aimbot_bones_feat.child_count = AIM_BONE_COUNT;
   for (int i = 0; i < ESP_OPT_COUNT; i++)
     g_esp_feat.children[i + 1] = &g_esp_opt_nodes[i];
   g_esp_feat.child_count = ESP_OPT_COUNT + 1;
@@ -115,6 +149,7 @@ static void flatten(MenuNode *node, int depth) {
   node->type = (depth == 0) ? NODE_ROOT
               : (node->child_count > 0) ? NODE_FEATURE
               : (node->active_ptr) ? NODE_TOGGLE
+              : (node->value_ptr) ? NODE_VALUE
               : NODE_BIND;
   g_depth[g_visible_count] = depth;
   g_visible[g_visible_count++] = node;
@@ -167,8 +202,13 @@ static void config_save() {
   fprintf(f, "[bhop]\nactive=%d\nbind=%d\nautostrafe=%d\n", g_bhop.active,
           g_bhop.bind, g_bhop_autostrafe);
   fprintf(f, "[abh]\nactive=%d\nbind=%d\n", g_abh.active, g_abh.bind);
-  fprintf(f, "[aimbot]\nactive=%d\nbind=%d\n", g_aimbot.active,
-          g_aimbot.bind);
+  fprintf(f, "[aimbot]\nactive=%d\nbind=%d\nsmooth=%d\n"
+             "bone_head=%d\nbone_torso=%d\nbone_larm=%d\nbone_rarm=%d\n"
+             "bone_lleg=%d\nbone_rleg=%d\n",
+          g_aimbot.active, g_aimbot.bind, g_aim_smooth,
+          g_aim_bones[AIM_BONE_HEAD], g_aim_bones[AIM_BONE_TORSO],
+          g_aim_bones[AIM_BONE_LARM], g_aim_bones[AIM_BONE_RARM],
+          g_aim_bones[AIM_BONE_LLEG], g_aim_bones[AIM_BONE_RLEG]);
   fprintf(f, "[esp]\nactive=%d\nbind=%d\nneutral=%d\nhealthbar=%d\n"
              "healthnum=%d\nname=%d\nskeleton=%d\n",
           g_esp.active, g_esp.bind, g_esp_opt[ESP_OPT_SHOW_NEUTRAL],
@@ -184,7 +224,7 @@ static void config_load() {
     return;
   char line[256];
   FeatureState *cur = nullptr;
-  bool in_esp = false, in_bhop = false;
+  bool in_esp = false, in_bhop = false, in_aim = false;
   while (fgets(line, sizeof(line), f)) {
     line[strcspn(line, "\r\n")] = 0;
     if (line[0] == '[') {
@@ -200,6 +240,7 @@ static void config_load() {
         cur = nullptr;
       in_esp = (cur == &g_esp);
       in_bhop = (cur == &g_bhop);
+      in_aim = (cur == &g_aimbot);
     } else if (cur) {
       char *eq = strchr(line, '=');
       if (!eq)
@@ -209,6 +250,26 @@ static void config_load() {
       if (in_bhop && strcmp(key, "autostrafe") == 0) {
         g_bhop_autostrafe = atoi(val) != 0;
       }
+      if (in_aim) {
+        if (strcmp(key, "smooth") == 0) {
+          int v = atoi(val);
+          if (v < 1) v = 1;
+          if (v > 45) v = 45;
+          g_aim_smooth = v;
+        } else if (strcmp(key, "bone_head") == 0) {
+          g_aim_bones[AIM_BONE_HEAD] = atoi(val) != 0;
+        } else if (strcmp(key, "bone_torso") == 0) {
+          g_aim_bones[AIM_BONE_TORSO] = atoi(val) != 0;
+        } else if (strcmp(key, "bone_larm") == 0) {
+          g_aim_bones[AIM_BONE_LARM] = atoi(val) != 0;
+        } else if (strcmp(key, "bone_rarm") == 0) {
+          g_aim_bones[AIM_BONE_RARM] = atoi(val) != 0;
+        } else if (strcmp(key, "bone_lleg") == 0) {
+          g_aim_bones[AIM_BONE_LLEG] = atoi(val) != 0;
+        } else if (strcmp(key, "bone_rleg") == 0) {
+          g_aim_bones[AIM_BONE_RLEG] = atoi(val) != 0;
+        }
+      }
       if (in_esp) {
         if (strcmp(key, "neutral") == 0)
           g_esp_opt[ESP_OPT_SHOW_NEUTRAL] = atoi(val) != 0;
@@ -346,6 +407,25 @@ bool esp_option(int opt) {
 
 bool bhop_autostrafe_enabled() { return g_bhop_autostrafe; }
 
+unsigned aimbot_bone_mask() {
+  unsigned m = 0;
+  if (g_aim_bones[AIM_BONE_HEAD])
+    m |= 1u << 1; // HITGROUP_HEAD
+  if (g_aim_bones[AIM_BONE_TORSO])
+    m |= (1u << 2) | (1u << 3); // chest + stomach
+  if (g_aim_bones[AIM_BONE_LARM])
+    m |= 1u << 4;
+  if (g_aim_bones[AIM_BONE_RARM])
+    m |= 1u << 5;
+  if (g_aim_bones[AIM_BONE_LLEG])
+    m |= 1u << 6;
+  if (g_aim_bones[AIM_BONE_RLEG])
+    m |= 1u << 7;
+  return m;
+}
+
+int aimbot_smooth_step() { return g_aim_smooth; }
+
 static void menu_handle_input() {
   poll_keyboard();
 
@@ -415,6 +495,19 @@ static void menu_handle_input() {
     g_selected++;
     return;
   }
+  if (key_pressed(SCAN_LEFT) || key_pressed(SCAN_RIGHT)) {
+    MenuNode *n = g_visible[g_selected];
+    if (n->type == NODE_VALUE && n->value_ptr) {
+      int dir = key_pressed(SCAN_RIGHT) ? 1 : -1;
+      int step = n->value_step ? n->value_step : 1;
+      int v = *n->value_ptr + dir * step;
+      if (v < n->value_min) v = n->value_min;
+      if (v > n->value_max) v = n->value_max;
+      *n->value_ptr = v;
+      config_save();
+    }
+    return;
+  }
   if (key_pressed(SCAN_ENTER)) {
     MenuNode *n = g_visible[g_selected];
     if (n->child_count > 0) {
@@ -599,6 +692,10 @@ void outer_draw() {
         snprintf(bn[fi], 32, "[%s]", scancode_name(*n->bind_ptr));
         draw_text(270, 10 + i * 20, bn[fi]);
       }
+    } else if (n->type == NODE_VALUE) {
+      char vb[32];
+      snprintf(vb, sizeof(vb), "%d deg/tick", *n->value_ptr);
+      draw_text(270, 10 + i * 20, vb);
     }
   }
   surf_pop_make_current(panel);