#!/bin/bash
# SPDX-License-Identifier: Apache-2.0
# Copyright (C) 2020 Fewtarius (https://github.com/fewtarius)
# Copyright (C) 2022 kkoshelev (https://github.com/kkoshelev)

. /etc/profile

if [ ! -d "" ]
then
  mkdir -p "/storage/.cache/connman"
fi

### ES won't save the configuration in time
### for it to be useable by the script, so
### we have to accept the ssid and key on the
### command line too.

if [ ! -z "${2}" ]
then
  SSID="${2}"
else
  SSID="$(get_setting wifi.ssid)" 2>/dev/null
fi

if [ ! -z "${3}" ]
then
  PSK="${3}"
else
  PSK="$(get_setting wifi.key)" 2>/dev/null
fi

WIFICFG="/storage/.cache/connman/wifi.config"

# lists all wifi services in service=ssid format
list_wifi() {
  connmanctl services | cut -b 5- | awk '/\S+.+\s+wifi/ {a=$0; sub(/\s+wifi_.*$/,"", a); b=$0; sub(a, "", b); sub(/\s+/, "", b); print b "=" a}' | sort | uniq
}

# Looksup connman service name based on ssid
# $1 - SSID to lookup
get_wifi_service() {
  list_wifi | awk -v ssid="${1}" '{ split($0, a, "="); if (a[2]==ssid) print a[1] }'
}

set_powersave() {
  ENABLED=$(get_setting system.power.wifi)
  if [ "${ENABLED}" = "1" ]
  then
    log $0 "Enabling WIFI power saving."
    iw wlan0 set power_save on 2>/dev/null
  else
    log $0 "Disabling WIFI power saving."
    iw wlan0 set power_save off 2>/dev/null
  fi
}

get_wifi_state() {
  STATE=$(connmanctl state | awk '/State/ {print $3}' 2>/dev/null)
  echo ${STATE}
}

check_wifi() {
  SSID_CHK=$(grep "${SSID}" ${WIFICFG} 2>/dev/null)
  KEY_CHK=$(grep "${PSK}" ${WIFICFG} 2>/dev/null)
  STATE=$(get_wifi_state)
  if [ -n "${SSID_CHK}" ] && \
     [ -n "${KEY_CHK}" ] && \
     [ "${STATE}" = "online" ]
  then
    exit 0
  fi
}

connect_wifi() {
  NETCHECK=$(systemctl status connman >/dev/null 2>&1)
  if [ ! "$?" = "0" ]
  then
    systemctl start connman
  fi
  STATE=$(get_wifi_state)
  if [ ! "${STATE}" = "online" ]
  then
    connmanctl enable wifi 2>/dev/null
  fi
  rfkill unblock wifi 2>/dev/null
  connmanctl connect $(get_wifi_service "${SSID}")
  set_powersave 2>/dev/null
}

set_profile() {
  cat > "${WIFICFG}" <<EOF
[service_${OS_NAME}_default]
Type = wifi
Name = ${SSID}
Passphrase = ${PSK}
EOF
}

case "${1}" in
  enable)
    check_wifi
    set_setting wifi.enabled 1
    set_profile
    connect_wifi 
  ;;
  disable)
    rfkill block wifi
    connmanctl disable wifi
    rm -f "${WIFICFG}" 2>/dev/null
    set_setting wifi.enabled 0
  ;;
  reconnect)
    /usr/bin/wifictl disable
    /usr/bin/wifictl enable
  ;;
  list)
    #connmanctl services | cut -b 5- | awk '/wifi/ {sub(/\s+wifi_.*$/,"",$0);print}' | sort | uniq
    list_wifi | awk '{sub(/\S+=/,"",$0);print}'
  ;;
  scan)
    connmanctl scan wifi 2>/dev/null
  ;;
  scanlist)
    set_wifi scan 2>/dev/null
    list_wifi | awk '{sub(/\S+=/,"",$0);print}'
  ;;
  service)
     get_wifi_service "${SSID}"
  ;;
  setpowersave)
    set_powersave
  ;;
  setprofile)
    set_profile
  ;;
esac