#!/bin/bash

set -e

source "$(
  cd "$(dirname "$0")" || exit 1
  pwd
)/../common.sh"

# 安装编译前必要应用
function install_development() {
  sudo yum install -y wget tar libatomic environment-modules
  source /etc/profile
}

# 下载zlib
function download_zlib() {
  echo "# zlib正在下载"
  wget -t 40 -c -P "$1" http://www.zlib.net/fossils/zlib-1.2.11.tar.gz
  if [[ $? -eq 0 ]]; then
    echo -e "\033[1;32;1m# zlib已经下载成功\033[0m"
  else
    echo -e "\033[31m# zlib下载失败,请重新执行脚本\033[0m"
    exit 1
  fi
}

# 安装zlib
function install_zlib() {
  echo "# zlib正在安装"
  tar xf "$1"/zlib-1.2.11.tar.gz -C "$1" && rm -rf "$1"/zlib-1.2.11.tar.gz
  cd "$1"/zlib-1.2.11

  CC=clang ./configure --prefix="$2"/zlib-1.2.11
  make -j$(nproc) && make -j$(nproc) install

  echo -e "\033[1;32;1m# zlib已经安装完毕\033[0m"
}

# 创建zlib的modulefiles
function write_zlib_module() {
  cat >"$1/zlib-1.2.11/zlib_modulefiles"<<EOF
#%Module1.0
conflict zlib
variable modfile [file normalize [info script]]
proc getModulefileDir {} {
  variable modfile
  set modfile_path [file dirname \$modfile]
  return \$modfile_path
}
set pwd [getModulefileDir]
set ZLIB \$pwd
setenv ZLIB \$ZLIB
prepend-path LD_LIBRARY_PATH \$ZLIB/lib
EOF
}

function main() {
  # 打印当前运行脚本信息
  print_current_running_script_info "zlib-1.2.11.sh"
  # 校验安装路径与构建路径是否一致
  is_equal_build_install_path "$1" "$2"
  # 安装编译前应用
  install_development
  # 下载zlib
  download_zlib "$1"
  # 安装zlib
  install_zlib "$1" "$2"
  # 写入zlib环境变量modulefiles文件
  write_zlib_module "$2"
  # 删除构建路径
  rm -rf "$1"/zlib-1.2.11
}

main "$@"