From e9eb9045b71f4702e2f6875ff1982529808826d3 Mon Sep 17 00:00:00 2001
From: Ahmad Rezaii <ahmad.rezaii@hpe.com>
Date: Tue, 10 Dec 2024 13:06:43 -0700
Subject: [PATCH] determine rpath for chapel-py shared lib when using --prefix

Signed-off-by: Ahmad Rezaii <ahmad.rezaii@hpe.com>
---
 tools/chapel-py/setup.py | 41 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/tools/chapel-py/setup.py b/tools/chapel-py/setup.py
index 30c270872467..6d8acdd92196 100644
--- a/tools/chapel-py/setup.py
+++ b/tools/chapel-py/setup.py
@@ -47,6 +47,41 @@
 
 host_bin_subdir = str(chpl_variables.get("CHPL_HOST_BIN_SUBDIR"))
 chpl_lib_path = os.path.join(chpl_home, "lib", "compiler", host_bin_subdir)
+# For installations using --prefix, the lib final lib path are going to differ
+# figure it out now and write it to the rpath
+chpl_install_lib_path = None
+if os.path.exists(os.path.join(chpl_home, "configured-prefix")):
+    with open(os.path.join(chpl_home, "CMakeLists.txt"), "r") as f:
+        # read CMakeLists.txt to get the CHPL_MAJOR_VERSION and
+        # CHPL_MINOR_VERSION and then construct the path from that
+        chpl_major_version = None
+        chpl_minor_version = None
+        for line in f:
+            if "set(CHPL_MAJOR_VERSION" in line:
+                chpl_major_version = line.split()[1].strip(")")
+            if "set(CHPL_MINOR_VERSION" in line:
+                chpl_minor_version = line.split()[1].strip(")")
+            if (
+                chpl_major_version is not None
+                and chpl_minor_version is not None
+            ):
+                break
+    assert chpl_major_version is not None and chpl_minor_version is not None
+    chpl_version_string = "{}.{}".format(
+        chpl_major_version,
+        chpl_minor_version,
+    )
+    chpl_prefix = None
+    with open(os.path.join(chpl_home, "configured-prefix"), "r") as f:
+        chpl_prefix = f.read().strip()
+    assert chpl_prefix is not None
+    chpl_install_lib_path = os.path.join(
+        chpl_prefix,
+        "lib",
+        "chapel",
+        chpl_version_string,
+        "compiler",
+    )
 
 CXXFLAGS = []
 if have_llvm and have_llvm != "none":
@@ -68,6 +103,12 @@
     "-lChplFrontendShared",
 ]
 
+if chpl_install_lib_path is not None:
+    LDFLAGS += [
+        "-L{}".format(chpl_install_lib_path),
+        "-Wl,-rpath,{}".format(chpl_install_lib_path),
+    ]
+
 if str(chpl_variables.get("CHPL_SANITIZE")) == "address":
     if str(chpl_variables.get("CHPL_HOST_PLATFORM")) == "darwin":
         sys.exit(