bookwyrm/bw-dev

#!/bin/bash

# exit on errors
set -e

# import our ENV variables
# catch exits and give a friendly error message
function showerr {
    echo "Failed to load configuration! You may need to update your .env and quote values with special characters in them."
}
trap showerr EXIT
source .env
trap - EXIT

# show commands as they're executed
set -x
function runweb {
    docker-compose run --rm web "$@"
}

function rundb {
    docker-compose run --rm db $@
}

function execweb {
    docker-compose exec web "$@"
}

function initdb {
    runweb python manage.py initdb "$@"
}

function migrate {
    runweb python manage.py migrate "$@"
}

function admin_code {
    runweb python manage.py admin_code
}

function awscommand {
    # expose env vars
    export AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID}
    export AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY}
    export AWS_DEFAULT_REGION=${AWS_S3_REGION_NAME}
    # first arg is mountpoint, second is the whole aws command
    docker run --rm -it -v $1\
        -e AWS_ACCESS_KEY_ID -e AWS_SECRET_ACCESS_KEY -e AWS_DEFAULT_REGION\
        amazon/aws-cli $2
}

CMD=$1
if [ -n "$CMD" ]; then
    shift
fi

# show commands as they're executed
set -x

case "$CMD" in
    up)
        docker-compose up --build "$@"
        ;;
    initdb)
        initdb "@"
        ;;
    migrate)
        migrate "$@"
        ;;
    bash)
        runweb bash
        ;;
    shell)
        runweb python manage.py shell
        ;;
    dbshell)
        rundb psql -U ${POSTGRES_USER} ${POSTGRES_DB}
        ;;
    restart_celery)
        docker-compose restart celery_worker
        ;;
    collectstatic)
        runweb python manage.py collectstatic --no-input
        ;;
    build)
        docker-compose build
        ;;
    clean)
        clean
        ;;
    black)
        docker-compose run --rm dev-tools black celerywyrm bookwyrm
        ;;
    prettier)
        docker-compose run --rm dev-tools npx prettier --write bookwyrm/static/js/*.js
        ;;
    stylelint)
        docker-compose run --rm dev-tools npx stylelint \
            bookwyrm/static/css/bookwyrm.scss bookwyrm/static/css/bookwyrm/**/*.scss --fix \
            --config dev-tools/.stylelintrc.js
        ;;
    formatters)
        docker-compose run --rm dev-tools black celerywyrm bookwyrm
        docker-compose run --rm dev-tools npx prettier --write bookwyrm/static/js/*.js
        docker-compose run --rm dev-tools npx stylelint \
            bookwyrm/static/css/bookwyrm.scss bookwyrm/static/css/bookwyrm/**/*.scss --fix \
            --config dev-tools/.stylelintrc.js
        ;;
    compilescss)
        runweb python manage.py compilescss
        runweb python manage.py collectstatic --no-input
        ;;
    collectstatic_watch)
        npm run --prefix dev-tools watch:static
        ;;
    update)
        git pull
        docker-compose build
        ./update.sh
        runweb python manage.py migrate
        runweb python manage.py collectstatic --no-input
        docker-compose up -d
        docker-compose restart web
        docker-compose restart celery_worker
        ;;
    populate_streams)
        runweb python manage.py populate_streams "$@"
        ;;
    populate_lists_streams)
        runweb python manage.py populate_lists_streams $@
        ;;
    populate_suggestions)
        runweb python manage.py populate_suggestions
        ;;
    generate_thumbnails)
        runweb python manage.py generateimages
        ;;
    generate_preview_images)
        runweb python manage.py generate_preview_images "$@"
        ;;
    copy_media_to_s3)
        awscommand "bookwyrm_media_volume:/images"\
            "s3 cp /images s3://${AWS_STORAGE_BUCKET_NAME}/images\
            --endpoint-url ${AWS_S3_ENDPOINT_URL}\
            --recursive --acl public-read" "$@"
        ;;
    sync_media_to_s3)
        awscommand "bookwyrm_media_volume:/images"\
            "s3 sync /images s3://${AWS_STORAGE_BUCKET_NAME}/images\
            --endpoint-url ${AWS_S3_ENDPOINT_URL}\
            --acl public-read" "$@"
        ;;
    set_cors_to_s3)
        set +x
        config_file=$1
        if [ -z "$config_file" ]; then
            echo "This command requires a JSON file containing a CORS configuration as an argument"
            exit 1
        fi
        set -x
        awscommand "$(pwd):/bw"\
            "s3api put-bucket-cors\
            --bucket ${AWS_STORAGE_BUCKET_NAME}\
            --endpoint-url ${AWS_S3_ENDPOINT_URL}\
            --cors-configuration file:///bw/$config_file" "$@"
        ;;
    admin_code)
        admin_code
        ;;
    setup)
        migrate
        migrate django_celery_beat
        initdb
        runweb python manage.py collectstatic --no-input
        admin_code
        ;;
    runweb)
        runweb "$@"
        ;;
    *)
        set +x # No need to echo echo
        echo "Unrecognised command. Try:"
        echo "    setup"
        echo "    up [container]"
        echo "    initdb"
        echo "    makemigrations [migration]"
        echo "    migrate [migration]"
        echo "    bash"
        echo "    shell"
        echo "    dbshell"
        echo "    restart_celery"
        echo "    collectstatic"
        echo "    build"
        echo "    clean"
        echo "    populate_streams [--stream=<stream name>]"
        echo "    populate_lists_streams"
        echo "    populate_suggestions"
        echo "    generate_thumbnails"
        echo "    generate_preview_images [--all]"
        echo "    copy_media_to_s3"
        echo "    sync_media_to_s3"
        echo "    set_cors_to_s3 [cors file]"
        echo "    runweb [command]"
        ;;
esac