From 569d8503a6058521cc86dab726755980ac452bbb Mon Sep 17 00:00:00 2001
From: gregory langlais <admin@gregorylanglais.com>
Date: Thu, 18 May 2023 17:00:13 -0500
Subject: [PATCH] add argument to node dir blder

---
 node-module-directory-builder/BUILD               | 1 +
 node-module-directory-builder/directory_rules.bzl | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/node-module-directory-builder/BUILD b/node-module-directory-builder/BUILD
index d5f4d6e..94ff7d3 100644
--- a/node-module-directory-builder/BUILD
+++ b/node-module-directory-builder/BUILD
@@ -12,6 +12,7 @@ nodejs_binary(
 directory_build(
     name = "node_module_directory_build",
     node_binary = ":node_module_directory_builder_binary",
+    number_of_files = 5,
     deps = [
         "@directory_builder//:directory_builder"
     ],
diff --git a/node-module-directory-builder/directory_rules.bzl b/node-module-directory-builder/directory_rules.bzl
index 9ef1eec..dbe4e96 100644
--- a/node-module-directory-builder/directory_rules.bzl
+++ b/node-module-directory-builder/directory_rules.bzl
@@ -6,7 +6,7 @@ def _directory_build_impl(ctx):
         inputs = ctx.files.deps,
         outputs = [output_dir],
         executable = node_binary,
-        arguments = [output_dir.path],
+        arguments = [output_dir.path, "{}".format(ctx.attr.number_of_files)],
     )
 
     return [
@@ -17,6 +17,7 @@ directory_build = rule(
     implementation = _directory_build_impl,
     attrs = {
         "node_binary": attr.label(executable = True, cfg = "host"),
+        "number_of_files": attr.int(default = 3),
         "deps": attr.label_list(allow_files = True),
     },
 )
\ No newline at end of file