#!/usr/bin/env bash
# Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

# BEGIN environment bootstrap section
# Do not edit between here and END as this section should stay identical in all scripts

findpath () {
    myname=${0}
    mypath=${myname%/*}
    myname=${myname##*/}
    empty_if_start_slash=${mypath%%/*}
    if [ "${empty_if_start_slash}" ]; then
        mypath=$(pwd)/${mypath}
    fi
    if [ "$mypath" ] && [ -d "$mypath" ]; then
        return
    fi
    mypath=$(pwd)
    if [ -f "${mypath}/${myname}" ]; then
        return
    fi
    echo "FATAL: Could not figure out the path where $myname lives from $0"
    exit 1
}

COMMON_ENV=libexec/vespa/common-env.sh

source_common_env () {
    if [ "$VESPA_HOME" ] && [ -d "$VESPA_HOME" ]; then
        export VESPA_HOME
        common_env=$VESPA_HOME/$COMMON_ENV
        if [ -f "$common_env" ]; then
            . $common_env
            return
        fi
    fi
    return 1
}

findroot () {
    source_common_env && return
    if [ "$VESPA_HOME" ]; then
        echo "FATAL: bad VESPA_HOME value '$VESPA_HOME'"
        exit 1
    fi
    if [ "$ROOT" ] && [ -d "$ROOT" ]; then
        VESPA_HOME="$ROOT"
        source_common_env && return
    fi
    findpath
    while [ "$mypath" ]; do
        VESPA_HOME=${mypath}
        source_common_env && return
        mypath=${mypath%/*}
    done
    echo "FATAL: missing VESPA_HOME environment variable"
    echo "Could not locate $COMMON_ENV anywhere"
    exit 1
}

findhost () {
    if [ "${VESPA_HOSTNAME}" = "" ]; then
        VESPA_HOSTNAME=$(vespa-detect-hostname || hostname -f || hostname || echo "localhost") || exit 1
    fi
    validate="${VESPA_HOME}/bin/vespa-validate-hostname"
    if [ -f "$validate" ]; then
        "$validate" "${VESPA_HOSTNAME}" || exit 1
    fi
    export VESPA_HOSTNAME
}

findroot
findhost

ROOT=${VESPA_HOME%/}
export ROOT

# END environment bootstrap section

set -euo pipefail

if [ $# -ne 2 ]; then
  echo "Usage: $0 <id> <output-directory>"
  echo ""
  vespa-sentinel-cmd list 2>/dev/null
  exit 1
fi

readonly service=$1
readonly output_directory=$2
if ! [ -d "${output_directory}" -a -w "${output_directory}" ]; then
  echo "Directory '${output_directory}' is not writable"
  exit 1
fi
jvm_pid=$(${VESPA_HOME}/libexec/vespa/find-pid "$service")
echo "Pid for '${service}' is '${jvm_pid}'"

echo "Creating heap dump"
readonly heap_dump_file=${output_directory}/jvm-heap-dump.hprof
if test -f "${heap_dump_file}"; then
    rm "${heap_dump_file}"
fi
jmap -dump:live,format=b,file=${heap_dump_file} ${jvm_pid}

echo "Trigger vespa-malloc to dump information to Vespa log"
kill -SIGPROF ${jvm_pid}

echo "Getting jmap information"
jhsdb jmap  --heap --pid ${jvm_pid} > ${output_directory}/jmap-output.txt

echo "Getting jstat information"
jstat -gcutil ${jvm_pid} > ${output_directory}/jstat-output.txt

echo "Getting jstack information"
jstack ${jvm_pid} > ${output_directory}/jstack-output.txt

echo "Getting pmap information"
pmap -x ${jvm_pid} | sort -nk3 | tail -10 > ${output_directory}/pmap-output.txt

echo "Copying vespa logs"
cp ${VESPA_HOME}/logs/vespa/vespa.log ${output_directory}/vespa.log

echo "Done!"