#!/bin/sh
#
# Usage: set_tx_power.sh <value 0-10> [direction] [--mcs <0-7>]
#
# This script:
#   1. Checks if TX power throttling is enabled.
#   2. Retrieves the current mcs_index from wfb_tx_cmd (or uses provided --mcs value).
#   3. Uses yaml-cli to get the TX power table for that mcs_index.
#   4. Interpolates the provided value (0–10) to select one of the table entries.
#   5. Checks the wifi adapter type and, if it is "88XXau", prepends a negative sign to the TX power.
#   6. Sets the TX power via iw.
#

# --- Argument Parsing ---
TX_INPUT=""
DIRECTION=""
MCS_OVERRIDE=""

while [ "$#" -gt 0 ]; do
    case "$1" in
        --mcs)
            shift
            if [ "$#" -gt 0 ]; then
                MCS_OVERRIDE="$1"
            else
                echo "Error: --mcs requires a value."
                exit 1
            fi
            ;;
        *)
            if [ -z "$TX_INPUT" ]; then
                TX_INPUT="$1"
            elif [ -z "$DIRECTION" ]; then
                DIRECTION="$1"
            else
                echo "Unknown argument: $1"
                exit 1
            fi
            ;;
    esac
    shift
done

if [ -z "$TX_INPUT" ]; then
    echo "Usage: $0 <value 0-10> [direction] [--mcs <0-7>]"
    exit 1
fi

if [ -z "$DIRECTION" ]; then
    DIRECTION="initiated"
fi

# --- Throttling Check ---
# For now, throttling is set to "disabled". (Update this logic as needed.)
TEMP_THROTTLE="disabled"
if [ "$TEMP_THROTTLE" = "enabled" ]; then
    echo "TX power throttling is enabled. Aborting TX power change."
    exit 0
fi

# --- Determine mcs_index ---
if [ -n "$MCS_OVERRIDE" ]; then
    # Sanity check: ensure MCS_OVERRIDE is between 0 and 7, else default to 7.
    if [ "$MCS_OVERRIDE" -ge 0 ] 2>/dev/null && [ "$MCS_OVERRIDE" -le 7 ] 2>/dev/null; then
        MCS_INDEX="$MCS_OVERRIDE"
    else
        MCS_INDEX=7
    fi
else
    # --- Get current mcs_index from wfb_tx_cmd ---
    MCS_INDEX=$(wfb_tx_cmd 8000 get_radio | grep '^mcs_index=' | cut -d '=' -f2)
    if [ -z "$MCS_INDEX" ]; then
        echo "Error: Unable to retrieve mcs_index."
        exit 1
    fi
fi

# --- Retrieve TX power table from YAML ---
# Expected output: e.g., mcs1: [1,5,10,15,20,25,30,35,40,45,50]
TABLE_OUTPUT=$(yaml-cli -i /etc/vtx_info.yaml -g ".wifi.tx_power.mcs${MCS_INDEX}")
# Remove the "mcsX:" prefix and the surrounding brackets.
POWER_LIST=$(echo "$TABLE_OUTPUT" | sed -e "s/^mcs${MCS_INDEX}:[[:space:]]*//" -e 's/[][]//g')

# --- Parse list into individual values ---
OLD_IFS="$IFS"
IFS=,
set -- $POWER_LIST
IFS="$OLD_IFS"
NUM_VALUES=$#

# --- Interpolate to select TX power ---
# Map TX_INPUT (0–10) to an index in the list.
DESIRED_INDEX=$(awk -v inVal="$TX_INPUT" -v n="$NUM_VALUES" 'BEGIN { print int(inVal/10*(n-1)+0.5) }')
ARRAY_INDEX=$((DESIRED_INDEX + 1))

# Retrieve the selected TX power value.
eval "MCS_TX_PWR=\${$ARRAY_INDEX}"
if [ -z "$MCS_TX_PWR" ]; then
    echo "Error: Could not determine TX power from table."
    exit 1
fi

# --- Parameters for TX power multiplier ---
DEFAULT_MULT=50   # Used for non-88XXau adapters
ALT_MULT=100      # Used for 88XXau adapter (TX power will be negative)

# --- Check wifi adapter type ---
WIFI_ADAPTER=$(yaml-cli -i /etc/vtx_info.yaml -g .wifi.wifi_adapter)
if [ "$WIFI_ADAPTER" = "88XXau" ]; then
    MULTIPLIER=$ALT_MULT
    # Force MCS_TX_PWR to be negative if not already.
    case "$MCS_TX_PWR" in
        -*)
            # Already negative.
            ;;
        *)
            MCS_TX_PWR="-${MCS_TX_PWR}"
            ;;
    esac
else
    MULTIPLIER=$DEFAULT_MULT
fi

# --- Set the TX power ---
# Multiplication in shell arithmetic handles negatives correctly.
TXPOWER=$(( MCS_TX_PWR * MULTIPLIER ))
echo "Setting TX power to $MCS_TX_PWR ($TXPOWER) (mcs${MCS_INDEX}, adapter: $WIFI_ADAPTER, direction: $DIRECTION)"
iw wlan0 set txpower fixed "$TXPOWER"

exit 0