determine rpath for chapel-py shared lib when using --prefix

Signed-off-by: Ahmad Rezaii <ahmad.rezaii@hpe.com>

Author: arezaii

tools/chapel-py/setup.py

@@ -47,6 +47,19 @@
 
 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 build and final lib paths are going to
+#  differ figure out the install location now and write it to the rpath
+chpl_home_utils = os.path.join(
+    chpl_home, "util", "chplenv", "chpl_home_utils.py"
+)
+chpl_install_lib_path = (
+    subprocess.check_output(
+        ["python", chpl_home_utils, "--configured-install-lib-prefix"],
+    )
+    .decode(sys.stdout.encoding)
+    .strip()
+    .splitlines()
+)
 
 CXXFLAGS = []
 if have_llvm and have_llvm != "none":
@@ -68,6 +81,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(

util/chplenv/chpl_home_utils.py

@@ -54,6 +54,44 @@ def get_chpl_third_party():
                                                   os.path.sep,
                                                   os.path.sep))
 
+@memoize
+def get_chpl_configured_install_lib_prefix():
+    # gets the path to the lib directory for a prefix install, or None if not
+    # a prefix install
+    chpl_home = str(os.getenv("CHPL_HOME"))
+    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
+        return os.path.join(
+            chpl_prefix,
+            "lib",
+            "chapel",
+            chpl_version_string,
+            "compiler",
+        )
+
 @memoize
 def get_chpl_version_from_install():
     if get_prefix_install_prefix():
@@ -189,6 +227,8 @@ def _main():
     )
     parser.add_option('--using-module', action='store_const',
                       dest='func', const=using_chapel_module)
+    parser.add_option('--configured-install-lib-prefix', action='store_const',
+                      dest='func', const=get_chpl_configured_install_lib_prefix)
     (options, args) = parser.parse_args()
 
     if options.func: