compose.sh

#!/bin/bash

set -Eeo pipefail

parent_path=$(
    cd "$(dirname "${BASH_SOURCE[0]}")"
    pwd -P
)
cd "$parent_path"

COMPOSECMD="docker compose -f docker-compose.yml"

mkdocs() {
    (
        DOCKER_BUILDKIT=1 docker build --file api/Dockerfile.openapi --output api/internal/server/docs api
        cd report && npm run generate >/dev/null
    )
}

if [ "$1" = "gen" ]; then
    docker run --rm -v $(pwd):/src -w /src sqlc/sqlc generate

elif [ "$1" = "watch" ]; then
    mkdocs -q
    if [ "$2" = "mock" ]; then
        DOCKER_BUILDKIT=1 $COMPOSECMD --profile=mock up --watch
    else
        DOCKER_BUILDKIT=1 $COMPOSECMD up --watch
    fi

elif [ "$1" = "up" ]; then
    mkdocs -q
    if [ "$2" = "mock" ]; then
        DOCKER_BUILDKIT=1 $COMPOSECMD --profile=mock up --remove-orphans -d --build
    else
        DOCKER_BUILDKIT=1 $COMPOSECMD up --remove-orphans -d --build
    fi

elif [ "$1" = "build" ]; then
    if [ "$2" = "local" ] || [ "$2" = "develop" ] || [ "$2" = "test" ] || [ "$2" = "prod" ]; then
        SCRATCH_BASE_IMAGE=scratch
        ALPINE_BASE_IMAGE=alpine:3.19
        FLYWAY_BASE_IMAGE=flyway/flyway:10.9.1

        if [ "$2" = "test" ] || [ "$2" = "prod" ]; then
            SCRATCH_BASE_IMAGE=registry1.dso.mil/ironbank/docker/scratch:ironbank
            ALPINE_BASE_IMAGE=registry1.dso.mil/ironbank/opensource/alpinelinux/alpine:3.19.1
            FLYWAY_BASE_IMAGE=registry1.dso.mil/ironbank/flyway/flyway-docker:v10.9.1
            AMD64_TARGET_PLATFORM=true
        fi

        for BUILD_TARGET in midas-api midas-sql midas-telemetry midas-alert midas-dcs-loader; do
            docker build \
                ${AMD64_TARGET_PLATFORM:+--platform=linux/amd64} \
                --build-arg="BASE_IMAGE=${SCRATCH_BASE_IMAGE}" \
                --build-arg="GO_VERSION=1.23" \
                --build-arg="BUILD_TAG=$2" \
                --build-arg="BUILD_TARGET=${BUILD_TARGET}" \
                -t $BUILD_TARGET:"$2" api
        done

        docker build \
            --build-arg="BASE_IMAGE=${ALPINE_BASE_IMAGE}" \
            -t midas-report:$2 report
    else
        echo -e "usage:\n\t./compose.sh build [local,develop,test,prod]"
        exit 1
    fi

    if [ "$3" = "push" ]; then
        if [ -z "$4" ]; then
            echo -e "usage:\n\t./compose.sh build [local,develop,test,prod] push <image_registry>"
            exit 1
        fi

        declare -a REGISTRIES=("midas-api" "midas-telemetry" "midas-alert" "midas-dcs-loader" "midas-sql")

        # tag
        for IMAGE in "${REGISTRIES[@]}"; do
            docker tag $IMAGE:"$2" $4/$IMAGE:"$2"
        done
        if [ "$2" = "develop" ]; then
            docker tag midas-report:"$2" $4/midas-report:"$2"
        fi

        # push
        for IMAGE in "${REGISTRIES[@]}"; do
            docker push $4/$IMAGE:"$2"
        done
        if [ "$2" = "develop" ]; then
            docker push $4/midas-report:"$2"
        fi
    fi

elif [ "$1" = "authdbdump" ]; then
    $COMPOSECMD exec authdb pg_dump postgres >auth/initdb/init2.sql

elif [ "$1" = "down" ]; then
    mkdocs -q
    $COMPOSECMD --profile=mock down

elif [ "$1" = "clean" ]; then
    $COMPOSECMD --profile=mock down -v

elif [ "$1" = "test" ]; then
    $COMPOSECMD up --remove-orphans -d --build db elasticmq api
    $COMPOSECMD up --build migrate
    shift

    TEARDOWN=false
    REST_ARGS=()

    while [[ $# -gt 0 ]]; do
        case $1 in
        -rm)
            TEARDOWN=true
            shift
            ;;
        *)
            REST_ARGS+=("$1")
            shift
            ;;
        esac
    done

    GOCMD="go test ${REST_ARGS[@]} github.com/USACE/instrumentation-api/api/internal/handler"

    set +e
    echo -e 'running test suite...'
    if [ "$REPORT" = true ]; then
        docker compose run --remove-orphans -e INSTRUMENTATION_AUTH_JWT_MOCKED=true --entrypoint="$GOCMD" api >$(pwd)/test.log
    else
        docker compose run --remove-orphans -e INSTRUMENTATION_AUTH_JWT_MOCKED=true --entrypoint="$GOCMD" api
    fi
    set -e

    if [ $TEARDOWN = true ]; then
        docker compose --profile=mock down -v
    fi

elif [ "$1" = "mkdocs" ]; then
    mkdocs

else
    echo -e "usage:\n\t./compose.sh watch\n\t./compose.sh up\n\t./compose.sh down\n\t./compose.sh clean\n\t./compose.sh test\n\t./compose.sh mkdocs"
fi