Make things more testable

CodegenBot/CodegenBotImports.cs

@@ -0,0 +1,31 @@
+using System.Text.Json;
+using System.Text.Json.Serialization.Metadata;
+using Extism;
+
+namespace CodegenBot;
+
+public interface ICodegenBotImports
+{
+    string GraphQL<T>(GraphQLRequest<T> request, JsonTypeInfo<GraphQLRequest<T>> jsonTypeInfo);
+    void Log(LogEvent logEvent);
+}
+
+public class CodegenBotImports : ICodegenBotImports
+{
+    public string GraphQL<T>(GraphQLRequest<T> request, JsonTypeInfo<GraphQLRequest<T>> jsonTypeInfo)
+    {
+        var json = request.ToJsonString(jsonTypeInfo);
+        using var block = Pdk.Allocate(json);
+        var ptr = Imports.ExternGraphQL(block.Offset);
+        var response = MemoryBlock.Find(ptr).ReadString();
+        response = JsonUtility.EnsureTypeDiscriminatorPropertiesComeFirst(response);
+        return response;
+    }
+
+    public void Log(LogEvent logEvent)
+    {
+        var json = JsonSerializer.Serialize(logEvent, LogEventJsonContext.Default.LogEvent);
+        using var block = Pdk.Allocate(json);
+        Imports.ExternLog(block.Offset);
+    }
+}

CodegenBot/Imports.cs

@@ -1,9 +1,6 @@
 using System;
 using System.Runtime.InteropServices;
-using System.Text.Json;
-using System.Text.Json.Serialization.Metadata;
 using CodegenBot;
-using Extism;
 
 namespace CodegenBot;
 
@@ -16,35 +13,6 @@ public class Imports
     [DllImport("extism", EntryPoint = "cgb_graphql")]
     public static extern ulong ExternGraphQL(ulong offset);
 
-    public static string GraphQL<T>(GraphQLRequest<T> request, JsonTypeInfo<GraphQLRequest<T>> jsonTypeInfo)
-    {
-        var json = request.ToJsonString(jsonTypeInfo);
-        using var block = Pdk.Allocate(json);
-        var ptr = ExternGraphQL(block.Offset);
-        var response = MemoryBlock.Find(ptr).ReadString();
-        response = JsonUtility.EnsureTypeDiscriminatorPropertiesComeFirst(response);
-        return response;
-    }
-
     [DllImport("extism", EntryPoint = "cgb_log")]
     public static extern void ExternLog(ulong offset);
-
-    public static void Log(LogEvent logEvent)
-    {
-        var json = JsonSerializer.Serialize(logEvent, LogEventJsonContext.Default.LogEvent);
-        using var block = Pdk.Allocate(json);
-        ExternLog(block.Offset);
-    }
-
-    [DllImport("extism", EntryPoint = "cgb_random")]
-    public static extern ulong ExternRandom(ulong offset);
-
-    public static byte[] GetRandomBytes(int numBytes)
-    {
-        var text = numBytes.ToString();
-        using var block = Pdk.Allocate(text);
-        var ptr = ExternRandom(block.Offset);
-        var response = MemoryBlock.Find(ptr).ReadBytes();
-        return response;
-    }
 }