Added support for unicode strings. (fixes #149)

Author: KN4CK3R

ReClass.NET/CodeGenerator/CSharpCodeGenerator.cs

@@ -37,7 +37,7 @@ public class CSharpCodeGenerator : ICodeGenerator
 			[typeof(Vector2Node)] = "Vector2",
 			[typeof(Vector3Node)] = "Vector3",
 			[typeof(Vector4Node)] = "Vector4"
-        };
+		};
 
 		public Language Language => Language.CSharp;
 
@@ -77,21 +77,42 @@ public string GenerateCode(IReadOnlyList<ClassNode> classes, IReadOnlyList<EnumD
 				.Where(c => c.Nodes.None(n => n is FunctionNode)) // Skip class which contains FunctionNodes because these are not data classes.
 				.Distinct();
 
+			var unicodeStringClassLengthsToGenerate = new HashSet<int>();
+
 			using (var en = classesToWrite.GetEnumerator())
 			{
 				if (en.MoveNext())
 				{
+					void FindUnicodeStringClasses(IEnumerable<BaseNode> nodes)
+					{
+						unicodeStringClassLengthsToGenerate.UnionWith(nodes.OfType<Utf16TextNode>().Select(n => n.Length));
+					}
+
+					FindUnicodeStringClasses(en.Current!.Nodes);
+
 					WriteClass(iw, en.Current, logger);
 
 					while (en.MoveNext())
 					{
 						iw.WriteLine();
 
+						FindUnicodeStringClasses(en.Current!.Nodes);
+
 						WriteClass(iw, en.Current, logger);
 					}
 				}
 			}
 
+			if (unicodeStringClassLengthsToGenerate.Any())
+			{
+				foreach (var length in unicodeStringClassLengthsToGenerate)
+				{
+					iw.WriteLine();
+
+					WriteUnicodeStringClass(iw, length);
+				}
+			}
+
 			return sw.ToString();
 		}
 
@@ -152,7 +173,7 @@ private static void WriteClass(IndentedTextWriter writer, ClassNode @class, ILog
 			Contract.Requires(@class != null);
 			Contract.Requires(logger != null);
 
-			writer.WriteLine("[StructLayout(LayoutKind.Explicit)]");
+			writer.WriteLine("[StructLayout(LayoutKind.Explicit, CharSet = CharSet.Ansi)]");
 			writer.Write("public struct ");
 			writer.Write(@class.Name);
 
@@ -199,10 +220,10 @@ private static void WriteClass(IndentedTextWriter writer, ClassNode @class, ILog
 		}
 
 		/// <summary>
-		/// Gets the type definition for the given node. If the node is not a simple node <c>null</c> is returned.
+		/// Gets the type definition for the given node. If the node is not expressible <c>null</c> as typename is returned.
 		/// </summary>
 		/// <param name="node">The target node.</param>
-		/// <returns>The type definition for the node or null if no simple type is available.</returns>
+		/// <returns>The type definition for the node or null as typename if the node is not expressible.</returns>
 		private static (string typeName, string attribute) GetTypeDefinition(BaseNode node)
 		{
 			Contract.Requires(node != null);
@@ -219,17 +240,36 @@ private static (string typeName, string attribute) GetTypeDefinition(BaseNode no
 				return (type, null);
 			}
 
-			if (node is EnumNode enumNode)
+			return node switch
 			{
-				return (enumNode.Enum.Name, null);
-			}
+				EnumNode enumNode => (enumNode.Enum.Name, null),
+				Utf8TextNode utf8TextNode => ("string", $"[MarshalAs(UnmanagedType.ByValTStr, SizeConst = {utf8TextNode.Length})]"),
+				Utf16TextNode utf16TextNode => (GetUnicodeStringClassName(utf16TextNode.Length), "[MarshalAs(UnmanagedType.Struct)]"),
+				_ => (null, null)
+			};
+		}
 
-			if (node is Utf8TextNode utf8TextNode)
-			{
-				return ("string", $"[MarshalAs(UnmanagedType.ByValTStr, SizeConst = {utf8TextNode.Length})]");
-			}
+		private static string GetUnicodeStringClassName(int length) => $"__UnicodeString{length}";
 
-			return (null, null);
+		/// <summary>
+		/// Writes a helper class for unicode strings with the specific length.
+		/// </summary>
+		/// <param name="writer">The writer to output to.</param>
+		/// <param name="length">The string length for this class.</param>
+		private static void WriteUnicodeStringClass(IndentedTextWriter writer, int length)
+		{
+			var className = GetUnicodeStringClassName(length);
+
+			writer.WriteLine("[StructLayout(LayoutKind.Sequential, CharSet=CharSet.Unicode)]");
+			writer.WriteLine($"public struct {className}");
+			writer.WriteLine("{");
+			writer.Indent++;
+			writer.WriteLine($"[MarshalAs(UnmanagedType.ByValTStr, SizeConst = {length})]");
+			writer.WriteLine("public string Value;");
+			writer.WriteLine();
+			writer.WriteLine($"public static implicit operator string({className} value) => value.Value;");
+			writer.Indent--;
+			writer.WriteLine("}");
 		}
 	}
 }