#!/usr/bin/env bash

# Network

# Adds to /boot configuration files that let you set up wifi network before booting

# Written by Guy Sheffer <guysoft at gmail dot com> and Gina Häußge <gina@octoprint.org>

# GPL V3

########

set -x

set -e

export LC_ALL=C

source /common.sh

install_cleanup_trap

if [ "${NETWORK_WPA_SUPPLICANT}" == "yes" ]; then

unpack /filesystem/wpa-supplicant/boot /"${BASE_BOOT_MOUNT_PATH}"

DIST_NETWORK_FILE=/"${BASE_BOOT_MOUNT_PATH}"/${DIST_NAME,,}-wpa-supplicant.txt

# allow configuring multiple wifi networks via /boot/DIST_NAME-wpa-supplicant.txt

mv /"${BASE_BOOT_MOUNT_PATH}"/custompios-wpa-supplicant.txt ${DIST_NETWORK_FILE}

if [ "${BASE_DISTRO}" == "ubuntu" ] || [ "${BASE_DISTRO}" == "armbian" ]; then

echo "ctrl_interface=DIR=/var/run/wpa_supplicant GROUP=netdev" > /etc/wpa_supplicant/wpa_supplicant.conf

echo "update_config=1" >> /etc/wpa_supplicant/wpa_supplicant.conf

fi

cat /etc/wpa_supplicant/wpa_supplicant.conf >> ${DIST_NETWORK_FILE} # append distributed conf to our own

rm -f /etc/wpa_supplicant/wpa_supplicant.conf # remove distributed conf

# create symlink

ln -s "${DIST_NETWORK_FILE}" /etc/wpa_supplicant/wpa_supplicant.conf

# copy /etc/wpa_supplicant/ifupdown.sh to /etc/ifplugd/action.d/ifupdown - for wlan auto reconnect

[ -f /etc/ifplugd/action.d/ifupdown ] && mv /etc/ifplugd/action.d/ifupdown /etc/ifplugd/action.d/ifupdown.original

[ -f /etc/wpa_supplicant/ifupdown.sh ] && ln -s /etc/wpa_supplicant/ifupdown.sh /etc/ifplugd/action.d/ifupdown

fi

if [ "${NETWORK_NETWORK_MANAGER}" == "yes" ]; then

unpack filesystem/network-manager/root / root

unpack filesystem/network-manager/boot /"${BASE_BOOT_MOUNT_PATH}"

if [ "${BASE_BOARD}" == "debian_lepotato" ]; then

apt_update_skip

apt-get install -y network-manager

fi

# mv /"${BASE_BOOT_MOUNT_PATH}"/wifi.nmconnection ${DIST_NETWORK_FILE}

systemctl_if_exists enable copy-network-manager-config@wlan0.service

fi

if [ "${BASE_DISTRO}" == "raspbian" ]; then

# Workaround rfkill not unblocking on boot

rm /var/lib/systemd/rfkill/*

if [ -f "/var/lib/NetworkManager/NetworkManager.state" ]; then

sed -i 's/WirelessEnabled=false/WirelessEnabled=true/g' /var/lib/NetworkManager/NetworkManager.state

fi

fi

if [ ! -f "/etc/rc.local" ];then

echo 'exit 0' >> /etc/rc.local

fi

# prevent ntp updates from failing due to some Rpi3 weirdness, see also "Fix SSH" further below

apt_update_skip

apt-get install -y iptables

sed -i 's@exit 0@@' /etc/rc.local

echo '/sbin/iptables -t mangle -I POSTROUTING 1 -o wlan0 -p udp --dport 123 -j TOS --set-tos 0x00' >> /etc/rc.local

echo 'exit 0' >> /etc/rc.local

# Install powersave option

if [ "$NETWORK_DISABLE_PWRSAVE" == "yes" ]; then

# Copy pwrsave script

unpack filesystem/usr/local/bin /usr/local/bin root

# Use rc.local

if [ "$NETWORK_PWRSAVE_TYPE" == "rclocal" ]; then

echo_green "Modifying /etc/rc.local ..."

sed -i 's@exit 0@@' /etc/rc.local

(echo "# Disable WiFi Power Management"; \

echo 'echo "Disabling power management for wlan0 ..."' ; \

echo "/usr/local/bin/pwrsave off"; echo "exit 0") >> /etc/rc.local

fi

# Use service

if [ "$NETWORK_PWRSAVE_TYPE" == "service" ]; then

echo_green "Installing disable-wifi-pwr-mgmt service ..."

unpack filesystem/etc/systemd/system /etc/systemd/system root

systemctl_if_exists enable disable-wifi-pwr-mgmt.service

fi

# Use udev rule

if [ "$NETWORK_PWRSAVE_TYPE" == "udev" ]; then

echo_green "Installing WiFi Power Management udev rule ..."

unpack filesystem/etc/udev/rules.d /etc/udev/rules.d root

fi

# strip out unneeded script, depending on choose

if [ "$NETWORK_PWRSAVE_TYPE" != "udev" ]; then

rm -f /usr/local/bin/pwrsave-udev

else

rm -f /usr/local/bin/pwrsave

fi

fi