Add --preferred-project arg for generator scripts

Author: GAMMACASE

README.md

@@ -40,6 +40,7 @@ This plugin comes with a set of particular python generator scripts located in t
    * ``-g`` ``--generate-classes``: A list of class/enum definitions to generate. Use "all" to generate all classes (Default).
    * ``-a`` ``--static-assert``: Generate static assertions of resulting class/enum definitions to ensure their validity.
    * ``-d`` ``--supply-hl2sdk``: Supplies hl2sdk class/enum definitions if applicable to the generated file.
+   * ``-p`` ``--preferred-project``: Prefer server or client project for generation. (Could only be ``server`` or ``client``). Default is ``server``.
 > [!NOTE]
 > Schema dumps with parent scope, might not generate correct code without supplying [hl2sdk](https://github.com/alliedmodders/hl2sdk) definitions!
     
@@ -51,6 +52,7 @@ This plugin comes with a set of particular python generator scripts located in t
    * ``-s`` ``--silent``: Disables stdout output.
    * ``-c`` ``--comments``: Generate help comments for resulting class/enum definitions.
    * ``-g`` ``--generate-classes``: A list of class/enum definitions to generate. Doesn't default to "all" and expects you to provide classes to generate.
+   * ``-p`` ``--preferred-project``: Prefer server or client project for generation. (Could only be ``server`` or ``client``). Default is ``server``.
 > [!NOTE]
 > This script is mostly an example of how you can automate dumping schema to macro headers for plugins that use [s2ze](https://github.com/Source2ZE/CS2Fixes) alike schema systems.
 

generator_scripts/generate_cpp.py

@@ -496,6 +496,7 @@ def main():
 	parser.add_argument('-g', '--generate-classes', help = 'A list of class/enum definitions to generate. Use "all" to generate all classes (Default).', nargs = '+', type = str, dest = 'generate_classes', default = 'all')
 	parser.add_argument('-a', '--static-assert', help = 'Generate static assertions of resulting class/enum definitions to ensure their validity.', action = 'store_true', dest = 'static_assert')
 	parser.add_argument('-d', '--supply-hl2sdk', help = 'Supplies hl2sdk class/enum definitions if applicable to the generated file.', action = 'store_true', dest = 'supply_hl2sdk')
+	parser.add_argument('-p', '--preferred-project', help = 'Prefer server or client project for generation.', type = str, dest = 'preferred_project', choices=['server', 'client'], default = 'server')
 
 	args = parser.parse_args()
 	flags = parse_args_as_flags(args)
@@ -629,6 +630,9 @@ def main():
 			print_stdout('Generating all class definitions...')
 
 			for defn in schema_file.defs:
+				if defn.project in ['server', 'client'] and defn.project != args.preferred_project:
+					continue
+
 				total_generated += 1
 				context.process_object(defn)
 		else:

generator_scripts/generate_cpp_defs.py

@@ -124,6 +124,7 @@ def main():
 	parser.add_argument('-s', '--silent', help = 'Disables stdout output.', action = 'store_true', dest = 'silent')
 	parser.add_argument('-c', '--comments', help = 'Generate help comments for resulting class/enum definitions.', action = 'store_true', dest = 'add_comments')
 	parser.add_argument('-g', '--generate-classes', help = 'A list of class/enum definitions to generate.', required = True, nargs = '+', type = str, dest = 'generate_classes', default = None)
+	parser.add_argument('-p', '--preferred-project', help = 'Prefer server or client project for generation.', type = str, dest = 'preferred_project', choices=['server', 'client'], default = 'server')
 
 	args = parser.parse_args()
 	flags = parse_args_as_flags(args)
@@ -144,14 +145,23 @@ def main():
 		print_stdout(f'Generating ({", ".join(args.generate_classes)}) class definitions...')
 
 		for class_name in args.generate_classes:
-			defn = schema_file.defs.get_def_at_name(class_name)
+			start_at = 0
 
-			if defn == None:
-				print_stdout(f'Class definition ({class_name}) not found in schema file, failed to generate!')
-				continue
+			while start_at < len(schema_file.defs):
+				defn = schema_file.defs.get_def_at_name(class_name, start_at)
 
-			total_generated += 1
-			context.process_object(defn)
+				if defn == None:
+					print_stdout(f'Class definition ({class_name}) not found in schema file, failed to generate! {start_at}')
+					break
+
+				start_at = schema_file.defs.get_def_idx(defn) + 1
+
+				if defn.project in ['server', 'client'] and defn.project != args.preferred_project:
+					continue
+
+				total_generated += 1
+				context.process_object(defn)
+				break
 
 	print_stdout(f'Successfully generated C++ file at {os.path.abspath(args.out_path)} (Total objects processed: {total_generated})')
 

generator_scripts/generator_scripts/obj_defs.py

@@ -669,8 +669,8 @@ def get_def_at_name_or_synth(self, name):
 		self.synth_defs.append(ObjectDefinition.generate_synthetic_class(name))
 		return self.synth_defs[-1]
 
-	def get_def_at_name(self, name):
-		for defn in self.defs:
+	def get_def_at_name(self, name, start_at = 0):
+		for defn in self.defs[start_at:]:
 			if defn.name == name:
 				return defn
 		return None