1 files changed, 20 insertions, 0 deletions

diff --git a/src/esp.cpp b/src/esp.cpp
index 1631df1..d32a8bc 100644
--- a/src/esp.cpp
+++ b/src/esp.cpp
@@ -187,6 +187,26 @@ bool esp_is_viewmodel_index(int idx) {
   return model && (strstr(model, "/v_") || strstr(model, "\\v_"));
 }
 
+// 0 = hostile, 1 = neutral, 2 = friendly, -1 = not an NPC.
+static bool npc_is_friendly(int idx, const char *model, int local_idx,
+                            bool &neutral);
+int esp_npc_relation(int idx) {
+  if (!esp_is_npc_index(idx))
+    return -1;
+  int local = engine_get_local_player();
+  if (local <= 0)
+    return 0;
+  void *ent = entitylist_get(idx);
+  const char *model = ent ? entity_model_name(ent) : nullptr;
+  bool neutral = false;
+  bool friendly = npc_is_friendly(idx, model, local, neutral);
+  if (neutral)
+    return 1;
+  if (friendly)
+    return 2;
+  return 0;
+}
+
 static bool model_is_friendly(const char *m) {
   static const char *patterns[] = {
       "citizen", "humans/",  "male_",   "female_", "barney", "alyx",