#!/bin/bash

# shared variables between the different services
#
# CircleCI set values:
#   $SERVER - points to the millennium build server
#   $AWS_SERVER - points to the aws manager instance
#   $CI_DIR - home directory on build server
#   $CI_AWS_DIR - home directory on aws

##################
# SHARED FUNCTIONS
##################

# shared across build server and aws

# copy files from server to server
# $1 - src
# $2 - dest
copy () {
    rsync -avzp -e 'ssh' $1 $2
}

# run command over ssh
# $1 - server login (user@IP)
# $2 - command
run_impl () {
    ssh -o "StrictHostKeyChecking no" -t $1 $2
}

# run script over ssh
# $1 - server login (user@IP)
# $2 - local script
# $2 - arguments to the script
run_script_impl () {
    ssh -o "StrictHostKeyChecking no" -t $1 'bash -s -l' < $2 "$3"
}

# build server calls

# run command on build server
# $1 - command
run () {
    run_impl $SERVER "$@"
}

# run script on build server
# $1 - script
# $1 - arguments to the script
run_script () {
    run_script_impl $SERVER $1 $2
}

# remove the work dir on the build server
clean () {
    run "rm -rf $REMOTE_WORK_DIR"
}

# aws calls

# run command on aws server
# $1 - command
run_aws () {
    run_impl $AWS_SERVER "$@"
}

# run script on aws server
# $1 - script
# $1 - arguments to the script
run_script_aws () {
    run_script_impl $AWS_SERVER $1 $2
}

# run script on aws server but detach after running
# $1 - session name
# $2 - local script path
# $3 - script name
run_detach_script_aws () {
    # remove old script if it exists
    run_impl $AWS_SERVER "rm -rf $REMOTE_AWS_WORK_DIR/$3"
    # copy new script to run to work dir
    copy $2/$3 $AWS_SERVER:$REMOTE_AWS_WORK_DIR/
    # run script and detach
    # NOTE: for some reason without re-invoking screen after this command it breaks
    # NOTE: AND, screen -list returns false by default (even if there is a session running)
    run_impl $AWS_SERVER "screen -S CI-$1-SESSION -dm $REMOTE_AWS_WORK_DIR/$3; screen -list || true"
}

# remove the work dir on the aws server
clean_aws () {
    run_aws "rm -rf $REMOTE_AWS_WORK_DIR"
}

#############
# SHARED VARS
#############

# make parallelism
CI_MAKE_NPROC=8
# chosen based on a 24c system shared with 1 other project
REMOTE_MAKE_NPROC=4

# verilator version
VERILATOR_VERSION=v4.034

# remote variables (on build instance)
REMOTE_PREFIX=$CI_DIR/$CIRCLE_PROJECT_REPONAME-$CIRCLE_BRANCH
REMOTE_WORK_DIR=$REMOTE_PREFIX-$CIRCLE_SHA1-$CIRCLE_JOB
REMOTE_RISCV_DIR=$REMOTE_WORK_DIR/riscv-tools-install
REMOTE_ESP_DIR=$REMOTE_WORK_DIR/esp-tools-install
REMOTE_CHIPYARD_DIR=$REMOTE_WORK_DIR/chipyard
REMOTE_VERILATOR_DIR=$REMOTE_PREFIX-$CIRCLE_SHA1-verilator-install
REMOTE_SIM_DIR=$REMOTE_CHIPYARD_DIR/sims/verilator
REMOTE_JAVA_OPTS="-Xmx16G -Xss8M"
REMOTE_SBT_OPTS="-Dsbt.ivy.home=$REMOTE_WORK_DIR/.ivy2 -Dsbt.supershell=false -Dsbt.global.base=$REMOTE_WORK_DIR/.sbt -Dsbt.boot.directory=$REMOTE_WORK_DIR/.sbt/boot"
REMOTE_SPEC=$CI_DIR/../abejgonza/cpu2017-1.0.1.iso # TODO: this is temporary until a better location is found

# remote variables (on manager instance)
REMOTE_AWS_WORK_DIR=$CI_AWS_DIR/$CIRCLE_PROJECT_REPONAME-$CIRCLE_BRANCH-$CIRCLE_SHA1
REMOTE_AWS_CHIPYARD_DIR=$REMOTE_AWS_WORK_DIR/chipyard
REMOTE_AWS_FSIM_DIR=$REMOTE_AWS_CHIPYARD_DIR/sims/firesim
REMOTE_AWS_MARSHAL_DIR=$REMOTE_AWS_FSIM_DIR/sw/firesim-software
REMOTE_AWS_FSIM_DEPLOY_DIR=$REMOTE_AWS_FSIM_DIR/deploy
REMOTE_AWS_FSIM_HWDB_DIR=$REMOTE_AWS_FSIM_DEPLOY_DIR/built-hwdb-entries
REMOTE_AWS_RESULTS_DIR=$REMOTE_AWS_FSIM_DEPLOY_DIR/results-workload
SPEC_DIR=$REMOTE_AWS_WORK_DIR/spec-2017
SPEC_SRC_DIR=$REMOTE_AWS_WORK_DIR/cpu2017-1.0.1

# local variables (aka within the docker container)
LOCAL_CHECKOUT_DIR=$HOME/project
LOCAL_RISCV_DIR=$HOME/riscv-tools-install
LOCAL_ESP_DIR=$HOME/esp-tools-install
LOCAL_CHIPYARD_DIR=$HOME/chipyard
LOCAL_SIM_DIR=$LOCAL_CHIPYARD_DIR/sims/verilator
LOCAL_FSIM_CFGS_DIR=$LOCAL_CHECKOUT_DIR/.circleci/firesim-configs

# api url to do curls
API_URL=https://circleci.com/api/v2

# key value store to get the build strings
declare -A mapping
mapping["smallboom"]="CONFIG=SmallBoomConfig"
mapping["mediumboom"]="CONFIG=MediumBoomConfig"
mapping["largeboom"]="CONFIG=LargeBoomConfig"
mapping["megaboom"]="CONFIG=MegaBoomConfig"
mapping["boomandrocket"]="CONFIG=SmallBoomAndRocketConfig"
mapping["rv32boom"]="CONFIG=SmallRV32BoomConfig"
mapping["hwachaboom"]="CONFIG=HwachaLargeBoomConfig"

declare -A afis
afis["largefireboom"]="firesim-boom-singlecore-no-nic-l2-llc4mb-ddr3"