exposes the already existing -timeout argument of mirror program to the bazel rule declaration (#17)

Author: michaelschiff

mirror/defs.bzl

@@ -56,6 +56,7 @@ def _mirror_image_impl(ctx):
             "{src_image}": src_image,
             "{digest}": digest,
             "{dst_image}": dst_without_hash,
+            "{timeout}": ctx.attr.push_timeout,
         },
         is_executable = True,
     )
@@ -103,6 +104,10 @@ mirror_image_rule = rule(
         "dst": attr.string(
             doc = "The destination image location, should include the registry and repository. Either dst_prefix or dst_image must be specified.",
         ),
+        "push_timeout": attr.string(
+            doc = "The allowed wait time for image pushes",
+            default = "30s",
+        ),
         "mirror_tool": attr.label(
             default = Label("//cmd/mirror"),
             executable = True,

mirror/mirror_image.sh

@@ -12,4 +12,4 @@ function guess_runfiles() {
 
 RUNFILES="${PYTHON_RUNFILES:-$(guess_runfiles)}"
 
-{mirror_tool} -from {src_image} -digest {digest} -to {dst_image}
+{mirror_tool} -from {src_image} -digest {digest} -to {dst_image} -timeout {timeout}

mirror/tests/BUILD.bazel

@@ -44,6 +44,7 @@ mirror_image(
     src_image = "localhost:1338/image",
     tags = ["manual"],  # test should be tagged manual since it requires a running registry
     visibility = ["//visibility:public"],
+    push_timeout = "60s",
 )
 
 mirror_image(