run

#!/bin/bash

MYDIR=`dirname $0` && [ ! `echo "$0" | grep '^\/'` ] && MYDIR=`pwd`/$MYDIR

DEV=0

# Local HTTP Port for web application
WEB_PORT=8087

# Path to the data
DATADIR=/opt/data/

# Docker images
DB_IMAGE=pgd-mmdt-db
SCAN_IMAGE=pgd-mmdt-scan
WEB_IMAGE=pgd-mmdt-web

# Docker containers
DB_CONTAINER=mmdt-db
SCAN_CONTAINER=mmdt-scan
WEB_CONTAINER=mmdt-web

# Volume name for Mondo DB
MONGO_VOL=mmdt-mongodb
MONGO_PORT=27017

# Docker account for retrieving the images
DOCKERACCOUNT=djacob65

# default user for htpasswd file
USER=admin

# Replace default configuration parameters with local ones if exist
[ -f $MYDIR/local.conf ] && . $MYDIR/local.conf

CMD=$1

usage() { echo "usage: sh $0 start|stop|restart|fullstart|ps|build|pull|initdb|scan";  exit 1; }

case "$CMD" in
   start)
        echo "Lauching ... "
        [ ! -d $MYDIR/web/tmp ] && sudo mkdir -p $MYDIR/web/tmp
        [ -d $MYDIR/web/tmp ] && sudo chmod 777 $MYDIR/web/tmp
        [ ! -d $MYDIR/web/cache ] && sudo mkdir -p $MYDIR/web/cache
        [ -d $MYDIR/web/cache ] && sudo chmod 777 $MYDIR/web/cache
        sudo chmod 777 $MYDIR/web/conf/*.json
        VOLDBSRC="-v $MYDIR/dockerdbpart/initialisation:/js"
        VOLSCANSRC=
        [ $DEV -eq 1 ] && VOLSRCSCAN="-v $MYDIR/dockerscanpart/scripts:/pgd"
        sudo docker run -d -v $MONGO_VOL:/data/db -v $DATADIR:/pgd_data -v $MYDIR/web/conf:/json $VOLDBSRC -p $MONGO_PORT:27017 --name $DB_CONTAINER $DB_IMAGE
        sudo docker run -d -v $DATADIR:/pgd_data -v $MYDIR/web/conf:/json $VOLSRCSCAN --link $DB_CONTAINER --name $SCAN_CONTAINER $SCAN_IMAGE
        sudo docker run -d -p $WEB_PORT:80 -v $MYDIR/web:/var/www/html -v $DATADIR:/mnt -v $MYDIR/etc:/etc/pgd-mmdt --link $DB_CONTAINER --name $WEB_CONTAINER $WEB_IMAGE
        for d in `find $MYDIR/web/cvlist/ -type d -mindepth 1 -maxdepth 1 2>/dev/null`; do f=$(basename $d); sudo chmod 777 "$d/$f.txt"; done
        [ $? -eq 0 ] && echo OK
        ;;
   fullstart)
        ( sh $0 start ; sh $0 initdb ; sh $0 scan )
        ;;
   stop)
        sudo docker rm -f $WEB_CONTAINER $SCAN_CONTAINER $DB_CONTAINER
        echo -n "Remove volumes ... "
        for v in $(sudo docker volume ls | grep local | grep -v $MONGO_VOL | cut -d' ' -f6); do
           sudo docker volume rm $v 1>/dev/null 2>/dev/null
        done
        sudo docker volume rm $MONGO_VOL 1>/dev/null 2>/dev/null
        sudo rm -f $MYDIR/web/tmp/*
        [ $? -eq 0 ] && echo OK
        ;;
   restart)
        ( sh $0 stop ; sh $0 fullstart )
        ;;
   ps)
        sudo docker ps | head -1
        sudo docker ps | grep "mmdt-"
        ;;
   pull)
        sudo docker pull $DOCKERACCOUNT/maggot-db:latest;
        sudo docker image tag $DOCKERACCOUNT/maggot-db $DB_IMAGE
        sudo docker pull $DOCKERACCOUNT/maggot-scan:latest;
        sudo docker image tag $DOCKERACCOUNT/maggot-scan $SCAN_IMAGE
        sudo docker pull $DOCKERACCOUNT/maggot-web:latest;
        sudo docker image tag $DOCKERACCOUNT/maggot-web $WEB_IMAGE
        ;;
   build)
        ( cd $MYDIR/dockerdbpart; sudo docker build -t $DB_IMAGE . )
        ( cd $MYDIR/dockerscanpart; sudo docker build -t $SCAN_IMAGE . )
        ( cd $MYDIR/dockerwebpart; sudo docker build -t $WEB_IMAGE . )
        ;;
   initdb)
        VER=$(sudo docker exec -t $DB_CONTAINER mongod --version | head -1)
        echo "Mongo $VER"
        # Mongo command (ver 4.0: mongo ; ver 5.0+: mongosh)
        MONGO=mongo
        RET=$(sudo docker exec -t $DB_CONTAINER which $MONGO)
        [ -z "$RET" ] && MONGO=mongosh
        RET=$(sudo docker exec -t $DB_CONTAINER cat /proc/cpuinfo | grep avx 1>/dev/null 2>/dev/null; echo $?)
        if [ $MONGO == "mongosh" -a $RET -ne 0 ]; then
            echo "ERROR: AVX support is disabled - See https://inrae.github.io/pgd-mmdt/avx_win11_vbox/"
            exit 1
        fi
        echo -n "Init DB ... "
        sudo docker exec -t $DB_CONTAINER /bin/bash -c "cd /js; python3 ./generate_setupdb.py"
        sudo docker exec -t $DB_CONTAINER $MONGO /js/setupdb.js 1>/dev/null 2>/dev/null
        [ $? -eq 0 ] && echo OK
        [ $? -ne 0 ] && echo FAILED
        ;;
   scan)
        echo -n "Scanning ... "
        sudo docker exec -t $SCAN_CONTAINER /bin/bash -c "cd /pgd; python3 ./pgd.py"
        sudo docker exec -t $SCAN_CONTAINER /bin/bash -c "cd /pgd; python3 ./db-insert.py"
        [ $? -eq 0 ] && echo OK
        [ $? -ne 0 ] && echo FAILED
        ;;
   passwd)
        [ $# -gt 1 ] && USER=$2
        echo "Account for $USER"
        sudo docker exec -it $WEB_CONTAINER /usr/bin/htpasswd /etc/pgd-mmdt/.htpasswd $USER
        ;;
   vol)
        sudo docker volume ls
        ;;
   reset)
        sudo docker volume rm $MONGO_VOL
        ;;
   delimages)
        sudo docker rmi -f $(sudo docker images | grep $WEB_IMAGE | sed -e "s/  \+/@/g" | cut -d'@' -f3)
        sudo docker rmi -f $(sudo docker images | grep $SCAN_IMAGE | sed -e "s/  \+/@/g" | cut -d'@' -f3)
        sudo docker rmi -f $(sudo docker images | grep $DB_IMAGE | sed -e "s/  \+/@/g" | cut -d'@' -f3)
        ;;
   *) usage
      exit 2
esac

exit 0