#!/bin/bash
set -e

CONFIG_PATH=/data/options.json

ASSISTANT=$(jq --raw-output '.assistant' $CONFIG_PATH)
LANG=$(jq --raw-output '.language // "en"' $CONFIG_PATH)
CUSTOMTTS=$(jq --raw-output '.custom_tts.active' $CONFIG_PATH)
MQTT_CONFIG=

echo "[INFO] LANG: ${LANG}"

echo "[INFO] Checking for /share/snips.toml"
if [ -f "/share/snips.toml" ]; then
    echo "[INFO] - Installing /share/snips.toml"
    cp -v /share/snips.toml /etc/
fi

if [ "${CUSTOMTTS}" == "true" ]; then
    PLATFORM=$(jq --raw-output '.custom_tts.platform' $CONFIG_PATH)

    if [ -z "${PLATFORM}" ]; then
        echo "[ERROR] - platform must be set to use custom tts!"
    else
        echo "[INFO] - Using custom tts"
        (
            echo "provider = \"customtts\""
            echo "customtts = { command = [\"/usr/bin/customtts.sh\", \"${PLATFORM}\", \"%%OUTPUT_FILE%%\", \"${LANG}\", \"%%TEXT%%\"] }"
        ) >> /etc/snips.toml
    fi
else
    echo "[INFO] - Using default tts (picotts)"
fi

# Use Hass.io mqtt services
if MQTT_CONFIG="$(curl -s -f -H "X-Hassio-Key: ${HASSIO_TOKEN}" http://hassio/services/mqtt)"; then
    HOST="$(echo "${MQTT_CONFIG}" | jq --raw-output '.data.host')"
    PORT="$(echo "${MQTT_CONFIG}" | jq --raw-output '.data.port')"
    USER="$(echo "${MQTT_CONFIG}" | jq --raw-output '.data.username')"
    PASSWORD="$(echo "${MQTT_CONFIG}" | jq --raw-output '.data.password')"

    echo "[INFO] Setup Hass.io mqtt service to ${HOST}"

    (
        echo "connection main-mqtt"
        echo "address ${HOST}:${PORT}"
    ) >> /etc/mosquitto.conf

    if [ -n "${USER}" ]; then
      (
          echo "username ${USER}"
          echo "password ${PASSWORD}"
      ) >> /etc/mosquitto.conf
    fi

    (
        echo "topic hermes/intent/# out"
        echo "topic hermes/hotword/toggleOn out"
        echo "topic hermes/hotword/toggleOff out"
        echo "topic hermes/hotword/+/detected out"
        echo "topic hermes/asr/stopListening out"
        echo "topic hermes/asr/startListening out"
        echo "topic hermes/nlu/intentNotParsed out"
        echo "topic hermes/audioServer/+/playBytes/# out"
        echo "topic hermes/audioServer/+/playFinished out"
        echo "topic # IN hermes/"
    ) >> /etc/mosquitto.conf
else
    echo "[ERROR] No Hass.io mqtt service found!"
fi

echo "[INFO] Start internal mqtt broker"
mosquitto -c /etc/mosquitto.conf &


echo "[INFO] Checking for updated ${ASSISTANT} in /share"
# check if a new assistant file exists
if [ -f "/share/${ASSISTANT}" ]; then
    echo "[INFO] Install/Update snips assistant"
    rm -rf /usr/share/snips/assistant
    unzip -o -u "/share/${ASSISTANT}" -d /usr/share/snips
# otherwise use the default
else
    echo "[INFO] Checking for /assistant_Hass_${LANG}.zip"
    if [ -f "/assistant_Hass_${LANG}.zip" ]; then
        echo "[INFO] - Using default assistant_Hass_${LANG}.zip"
        rm -rf /usr/share/snips/assistant
        unzip -o -u "/assistant_Hass_${LANG}.zip" -d /usr/share/snips
    else
        echo "[ERROR] Could not find assistant!"
    fi
fi

echo "[INFO] Starting snips-watch"
( sleep 2; /usr/bin/snips-watch -vvv --no_color ) &

/usr/bin/snips-asr --version
/usr/bin/snips-injection --version
/usr/bin/snips-audio-server --version
/usr/bin/snips-dialogue --version
/usr/bin/snips-hotword --version
/usr/bin/snips-nlu --version

/snips-entrypoint.sh --mqtt localhost:1883