#!/bin/bash

wfc_work_dir=/opt/wfc
docker_work_dir=${wfc_work_dir}/docker
src_service_dir=${wfc_work_dir}/systemd/system
dst_service_dir=/etc/systemd/system
java_work_dir=${docker_work_dir}/java

base_dockers="wfc-nacos wfc-mysql wfc-redis"
jar_dockers="wfc-auth wfc-gateway wfc-modules-system wfc-modules-user wfc-modules-job wfc-modules-file wfc-modules-payment"
modules_dockers="wfc-modules-system wfc-modules-user wfc-modules-job wfc-modules-file wfc-modules-payment"
fe_docker=wfc-nginx

docker_mysql_conf_d=/etc/mysql/conf.d
docker_my_conf_file=${docker_mysql_conf_d}/my.cnf

wfc_jar_images=$(docker images --format "{{.Repository}}"|grep -E "docker-wfc-|docker_wfc-")
wfc_modules_images=$(docker images --format "{{.Repository}}"|grep -E "docker-wfc-modules-|docker_wfc-modules-")

case "$1" in
    env)
        # prepare directory
        mkdir -p ${docker_work_dir}/mysql/tmp
        chown -R root:root ${docker_work_dir}
        if [ -d ${docker_work_dir}/mysql/data ]; then
            chown -R 999:999 ${docker_work_dir}/mysql/data
        fi
        find ${docker_work_dir} -type d -exec chmod 775 {} \;
        find ${docker_work_dir} -type f -exec chmod 664 {} \;
        chown root:root ${docker_work_dir}/mysql/conf.d/my.cnf
        chmod 600 ${docker_work_dir}/mysql/conf.d/my.cnf
        chmod 1777 ${docker_work_dir}/mysql/tmp

        # setup system serivce 
        cp -rf ${src_service_dir}/wfccontrol.service ${dst_service_dir}/
        systemctl daemon-reload
        systemctl enable wfccontrol.service

        # 获取传入的IP地址
        new_ip=$2

        # env定义原始文件和临时文件
        def_env_file=${docker_work_dir}/env/default.env
        org_env_file=${docker_work_dir}/.env
        tmp_env_file=${docker_work_dir}/temp.env

        if [ ! -f ${org_env_file} ]; then
            cp ${def_env_file} ${org_env_file}
        fi

        if [ ! -z "${new_ip}" ]; then
            # 使用sed命令替换IP地址
            # 替换.env文件
            sed "s/WFC_SERVER_IP=.*/WFC_SERVER_IP=${new_ip}/" ${org_env_file} > $tmp_env_file
            # sed -i "s/GATEWAY_ADDR=.*/GATEWAY_ADDR=${new_ip}:8080/" $tmp_env_file
            mv $tmp_env_file $org_env_file 
        fi

        # replace nginx.conf gateway server ip and port

        while IFS= read -r line; do
            if [[ ! $line =~ ^# && $line =~ .*=.* ]]; then
                eval "export $line"
            fi
        done < $org_env_file

        # nginx定义原始文件和临时文件
        org_nginx_conf=${docker_work_dir}/nginx/conf/nginx.conf
        tmp_nginx_conf=${docker_work_dir}/nginx/conf/tmp_nginx.conf

        # 使用sed命令替换proxy_pass行中的IP地址
        sed "s|proxy_pass http://.*/;|proxy_pass http://${GATEWAY_SERVER_IP}:${GATEWAY_SERVER_PORT}/;|" $org_nginx_conf > $tmp_nginx_conf
        mv $tmp_nginx_conf $org_nginx_conf

        ;;
    initdb)
        # init database
        cd ${docker_work_dir}
        docker-compose up -d wfc-mysql
        sleep 10
        for sql in ${docker_work_dir}/mysql/db/*.sql; do
            if [ -f "$sql" ]; then
                echo -n "Execute SQL script: ${sql} ... "
                docker exec -i wfc-mysql mysql --defaults-extra-file=${docker_my_conf_file} < ${sql}
                if [ $? = 0 ]; then
                    echo "done" 
                fi
            else
                echo "No SQL files found in ${docker_work_dir}/mysql/db/"
            fi
        done
        ;;
    base)
        # build docker compose
        cd ${docker_work_dir}
        docker-compose stop ${base_dockers}
        docker-compose build ${base_dockers}   
        cd ${java_work_dir}
        if [[ "$(docker images -q wfc-java:jre8 2> /dev/null)" == "" ]]; then
            echo "Building wfc-java:jre8"
            docker build -t wfc-java:jre8 -f dockerfile .
        fi        
        ;;
    jar)
        # build docker compose
        cd ${docker_work_dir}
        docker-compose stop ${jar_dockers}
        docker-compose build ${jar_dockers}
        ;;
    modules)
        # build docker compose
        cd ${docker_work_dir}
        docker-compose stop ${modules_dockers}
        docker-compose build ${modules_dockers}  
        ;;
    fe)
        # build docker compose
        cd ${docker_work_dir}
        docker-compose stop ${fe_docker}
        docker-compose build ${fe_docker}
        ;;
    all)
        $0 env $2
        $0 base
        $0 jar
        $0 fe
        ;;                                  
    rm)
        case "$2" in
            base)
                cd ${docker_work_dir}
                docker-compose stop ${base_dockers}
                for base_docker in ${base_dockers}; do
                    docker rm ${base_docker}
                done   
                ;;
            jar)
                cd ${docker_work_dir}
                docker-compose stop ${jar_dockers}
                docker rm ${jar_dockers}
                if [ ! -z "${wfc_jar_images}" ]; then
                    docker rmi ${wfc_jar_images} 
                fi
               
                # for jar_docker in ${jar_dockers}; do
                #     docker rm ${jar_docker}
                #     docker rmi docker_${jar_docker}                
                # done               
                ;;
            modules)
                cd ${docker_work_dir}
                docker-compose stop ${modules_dockers}
                docker rm ${modules_dockers}
                if [ ! -z "${wfc_modules_images}" ]; then
                    docker rmi ${wfc_modules_images} 
                fi
           
                # for modules_docker in ${modules_dockers}; do
                #     docker rm ${modules_docker}
                #     docker rmi docker_${modules_docker}
                #     docker rmi docker-${modules_docker}
                # done               
                ;;
            fe)
                cd ${docker_work_dir}
                docker-compose stop ${fe_docker}
                docker rm ${fe_docker}
                # docker rmi nginx                    
                ;;
            network)
                cd ${docker_work_dir}
                docker-compose stop
                docker network rm wfc-fe-network
                docker network rm wfc-be-network
                ;;
            all)
                $0 rm base
                $0 rm jar
                $0 rm fe
                $0 rm network
                ;;
            *)
                echo "WANFi Control and Billing Management System Setup ... "
                echo "Usage: $0 rm [base|jar|modules|fe|network|all]"
                exit 1 
                ;;               
        esac
        ;;
    prune)
        cd ${docker_work_dir}
        docker-compose down
        docker system prune -a
        ;;
    *)
        echo "WANFi Control and Billing Management System Setup ... "
        echo "Usage: $0 env|initdb|base|jar|modules|fe|all|rm [base|jar|modules|fe|network|all]|prune"
        echo "       $0 env [IP]"
        echo "       $0 initdb"
        echo "       $0 base|jar|modules|fe"
        echo "       $0 all [IP]"
        echo "       $0 rm [base|jar|modules|fe|network|all]"
        echo "       $0 prune"
        exit 1
        ;;
esac