#!/bin/sh

set -e

REPO=${REPO:-d7yio}
TAG=${TAG:-latest}

DIR=$(cd "$(dirname "$0")" && pwd)
cd $DIR

prepare(){
    mkdir -p config

    cat template/cdn.template.yaml > config/cdn.yaml
    cat template/dfget.template.yaml > config/dfget.yaml
    cat template/scheduler.template.yaml > config/scheduler.yaml
    cat template/manager.template.yaml > config/manager.yaml

    ip=${IP:-$(hostname -i)}

    sed -i "s,__IP__,$ip," config/dfget.yaml
    sed -i "s,__IP__,$ip," config/cdn.yaml
    sed -i "s,__IP__,$ip," config/scheduler.yaml
    sed -i "s,__IP__,$ip," config/manager.yaml
}

run_container(){
    RUNTIME=${RUNTIME:-docker}
    echo use container runtime: ${RUNTIME}

    echo try to clean old containers
    ${RUNTIME} rm -f dragonfly-cdn dragonfly-scheduler dragonfly-dfdaemon

    printf "create dragonfly-manager "
    ${RUNTIME} run -d --name dragonfly-cdn --net=host \
        -v /tmp/log/dragonfly:/var/log/dragonfly \
        -v ${DIR}/config/manager.yaml:/etc/dragonfly/manager.yaml \
        ${REPO}/manager:${TAG}

    printf "create dragonfly-cdn "
    ${RUNTIME} run -d --name dragonfly-cdn --net=host \
        -v /tmp/log/dragonfly:/var/log/dragonfly \
        -v ${DIR}/config/cdn.yaml:/etc/dragonfly/cdn.yaml \
        -v ${DIR}/config/nginx.conf:/etc/nginx/nginx.conf \
        ${REPO}/cdn:${TAG}

    printf "create dragonfly-scheduler "
    ${RUNTIME} run -d --name dragonfly-scheduler --net=host \
        -v /tmp/log/dragonfly:/var/log/dragonfly \
        -v ${DIR}/config/scheduler.yaml:/etc/dragonfly/scheduler.yaml \
        ${REPO}/scheduler:${TAG}

    printf "create dragonfly-dfdaemon "
    ${RUNTIME} run -d --name dragonfly-dfdaemon --net=host \
        -v /tmp/log/dragonfly:/var/log/dragonfly \
        -v ${DIR}/config/dfget.yaml:/etc/dragonfly/dfget.yaml \
        ${REPO}/dfdaemon:${TAG}
}

prepare

case "$1" in
  container)
    run_container
    ;;

  *)
    if [ -z "$1" ]; then

        # start all of docker-compose defined service
        docker-compose up -d

        # docker-compose version 3 depends_on does not wait for redis and mysql to be “ready” before starting manager ...
        # doc https://docs.docker.com/compose/compose-file/compose-file-v3/#depends_on
        for i in $(seq 0 10); do
          service_num=$(docker-compose ps --services |wc -l)
          ready_num=$(docker-compose ps | grep healthy | wc -l)
           if [ "$service_num" -eq "$ready_num" ]; then
             break
           fi
           echo "wait for all service ready: $ready_num/$service_num,$i times check"
           sleep 2
        done

        # print service list info
        docker-compose ps
        exit 0
    fi
    echo "unknown argument: $1"
    exit 1
    ;;
esac