#!/bin/bash
set -eu

usage() {
    printf 'usage: build_static_site.sh --target TARGET\n'
    printf '                            [--project PROJECT [...]]\n'
    printf '                            [--project-file PROJECT_FILE [...]]\n'
    printf '                            [--weights WEIGHTS_FILE]\n'
    printf '                            [--cname DOMAIN]\n'
    printf '                            [--no-backend]\n'
    printf '                            [-h|--help]\n'
    printf '\n'
    printf 'Build the static SourceCred website, including example data.\n'
    printf '\n'
    printf '%s\n' '--target TARGET'
    printf '\t%s\n' 'an empty directory into which to build the site'
    printf '%s\n' '--project PROJECT'
    printf '\t%s\n' 'a project spec; see help for cli/load.js for details'
    printf '%s\n' '--project-file PROJECT_FILE'
    printf '\t%s\n' 'the path to a file containing a project config'
    printf '%s\n' '--weights WEIGHTS_FILE'
    printf '\t%s\n' 'path to a json file which contains a weights configuration.'
    printf '\t%s\n' 'This will be used instead of the default weights and persisted.'
    printf '%s\n' '--cname DOMAIN'
    printf '\t%s\n' 'configure DNS for a GitHub Pages site to point to'
    printf '\t%s\n' 'the provided custom domain'
    printf '%s\n' '--no-backend'
    printf '\t%s\n' 'do not run "yarn backend"; see also the SOURCECRED_BIN'
    printf '\t%s\n' 'environment variable'
    printf '%s\n' '-h|--help'
    printf '\t%s\n' 'show this message'
    printf '\n'
    printf 'Environment variables:\n'
    printf '\n'
    printf '%s\n' 'SOURCECRED_BIN'
    printf '\t%s\n' 'When using --no-backend, directory containing the'
    printf '\t%s\n' 'SourceCred executables (output of "yarn backend").'
    printf '\t%s\n' 'Default is ./bin. Ignored without --no-backend.'
}

main() {
    parse_args "$@"

    toplevel="$(git -C "$(dirname "$0")" rev-parse --show-toplevel)"
    cd "${toplevel}"

    sourcecred_data=
    sourcecred_bin=
    trap cleanup EXIT

    build
}

parse_args() {
    BACKEND=1
    target=
    cname=
    weights=
    repos=( )
    projects=( )
    project_files=( )
    while [ $# -gt 0 ]; do
        case "$1" in
            --target)
                if [ -n "${target}" ]; then
                    die '--target specified multiple times'
                fi
                shift
                if [ $# -eq 0 ]; then die 'missing value for --target'; fi
                target="$1"
                ;;
            --weights)
                if [ -n "${weights}" ]; then
                    die '--weights specified multiple times'
                fi
                shift
                if [ $# -eq 0 ]; then die 'missing value for --weights'; fi
                weights="$1"
                ;;
            --project)
                shift
                if [ $# -eq 0 ]; then die 'missing value for --project'; fi
                projects+=( "$1" )
                ;;
            --project-file)
                shift
                if [ $# -eq 0 ]; then die 'missing value for --project-file'; fi
                project_files+=( "$1" )
                ;;
            --cname)
                shift
                if [ $# -eq 0 ]; then die 'missing value for --cname'; fi
                if [ -n "${cname}" ]; then
                    die '--cname specified multiple times'
                fi
                cname="$1"
                if [ -z "${cname}" ]; then
                    die 'empty value for --cname'
                fi
                ;;
            --no-backend)
                BACKEND=0
                ;;
            -h|--help)
                usage
                exit 0
                ;;
            *)
                printf >&2 'fatal: unknown argument: %s\n' "$1"
                exit 1
                ;;
        esac
        shift
    done
    if [ -z "${target}" ]; then
        die 'target directory not specified'
    fi
    if ! [ -e "${target}" ]; then
        mkdir -p -- "${target}"
    fi
    if ! [ -d "${target}" ]; then
        die "target is not a directory: ${target}"
    fi
    if [ "$(command ls -A "${target}" | wc -l)" != 0 ]; then
        die "target directory is nonempty: ${target}"
    fi
    target="$(readlink -e "${target}")"
    : "${SOURCECRED_BIN:=./bin}"
}

build() {
    sourcecred_data="$(mktemp -d --suffix ".sourcecred-data")"

    if [ -n "${SOURCECRED_DIRECTORY:-}" ]; then
        # If $SOURCECRED_DIRECTORY is available, then give sourcecred access to
        # the cache. This will greatly speed up site builds on repos that have
        # already been loaded.
        # Note this speedup will only apply if the SOURCECRED_DIRECTORY has been
        # explicitly set.
        ln -s "${SOURCECRED_DIRECTORY}/cache" "${sourcecred_data}/cache"
    fi

    export SOURCECRED_DIRECTORY="${sourcecred_data}"

    if [ "${BACKEND}" -ne 0 ]; then
        sourcecred_bin="$(mktemp -d --suffix ".sourcecred-bin")"
        export SOURCECRED_BIN="${sourcecred_bin}"
        yarn
        yarn -s backend --output-path "${SOURCECRED_BIN}"
    fi

    if [ "${#projects[@]}" -ne 0 ]; then
        local weightsStr=""
        if [ -n "${weights}" ]; then
            weightsStr="--weights ${weights}"
        fi
        for project in "${projects[@]}"; do
            NODE_PATH="./node_modules${NODE_PATH:+:${NODE_PATH}}" \
                node "${SOURCECRED_BIN:-./bin}/sourcecred.js" load "${project}" $weightsStr
        done
    fi

    if [ "${#project_files[@]}" -ne 0 ]; then
        local weightsStr=""
        if [ -n "${weights}" ]; then
            weightsStr="--weights ${weights}"
        fi
        for project_file in "${project_files[@]}"; do
            NODE_PATH="./node_modules${NODE_PATH:+:${NODE_PATH}}" \
                node "${SOURCECRED_BIN:-./bin}/sourcecred.js" load --project "${project_file}" $weightsStr
        done
    fi

    yarn -s build --output-path "${target}"

    # Copy the SourceCred data into the appropriate API route. Using
    # `mkdir` here will fail in the case where an `api/` folder exists,
    # which is the correct behavior. (In this case, our site's
    # architecture conflicts with the required static structure, and we
    # must fail.)
    mkdir "${target}/api/"
    mkdir "${target}/api/v1/"
    # Eliminate the cache, which is only an intermediate target used to
    # load the actual data. The development server similarly forbids
    # access to the cache so that the dev and prod environments have the
    # same semantics.
    rm -rf "${sourcecred_data}/cache"
    cp -r "${sourcecred_data}" "${target}/api/v1/data"

    if [ -n "${cname:-}" ]; then
        cname_file="${target}/CNAME"
        if [ -e "${cname_file}" ]; then
            die 'CNAME file exists in static site output'
        fi
        printf '%s' "${cname}" >"${cname_file}"  # no newline
    fi
}

cleanup() {
    if [ -d "${sourcecred_data:-}" ]; then rm -rf "${sourcecred_data}"; fi
    if [ -d "${sourcecred_bin:-}" ]; then rm -rf "${sourcecred_bin}"; fi
}

die() {
    printf >&2 'fatal: %s\n' "$@"
    exit 1
}

main "$@"