Radix/bazel/patchelf.bzl

def _patchelf_impl(ctx):
    output_name = ctx.file.src.basename
    if ctx.attr.soname:
        output_name = ctx.attr.soname
    output = ctx.actions.declare_file("{}/{}".format(ctx.attr.name, output_name))

    commands = [
        "set -e",
        'cp -f "$2" "$3"',
        'chmod +w "$3"',
    ]

    if ctx.attr.soname:
        commands.append(""" "$1" --set-soname '{}' "$3" """.format(ctx.attr.soname))
    if ctx.attr.remove_needed:
        for v in ctx.attr.remove_needed:
            commands.append(""" "$1" --remove-needed '{}' "$3" """.format(v))
    if ctx.attr.add_needed:
        for v in ctx.attr.add_needed:
            commands.append(""" "$1" --add-needed '{}' "$3" """.format(v))
    if ctx.attr.replace_needed:
        for k, v in ctx.attr.replace_needed.items():
            commands.append(""" "$1" --replace-needed '{}' '{}' "$3" """.format(k, v))

    if ctx.attr.set_rpath:
        commands.append(""" "$1" --set-rpath '{}' "$3" """.format(ctx.attr.set_rpath))
    if ctx.attr.add_rpath:
        for path in ctx.attr.add_rpath:
            commands.append(""" "$1" --add-rpath '{}' "$3" """.format(path))
    if ctx.attr.remove_rpath:
        for path in ctx.attr.remove_rpath:
            commands.append(""" "$1" --remove-rpath '{}' "$3" """.format(path))

    renamed_syms = ctx.actions.declare_file("{}.rename.txt".format(ctx.label.name))
    if ctx.attr.rename_dynamic_symbols:
        content = "\n".join([
            "{} {}".format(k, v)
            for k, v in ctx.attr.rename_dynamic_symbols.items()
        ])
        ctx.actions.write(renamed_syms, content)
        commands.append(""" "$1" --rename-dynamic-symbols '{}' "$3" """.format(renamed_syms.path))
    else:
        ctx.actions.write(renamed_syms, "")

    ctx.actions.run_shell(
        inputs = [ctx.file.src, renamed_syms],
        outputs = [output],
        arguments = [ctx.executable._patchelf.path, ctx.file.src.path, output.path],
        command = "\n".join(commands),
        tools = [ctx.executable._patchelf],
    )

    return [
        DefaultInfo(
            files = depset([output]),
        ),
    ]

patchelf = rule(
    implementation = _patchelf_impl,
    attrs = {
        "src": attr.label(allow_single_file = True, mandatory = True),
        "soname": attr.string(),
        "add_needed": attr.string_list(),
        "remove_needed": attr.string_list(),
        "replace_needed": attr.string_dict(),
        "rename_dynamic_symbols": attr.string_dict(),
        "set_rpath": attr.string(),
        "add_rpath": attr.string_list(),
        "remove_rpath": attr.string_list(),
        "_patchelf": attr.label(
            default = "@patchelf",
            allow_single_file = True,
            executable = True,
            cfg = "exec",
        ),
    },
)