#!/bin/sh

CFG=/etc/board.json

. /usr/share/libubox/jshn.sh

[ -f $CFG ] || exit 1

generate_static_network() {
	uci -q batch <<EOF
delete network.loopback
set network.loopback='interface'
set network.loopback.ifname='lo'
set network.loopback.proto='static'
set network.loopback.ipaddr='127.0.0.1'
set network.loopback.netmask='255.0.0.0'
delete network.globals
set network.globals='globals'
set network.globals.ula_prefix='auto'
EOF
}

next_vlan=3
generate_network() {
	local vlan

	json_select network
	json_select $1
	json_get_vars ifname create_vlan macaddr
	json_select ..
	json_select ..

	[ -n "$ifname" ] || return
	[ "${create_vlan:-0}" -eq 1 ] && case $1 in
	lan) vlan=1;;
	wan) vlan=2;;
	*)
		vlan=$next_vlan
		next_vlan=$((next_vlan + 1))
		;;
	esac
	[ -n "$vlan" ] && ifname=${ifname}.${vlan}
	uci -q batch <<EOF
delete network.$1
set network.$1='interface'
set network.$1.ifname='$ifname'
set network.$1.force_link=1
set network.$1.proto='none'
set network.$1.macaddr='$macaddr'
EOF

	case $1 in
	lan) uci -q batch <<EOF
set network.$1.type='bridge'
set network.$1.proto='static'
set network.$1.ipaddr='192.168.1.1'
set network.$1.netmask='255.255.255.0'
set network.$1.ip6assign='60'
EOF
		;;
	wan) uci -q batch <<EOF
set network.$1.proto='dhcp'
delete network.wan6
set network.wan6='interface'
set network.wan6.ifname='$ifname'
set network.wan6.proto='dhcpv6'
EOF
		;;
	esac
}

generate_switch_vlan() {
	local device=$1
	local vlan=$2
	local cpu_port=$3

	case $vlan in
	lan)	vlan=1;;
	wan)	vlan=2;;
	*)	vlan=${vlan##vlan};;
	esac

	json_select vlans
	json_select $2
	json_get_values ports
	json_select ..
	json_select ..

	uci -q batch <<EOF
add network switch_vlan
set network.@switch_vlan[-1].device='$device'
set network.@switch_vlan[-1].vlan='$vlan'
set network.@switch_vlan[-1].ports='$ports ${cpu_port}t'
EOF
}

calculate_switch_vlans() {
	local switch=$1
	local ports port attr val
	local prev_role
	local num device role index
	local n_cpu=0 cpu0 cpu1 cpu2 cpu3
	local dev0 dev1 dev2 dev3 role0 role1 role2 role3
	local n_vlan=0 vlan_off=-1
	local vlan_ports cpu_port

	json_get_keys ports ports

	json_select ports

	# gather all cpu ports and count vlans
	for port in $ports; do
		json_select "$port"
		json_get_vars num device role

		if json_is_a attr object; then
			json_get_keys attr attr
			json_select attr

			uci -q batch <<-EOF
				add network switch_port
				set network.@switch_port[-1].device='$switch'
				set network.@switch_port[-1].port=$num
			EOF

			for attr in $attr; do
				json_get_var val "$attr"
				uci -q set network.@switch_port[-1].$attr="$val"
			done

			json_select ..
		fi

		json_select ..

		if [ -n "$num" ] && [ -n "$device" ]; then
			export "cpu$n_cpu=$num"
			export "dev$n_cpu=$device"
			n_cpu=$((n_cpu + 1))
		elif [ -n "$num" ] && [ -n "$role" ] && [ "$role" != "$prev_role" ]; then
			export "role$n_vlan=$role"
			n_vlan=$((n_vlan + 1))
			prev_role="$role"
		fi
	done

	unset prev_role

	# autogenerate vlans
	for port in $ports ""; do
		if [ -n "$port" ]; then
			json_select "$port"
			json_get_vars num device role
			json_select ..
		else
			num="-"; role="-"
		fi

		if [ -n "$num" ] && [ -n "$role" ]; then
			if [ "$role" != "$prev_role" ]; then
				if [ -n "$vlan_ports" ]; then
					let cpu_port="cpu$((vlan_off % n_cpu))"
					[ $n_vlan -gt $n_cpu ] && cpu_port="${cpu_port}t"

					uci -q batch <<-EOF
						add network switch_vlan
						set network.@switch_vlan[-1].device='$switch'
						set network.@switch_vlan[-1].vlan='$((vlan_off + 1))'
						set network.@switch_vlan[-1].ports='$vlan_ports $cpu_port'
					EOF
				fi

				vlan_off=$((vlan_off + 1))
				vlan_ports="$num"
				prev_role="$role"
			else
				vlan_ports="$vlan_ports $num"
			fi

		fi
	done

	json_select ..

	# autogenerate interfaces
	vlan_off=0; while [ $vlan_off -lt $n_vlan ]; do
		eval role="\$role$((vlan_off))"
		eval device="\$dev$((vlan_off++ % n_cpu))"
		[ $n_vlan -gt $n_cpu ] && device="$device.$vlan_off"

		# quirk: append ifnames for subsequent switches
		case "$switch" in switch[1-9])
			local prev_devs="$(uci -q get "network.$role.ifname")"
			if echo "$prev_devs" | grep -wq "$device"; then
				device="$prev_devs"
			else
				device="$prev_devs $device"
			fi
		;; esac

		uci -q batch <<-EOF
			set network.$role='interface'
			set network.$role.ifname='$device'
		EOF

		case $role in
			lan) uci -q batch <<-EOF
				set network.lan.type='bridge'
				set network.lan.proto='static'
				set network.lan.ipaddr='192.168.1.1'
				set network.lan.netmask='255.255.255.0'
				set network.lan.ip6assign='60'
			EOF
			;;
			wan) uci -q batch <<-EOF
				set network.wan.proto='dhcp'
				set network.wan6='interface'
				set network.wan6.ifname='$device'
				set network.wan6.proto='dhcpv6'
			EOF
			;;
			*) uci -q batch <<-EOF
				set network.$role.force_link='1'
				set network.$role.proto='none'
			EOF
			;;
		esac
	done
}

generate_switch() {
	local key=$1
	local vlans

	json_select switch
	json_select $key
	json_get_vars enable reset blinkrate cpu_port

	uci -q batch <<EOF
add network switch
set network.@switch[-1].name='$key'
set network.@switch[-1].reset='$reset'
set network.@switch[-1].enable_vlan='$enable'
set network.@switch[-1].blinkrate='$blinkrate'
EOF

	if [ -n "$cpu_port" ]; then
		json_get_keys vlans vlans
		for vlan in $vlans; do generate_switch_vlan $1 $vlan $cpu_port; done
	elif json_is_a ports array; then
		calculate_switch_vlans $1
	fi

	json_select ..
	json_select ..
}

generate_led() {
	local key=$1
	local cfg="led_$key"

	json_select led
	json_select $key
	json_get_vars name sysfs type trigger device interface default port_mask
	json_select ..
	json_select ..

	uci -q batch <<EOF
delete system.$cfg
set system.$cfg='led'
set system.$cfg.name='$name'
set system.$cfg.sysfs='$sysfs'
set system.$cfg.dev='$device'
set system.$cfg.trigger='$trigger'
set system.$cfg.port_mask='$port_mask'
set system.$cfg.default='$default'
EOF
	case $type in
	netdev)
		uci -q batch <<EOF
set system.$cfg.trigger='netdev'
set system.$cfg.mode='link tx rx'
EOF
	;;

	usb)
		uci -q batch <<EOF
set system.$cfg.trigger='usbdev'
set system.$cfg.interval='50'
EOF
	;;

	esac
}

json_init
json_load "$(cat ${CFG})"

generate_static_network

json_get_keys keys network
for key in $keys; do generate_network $key; done

json_get_keys keys switch
for key in $keys; do generate_switch $key; done

json_get_keys keys led
for key in $keys; do generate_led $key; done

uci commit
