feat: include Go codegen

examples/proto/BUILD.bazel

@@ -1,8 +1,8 @@
 load("@aspect_rules_ts//ts:proto.bzl", "ts_proto_library")
-load("@rules_go//proto:def.bzl", "go_proto_library")
 load("@rules_proto//proto:defs.bzl", "proto_library")
 load("@rules_rust_prost//:defs.bzl", "rust_prost_library")
 load("@protobuf//bazel:py_proto_library.bzl", "py_proto_library")
+load("//tools:go_proto_library.bzl", "go_proto_library")
 
 package(default_visibility = ["//visibility:public"])
 

examples/tools/go_proto_library.bzl

@@ -0,0 +1,44 @@
+"go_proto_library wrapper macro to check in the resulting stub files for the editor"
+
+load("@aspect_bazel_lib//lib:write_source_files.bzl", "write_source_files")
+load("@aspect_bazel_lib//lib:output_files.bzl", "make_output_files")
+load("@bazel_skylib//lib:paths.bzl", "paths")
+load("@rules_go//proto:def.bzl", _go_proto_library = "go_proto_library")
+
+def go_proto_library(name, importpath, proto_srcs = [], **kwargs):
+    """Wrap go_proto_library with write_source_files.
+
+    This causes the resulting .pb.go files to be checked into the source tree.
+    Args:
+        name: name of the go_proto_library rule produced
+        importpath: passed to go_proto_library#importpath
+        proto_srcs: the srcs of the proto_library target passed to go_proto_library#proto
+            If unset, a glob() of all ".proto" files in the package is used.
+        **kwargs: remaining arguments to go_proto_library
+    """
+
+    # Based on our knowledge of the rule implementation,
+    # predict the output paths it writes.
+    proto_out_path = "{0}/{1}_/{2}/%s.pb.go".format(
+        native.package_name(),
+        name,
+        importpath,
+    )
+
+    if len(proto_srcs) < 1:
+        proto_srcs = native.glob(["*.proto"])
+
+    _go_proto_library(
+        name = name,
+        importpath = importpath,
+        **kwargs
+    )
+
+    write_source_files(
+        name = name + ".update_go_pb",
+        files = {
+            base + ".pb.go": make_output_files(base + "_pb_go", name, [proto_out_path % base], output_group = "go_generated_srcs")
+            for base in [paths.replace_extension(p, "") for p in proto_srcs]
+        },
+        visibility = ["//write_source_files:__pkg__"],
+    )