1 files changed, 77 insertions, 0 deletions
diff --git a/tools/go_stateify/defs.bzl b/tools/go_stateify/defs.bzl
new file mode 100644
index 000000000..87fdc0d28
--- /dev/null
+++ b/tools/go_stateify/defs.bzl
@@ -0,0 +1,77 @@
+"""Stateify is a tool for generating state wrappers for Go types.
+
+The go_stateify rule is used to generate a file that will appear in a Go
+target; the output file should appear explicitly in a srcs list. For example:
+
+go_stateify(
+    name = "foo_state",
+    srcs = ["foo.go"],
+    out = "foo_state.go",
+    package = "foo",
+)
+
+go_library(
+    name = "foo",
+    srcs = [
+        "foo.go",
+        "foo_state.go",
+    ],
+    deps = [
+        "//pkg/state",
+    ],
+)
+"""
+
+def _go_stateify_impl(ctx):
+  """Implementation for the stateify tool."""
+  output = ctx.outputs.out
+
+  # Run the stateify command.
+  args = ["-output=%s" % output.path]
+  args += ["-pkg=%s" % ctx.attr.package]
+  if ctx.attr._statepkg:
+    args += ["-statepkg=%s" % ctx.attr._statepkg]
+  if ctx.attr.imports:
+    args += ["-imports=%s" % ",".join(ctx.attr.imports)]
+  args += ["--"]
+  for src in ctx.attr.srcs:
+    args += [f.path for f in src.files]
+  ctx.actions.run(
+      inputs = ctx.files.srcs,
+      outputs = [output],
+      mnemonic = "GoStateify",
+      progress_message = "Generating state library %s" % ctx.label,
+      arguments = args,
+      executable = ctx.executable._tool,
+  )
+
+"""
+Generates save and restore logic from a set of Go files.
+
+
+Args:
+  name: the name of the rule.
+  srcs: the input source files. These files should include all structs in the package that need to be saved.
+  imports: an optional list of extra non-aliased, Go-style absolute import paths.
+  out: the name of the generated file output. This must not conflict with any other files and must be added to the srcs of the relevant go_library.
+  package: the package name for the input sources.
+"""
+
+go_stateify = rule(
+    attrs = {
+        "srcs": attr.label_list(
+            mandatory = True,
+            allow_files = True,
+        ),
+        "imports": attr.string_list(mandatory = False),
+        "package": attr.string(mandatory = True),
+        "out": attr.output(mandatory = True),
+        "_tool": attr.label(
+            executable = True,
+            cfg = "host",
+            default = Label("//tools/go_stateify:stateify"),
+        ),
+        "_statepkg": attr.string(default = "gvisor.googlesource.com/gvisor/pkg/state"),
+    },
+    implementation = _go_stateify_impl,
+)