commands

#!/bin/bash
set -e;

# Check if name is specified
if [[ $1 == mariadb:* ]]; then
    if [[ -z $2 ]]; then
        echo "You must specify an app name"
        exit 1
    else
        APP="$2"
        # Check if app exists with the same name
        if [[ -d "$DOKKU_ROOT/$APP" ]]; then
            APP_EXISTS=true
        else
            APP_EXISTS=false
        fi
    fi
fi

case "$1" in

  mariadb:create)
    DB_IMAGE=mariadb/$APP
    # Check if DB container is installed
    IMAGE=$(docker images | grep "kloadut/mariadb " |  awk '{print $3}')
    if [[ -z $IMAGE ]]; then
        echo "MariaDB image not found... Did you run 'dokku plugins-install' ?"
        exit 1
    fi
    # Check if an existing DB volume exists
    if [[ -d "/var/lib/docker/volumes/mariadb-$APP" ]]; then
        VOLUME="/var/lib/docker/volumes/mariadb-$APP/layer:/opt/mysql"
        echo
        echo "-----> Reusing mariadb/$APP database"
    else
        VOLUME="/opt/mysql"
        # Generate a random password for DB user
        DB_PASSWORD=$(< /dev/urandom tr -dc A-Za-z0-9 | head -c 16)
        # Temporarily store it to a file
        if [[ ! -d $DOKKU_ROOT/.mariadb ]]; then
            mkdir -p $DOKKU_ROOT/.mariadb
            chown -R dokku: $DOKKU_ROOT/.mariadb
        fi
        echo $DB_PASSWORD > "$DOKKU_ROOT/.mariadb/pwd_$APP"
        chown dokku: "$DOKKU_ROOT/.mariadb/pwd_$APP"
    fi
    # Stop existing container with the same persistent DB
    ID=$(docker ps | grep "$DB_IMAGE":latest |  awk '{print $1}')
    if [[ ! -z "$ID" ]]; then
        docker stop $ID > /dev/null
    fi
    # Fork DB image
    ID=$(docker run -d kloadut/mariadb exit 0)
    docker wait $ID > /dev/null
    IMAGE=$(docker commit $ID)
    docker tag $IMAGE $DB_IMAGE
    # Launch container
    ID=$(docker run -v $VOLUME -p 3306 -d $DB_IMAGE /usr/bin/start_mariadb.sh $DB_PASSWORD)
    sleep 4
    # Rename persistent volume
    if [[ ! -d "/var/lib/docker/volumes/mariadb-$APP" ]]; then
        VOLUME_PATH=$(docker inspect $ID | grep "/var/lib/docker/volumes/" | awk '{print $2}' | sed -e"s/\/layer//" | sed -e's/"//g')
        mv $VOLUME_PATH "/var/lib/docker/volumes/mariadb-$APP"
        sleep 1
    fi
    # Link to a potential existing app
    dokku mariadb:link $APP $APP
    echo
    echo "-----> MariaDB container created: $DB_IMAGE"
    sleep 1
    dokku mariadb:info $APP
    ;;

  mariadb:delete)
    DB_IMAGE=mariadb/$APP
    # Stop the container
    ID=$(docker ps -a | grep "$DB_IMAGE":latest |  awk '{print $1}')
    if [[ ! -z $ID ]]; then
        docker stop $ID
    fi
    # Remove image
    IMAGE=$(docker images | grep "$DB_IMAGE " |  awk '{print $1}')
    if [[ ! -z $IMAGE ]]; then
        docker rmi $IMAGE
    fi
    # Remove persistent volume
    if [[ -d "/var/lib/docker/volumes/mariadb-$APP" ]]; then
        rm -rf "/var/lib/docker/volumes/mariadb-$APP"
    fi
    echo
    echo "-----> MariaDB container deleted: $DB_IMAGE"
    ;;

  mariadb:info)
    DB_IMAGE=mariadb/$APP
    ID=$(docker ps -a | grep "$DB_IMAGE":latest |  awk '{print $1}')
    IP=$(docker inspect $ID | grep IPAddress | awk '{ print $2 }' | tr -d ',"')
    PORT=$(docker port $ID 3306)
    echo
    echo "       Host: $IP"
    echo "       User: 'root'"
    if [[ -f "$DOKKU_ROOT/.mariadb/pwd_$APP" ]]; then
        echo "       Password: '$(cat "$DOKKU_ROOT/.mariadb/pwd_$APP")'"
    fi
    echo "       Database: 'db'"
    echo "       Public port: $PORT"
    echo
    ;;

  mariadb:link)
    if $APP_EXISTS; then
        # Check argument
        if [[ -z $3 ]]; then
            echo "You must specify a database name"
            exit 1
        fi
        DB_IMAGE="mariadb/$3"
        # Check temporarily file
        if [[ ! -f "$DOKKU_ROOT/.mariadb/pwd_$APP" ]]; then
            echo "Database is already linked"
            exit 1
        fi
        DB_PASSWORD=$(cat "$DOKKU_ROOT/.mariadb/pwd_$APP")
        ID=$(docker ps -a | grep "$DB_IMAGE":latest |  awk '{print $1}')
        IP=$(docker inspect $ID | grep IPAddress | awk '{ print $2 }' | tr -d ',"')
        # Write DATABASE_URL to app's ENV file
        URL="DATABASE_URL=mysql://root:$DB_PASSWORD@$IP/db"
        cat "$DOKKU_ROOT/$APP/ENV" | grep "$URL" || echo "export $URL" >> "$DOKKU_ROOT/$APP/ENV"
        echo
        echo "-----> $APP linked to $DB_IMAGE database"
        rm "$DOKKU_ROOT/.mariadb/pwd_$APP"
    fi
    ;;

  mariadb:logs)
    DB_IMAGE=mariadb/$APP
    ID=$(docker ps -a | grep "$DB_IMAGE" |  awk '{print $1}')
    docker logs $ID | tail -n 100
    ;;

  help)
    cat && cat<<EOF
    mariadb:create <app>     Create a MariaDB container
    mariadb:delete <app>     Delete specified MariaDB container
    mariadb:info <app>       Display database informations
    mariadb:link <app> <db>  Link an app to a MariaDB database
    mariadb:logs <app>       Display last logs from MariaDB container
EOF
    ;;

esac
cat