Added messages.get_message_size()

Source-Python-Dev-Team · Jan 6, 2018 · a2fbe74 · a2fbe74
1 parent 661d1fc
commit a2fbe74
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 3 deletions.
diff --git a/addons/source-python/packages/source-python/messages/__init__.py b/addons/source-python/packages/source-python/messages/__init__.py
@@ -39,6 +39,7 @@
 from _messages import FadeFlags
 from _messages import get_message_index
 from _messages import get_message_name
+from _messages import get_message_size
 
 
 # =============================================================================
@@ -54,6 +55,7 @@
            'FadeFlags',
            'get_message_index',
            'get_message_name',
+           'get_message_size',
            'HintText',
            'HudDestination',
            'HudMsg',

diff --git a/src/core/modules/messages/messages.cpp b/src/core/modules/messages/messages.cpp
@@ -141,3 +141,17 @@ object GetMessageName(int index)
 	return str(result);
 #endif
 }
+
+object GetMessageSize(int index)
+{
+#ifdef USE_PROTOBUF
+	BOOST_RAISE_EXCEPTION(PyExc_NotImplementedError, "")
+#else
+	char result[256];
+	int size;
+	if (!servergamedll->GetUserMessageInfo(index, result, 255, size))
+		return object();
+
+	return object(size);
+#endif
+}
diff --git a/src/core/modules/messages/messages.h b/src/core/modules/messages/messages.h
@@ -382,8 +382,9 @@ class CUserMessage
 //-----------------------------------------------------------------------------
 // Functions.
 //-----------------------------------------------------------------------------
-void CreateMessage( edict_t *pEdict, DIALOG_TYPE type, KeyValues *data );
-int GetMessageIndex(const char* name);
-object GetMessageName(int index);
+void	CreateMessage( edict_t *pEdict, DIALOG_TYPE type, KeyValues *data );
+int		GetMessageIndex(const char* name);
+object	GetMessageName(int index);
+object	GetMessageSize(int index);
 
 #endif // _MESSAGES_H
diff --git a/src/core/modules/messages/messages_wrap.cpp b/src/core/modules/messages/messages_wrap.cpp
@@ -205,6 +205,12 @@ void export_message_functions(scope _messages)
 		"Return the name of the user message. Return None if the user message wasn't found.",
 		args("index")
 	);
+
+	def("get_message_size",
+		&GetMessageSize,
+		"Return the size of the user message. Return None if the user message wasn't found.",
+		args("index")
+	);
 }