SERVER_REQS_INSTALLED="FALSE"
SERVER_PID=""
DEFAULT_DB=""
PASSWORD=""
USERNAME=""

# wait_for_connection(<PORT>, <TIMEOUT IN MS>) attempts to connect to the sql-server at the specified
# port on localhost, using the $SQL_USER (or 'postgres' if unspecified) as the user name, and trying once
# per second until the millisecond timeout is reached. If a connection is successfully established,
# this function returns 0. If a connection was not able to be established within the timeout period,
# this function returns 1.
wait_for_connection() {
  port=$1
  timeout=$2
  end_time=$((SECONDS+($timeout/1000)))
  USERNAME="${USERNAME:=postgres}"
  PASSWORD="${PASSWORD:=password}"

  nativevar PGPASSWORD "$PASSWORD" /w
  
  while [ $SECONDS -lt $end_time ]; do
    run psql -U $USERNAME -h localhost -p $port -c "SELECT 1;" $DEFAULT_DB
    if [ $status -eq 0 ]; then
      echo "Connected successfully!"
      return 0
    else
      echo "$output"
    fi
    sleep 1
  done

  echo "Failed to connect to database $DEFAULT_DB on port $port within $timeout ms."
  return 1
}

start_sql_server() {
    DEFAULT_DB="$1"
    logFile=$2
    USERNAME=$3
    PASSWORD=$4

    if [ -n "$DEFAULT_DB" ]; then
        nativevar DOLTGRES_DB "$DEFAULT_DB" /w
    fi

    if [ -n "$PASSWORD" ]; then
        nativevar PGPASSWORD "password" /w
        nativevar DOLTGRES_PASSWORD "$PASSWORD" /w
    else
        nativevar PGPASSWORD "$PASSWORD" /w
        PASSWORD="password"
    fi

    if [ -n "$USERNAME" ]; then
        nativevar DOLTGRES_USER "$USERNAME" /w
        if [ -z "$DEFAULT_DB" ]; then
            DEFAULT_DB="$USERNAME"
        fi
    else
        USERNAME="postgres"
    fi

    DEFAULT_DB="${DEFAULT_DB:=postgres}"

    PORT=$( definePORT )
    CONFIG=$( defineCONFIG $PORT )
    echo "$CONFIG" > config.yaml
    if [[ $logFile ]]
    then
        doltgres -data-dir=. -config=config.yaml> $logFile 2>&1 &
    else
        doltgres -data-dir=. -config=config.yaml &
    fi
    SERVER_PID=$!
    wait_for_connection $PORT 7500
}

# like start_sql_server, but the second argument is a string with all arguments to doltgres. The
# port argument is handled separately: if the variable $PORT is not defined and the --port argument
# is not included in the argument list, a random port is chosen for $PORT and the argument --port is
# appended to the argument list.
start_sql_server_with_args() {
    DEFAULT_DB=""
    nativevar DEFAULT_DB "$DEFAULT_DB" /w
    nativevar PGPASSWORD "password" /w

    echo "running doltgres $@"
    doltgres "$@" &
    SERVER_PID=$!
    wait_for_connection $PORT 7500
}

# stop_sql_server stops the SQL server. For cases where it's important
# to wait for the process to exit after the kill signal (e.g. waiting
# for an async replication push), pass 1.
# kill the process if it's still running
stop_sql_server() {
    # Clean up any mysql.sock file in the default, global location
    if [ -f "/tmp/mysql.sock" ]; then rm -f /tmp/mysql.sock; fi
    if [ -f "/tmp/postgres.sock" ]; then rm -f /tmp/mysql.sock; fi
    if [ -f "/tmp/dolt.$PORT.sock" ]; then rm -f /tmp/dolt.$PORT.sock; fi

    wait=$1
    if [ ! -z "$SERVER_PID" ]; then
        # ignore failures of kill command in the case the server is already dead
        run kill $SERVER_PID
        if [ $wait ]; then
            while ps -p $SERVER_PID > /dev/null; do
                sleep .1;
            done
        fi;
    fi
    SERVER_PID=
    PORT=
}

definePORT() {
  for i in {0..99}
  do
    port=$((RANDOM % 4096 + 2048))
    # nc (netcat) returns 0 when it _can_ connect to a port (therefore in use), 1 otherwise.
    run nc -z localhost $port
    if [ "$status" -eq 1 ]; then
      echo $port
      break
    fi
  done
}

defineCONFIG() {
    PORT=$1
    cat <<EOF
    log_level: debug

    behavior:
      read_only: false
      disable_client_multi_statements: false
      dolt_transaction_commit: false

    listener:
      host: localhost
      port: $PORT
EOF
}