Add demo of declare_directory

aaliddell · aaliddell · commit 7b1f2cd1e4dd · 2020-10-29T21:13:48.000Z
diff --git a/experimental/examples/wheel/BUILD b/experimental/examples/wheel/BUILD
@@ -14,11 +14,17 @@
 
 load("//experimental/python:wheel.bzl", "py_package", "py_wheel")
 load("//python:defs.bzl", "py_library", "py_test")
+load(":dir_generator.bzl", "dir_generator")
 
 package(default_visibility = ["//visibility:public"])
 
 licenses(["notice"])  # Apache 2.0
 
+
+dir_generator(
+    name = "generated"
+)
+
 py_library(
     name = "main",
     srcs = ["main.py"],
@@ -55,6 +61,7 @@ py_wheel(
     deps = [
         "//experimental/examples/wheel/lib:module_with_data",
         "//experimental/examples/wheel/lib:simple_module",
+        ":generated",
     ],
 )
 
diff --git a/experimental/examples/wheel/dir_generator.bzl b/experimental/examples/wheel/dir_generator.bzl
@@ -0,0 +1,16 @@
+def _dir_generator_impl(ctx):
+    out_dir = ctx.actions.declare_directory('dir_{}'.format(ctx.label.name))
+    out_file = ctx.actions.declare_file('file_{}.py'.format(ctx.label.name))
+    ctx.actions.run_shell(
+        outputs=[out_dir],
+        command='echo hi > {}/demo.py'.format(out_dir.path)
+    )
+    ctx.actions.run_shell(
+        outputs=[out_file],
+        command='echo hi > {}'.format(out_file.path)
+    )
+    return [DefaultInfo(files = depset([out_dir, out_file]))]
+
+dir_generator = rule(
+    implementation = _dir_generator_impl,
+)
