1 files changed, 22 insertions, 0 deletions
diff --git a/primedev/scripts/client/cursorposition.cpp b/primedev/scripts/client/cursorposition.cpp
new file mode 100644
index 00000000..c0e8623c
--- /dev/null
+++ b/primedev/scripts/client/cursorposition.cpp
@@ -0,0 +1,22 @@
+#include "squirrel/squirrel.h"
+#include "util/wininfo.h"
+
+ADD_SQFUNC("vector ornull", NSGetCursorPosition, "", "", ScriptContext::UI)
+{
+	RECT rcClient;
+	POINT p;
+	if (GetCursorPos(&p) && ScreenToClient(*g_gameHWND, &p) && GetClientRect(*g_gameHWND, &rcClient))
+	{
+		if (GetAncestor(GetForegroundWindow(), GA_ROOTOWNER) != *g_gameHWND)
+			return SQRESULT_NULL;
+
+		g_pSquirrel<context>->pushvector(
+			sqvm,
+			{p.x > 0 ? p.x > rcClient.right ? rcClient.right : (float)p.x : 0,
+			 p.y > 0 ? p.y > rcClient.bottom ? rcClient.bottom : (float)p.y : 0,
+			 0});
+		return SQRESULT_NOTNULL;
+	}
+	g_pSquirrel<context>->raiseerror(sqvm, "Failed retrieving cursor position of game window");
+	return SQRESULT_ERROR;
+}