#!/bin/bash

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

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
}

populate_drive_info()
{
    source $(get_drive_config_file) > /dev/null 2>&1
    DRIVE_NAME="${1}"
    DEV_USCORES="DP_$(dev_name_to_underscores ${DRIVE_NAME})"
    
    if [ ! -z ${!DEV_USCORES} ]; then
        # Check if there's a user-defined purpose first...
        DRIVE_PURPOSE=${!DEV_USCORES}
        return
    else
        # ... then check if it's a boot drive...
        BOOT_DRIVES=$(${MY_PATH}/get_boot_drives.bash | cut -d':' -f2 | sed 's/,/ /g')
        for dev in ${BOOT_DRIVES}; do
            if [ "${DRIVE_NAME}" == "${dev}" ]; then
                DRIVE_PURPOSE="boot"
                return
            fi 
        done 

        # ... then check if it's a data drive.
        DATA_DRIVES=$(${MY_PATH}/get_data_drives.bash | cut -d':' -f2 | sed 's/,/ /g')
        for dev in ${DATA_DRIVES}; do
            if [ "${DRIVE_NAME}" == "${dev}" ]; then
                DRIVE_PURPOSE="data"
                return
            fi 
        done 
    fi 
}

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

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

populate_drive_info "${MY_DEV}"
output_results ${DO_JSON}

exit 0