#!/bin/bash

WFCWorkDir=/opt/wfc
DockerWorkDir=${WFCWorkDir}/docker
SrcServiceDir=${WFCWorkDir}/systemd/system
DstServiceDir=/etc/systemd/system

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

case "$1" in
    env)
        # prepare directory
        mkdir -p ${DockerWorkDir}/mysql/tmp
        chown -R root:root ${DockerWorkDir}/*
        chown -R root:root ${DockerWorkDir}/.*
        chown -R mysql:mysql ${DockerWorkDir}/mysql
        chmod -R 775 ${DockerWorkDir}/nacos
        chmod -R 775 ${DockerWorkDir}/nginx
        chmod -R 775 ${DockerWorkDir}/redis
        chmod -R 775 ${DockerWorkDir}/mysql
        chmod 1777 ${DockerWorkDir}/mysql/tmp

        # setup system serivce 
        cp -rf ${SrcServiceDir}/wfccontrol.service ${DstServiceDir}/
        systemctl daemon-reload
        systemctl enable wfccontrol.service    
    ;;
    base)
        # build docker compose
        cd ${DockerWorkDir}
        docker-compose stop ${base_dockers}
        docker-compose build ${base_dockers}   
    ;;
    jar)
        # build docker compose
        cd ${DockerWorkDir}
        docker-compose stop ${jar_dockers}
        docker-compose build ${jar_dockers}
    ;;
    modules)
        # build docker compose
        cd ${DockerWorkDir}
        docker-compose stop ${modules_dockers}
        docker-compose build ${modules_dockers}  
    ;;
    fe)
        # build docker compose
        cd ${DockerWorkDir}
        docker-compose stop ${fe_docker}
        docker-compose build ${fe_docker}
    ;;
    all)
        $0 env
        $0 base
        $0 jar
        $0 fe
    ;;                                  
    rm)
        case "$2" in
            base)
                cd ${DockerWorkDir}
                docker-compose stop ${base_dockers}
                for base_docker in ${base_dockers}; do
                    docker rm ${base_docker}
                done   
            ;;
            jar)
                cd ${DockerWorkDir}
                docker-compose stop ${jar_dockers}
                for jar_docker in ${jar_dockers}; do
                    docker rm ${jar_docker}
                    docker rmi docker_${jar_docker}
                done               
            ;;
            modules)
                cd ${DockerWorkDir}
                docker-compose stop ${modules_dockers}
                for modules_docker in ${modules_dockers}; do
                    docker rm ${modules_docker}
                    docker rmi docker_${modules_docker}
                done               
            ;;
            fe)
                cd ${DockerWorkDir}
                docker-compose stop ${fe_docker}
                docker rm ${fe_docker}
                docker rmi nginx                    
            ;;
            *)
                echo "WANFi Controller Setup "
                echo "Usage: $0 rm [base|jar|modules|fe]"
                exit 1 
            ;;               
        esac
    ;;
    *)
        echo "WANFi Controller Setup "
        echo "Usage: $0 all|env|base|jar|modules|fe|rm [base|jar|modules|fe]"
        exit 1
    ;;
esac