#!/bin/bash

#### 从临时服务器发布站点到正式环境
#### Author: chenpeiqing
#### Date: 2025-01-07

SITE_CODE=$1
DOMAIN_NAME=$2
DATABASE_NAME=$3
# 最大重试次数
MAX_RETRIES=3
# 初始重试次数
RETRY_COUNT=0
TEMP_DOMAIN='v3-site.adwebcloud.com'

RDS_HOST='127.0.0.1'
RDS_ADMIN='root'
RDS_PWD='Admin@123.com'

TEMP_USER_NAME='ubuntu'
TEMP_IP='35.87.155.71'
TEMP_PEM='/opt/adweb3/pem/adweb3.pem'

first_time=$(date +'%s')

#打印入参
echo "SITE_CODE":$SITE_CODE
echo "DOMAIN_NAME":$DOMAIN_NAME
echo "DATABASE_NAME":$DATABASE_NAME

#临时服务器打包项目代码和数据库文件
start_time=$(date +'%s')
echo "打包站点代码和数据库文件开始："
ssh -i $TEMP_PEM -p 22 -o IdentitiesOnly=yes $TEMP_USER_NAME@$TEMP_IP << EOF
        sudo -i
        sh /opt/adweb3/shell/bak-wpsite.sh $SITE_CODE
   exit
EOF
end_time=$(date +'%s')
echo "打包站点代码和数据库文件完成，花费："$((end_time - start_time))"s"


#创建空数据库以及指定用户
start_time=$(date +'%s')
echo "创建生产目录和数据库开始："
mysql -h $RDS_HOST -u$RDS_ADMIN -p$RDS_PWD<< EOF
  DROP DATABASE IF EXISTS $DATABASE_NAME;
  CREATE DATABASE $DATABASE_NAME CHARACTER SET=utf8mb4 COLLATE=utf8mb4_unicode_ci;
  DROP USER IF EXISTS $SITE_CODE;
  CREATE USER '$SITE_CODE'@'%' IDENTIFIED BY 'Admin@123.com';
  GRANT ALL ON $DATABASE_NAME.* TO '$SITE_CODE'@'%';
  flush privileges;
  exit
EOF

#创建站点文件夹
if [ ! -d  "/usr/local/wwwroot/$DOMAIN_NAME" ]; then
     mkdir -p /usr/local/wwwroot/$DOMAIN_NAME
fi
end_time=$(date +'%s')
echo "创建生产目录和数据库完成，花费："$((end_time - start_time))"s"

##下载SQL文件
start_time=$(date +'%s')
echo "下载数据库文件并导入数据库开始："
wget -P /usr/local/wwwroot/$DOMAIN_NAME/ http://download-template-v3.adwebcloud.com/$SITE_CODE.sql
mysql -h $RDS_HOST -u$SITE_CODE -p'Admin@123.com' $DATABASE_NAME</usr/local/wwwroot/$DOMAIN_NAME/$SITE_CODE.sql
end_time=$(date +'%s')
echo "下载数据库文件并导入数据库完成，花费："$((end_time - start_time))"s"


#下载站点文件
start_time=$(date +'%s')

# 函数：下载文件并显示进度
download_file() {
    echo "正在下载: http://download-template-v3.adwebcloud.com/$SITE_CODE.tar.gz"
    wget -c "http://download-template-v3.adwebcloud.com/$SITE_CODE.tar.gz" -O "/usr/local/wwwroot/$DOMAIN_NAME/$SITE_CODE.tar.gz"
    if [ $? -eq 0 ]; then
        echo "下载成功: /usr/local/wwwroot/$DOMAIN_NAME/$SITE_CODE.tar.gz"
        return 0
    else
        echo "下载失败: http://download-template-v3.adwebcloud.com/$SITE_CODE.tar.gz"
        return 1
    fi
}

# 下载文件，直到成功或达到最大重试次数
while [ $RETRY_COUNT -lt $MAX_RETRIES ]; do
    download_file
    if [ $? -eq 0 ]; then
        break
    else
        RETRY_COUNT=$((RETRY_COUNT + 1))
        echo "重试 $RETRY_COUNT/$MAX_RETRIES..."
    fi
done

# 检查是否下载成功
if [ $RETRY_COUNT -ge $MAX_RETRIES ]; then
    echo "下载失败，已达到最大重试次数 $MAX_RETRIES"
else
    echo "文件已成功下载到 /usr/local/wwwroot/$DOMAIN_NAME/"
fi

#wget -P /usr/local/wwwroot/$DOMAIN_NAME/ http://download-template-v3.adwebcloud.com/$SITE_CODE.tar.gz
end_time=$(date +'%s')
echo "下载代码包完成，花费："$((end_time - start_time))"s"

#下载站点文件
start_time=$(date +'%s')
echo "解压缩到指定位置开始："
tar -zxf /usr/local/wwwroot/$DOMAIN_NAME/$SITE_CODE.tar.gz -C /usr/local/wwwroot/$DOMAIN_NAME/
mv /usr/local/wwwroot/$DOMAIN_NAME/$SITE_CODE/* /usr/local/wwwroot/$DOMAIN_NAME/
rm -rf /usr/local/wwwroot/$DOMAIN_NAME/$SITE_CODE
end_time=$(date +'%s')
echo "解压缩到指定位置完成，花费："$((end_time - start_time))"s"

#更新站点配置文件
start_time=$(date +'%s')
echo "更新站点配置开始："
chown -R www-data:www-data /usr/local/wwwroot/$DOMAIN_NAME

#替换redis object cache 键值
sed -i "s:'$SITE_CODE.$TEMP_DOMAIN':'$DOMAIN_NAME':g" /usr/local/wwwroot/$DOMAIN_NAME/wp-config.php

#替换sql数据库名称
sed -i "s:'$SITE_CODE':'$DATABASE_NAME':g" /usr/local/wwwroot/$DOMAIN_NAME/wp-config.php

#替换sql数据库用户名
sed -i "s:'adweb':'$SITE_CODE':g" /usr/local/wwwroot/$DOMAIN_NAME/wp-config.php

#替换sql数据库密码
sed -i "s:'2024_RqBQxctjakHCJVq8kmmYHQKZQGU4YMUH':'Admin@123.com':g" /usr/local/wwwroot/$DOMAIN_NAME/wp-config.php

#替换sql数据库链接地址
sed -i "s:'127.0.0.1':'$RDS_HOST':g" /usr/local/wwwroot/$DOMAIN_NAME/wp-config.php
sed -i "s:'localhost':'$RDS_HOST':g" /usr/local/wwwroot/$DOMAIN_NAME/wp-config.php

#删除下载站点文件和sql文件压缩包
cd /usr/local/wwwroot/$DOMAIN_NAME/
rm -rf $SITE_CODE.sql.tar.gz $SITE_CODE.tar.gz $SITE_CODE.sql
end_time=$(date +'%s')

#更新站点域名
cp /opt/adweb3/config/wp-cli.phar /usr/local/wwwroot/$DOMAIN_NAME
cd /usr/local/wwwroot/$DOMAIN_NAME

# 解析域名
result=$(nslookup $DOMAIN_NAME | grep -Po 'Address: \K[\d.]+')
result_www=$(nslookup www.$DOMAIN_NAME | grep -Po 'Address: \K[\d.]+')

# 判断解析结果是否为空
if [ ! -z "$result" ] && [ ! -z "$result_www" ]; then
  #替换为www域名
  php ./wp-cli.phar search-replace "$SITE_CODE.$TEMP_DOMAIN" "www.$DOMAIN_NAME" --allow-root
  echo "result:"$result
  echo "result_www:"$result_www
elif [ ! -z "$result" ] && [ -z "$result_www" ]; then
  #替换域名
  php ./wp-cli.phar search-replace "$SITE_CODE.$TEMP_DOMAIN" "$DOMAIN_NAME" --allow-root
  echo "result":$result
elif [ -z "$result" ] && [ ! -z "$result_www" ]; then
  #替换为www域名
  php ./wp-cli.phar search-replace "$SITE_CODE.$TEMP_DOMAIN" "www.$DOMAIN_NAME" --allow-root
  echo "result:"$result
  echo "result_www:"$result_www
else
  echo "Parsing failed"
fi

#清除站点缓存
php ./wp-cli.phar cache flush --allow-root
rm -rf /usr/local/wwwroot/$DOMAIN_NAME/wp-cli.phar
echo "更新站点配置完成，花费："$((end_time - start_time))"s"

# 重启nginx
cp /opt/adweb3/nginx_template/wp-site.conf /usr/local/openresty/nginx/vhost/$DOMAIN_NAME.conf
sed -i "s:DOMAIN_NAME:$DOMAIN_NAME:g" /usr/local/openresty/nginx/vhost/$DOMAIN_NAME.conf
echo "开始重启nginx配置"
/usr/local/openresty/nginx/sbin/nginx -s reload
echo "nginx配置重启完成"echo "开始重启nginx配置"

#生成ssl证书

echo "生成$DOMAIN_NAME的ssl证书开始："
start_time=$(date +'%s')

# 解析域名
result=$(nslookup $DOMAIN_NAME | grep -Po 'Address: \K[\d.]+')
result_www=$(nslookup www.$DOMAIN_NAME | grep -Po 'Address: \K[\d.]+')

# 判断解析结果是否为空
if [ ! -z "$result" ] && [ ! -z "$result_www" ]; then
  #生成SSL证书
  #生成位置 /root/.acme.sh/$DOMAIN_NAME
  /root/.acme.sh/acme.sh --issue -d $DOMAIN_NAME -d "www.$DOMAIN_NAME" --webroot /usr/local/wwwroot/$DOMAIN_NAME
  echo "result:"$result
  echo "result_www:"$result_www
elif [ ! -z "$result" ] && [ -z "$result_www" ]; then
  /root/.acme.sh/acme.sh --issue -d $DOMAIN_NAME  --webroot /usr/local/wwwroot/$DOMAIN_NAME
  echo "result":$result
elif [ -z "$result" ] && [ ! -z "$result_www" ]; then
  /root/.acme.sh/acme.sh --issue -d "www.$DOMAIN_NAME" --webroot /usr/local/wwwroot/$DOMAIN_NAME
  echo "result:"$result
  echo "result_www:"$result_www
else
  echo "Parsing failed"
fi


#等待30s,确保SSL证书目录生成
sleep 30

if [ -d "/root/.acme.sh/$DOMAIN_NAME" ]; then
fullchain=$(cat /root/.acme.sh/$DOMAIN_NAME/fullchain.cer)
awk 'split_after==1{n++;split_after=0} /-----END CERTIFICATE-----/ {split_after=1} {print > "fullchain" n ".pem"}' < /root/.acme.sh/$DOMAIN_NAME/fullchain.cer
Certificate_body=$(cat fullchain.pem)
echo '[start cert.pem]'
echo $Certificate_body
echo '[end cert.pem]'

Certificate_chain=$(cat fullchain1.pem)
Certificate_chain1=$(cat fullchain2.pem)
echo '[start chain.pem]'
echo ${Certificate_chain}" "${Certificate_chain1}
echo '[end chain.pem]'

Certificate_private_key=$(cat /root/.acme.sh/$DOMAIN_NAME/*key)
echo '[start privkey.pem]'
echo $Certificate_private_key
echo '[end privkey.pem]'

rm -rf *.pem
elif [ -d "/root/.acme.sh/${DOMAIN_NAME}_ecc" ]; then
fullchain=$(cat /root/.acme.sh/${DOMAIN_NAME}_ecc/fullchain.cer)
awk 'split_after==1{n++;split_after=0} /-----END CERTIFICATE-----/ {split_after=1} {print > "fullchain" n ".pem"}' < /root/.acme.sh/${DOMAIN_NAME}_ecc/fullchain.cer
Certificate_body=$(cat fullchain.pem)
echo '[start cert.pem]'
echo $Certificate_body
echo '[end cert.pem]'

Certificate_chain=$(cat fullchain1.pem)
Certificate_chain1=$(cat fullchain2.pem)
echo '[start chain.pem]'
echo ${Certificate_chain}" "${Certificate_chain1}
echo '[end chain.pem]'

Certificate_private_key=$(cat /root/.acme.sh/${DOMAIN_NAME}_ecc/*key)
echo '[start privkey.pem]'
echo $Certificate_private_key
echo '[end privkey.pem]'
rm -rf *.pem
else
echo "SSL Certificate not found"
fi

end_time=$(date +'%s')
echo "生成$DOMAIN_NAME的ssl证书开始花费："$((end_time - start_time))"s"

#开启nginx证书 ssl 配置
#重启nginx
sed -i "s:####::g" /usr/local/openresty/nginx/vhost/$DOMAIN_NAME.conf
echo "开始重启nginx配置"
/usr/local/openresty/nginx/sbin/nginx -s reload
echo "nginx配置重启完成"echo "开始重启nginx配置"

#临时服务器清空之前的临时资源
ssh -i $TEMP_PEM -p 22 -o IdentitiesOnly=yes $TEMP_USER_NAME@$TEMP_IP << EOF
        sudo -i
        sh /opt/adweb3/shell/remove-wpsite.sh $SITE_CODE
   exit
EOF

echo "[Creation completed]"