From b18567003c8e7f4c22bdec5a8d279f6aa8c845d2 Mon Sep 17 00:00:00 2001
From: peter-r-g <petergorman11@gmail.com>
Date: Mon, 15 May 2023 12:15:49 -0400
Subject: [PATCH] Add IsServerCommandSupported extension method

---
 .../Extensions/ITypeSymbolExtensions.cs       | 35 +++++++++++++++++++
 .../Extensions/TypeSyntaxExtensions.cs        | 17 ++++++++-
 2 files changed, 51 insertions(+), 1 deletion(-)

diff --git a/SboxAnalyzers/Extensions/ITypeSymbolExtensions.cs b/SboxAnalyzers/Extensions/ITypeSymbolExtensions.cs
index 178b109..2a32cfa 100644
--- a/SboxAnalyzers/Extensions/ITypeSymbolExtensions.cs
+++ b/SboxAnalyzers/Extensions/ITypeSymbolExtensions.cs
@@ -40,6 +40,27 @@ internal static class ITypeSymbolExtensions
 		"System.Collections.Generic.IDictionary<TKey, TValue>"
 		);
 
+	/// <summary>
+	/// Contains a set of all types that are supported for server commands in S&box.
+	/// </summary>
+	internal static ImmutableHashSet<string> ServerCommandSupportedTypes = ImmutableHashSet.Create(
+		typeof( int ).FullName,
+		typeof( uint ).FullName,
+		typeof( long ).FullName,
+		typeof( ulong ).FullName,
+		typeof( float ).FullName,
+		typeof( double ).FullName,
+		typeof( bool ).FullName,
+		typeof( string ).FullName,
+		"Sandbox.Vector2",
+		"Sandbox.Vector3",
+		"Sandbox.Vector4",
+		"Sandbox.Rotation",
+		"Sandbox.Angles",
+		"Sandbox.Color",
+		"Sandbox.RangedFloat"
+		);
+
 	/// <summary>
 	/// Returns whether or not the <see cref="ITypeSymbol"/> can be networked.
 	/// </summary>
@@ -91,6 +112,20 @@ internal static bool IsNetworkable( this ITypeSymbol symbol)
 		return false;
 	}
 
+	/// <summary>
+	/// Returns whether or not the <see cref="ITypeSymbol"/> is supported in server commands.
+	/// </summary>
+	/// <param name="symbol">The symbol to check if it is networkable in server commands.</param>
+	/// <returns>Whether or not the <see cref="ITypeSymbol"/> is supported in server commands.</returns>
+	internal static bool IsServerCommandSupported( this ITypeSymbol symbol )
+	{
+		// Fast path, enums are supported.
+		if ( symbol.TypeKind == TypeKind.Enum )
+			return true;
+
+		return ServerCommandSupportedTypes.Contains( symbol.ToNameString( false ) );
+	}
+
 	/// <summary>
 	/// Returns a type string of a <see cref="ITypeSymbol"/>.
 	/// </summary>
diff --git a/SboxAnalyzers/Extensions/TypeSyntaxExtensions.cs b/SboxAnalyzers/Extensions/TypeSyntaxExtensions.cs
index 25cc655..060c1db 100644
--- a/SboxAnalyzers/Extensions/TypeSyntaxExtensions.cs
+++ b/SboxAnalyzers/Extensions/TypeSyntaxExtensions.cs
@@ -22,7 +22,22 @@ internal static bool IsNetworkable( this TypeSyntax syntax, SemanticModel semant
 
 		return namedTypeSymbol.IsNetworkable();
 	}
-	
+
+	/// <summary>
+	/// Returns whether or not a <see cref="TypeSyntax"/> is supported in server commands.
+	/// </summary>
+	/// <param name="syntax">The <see cref="TypeSyntax"/> to check.</param>
+	/// <param name="semanticModel">The semantic model context around the compilation.</param>
+	/// <returns>Whether or not a <see cref="TypeSyntax"/> is supported in server commands.</returns>
+	internal static bool IsServerCommandSupported( this TypeSyntax syntax, SemanticModel semanticModel )
+	{
+		var symbol = semanticModel.GetSymbolInfo( syntax );
+		if ( symbol.Symbol is not INamedTypeSymbol namedTypeSymbol )
+			return false;
+
+		return namedTypeSymbol.IsServerCommandSupported();
+	}
+
 	/// <summary>
 	/// Returns whether or not a <see cref="TypeSyntax"/> is equal to another <see cref="TypeSyntax"/>.
 	/// </summary>