Add support for extvars in jsonnet_library

examples/BUILD

@@ -120,23 +120,23 @@ jsonnet_to_json_test(
 jsonnet_library(
     name = "code_library_lib",
     srcs = ["code_library.libsonnet"],
-    deps = [":workflow"]
+    deps = [":workflow"],
 )
 
 jsonnet_to_json_test(
     name = "extvar_code_library_test",
     size = "small",
     src = "extvar_code_library.jsonnet",
-    ext_code_libraries = { ":code_library_lib": "codefile" },
+    ext_code_libraries = {":code_library_lib": "codefile"},
     golden = "extvar_files_library_golden.json",
 )
 
 jsonnet_to_json_test(
     name = "tla_code_library_test",
     size = "small",
     src = "tla_code_library.jsonnet",
-    tla_code_libraries = { ":code_library_lib": "tla_code" },
     golden = "tla_code_library_golden.json",
+    tla_code_libraries = {":code_library_lib": "tla_code"},
 )
 
 jsonnet_to_json_test(
@@ -336,3 +336,46 @@ jsonnet_to_json(
         "multiple_outs_nested_asymmetric/file.json",
     ],
 )
+
+jsonnet_library(
+    name = "extvar-lib",
+    srcs = [
+        "extvar.libsonnet",
+    ],
+    ext_code_envs = ["MYJSONNET"],
+    ext_code_libraries = {
+        ":code_library_lib": "ext_lib",
+    },
+    ext_str_envs = ["MYTEST"],
+    ext_str_files = {
+        ":test_str_files": "str_files",
+    },
+    ext_strs = {
+        "non_stamp": "non_stamp",
+        "mydefine": "$(mydefine)",
+        "k8s": "{STABLE_K8S_CLUSTER}",
+    },
+)
+
+jsonnet_to_json_test(
+    name = "extvar-lib-test",
+    src = "extvar-library.jsonnet",
+    golden = "extvar-library.json",
+    deps = [
+        ":extvar-lib",
+    ],
+)
+
+jsonnet_to_json(
+    name = "extvar-lib-json",
+    src = "extvar-library.jsonnet",
+    outs = ["ext-lib.json"],
+    ext_code_envs = ["MYJSONNET"],
+    ext_str_envs = ["MYTEST"],
+    ext_strs = {
+        "non_stamp": "non_stamp",
+    },
+    deps = [
+        ":extvar-lib",
+    ],
+)

examples/extvar-library.json

@@ -0,0 +1,13 @@
+{
+   "code_env": {
+      "code": "some code"
+   },
+   "library": {
+      "workflow": { }
+   },
+   "k8s": "{STABLE_K8S_CLUSTER}",
+   "mydefine": "",
+   "non_stamp": "non_stamp",
+   "str_env": "test",
+   "str_files": "this is great\n"
+}

examples/extvar-library.jsonnet

@@ -0,0 +1 @@
+import 'extvar.libsonnet'

examples/extvar.libsonnet

@@ -0,0 +1,9 @@
+{
+  non_stamp: std.extVar('non_stamp'),
+  mydefine: std.extVar('mydefine'),
+  k8s: std.extVar('k8s'),
+  str_files: std.extVar('str_files'),
+  code_env: std.extVar('MYJSONNET'),
+  str_env: std.extVar('MYTEST'),
+  library: std.extVar('ext_lib'),
+}