try adding an sh_wrapper around heir-opt to populate env vars

Author: j2kun

MODULE.bazel

@@ -27,6 +27,7 @@ bazel_dep(name = "rules_license", version = "1.0.0")
 bazel_dep(name = "rules_foreign_cc", version = "0.14.0")
 bazel_dep(name = "rules_cc", version = "0.2.0")
 bazel_dep(name = "rules_go", version = "0.53.0")
+bazel_dep(name = "rules_shell", version = "0.6.0")
 bazel_dep(name = "rules_python", version = "1.2.0")
 bazel_dep(name = "googletest", version = "1.16.0")
 bazel_dep(name = "google_benchmark", version = "1.9.1")

MODULE.bazel.lock

@@ -1,6 +1,7 @@
 # HEIR tools
 load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
 load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
+load("@rules_shell//shell:sh_binary.bzl", "sh_binary")
 
 package(
     default_applicable_licenses = ["@heir//:license"],
@@ -12,7 +13,7 @@ WORKSPACE_PATH = ""
 
 # Custom `mlir-opt` replacement that links our dialect and passes
 cc_binary(
-    name = "heir-opt",
+    name = "heir-opt-bin",
     srcs = ["heir-opt.cpp"],
     data = select({
         "@heir//:config_enable_yosys": [
@@ -198,6 +199,12 @@ cc_binary(
     }),
 )
 
+sh_binary(
+    name = "heir-opt",
+    srcs = ["heir-opt-wrapper.sh"],
+    data = [":heir-opt-bin"],
+)
+
 # Custom `mlir-translate` replacement that adds our custom translations
 cc_binary(
     name = "heir-translate",

tools/BUILD

@@ -0,0 +1,23 @@
+#!/bin/bash
+# This script is a wrapper for heir-opt-bin that sets up the environment
+# for standalone execution by relying on the predictable structure of the
+# bazel runfiles directory.
+
+set -e
+
+# When run via `bazel run`, $0 is the path to this script. The runfiles
+# directory is created by bazel as a sibling directory with a .runfiles suffix.
+RUNFILES_DIR="${0}.runfiles"
+
+# The main workspace is symlinked as '_main' inside the runfiles dir.
+HEIR_WORKSPACE="_main"
+ABC_WORKSPACE="edu_berkeley_abc"
+
+HEIR_OPT_BIN="${RUNFILES_DIR}/${HEIR_WORKSPACE}/tools/heir-opt-bin"
+
+# These paths correspond to the `data` dependencies of the `heir-opt-bin` target.
+export HEIR_ABC_BINARY="${RUNFILES_DIR}/${ABC_WORKSPACE}/abc"
+export HEIR_YOSYS_SCRIPTS_DIR="${RUNFILES_DIR}/${HEIR_WORKSPACE}/lib/Transforms/YosysOptimizer/yosys"
+
+# Use exec to replace this script's process with the actual binary.
+exec "${HEIR_OPT_BIN}" "$@"