#!/bin/bash

INS_TYPE_URL="http://metadata.tencentyun.com/latest/meta-data/instance/instance-type"
GN7VI_TYPE="gn7vi"

SDK_URL_PATH="https://ie-mps-1258344699.cos.ap-nanjing.myqcloud.com/common/gn7vi/tscsdk-center.tar.gz"
SDK_FILE_NAME="tscsdk-center.tar.gz"
SDK_LOCATION="/usr/local/qcloud"
SDK_PATH="$SDK_LOCATION/tscsdk-center"

PROFILE="/etc/profile"

log_file="/tmp/nvenc_ai_sdk_install.log"


log() {
    local msg=$1
    [ -z "$msg" ] && return 1
    local now=`date +"%F %H:%M:%S"` 
    local ms=`date +%N`
    ms=`echo | awk "{printf(\"%03d\", $ms/1000000)}"`
    echo "[$now,$ms]: $msg" | tee -a $log_file
}

exitlog() {
    log "$1"
    exit 0
}


type=`curl $INS_TYPE_URL`
echo "ins-type:$type"
[ -z "$type" ] && exitlog "instance type($type) is invalid"

found=`echo $type | grep -i $GN7VI_TYPE`
[ -z "$found" ] && exitlog "ins-type:$type no need install"


wget -t 100 --timeout 10 "$SDK_URL_PATH" -O "/tmp/$SDK_FILE_NAME" 
[ $? != 0 ] && exitlog "wget sdk file fail: $SDK_URL_PATH"

tar zxvf /tmp/$SDK_FILE_NAME -C $SDK_LOCATION  > /dev/null 2>&1
rm -f /tmp/$SDK_FILE_NAME > /dev/null 2>&1

firsteth=`ifconfig | grep ^[a-z] | awk -F: '{print $1}' | head -n 1`
if [ "$firsteth" != "eth0" ]; then
    sed -i "2s/eth0/${firsteth}/" $SDK_PATH/sdk_config
fi

echo "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$SDK_PATH/fflib_gpu/" >> ${PROFILE}
source ${PROFILE}

log "ai sdk install succuess"