#!/bin/bash

set -e

PACKAGE_NAME=@PACKAGE_NAME@
INSTALL_PATH=@INSTALL_PATH@
LIBRARY_DIR=@LIBRARY_DIR@
PROCESSOR=@PROCESSOR@
HCC_VERSION_MAJOR=@HCC_VERSION_MAJOR@
HCC_VERSION_MINOR=@HCC_VERSION_MINOR@

ROCM_PATH="/opt/rocm"

# NOTE: if you modify this table, also update the same table in prerm
SOFTLINKS=(
# <SOURCE DIRECTORY> <DESTINATION DIRECTORY> <FILE>
  "bin" "bin" "clamp-config"
  "bin" "bin" "extractkernel"
  "bin" "bin" "hcc"
  "bin" "bin" "hcc-config"
  "$LIBRARY_DIR" "lib" "libhc_am.so"
  "$LIBRARY_DIR" "lib" "libhc_am.so.${HCC_VERSION_MAJOR}"
  "$LIBRARY_DIR" "lib" "libhc_am.so.${HCC_VERSION_MAJOR}.${HCC_VERSION_MINOR}"
  "$LIBRARY_DIR" "lib" "libmcwamp.so"
  "$LIBRARY_DIR" "lib" "libmcwamp.so.${HCC_VERSION_MAJOR}"
  "$LIBRARY_DIR" "lib" "libmcwamp.so.${HCC_VERSION_MAJOR}.${HCC_VERSION_MINOR}"
  "$LIBRARY_DIR" "lib" "libmcwamp_cpu.so"
  "$LIBRARY_DIR" "lib" "libmcwamp_cpu.so.${HCC_VERSION_MAJOR}"
  "$LIBRARY_DIR" "lib" "libmcwamp_cpu.so.${HCC_VERSION_MAJOR}.${HCC_VERSION_MINOR}"
  "$LIBRARY_DIR" "lib" "libmcwamp_hsa.so"
  "$LIBRARY_DIR" "lib" "libmcwamp_hsa.so.${HCC_VERSION_MAJOR}"
  "$LIBRARY_DIR" "lib" "libmcwamp_hsa.so.${HCC_VERSION_MAJOR}.${HCC_VERSION_MINOR}"
  "$LIBRARY_DIR" "lib" "libclang_rt.builtins-${PROCESSOR}.a"
)

do_softlinks() {

  mkdir -p "$ROCM_PATH/lib"
  mkdir -p "$ROCM_PATH/lib/cmake"
  mkdir -p "$ROCM_PATH/bin"

  if [ -L "$ROCM_PATH/hcc" ] ; then
    rm -f "$ROCM_PATH/hcc" 
  fi

  ln -sf $ROCM_PATH/hcc/$LIBRARY_DIR/cmake/hcc $ROCM_PATH/lib/cmake/hcc

  if [ ! -e $ROCM_PATH/include/hcc ] ; then
    ln -sf $ROCM_PATH/hcc/include $ROCM_PATH/include/hcc
  fi

  for (( i=0; i<${#SOFTLINKS[@]}; i+=3 ))
  do
    SOURCE_DIR=${SOFTLINKS[i]}
    DEST_DIR=${SOFTLINKS[i+1]}
    FILE=${SOFTLINKS[i+2]}
    ln -sf $ROCM_PATH/hcc/$SOURCE_DIR/$FILE $ROCM_PATH/$DEST_DIR/$FILE
  done
}

if [ $1 == "configure" ] || [ $1 == "upgrade" ] || [ "$1" -ge "1" ] ; then
  do_softlinks
fi