#!/bin/bash
date=`date +"%Y%m%d%H%M"`
source_path=/mnt/tnbs/gh_install/test
# sofrware_path=/data/app
registry='cr.registry.hn-1.res.sgmc.sgcc.com.cn/pdwswszhghkyythglxt/pdwswszhghkyythglxt_dev'

cat<<EOF
                                    _oo0oo_
                                   o8888888o
                                   88" . "88
                                   (| -_- |)
                                   0\  =  /0
                                ____/'---'\____
                              .'  \|       ||/ '.
                             /  \|||   :   |||/  \
                            /  _||||| -:- |||||_  \
                            |   |   \  -  /   |   |
                            | \_|  ''\---/''  |_/ |
                            \  .-\__  '-'  __/-.  /
                          ___'. .'  /--.--\  '. .'___
                       ."" '<  '.___\_<|>_/___.'  >' "".
                      | | : '-  \'.;'\ _ /';.'/  -  ' : | |
                      \ \ '_.   \_ ___\_/ __ _/   .-'  / /
                  ====='-.____'.___ \_____/ ___.-'____.-'=====
                                    '=---='

                 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
                              佛祖保佑      永不宕機(jī)
EOF


# 打包
function package(){
  echo "------------------------update package ${service}------------------------"
  cd ${source_path}
  mkdir -p ./${service}
  rm -f ./${service}/app.jar #刪除config中更名為app.jar的包
  if [ ${service} == 'gclfw' ]
  then
    if [ -e ./quantity*.jar ];then
      mv ./quantity*.jar ./${service}
    elif [ -e ./${service}.jar ]; then
      mv ./${service}.jar ./${service}
    else
      echo "No suitable jar file found. Exiting the program."
      exit 1
    fi
  elif [ -e ./${service}*.jar ];then
    mv ./${service}*.jar ./${service}
  else
    echo "No suitable jar file found. Exiting the program."
      exit 1
  fi
}

function config(){
   echo "------------------------開始修改服務(wù) ${service}配置------------------------"
  # 確認(rèn)是否存在更新配置
  cd ${source_path}
  if [ -f ${service}.yml ];then
    mv ${source_path}/${service}*.yml ${source_path}/${service}/application.yml
  elif [ -f ${app}.properties ];then
    mv ${source_path}/${service}*.properties ${source_path}/${service}/application.properties
  fi
  
  cd ${source_path}/${service}
  ### rm -f app.jar
  rm -rf ./BOOT-INF ./META-INF ./org

# 確保找到正確的文件名
  jar_file=$(find . -name "*.jar" -print -quit)
  if [ -z "$jar_file" ]; then
    echo "未找到匹配的 .jar 文件"
    exit 1
  fi
  unzip $jar_file > /dev/null
  rm -f $jar_file
  rm -f ./BOOT-INF/classes/application*

  if [ -f application.yml ];then
    cp ./application.yml ./BOOT-INF/classes/application.yml
  elif [ -f application.properties ];then
    cp ./application.properties ./BOOT-INF/classes/application.properties
  fi

  jar cvfm0 app.jar META-INF/MANIFEST.MF ./BOOT-INF ./META-INF ./org >> /dev/null
  rm -rf ./BOOT-INF ./META-INF ./org
}

# 構(gòu)建鏡像
function build_image {
  echo "------------------------構(gòu)建 package ${service}鏡像版本------------------------"
if [ ! -d "${source_path}/${service}" ]; then
  echo "${service}目錄不存在杉适，跳過操作"
  exit 1
else
  cd ${source_path}/${service}/
  if [!  -f dockerfile ]
  then
    cp -n /mnt/tnbs/gh_install/test/dockerfile ${source_path}/${service}/dockerfile
  fi
  
  sudo docker build -t ${registry}:${service}${date} .
  sudo docker push ${registry}:${service}${date}
  sudo docker rmi ${registry}:${service}${date}
fi
}

# 切換鏡像
function  redeploy_service {
  echo "------------------------替換 package ${service}鏡像版本------------------------"
  echo -e "\033[1;31m警告"'!!!'"此操作較危險(xiǎn),終止請(qǐng)按CTRL+z挂谍！如未終止導(dǎo)致服務(wù)異常审编，可執(zhí)行命令：sudo kubectl rollout undo deploy進(jìn)行全量回滾"'!'"\033[0m"
  if [ ${service} == 'dpglfw' ]
  then
    # sudo kubectl get deploy | grep '^lcb-' | tail -1 | awk -F " " '{ print $1 }' | xargs sudo kubectl edit deploy
    # sudo kubectl get deploy | grep 'dpglfw-group' | tail -1 | awk -F " " '{ print $1 }' | xargs sudo kubectl get deploy -o yaml > ${service}.yaml
     #
    deploy_name=$(sudo kubectl get deploy | grep 'dpglfw-group' | tail -1)
  else
    # sudo kubectl get deploy | grep ${service} | tail -1 | awk -F " " '{ print $1 }' | xargs sudo kubectl get deploy -o yaml > ${service}.yaml
    # deploy_name=$(sudo kubectl get deploy | grep ${service} | tail -1)
    deploy_name=$(kubectl get deploy -l app=<service-name> | awk 'NR > 1 && $2 > 0 {print $1}'）
  fi

  if [ -z $deploy_name ];then
    echo echo "未找到指定的 Deployment：${service}"
    exit 1
  else
    sudo kubectl get deploy $deploy_name -o yaml > ${service}.yaml
    # 修改鏡像
    sed -i "s/cr.registry.hn-1.res.sgmc.sgcc.com.cn\/pdwswszhghkyythglxt\/pdwswszhghkyythglxt_dev\:\(.*\)/cr.registry.hn-1.res.sgmc.sgcc.com.cn\/pdwswszhghkyythglxt\/pdwswszhghkyythglxt_dev\:${service}${date}/g" ${service}.yaml
    sudo kubectl apply -f ${service}.yaml
    rm ${service}.yaml
  fi
}

# 檢查輸入?yún)?shù)是否為空
if [ $# -eq 0 ];then
  echo -e "\033[1;31m警告"'!!!',請(qǐng)輸入至少一個(gè)服務(wù)名'!'"\033[0m"
  exit 1
fi

# 獲取用戶選擇的部署場(chǎng)景
echo "請(qǐng)選擇要執(zhí)行的部署場(chǎng)景："
echo "1. 上傳jar包>替換配置>構(gòu)建docker鏡像"
echo "2. 替換配置>構(gòu)建docker鏡像"
echo "3. 上傳jar包>替換配置>構(gòu)建docker鏡像>切換服務(wù)鏡像"
echo "4. 替換配置>構(gòu)建docker鏡像>kubectl切換服務(wù)鏡像"
read -p "輸入數(shù)字選擇場(chǎng)景：" choice

# 循環(huán)處理每個(gè)輸入的服務(wù)名
for service in $@
do
  echo "開始處理服務(wù)：$service"
  
  case $choice in 
    1)
       #場(chǎng)景1：上傳jar包>替換配置>構(gòu)建docker鏡像
       package $service
       config $service
       build_image $service
      ;;
    2)
      #場(chǎng)景1：替換配置>構(gòu)建docker鏡像
      config $service
      build_image $service
      ;;
    3)
      #場(chǎng)景3：上傳jar包>替換配置>構(gòu)建docker鏡像>切換服務(wù)鏡像
      package $service
      config $service
      build_image $service
      redeploy_service $service
      ;;
    4)
      #場(chǎng)景4：替換配置>構(gòu)建docker鏡像>kubectl切換服務(wù)鏡像
      config $service
      build_image $service
      redeploy_service $service
      ;;
    *)
      echo "無效的選擇"
      ;;
  esac
    
  echo -e "\033[1;31m"服務(wù) $service 處理完成"\033[0m"
done