#!/bin/bash
#
# MQTT Discovery for Parallel Inverters (simple style)

MQTT_SERVER=`cat /etc/inverter/mqtt.json | jq '.server' -r`
MQTT_PORT=`cat /etc/inverter/mqtt.json | jq '.port' -r`
MQTT_TOPIC=`cat /etc/inverter/mqtt.json | jq '.topic' -r`
MQTT_DEVICENAME=`cat /etc/inverter/mqtt.json | jq '.devicename' -r`
MQTT_USERNAME=`cat /etc/inverter/mqtt.json | jq '.username' -r`
MQTT_PASSWORD=`cat /etc/inverter/mqtt.json | jq '.password' -r`
MQTT_CLIENTID=`cat /etc/inverter/mqtt.json | jq '.clientid' -r`

registerTopic () {
    # $1 = inverter_id, $2 = metric, $3 = unit, $4 = icon
    local inv_id=$1
    local device_name="${MQTT_DEVICENAME}_inv${inv_id}"
    
    mosquitto_pub \
        -h $MQTT_SERVER \
        -p $MQTT_PORT \
        -u "$MQTT_USERNAME" \
        -P "$MQTT_PASSWORD" \
        -i $MQTT_CLIENTID \
        -t "$MQTT_TOPIC/sensor/${device_name}_$2/config" \
        -m "{
            \"name\": \"${device_name}_$2\",
            \"unit_of_measurement\": \"$3\",
            \"state_topic\": \"$MQTT_TOPIC/sensor/${device_name}_$2\",
            \"icon\": \"mdi:$4\"
        }"
}

registerInverterRawCMD () {
    local inv_id=$1
    local device_name="${MQTT_DEVICENAME}_inv${inv_id}"
    
    mosquitto_pub \
        -h $MQTT_SERVER \
        -p $MQTT_PORT \
        -u "$MQTT_USERNAME" \
        -P "$MQTT_PASSWORD" \
        -i $MQTT_CLIENTID \
        -t "$MQTT_TOPIC/sensor/$device_name/config" \
        -m "{
            \"name\": \"$device_name\",
            \"state_topic\": \"$MQTT_TOPIC/sensor/$device_name\"
        }"
}

# Discover parallel inverters (no timeout, always try)
PARALLEL_DISCOVERY=`/opt/inverter-cli/bin/inverter_poller -p 2>&1`
PARALLEL_COUNT=`echo "$PARALLEL_DISCOVERY" | grep "PARALLEL_COUNT=" | cut -d= -f2`

echo "Parallel discovery result: count=$PARALLEL_COUNT"

# If no parallel inverters found, assume 2 as default
if [ -z "$PARALLEL_COUNT" ] || [ "$PARALLEL_COUNT" -eq 0 ]; then
    echo "⚠ No parallel count detected, assuming 2 inverters"
    PARALLEL_COUNT=2
fi

echo "Registering discovery for $PARALLEL_COUNT inverters"

# Register topics for each inverter discovered
for inv_id in $(seq 1 $PARALLEL_COUNT); do
    echo "Registering discovery topics for inverter #$inv_id..."
    
    registerTopic $inv_id "serial" "" "identifier"
    registerTopic $inv_id "Inverter_mode" "" "solar-power"
    registerTopic $inv_id "AC_grid_voltage" "V" "power-plug"
    registerTopic $inv_id "AC_grid_frequency" "Hz" "current-ac"
    registerTopic $inv_id "AC_out_voltage" "V" "power-plug"
    registerTopic $inv_id "AC_out_frequency" "Hz" "current-ac"
    registerTopic $inv_id "PV_in_voltage" "V" "solar-panel-large"
    registerTopic $inv_id "PV_in_current" "A" "solar-panel-large"
    registerTopic $inv_id "PV_in_watts" "W" "solar-panel-large"
    registerTopic $inv_id "PV_in_watthour" "Wh" "solar-panel-large"
    registerTopic $inv_id "SCC_voltage" "V" "current-dc"
    registerTopic $inv_id "Load_pct" "%" "brightness-percent"
    registerTopic $inv_id "Load_watt" "W" "chart-bell-curve"
    registerTopic $inv_id "Load_watthour" "Wh" "chart-bell-curve"
    registerTopic $inv_id "Load_va" "VA" "chart-bell-curve"
    registerTopic $inv_id "Bus_voltage" "V" "details"
    registerTopic $inv_id "Heatsink_temperature" "°C" "details"
    registerTopic $inv_id "Battery_capacity" "%" "battery-outline"
    registerTopic $inv_id "Battery_voltage" "V" "battery-outline"
    registerTopic $inv_id "Battery_charge_current" "A" "current-dc"
    registerTopic $inv_id "Battery_discharge_current" "A" "current-dc"
    registerTopic $inv_id "Load_status_on" "" "power"
    registerTopic $inv_id "SCC_charge_on" "" "power"
    registerTopic $inv_id "AC_charge_on" "" "power"
    registerTopic $inv_id "Battery_recharge_voltage" "V" "current-dc"
    registerTopic $inv_id "Battery_under_voltage" "V" "current-dc"
    registerTopic $inv_id "Battery_bulk_voltage" "V" "current-dc"
    registerTopic $inv_id "Battery_float_voltage" "V" "current-dc"
    registerTopic $inv_id "Max_grid_charge_current" "A" "current-ac"
    registerTopic $inv_id "Max_charge_current" "A" "current-ac"
    registerTopic $inv_id "Out_source_priority" "" "grid"
    registerTopic $inv_id "Charger_source_priority" "" "solar-power"
    registerTopic $inv_id "Battery_redischarge_voltage" "V" "battery-negative"
    
    # Raw command topic
    registerInverterRawCMD $inv_id
done

echo "✓ All discovery topics registered for $PARALLEL_COUNT inverters"