cat on a spaceship

2025-11-01 22:38:46 -04:00
parent 747fa227a1
commit f6dccf46d7
89 changed files with 2663 additions and 581 deletions
@@ -0,0 +1,96 @@
+using System.Collections.Generic;
+
+namespace Svrnty.CQRS.Grpc.Generators.Helpers
+{
+    internal static class ProtoTypeMapper
+    {
+        private static readonly Dictionary<string, string> TypeMap = new Dictionary<string, string>
+        {
+            // Primitives
+            { "System.String", "string" },
+            { "System.Boolean", "bool" },
+            { "System.Int32", "int32" },
+            { "System.Int64", "int64" },
+            { "System.UInt32", "uint32" },
+            { "System.UInt64", "uint64" },
+            { "System.Single", "float" },
+            { "System.Double", "double" },
+            { "System.Byte", "uint32" },
+            { "System.SByte", "int32" },
+            { "System.Int16", "int32" },
+            { "System.UInt16", "uint32" },
+            { "System.Decimal", "string" }, // Decimal as string to preserve precision
+            { "System.DateTime", "int64" }, // Unix timestamp
+            { "System.DateTimeOffset", "int64" }, // Unix timestamp
+            { "System.Guid", "string" },
+            { "System.TimeSpan", "int64" }, // Ticks
+
+            // Nullable variants
+            { "System.Boolean?", "bool" },
+            { "System.Int32?", "int32" },
+            { "System.Int64?", "int64" },
+            { "System.UInt32?", "uint32" },
+            { "System.UInt64?", "uint64" },
+            { "System.Single?", "float" },
+            { "System.Double?", "double" },
+            { "System.Byte?", "uint32" },
+            { "System.SByte?", "int32" },
+            { "System.Int16?", "int32" },
+            { "System.UInt16?", "uint32" },
+            { "System.Decimal?", "string" },
+            { "System.DateTime?", "int64" },
+            { "System.DateTimeOffset?", "int64" },
+            { "System.Guid?", "string" },
+            { "System.TimeSpan?", "int64" },
+        };
+
+        public static string MapToProtoType(string csharpType, out bool isRepeated, out bool isOptional)
+        {
+            isRepeated = false;
+            isOptional = false;
+
+            // Handle arrays
+            if (csharpType.EndsWith("[]"))
+            {
+                isRepeated = true;
+                var elementType = csharpType.Substring(0, csharpType.Length - 2);
+                return MapToProtoType(elementType, out _, out _);
+            }
+
+            // Handle generic collections
+            if (csharpType.StartsWith("System.Collections.Generic.List<") ||
+                csharpType.StartsWith("System.Collections.Generic.IList<") ||
+                csharpType.StartsWith("System.Collections.Generic.IEnumerable<") ||
+                csharpType.StartsWith("System.Collections.Generic.ICollection<"))
+            {
+                isRepeated = true;
+                var startIndex = csharpType.IndexOf('<') + 1;
+                var endIndex = csharpType.LastIndexOf('>');
+                var elementType = csharpType.Substring(startIndex, endIndex - startIndex);
+                return MapToProtoType(elementType, out _, out _);
+            }
+
+            // Handle nullable value types
+            if (csharpType.EndsWith("?"))
+            {
+                isOptional = true;
+            }
+
+            // Check if it's a known primitive type
+            if (TypeMap.TryGetValue(csharpType, out var protoType))
+            {
+                return protoType;
+            }
+
+            // For unknown types, assume it's a custom message type
+            // Extract just the type name without namespace
+            var lastDot = csharpType.LastIndexOf('.');
+            if (lastDot >= 0)
+            {
+                return csharpType.Substring(lastDot + 1).Replace("?", "");
+            }
+
+            return csharpType.Replace("?", "");
+        }
+    }
+}