#!/bin/bash

DO_JSON=0
MY_PATH=$(dirname $0)
DRIVE_CONFIG_FILE="unknown"
DRIVE_NAME="unknown"
DRIVE_PURPOSE="unknown"

write_config()
{
    D_KEY="DP_$(dev_name_to_underscores ${1})"
    D_VAL="${2}"
    DRIVE_CONFIG_FILE=$(get_drive_config_file)

    if [ -f ${DRIVE_CONFIG_FILE} ]; then
        sed -i "\|${D_KEY}|d" ${DRIVE_CONFIG_FILE}
    fi

    SQUELCH=$(echo "${D_KEY}=\"${D_VAL}\"" | tee -a ${DRIVE_CONFIG_FILE})
}

set_drive_config()
{
    DRIVE_NAME="${1}"
    DRIVE_PURPOSE="${2}"

    write_config "${DRIVE_NAME}" "${DRIVE_PURPOSE}"
}

output_results_json()
{
    echo "{"
    echo "  \"drive_name\": \"${DRIVE_NAME}\"",
    echo "  \"drive_purpose\": \"${DRIVE_PURPOSE}\""
    echo "}"
}

output_results()
{
    do_json=${1}

    if [ ${do_json} -eq 1 ]; then
        output_results_json
    else
        echo "drive_name:${DRIVE_NAME}"
        echo "drive_purpose:${DRIVE_PURPOSE}"
    fi
}

##### MAIN #####
source ${MY_PATH}/general_funcs.bash
source ${MY_PATH}/device_funcs.bash
must_run_as_root

while getopts "jd:p:" opt; do
    case ${opt} in
        p)
            MY_PURP=${OPTARG}
        ;;
        d)
            MY_DEV=$(prepend_dev ${OPTARG})
        ;;
        j)
            DO_JSON=1
        ;;
    esac
done

if [[ ! -z ${MY_DEV} && ! -z ${MY_PURP} ]]; then
    set_drive_config "${MY_DEV}" "${MY_PURP}"
fi
output_results ${DO_JSON}

exit 0