#!/bin/sh /etc/rc.common
# Copyright (C) 2008 OpenWrt.org

START=96

. /lib/functions.sh
. /lib/functions/service.sh

config_load wireless

wlan_2g_flag=0
wlan_5g_flag=0

led_set_attr() {
	[ -f "/sys/class/leds/$1/$2" ] && echo "$3" > "/sys/class/leds/$1/$2"
}

check_wlan_led() {
	wifi_status_24g="$(iwconfig `iwconfig 2>/dev/null|grep ath|grep -E "(11b|11g|11ng)"|head -n 1|cut -d ' ' -f 1`|grep -o -E '([[:xdigit:]]{1,2}:){5}[[:xdigit:]]{1,2}')"
	wifi_status_5g="$(iwconfig `iwconfig 2>/dev/null|grep -E "(ath|dummy|enjet)"|grep -E "(11a|11na)"|head -n 1|cut -d ' ' -f 1`|grep -o -E '([[:xdigit:]]{1,2}:){5}[[:xdigit:]]{1,2}')"
	wifi_status_5g_iface="$(iwconfig 2>/dev/null|grep -E "(ath|dummy|enjet)"|grep -E "(11a|11na)"|head -n 1|cut -d ' ' -f 1)"

	case "$1" in
		wifi0)
			[ -n "${wifi_status_24g}" ] && {
				wlan_2g_flag=1
			} || {
				# All wifi 2.4G disabled
				wlan_2g_flag=0
			}
		;;
		wifi1)
			[ -n "${wifi_status_5g}" -a "${wifi_status_5g_iface}" != "ath54" ] && {
				wlan_5g_flag=1
			} || {
				# All wifi 5G disabled
				wlan_5g_flag=0
			}
		;;
	esac
}

load_led() {
	local name
	local sysfs
	local trigger
	local dev
	local mode
	local default
	local delayon
	local delayoff
	local interval

	config_get sysfs $1 sysfs
	config_get name $1 name "$sysfs"
	config_get trigger $1 trigger "none"
	config_get dev $1 dev
	config_get mode $1 mode "link"
	config_get_bool default $1 default "nil"
	config_get delayon $1 delayon
	config_get delayoff $1 delayoff
	config_get interval $1 interval "50"
	config_get port_state $1 port_state
	config_get delay $1 delay "150"
	config_get message $1 message ""

	if [ "$trigger" = "rssi" ]; then
		# handled by rssileds userspace process
		return
	fi

	[ -e /sys/class/leds/${sysfs}/brightness ] && {
		echo "setting up led ${name}"

		[ ${sysfs} == "wlan_2g_led" ] && {
			check_wlan_led "wifi0"
			[ "$wlan_2g_flag" -eq 1 -a "$default" = 0 ] && {
				echo 1 >/sys/class/leds/${sysfs}/brightness
			} || {
				echo 0 >/sys/class/leds/${sysfs}/brightness
			}
			return;
		}
		[ ${sysfs} == "WLAN_LED" ] && {
			check_wlan_led "wifi1"
			[ "$wlan_5g_flag" -eq 1 -a "$default" = 0 ] && {
				echo 1 >/sys/class/leds/${sysfs}/brightness
			} || {
				echo 0 >/sys/class/leds/${sysfs}/brightness
			}
			return;
		}

		[ "$default" = 0 ] &&
			echo 1 >/sys/class/leds/${sysfs}/brightness

		echo $trigger > /sys/class/leds/${sysfs}/trigger 2> /dev/null
		ret="$?"

		[ $default = 1 ] &&
			echo 0 >/sys/class/leds/${sysfs}/brightness

		[ $ret = 0 ] || {
			echo >&2 "Skipping trigger '$trigger' for led '$name' due to missing kernel module"
			return 1
		}
		case "$trigger" in
		"netdev")
			[ -n "$dev" ] && {
				echo $dev > /sys/class/leds/${sysfs}/device_name
				echo $mode > /sys/class/leds/${sysfs}/mode
			}
			;;

		"timer")
			[ -n "$delayon" ] && \
				echo $delayon > /sys/class/leds/${sysfs}/delay_on
			[ -n "$delayoff" ] && \
				echo $delayoff > /sys/class/leds/${sysfs}/delay_off
			;;

		"usbdev")
			[ -n "$dev" ] && {
				echo $dev > /sys/class/leds/${sysfs}/device_name
				echo $interval > /sys/class/leds/${sysfs}/activity_interval
			}
			;;

		"port_state")
			[ -n "$port_state" ] && \
				echo $port_state > /sys/class/leds/${sysfs}/port_state
			;;

		"morse")
			echo $message > /sys/class/leds/${sysfs}/message
			echo $delay > /sys/class/leds/${sysfs}/delay
			;;

		switch[0-9]*)
			local port_mask

			config_get port_mask $1 port_mask
			[ -n "$port_mask" ] && \
				echo $port_mask > /sys/class/leds/${sysfs}/port_mask
			;;
		esac
	}

	#
	# 1 interface(eth0) + 2 phy port, LAN1/LAN2 LED trigger by ifplugd.
	#
	[ -e /proc/LAN_STATE ] && {
		local lan_state=$(cat /proc/LAN_STATE)

		[ $(($lan_state & 2)) -eq 2 ] && {
			led_set_attr LAN1_LED "brightness" "1"
		} || {
			led_set_attr LAN1_LED "brightness" "0"
		}
		[ $(($lan_state & 4)) -eq 4 ] && {
			led_set_attr LAN2_LED "brightness" "1"
		} || {			     
			led_set_attr LAN2_LED "brightness" "0" 
		}
	}
}

start() {
	[ -e /sys/class/leds/ ] && {
		config_load system
		config_foreach load_led led
		#
		# Control WLAN_STRONG_LED/WLAN_GOOD_LED/WLAN_LOW_LED
		#
		local opmode=$(uci get wireless.wifi1.opmode)
		local if_process_running=$(ps | grep "snrledcontrol" | grep -v "grep" | wc -l)
		#echo $if_process_running
		[ "$opmode" == "sta" -o "$opmode" == "wds_bridge" -o "$opmode" == "wds_sta" ] && {
			if [ $if_process_running -eq 0 ]; then
				[ -x "/usr/sbin/snrledcontrol" ] && /usr/sbin/snrledcontrol &
			fi
		} || {
			if [ $if_process_running -ne 0 ]; then
				killall -9 snrledcontrol
				echo 0 > /sys/class/leds/WLAN_LOW_LED/brightness 2>/dev/null
				echo 0 > /sys/class/leds/WLAN_GOOD_LED/brightness 2>/dev/null
				echo 0 > /sys/class/leds/WLAN_STRONG_LED/brightness 2>/dev/null
			fi
		}
	}
}

boot() {
	start
}
