refactor mod to account for changes in the IC10Editor mod interface

csharp_mod/Extensions.cs

@@ -1,66 +1,103 @@
+namespace Slang;
+
 using System;
 using System.Text;
 using StationeersIC10Editor;
 
-namespace Slang
+public static unsafe class SlangExtensions
 {
-    public static unsafe class SlangExtensions
+    /**
+     * <summary>
+     * This is a helper method to convert a Rust struct for a string pointer
+     * into a C# style string.
+     * </summary>
+     */
+    public static string AsString(this Vec_uint8_t vec)
     {
-        /**
-         * <summary>
-         * This is a helper method to convert a Rust struct for a string pointer
-         * into a C# style string.
-         * </summary>
-         */
-        public static string AsString(this Vec_uint8_t vec)
+        if (vec.ptr == null || vec.len == UIntPtr.Zero)
         {
-            if (vec.ptr == null || vec.len == UIntPtr.Zero)
-            {
-                return string.Empty;
-            }
-
-            // Rust strings are UTF-8. Read bytes from raw pointer.
-            var toReturn = Encoding.UTF8.GetString(vec.ptr, (int)vec.len);
-
-            return toReturn;
+            return string.Empty;
         }
 
-        /**
-         * <summary>This will free a Rust string struct. Because this is a pointer to a struct, this memory
-         * is managed by Rust, therefor it must be freed by Rust
-         * </summary>
-         */
-        public static void Drop(this Vec_uint8_t vec)
+        // Rust strings are UTF-8. Read bytes from raw pointer.
+        var toReturn = Encoding.UTF8.GetString(vec.ptr, (int)vec.len);
+
+        return toReturn;
+    }
+
+    /**
+     * <summary>This will free a Rust string struct. Because this is a pointer to a struct, this memory
+     * is managed by Rust, therefor it must be freed by Rust
+     * </summary>
+     */
+    public static void Drop(this Vec_uint8_t vec)
+    {
+        Ffi.free_string(vec);
+    }
+
+    /**
+     * <summary>This helper converts a Rust vec to a C# List. This handles freeing the
+     * Rust allocation after the List is created, there is no need to Drop this memory.
+     * </summary>
+     */
+    public static Line ToLine(this Vec_FfiToken_t vec, string sourceText)
+    {
+        var list = new Line(sourceText);
+
+        var currentPtr = vec.ptr;
+
+        // Iterate through the raw memory array
+        for (int i = 0; i < (int)vec.len; i++)
         {
-            Ffi.free_string(vec);
+            var token = currentPtr[i];
+
+            var color = GetColorForKind(token.token_kind);
+
+            int colIndex = token.column;
+            if (colIndex < 0)
+                colIndex = 0;
+
+            var semanticToken = new SemanticToken(
+                0,
+                colIndex,
+                token.length,
+                color,
+                token.token_kind
+            );
+
+            string errMsg = token.error.AsString();
+            if (!string.IsNullOrEmpty(errMsg))
+            {
+                semanticToken.IsError = true;
+                semanticToken.Data = errMsg;
+                semanticToken.Color = ICodeFormatter.ColorError;
+            }
+            list.AddToken(semanticToken);
         }
 
-        /**
-         * <summary>This helper converts a Rust vec to a C# List. This handles freeing the
-         * Rust allocation after the List is created, there is no need to Drop this memory.
-         * </summary>
-         */
-        public static Line AsList(this Vec_FfiToken_t vec)
+        Ffi.free_ffi_token_vec(vec);
+
+        return list;
+    }
+
+    private static uint GetColorForKind(uint kind)
+    {
+        switch (kind)
         {
-            var list = new Line();
-            list.Capacity = (int)vec.len;
-
-            var currentPtr = vec.ptr;
-
-            // Iterate through the raw memory array
-            for (int i = 0; i < (int)vec.len; i++)
-            {
-                // Dereference pointer to get the struct at index i
-                FfiToken_t token = currentPtr[i];
-
-                var newToken = new Token(token.text.AsString(), token.column);
-
-                list.Add(newToken);
-            }
-
-            Ffi.free_ffi_token_vec(vec);
-
-            return list;
+            case 1:
+                return SlangFormatter.ColorInstruction; // Keyword
+            case 2:
+                return SlangFormatter.ColorDefault; // Identifier
+            case 3:
+                return SlangFormatter.ColorNumber; // Number
+            case 4:
+                return SlangFormatter.ColorString; // String
+            case 5:
+                return SlangFormatter.ColorInstruction; // Boolean
+            case 6:
+                return SlangFormatter.ColorDefault; // Symbol
+            default:
+                return SlangFormatter.ColorDefault;
         }
     }
 }