From f34e04ee48f8ec0b5e72ee530dbe0693f072ee0e Mon Sep 17 00:00:00 2001
From: Rubidium <rubidium@openttd.org>
Date: Sun, 19 Jan 2025 14:52:44 +0100
Subject: [PATCH] Codechange: allow mapping enums as parameter and return type
 from scripts

---
 src/script/squirrel_helper.hpp | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/script/squirrel_helper.hpp b/src/script/squirrel_helper.hpp
index 3adf637500..51ce3bf9cb 100644
--- a/src/script/squirrel_helper.hpp
+++ b/src/script/squirrel_helper.hpp
@@ -44,6 +44,14 @@ namespace SQConvert {
 	template <> struct Return<void *>       { static inline int Set(HSQUIRRELVM vm, void *res)       { sq_pushuserpointer(vm, res); return 1; } };
 	template <> struct Return<HSQOBJECT>    { static inline int Set(HSQUIRRELVM vm, HSQOBJECT res)   { sq_pushobject(vm, res); return 1; } };
 
+	template <typename T> requires std::is_enum_v<T> struct Return<T> {
+		static inline int Set(HSQUIRRELVM vm, T res)
+		{
+			sq_pushinteger(vm, to_underlying(res));
+			return 1;
+		}
+	};
+
 	template <> struct Return<std::optional<std::string>> {
 		static inline int Set(HSQUIRRELVM vm, std::optional<std::string> res)
 		{
@@ -75,6 +83,15 @@ namespace SQConvert {
 	template <> struct Param<const char *> { /* Do not use const char *, use std::string& instead. */ };
 	template <> struct Param<void *>       { static inline void       *Get(HSQUIRRELVM vm, int index) { SQUserPointer tmp; sq_getuserpointer(vm, index, &tmp); return tmp; } };
 
+	template <typename T> requires std::is_enum_v<T> struct Param<T> {
+		static inline T Get(HSQUIRRELVM vm, int index)
+		{
+			SQInteger tmp;
+			sq_getinteger(vm, index, &tmp);
+			return static_cast<T>(tmp);
+		}
+	};
+
 	template <> struct Param<const std::string &> {
 		static inline const std::string Get(HSQUIRRELVM vm, int index)
 		{