Add imports path fixer

When bumping source proto files, a single build will change the necessary imports or fail as needed

Author: kzu

src/GrokClient/GrokClient.csproj

@@ -14,8 +14,9 @@
     <Protobuf Include="*.proto" GrpcServices="Client" />
   </ItemGroup>
 
-  <Target Name="FixProto" BeforeTargets="Build">
+  <Target Name="FixProto" BeforeTargets="Protobuf_BeforeCompile">
     <Exec Command="dotnet run --file ..\protofix.cs" />
+
   </Target>
 
 </Project>

src/GrokClient/chat.proto

@@ -3,10 +3,10 @@ syntax = "proto3";
 package xai_api;
 
 import "google/protobuf/timestamp.proto";
-import "xai/api/v1/deferred.proto";
-import "xai/api/v1/image.proto";
-import "xai/api/v1/sample.proto";
-import "xai/api/v1/usage.proto";
+import "deferred.proto";
+import "image.proto";
+import "sample.proto";
+import "usage.proto";
 
 // An API that exposes our language models via a Chat interface.
 service Chat {

src/GrokClient/embed.proto

@@ -2,8 +2,8 @@ syntax = "proto3";
 
 package xai_api;
 
-import "xai/api/v1/image.proto";
-import "xai/api/v1/usage.proto";
+import "image.proto";
+import "usage.proto";
 
 // An API service for interaction with available embedding models.
 service Embedder {

src/GrokClient/sample.proto

@@ -3,7 +3,7 @@ syntax = "proto3";
 package xai_api;
 
 import "google/protobuf/timestamp.proto";
-import "xai/api/v1/usage.proto";
+import "usage.proto";
 
 // An API service for sampling the responses of available language models.
 service Sample {

src/protofix.cs

@@ -0,0 +1,67 @@
+#:package Spectre.Console@0.*
+#:package ConsoleAppFramework@5.*
+#:property Nullable=enable
+#:property ImportDirectoryBuildProps=false
+#:property ImportDirectoryBuildTargets=false
+
+using System.IO;
+using System.Linq;
+using System.Text.RegularExpressions;
+using ConsoleAppFramework;
+using Spectre.Console;
+
+ConsoleApp.Run(args, FixProto);
+
+/// <summary>Check and fix imports in .proto files.</summary>
+/// <para name="dir">Optional directory, defaults to current directory.</para>
+static int FixProto(bool dryRun, [Argument] string? dir = default)
+{
+    dir ??= Directory.GetCurrentDirectory();
+    var regex = ImportExpr();
+    var result = 0;
+    foreach (var file in Directory.EnumerateFiles(dir, "*.proto", SearchOption.AllDirectories))
+    {
+        var lines = File.ReadAllLines(file).ToList();
+        var changed = false;
+        for (var i = 0; i < lines.Count; i++)
+        {
+            var line = lines[i];
+            if (regex.Match(line) is { Success: true } match)
+            {
+                var path = match.Groups[1].Value;
+                var baseDir = Path.GetDirectoryName(file)!;
+                if (File.Exists(Path.Combine(baseDir, path)))
+                {
+                    AnsiConsole.MarkupLine($":check_mark_button: {Path.GetRelativePath(dir, file)} [lime]{path}[/]");
+                    continue;
+                }
+
+                if (File.Exists(Path.Combine(baseDir, Path.GetFileName(path))))
+                {
+                    AnsiConsole.MarkupLine($":pencil:{(dryRun ? ":ghost:" : "")} {Path.GetRelativePath(dir, file)}: {path} > {Path.GetFileName(path)}");
+                    if (!dryRun)
+                    {
+                        lines[i] = line.Replace(path, Path.GetFileName(path));
+                        changed = true;
+                    }
+                    continue;
+                }
+
+                result = 1;
+                AnsiConsole.MarkupLine($":cross_mark: {Path.GetRelativePath(dir, file)}: import not found [yellow]{path}[/]");
+            }
+        }
+        if (changed && !dryRun)
+        {
+            File.WriteAllLines(file, lines);
+        }
+    }
+
+    return result;
+}
+
+partial class Program
+{
+    [GeneratedRegex(@"^import\s+""([^""]+\.proto)"";$", RegexOptions.Multiline)]
+    private static partial Regex ImportExpr();
+}