#!/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 postinst
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"
)

remove_all_softlinks() {

  rm -f $INSTALL_PATH/include/hcc

  for (( i=0; i<${#SOFTLINKS[@]}; i+=3 ))
  do
    SOURCE_DIR=${SOFTLINKS[i]}
    DEST_DIR=${SOFTLINKS[i+1]}
    FILE=${SOFTLINKS[i+2]}

    rm -f  $ROCM_PATH/$DEST_DIR/$FILE
  done
  rm -f $ROCM_PATH/include/hcc
  rm -f $ROCM_PATH/lib/cmake/hcc

  rmdir --ignore-fail-on-non-empty /opt/rocm/lib 
  rmdir --ignore-fail-on-non-empty /opt/rocm/lib/cmake
  rmdir --ignore-fail-on-non-empty /opt/rocm/include
  rmdir --ignore-fail-on-non-empty /opt/rocm/bin 
  rmdir --ignore-fail-on-non-empty /opt/rocm     
}

if [ $1 == "remove" ] || [ $1 == "purge" ] || [ $1 == "upgrade" ] || [ "$1" -eq "0" ] ; then
  remove_all_softlinks
fi