Merge pull request #51 from rabbitmq/source-archive-rule

Source archive rule

Author: HoloRin

MODULE.bazel

@@ -4,6 +4,11 @@ module(
     compatibility_level = 3,
 )
 
+bazel_dep(
+    name = "rules_pkg",
+    version = "0.5.1",
+)
+
 bazel_dep(
     name = "bazel_skylib",
     version = "1.2.0",

WORKSPACE.bazel

@@ -1,5 +1,18 @@
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
+http_archive(
+    name = "rules_pkg",
+    sha256 = "038f1caa773a7e35b3663865ffb003169c6a71dc995e39bf4815792f385d837d",
+    urls = [
+        "https://mirror.bazel.build/github.com/bazelbuild/rules_pkg/releases/download/0.4.0/rules_pkg-0.4.0.tar.gz",
+        "https://github.com/bazelbuild/rules_pkg/releases/download/0.4.0/rules_pkg-0.4.0.tar.gz",
+    ],
+)
+
+load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
+
+rules_pkg_dependencies()
+
 http_archive(
     name = "bazel_skylib",
     sha256 = "af87959afe497dc8dfd4c6cb66e1279cb98ccc84284619ebfec27d9c09a903de",

erlang_app.bzl

@@ -44,10 +44,12 @@ def erlang_app(
         deps = [],
         runtime_deps = [],
         stamp = -1):
+    srcs = native.glob(["src/**/*.erl"]) + extra_srcs
+
     erlang_bytecode(
         name = "beam_files",
         hdrs = native.glob(["include/**/*.hrl", "src/**/*.hrl"]) + extra_hdrs,
-        srcs = native.glob(["src/**/*.erl"]) + extra_srcs,
+        srcs = srcs,
         erlc_opts = erlc_opts,
         dest = "ebin",
         deps = build_deps + deps,
@@ -82,6 +84,7 @@ def erlang_app(
         beam = [":beam_files"],
         priv = native.glob(["priv/**/*"]) + extra_priv,
         license_files = native.glob(["LICENSE*"]) + extra_license_files,
+        srcs = srcs,
         deps = deps + runtime_deps,
         visibility = ["//visibility:public"],
     )
@@ -103,10 +106,12 @@ def test_erlang_app(
         build_deps = [],
         deps = [],
         runtime_deps = []):
+    srcs = native.glob(["src/**/*.erl"]) + extra_srcs
+
     erlang_bytecode(
         name = "test_beam_files",
         hdrs = native.glob(["include/**/*.hrl", "src/**/*.hrl"]) + extra_hdrs,
-        srcs = native.glob(["src/**/*.erl"]) + extra_srcs,
+        srcs = srcs,
         erlc_opts = erlc_opts,
         dest = "test",
         deps = build_deps + deps,
@@ -126,6 +131,7 @@ def test_erlang_app(
         beam = [":test_beam_files"],
         priv = native.glob(["priv/**/*"]) + extra_priv,
         license_files = native.glob(["LICENSE*"]) + extra_license_files,
+        srcs = srcs,
         deps = deps + runtime_deps,
         visibility = ["//visibility:public"],
         testonly = True,

erlang_app_info.bzl

@@ -6,6 +6,7 @@ ErlangAppInfo = provider(
         "beam": "Compiled bytecode (.beam) files, or a single ebin directory",
         "priv": "Additional files",
         "license_files": "License files",
+        "srcs": "Source files",
         "deps": "Runtime dependencies of the compiled sources",
     },
 )
@@ -43,6 +44,7 @@ def _impl(ctx):
             beam = compiled_files,
             priv = ctx.files.priv,
             license_files = ctx.files.license_files,
+            srcs = ctx.files.srcs,
             deps = deps,
         ),
         DefaultInfo(
@@ -60,6 +62,7 @@ erlang_app_info = rule(
         "beam": attr.label_list(allow_files = [".beam", ".appup"]),
         "priv": attr.label_list(allow_files = True),
         "license_files": attr.label_list(allow_files = True),
+        "srcs": attr.label_list(allow_files = True),
         "deps": attr.label_list(providers = [ErlangAppInfo]),
     },
 )

private/source_tree.bzl

@@ -0,0 +1,40 @@
+load(
+    "//:erlang_app_info.bzl",
+    "ErlangAppInfo",
+    "flat_deps",
+)
+load(
+    "//:util.bzl",
+    "path_join",
+)
+load(
+    ":util.bzl",
+    "additional_file_dest_relative_path",
+)
+
+def _impl(ctx):
+    deps = flat_deps(ctx.attr.deps)
+
+    files = []
+    for dep in deps:
+        lib_info = dep[ErlangAppInfo]
+        dep_path = path_join(ctx.label.name, lib_info.app_name)
+        for src in lib_info.include + lib_info.srcs + lib_info.priv + lib_info.license_files:
+            rp = additional_file_dest_relative_path(dep.label, src)
+            dest = ctx.actions.declare_file(path_join(dep_path, rp))
+            ctx.actions.symlink(output = dest, target_file = src)
+            files.append(dest)
+
+    return [DefaultInfo(
+        files = depset(files),
+    )]
+
+source_tree = rule(
+    implementation = _impl,
+    attrs = {
+        "deps": attr.label_list(
+            providers = [ErlangAppInfo],
+            mandatory = True,
+        ),
+    },
+)

source_archive.bzl

@@ -0,0 +1,19 @@
+load("@rules_pkg//:pkg.bzl", "pkg_tar")
+load(
+    "//private:source_tree.bzl",
+    "source_tree",
+)
+
+def source_archive():
+    source_tree(
+        name = "source_tree",
+        deps = [":erlang_app"],
+    )
+
+    pkg_tar(
+        name = "source_archive",
+        srcs = [":source_tree"],
+        package_dir = "deps",
+        strip_prefix = "source_tree",
+        visibility = ["//visibility:public"],
+    )

source_tree.bzl

@@ -0,0 +1,7 @@
+load(
+    "//private:source_tree.bzl",
+    _source_tree = "source_tree",
+)
+
+def source_tree(**kwargs):
+    _source_tree(**kwargs)

test/BUILD.bazel

@@ -12,8 +12,7 @@ platform(
     constraint_values = [
         erlang_external_constraint,
     ],
-    parents = ["@rbe//config:platform"],
-    # parents = ["@local_config_platform//:host"],
+    parents = ["@local_config_platform//:host"],
 )
 
 erlang_25_constraint = erlang_toolchain_from_github_release(

test/WORKSPACE.bazel

@@ -1,5 +1,18 @@
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
+http_archive(
+    name = "rules_pkg",
+    sha256 = "038f1caa773a7e35b3663865ffb003169c6a71dc995e39bf4815792f385d837d",
+    urls = [
+        "https://mirror.bazel.build/github.com/bazelbuild/rules_pkg/releases/download/0.4.0/rules_pkg-0.4.0.tar.gz",
+        "https://github.com/bazelbuild/rules_pkg/releases/download/0.4.0/rules_pkg-0.4.0.tar.gz",
+    ],
+)
+
+load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
+
+rules_pkg_dependencies()
+
 http_archive(
     name = "bazel_skylib",
     sha256 = "af87959afe497dc8dfd4c6cb66e1279cb98ccc84284619ebfec27d9c09a903de",

test/shard_suite/BUILD.bazel

@@ -1,19 +1,38 @@
-load("@rules_erlang//:app_file.bzl", "app_file")
-load("@rules_erlang//:erlang_bytecode.bzl", "erlang_bytecode")
-load("@rules_erlang//:erlang_app_info.bzl", "erlang_app_info")
+load(
+    "@rules_erlang//:app_file.bzl",
+    "app_file",
+)
+load(
+    "@rules_erlang//:erlang_bytecode.bzl",
+    "erlang_bytecode",
+)
+load(
+    "@rules_erlang//:erlang_app_info.bzl",
+    "erlang_app_info",
+)
 load(
     "@rules_erlang//:erlang_app.bzl",
     "DEFAULT_ERLC_OPTS",
     "DEFAULT_TEST_ERLC_OPTS",
 )
-load("@rules_erlang//:xref2.bzl", "xref")
+load(
+    "@rules_erlang//:xref2.bzl",
+    "xref",
+)
 load(
     "@rules_erlang//:dialyze.bzl",
     "DEFAULT_PLT_APPS",
     "dialyze",
     "plt",
 )
-load("@rules_erlang//:ct.bzl", "ct_suite")
+load(
+    "@rules_erlang//:ct.bzl",
+    "ct_suite",
+)
+load(
+    "@rules_erlang//:source_archive.bzl",
+    "source_archive",
+)
 
 APP_NAME = "shard_suite"
 
@@ -44,6 +63,7 @@ app_file(
 
 erlang_app_info(
     name = "erlang_app",
+    srcs = ["@rules_erlang//tools/shard_suite:src/shard_suite.erl"],
     app = ":app_file",
     app_name = APP_NAME,
     beam = [":beam_files"],
@@ -52,6 +72,7 @@ erlang_app_info(
 erlang_app_info(
     name = "test_erlang_app",
     testonly = True,
+    srcs = ["@rules_erlang//tools/shard_suite:src/shard_suite.erl"],
     app = ":app_file",
     app_name = APP_NAME,
     beam = [":test_beam_files"],
@@ -77,3 +98,5 @@ ct_suite(
         "test/example_suite.erl",
     ],
 )
+
+source_archive()