aboutsummaryrefslogtreecommitdiff
path: root/tests/core/go_path/pkg/lib/generated_embeded.bzl
diff options
context:
space:
mode:
Diffstat (limited to 'tests/core/go_path/pkg/lib/generated_embeded.bzl')
-rw-r--r--tests/core/go_path/pkg/lib/generated_embeded.bzl71
1 files changed, 71 insertions, 0 deletions
diff --git a/tests/core/go_path/pkg/lib/generated_embeded.bzl b/tests/core/go_path/pkg/lib/generated_embeded.bzl
new file mode 100644
index 00000000..42bcf595
--- /dev/null
+++ b/tests/core/go_path/pkg/lib/generated_embeded.bzl
@@ -0,0 +1,71 @@
+# Copyright 2018 The Bazel Authors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+load(
+ "@io_bazel_rules_go//go:def.bzl",
+ "go_context",
+)
+
+def _gen_library_impl(ctx):
+ go = go_context(ctx)
+ libname = getattr(ctx.attr, "libname")
+ src = go.actions.declare_file(ctx.label.name + ".go")
+
+ embedsrcs = getattr(ctx.attr, "embedsrcs", [])
+
+ lines = [
+ "package " + libname,
+ "",
+ 'import _ "embed"',
+ "",
+ ]
+
+ i = 0
+ for e in embedsrcs:
+ for f in e.files.to_list():
+ lines.extend([
+ "//go:embed {}".format(f.basename),
+ "var embeddedSource{} string".format(i),
+ ])
+ i += 1
+
+ ctx.actions.write(src, "\n".join(lines))
+
+ library = go.new_library(go, srcs = [src])
+ source = go.library_to_source(go, ctx.attr, library, ctx.coverage_instrumented())
+ archive = go.archive(go, source)
+ return [
+ library,
+ source,
+ archive,
+ DefaultInfo(files = depset([archive.data.file])),
+ ]
+
+generated_embeded = rule(
+ _gen_library_impl,
+ attrs = {
+ "importpath": attr.string(mandatory = True),
+ "_go_context_data": attr.label(
+ default = "//:go_context_data",
+ ),
+ "srcs": attr.label_list(
+ allow_files = True,
+ ),
+ "embedsrcs": attr.label_list(
+ allow_files = True,
+ ),
+ "libname": attr.string(default = "lib"),
+ },
+ toolchains = ["@io_bazel_rules_go//go:toolchain"],
+)