#!/bin/sh
#
# Copyright (C) 2022 Broadcom. All rights reserved. The term "Broadcom"
# refers solely to the Broadcom Inc. corporate affiliate that owns
# the software below. This work is licensed under the OpenAFC Project License,
# a copy of which is included with this software program
#
AFC_DEVEL_ENV=${AFC_DEVEL_ENV:-production}

if [ -z "$AFC_WORKER_CELERY_CONCURRENCY" ]; then
    export AFC_WORKER_CELERY_CONCURRENCY=$(nproc)
fi
case "$AFC_DEVEL_ENV" in
  "devel")
    echo "Running debug profile" 
    echo "AFC_WORKER_CELERY_OPTS = ${AFC_WORKER_CELERY_OPTS}"
    echo "AFC_WORKER_CELERY_WORKERS = ${AFC_WORKER_CELERY_WORKERS}"
    echo "AFC_WORKER_CELERY_LOG = ${AFC_WORKER_CELERY_LOG}"
    echo "AFC_WORKER_ENG_TOUT = ${AFC_WORKER_ENG_TOUT}"
    echo "AFC_WORKER_CELERY_CONCURRENCY = ${AFC_WORKER_CELERY_CONCURRENCY}"
    ;;
  "production")
    echo "Running production profile"
    ;;
  *)
    echo "Uknown profile"
    ;;
esac

if [ ! -z ${AFC_AEP_ENABLE+x} ]; then
    if [ -z "$AFC_AEP_DEBUG" ]; then
        export AFC_AEP_DEBUG=0
    fi
    if [ -z "$AFC_AEP_LOGFILE" ]; then
        export AFC_AEP_LOGFILE=/aep/log/aep.log
    fi
    mkdir -p $(dirname "$AFC_AEP_LOGFILE")
    if [ -z "$AFC_AEP_REAL_MOUNTPOINT" ]; then
        export AFC_AEP_REAL_MOUNTPOINT=/mnt/nfs/rat_transfer
    fi
    if [ -z "$AFC_AEP_ENGINE_MOUNTPOINT" ]; then
        export AFC_AEP_ENGINE_MOUNTPOINT=$AFC_AEP_REAL_MOUNTPOINT
    fi
    if [ -z "$AFC_AEP_FILELIST" ]; then
        export AFC_AEP_FILELIST=/aep/list/aep.list
    fi
    mkdir -p $(dirname "$AFC_AEP_FILELIST")
    if [ -z "$AFC_AEP_CACHE_MAX_FILE_SIZE" ]; then
        #50M
        export AFC_AEP_CACHE_MAX_FILE_SIZE=50000000
    fi
    if [ -z "$AFC_AEP_CACHE_MAX_SIZE" ]; then
        #1G
        export AFC_AEP_CACHE_MAX_SIZE=1000000000
    fi
    if [ -z "$AFC_AEP_CACHE" ]; then
        export AFC_AEP_CACHE=/aep/cache
    fi
    mkdir -p $AFC_AEP_CACHE
    /usr/bin/parse_fs.py "$AFC_AEP_REAL_MOUNTPOINT" "$AFC_AEP_FILELIST"
    export AFC_ENGINE="/usr/bin/afc-engine.sh"
else
    export AFC_ENGINE="/usr/bin/afc-engine"
fi

celery multi start $AFC_WORKER_CELERY_WORKERS $AFC_WORKER_CELERY_OPTS -A afc_worker --concurrency=$AFC_WORKER_CELERY_CONCURRENCY --pidfile=/var/run/celery/%n.pid --logfile=/proc/1/fd/2 --loglevel=$AFC_WORKER_CELERY_LOG &

sleep infinity