#!/bin/sh
# 2020-06-13, Nico Schottelius
# See https://ungleich.ch/u/products/vigir/

if [ $# -lt 2 ]; then
    echo "$0 vigir-ip-address vigir-id [wgprivkey]"
    echo "    vigir-ip-address: where to find the vigir"
    echo "    vigir-id: number in decimal format"
    echo "    wgprivkey: if specified, use this private key"
    exit 1
fi


set -x
vigir_ip=$1; shift
id=$1; shift
hex_id=$(printf "%0.2x\n" "$id")
vigir_hostname=vigir${id}

prefix_base=2a0a:e5c1:5
my_prefix=${prefix_base}${hex_id}
my_network=${my_prefix}::/48

my_wireguard_ip=${my_prefix}::42
my_lan_ip=${my_prefix}:cafe::42
my_wifi_ip=${my_prefix}:7ea::42

# wireguard
if [ $# -eq 1 ]; then
    private_key=$1; shift
else
    private_key=$(wg genkey)
fi
public_key=$(echo $private_key | wg pubkey)

vpn_endpoint_host=vpn-2a0ae5c1500.ungleich.ch
vpn_endpoint_pubkey=oaFiIVV1NjvDcfdtwJqR4F3k2XIC07npNgj0YjIEem4=

i=0
found=""
while [ $i -lt 30 ]; do
    echo "Trying to reach ${vigir_ip} ($i)"
    if ping -c1 ${vigir_ip} >/dev/null; then
        found="yes"
        break
    fi
    i=$((i+1))
done

if [ -z "$found" ]; then
    echo "Unable to contact vigir. Exiting"
    exit 1
fi


cat <<EOF | ssh -t "root@${vigir_ip}"
set -x

# Set IPv6 address on LAN
uci set network.lan.ip6addr='${my_lan_ip}/64'

# Do not announce ULA - we have GUA
uci delete network.globals.ula_prefix

# Setup Router Advertisements / remove IPv4 dhcp
uci set dhcp.lan.ra='server'

# We do stateless by default everywhere
uci delete dhcp.lan.dhcpv6

# Fix DNS: make dnsmasq NOT use a resolv.conf
# so that it only reads from our servers with DNS64 enabled
uci set dhcp.@dnsmasq[0].noresolv='1'

# Fix DNS: make the OS use the locally provided DNS servers
# otherwise the VPN tunnel cannot be established
uci set dhcp.@dnsmasq[0].localuse='0'

# DNS upstream over VPN gives DNS64
uci add_list dhcp.@dnsmasq[0].server='2a0a:e5c0:0:a::a'
uci add_list dhcp.@dnsmasq[0].server='2a0a:e5c0:2:a::a'

uci set system.@system[0].hostname="${vigir_hostname}"

# Wifi configuration
uci set wireless.radio0=wifi-device
uci set wireless.radio0.htmode='HT40'
uci set wireless.radio0.country='CH'
uci set wireless.radio0.channel='6'

uci set wireless.default_radio0.encryption='psk2'
uci set wireless.default_radio0.key='iloveipv6'
uci set wireless.default_radio0.ssid='IPv6 everywhere ${vigir_hostname}'

uci set wireless.radio1.country='CH'
uci set wireless.default_radio1.encryption='psk2'
uci set wireless.default_radio1.key='iloveipv6'
uci set wireless.default_radio1.ssid='IPv6 everywhere ${vigir_hostname} 5Ghz'


# Ensure it is not disabled
uci delete wireless.radio0.disabled

# Create temporary IPV4 CLIENT on LAN
# This way we get Internet/upstream from the LAN port
# Can we do this via IPv6 instead?
# this breaks if multiple vigir are setup, as we have fake / wrong
# dhcp server configurations.
# easy solution: using two different networks...

uci commit

ping -c5 ungleich.ch || exit 1

# update the sources
opkg update

# install wireguard + gui
opkg install wireguard
opkg install luci-app-wireguard

# VPN / Wireguard
uci set network.wg0=interface
uci set network.wg0.proto='wireguard'
uci set network.wg0.private_key='${private_key}'
uci set network.wg0.listen_port='51820'
uci set network.wg0.addresses='${my_wireguard_ip}/64'

if ! uci get network.@wireguard_wg0[0]; then
   uci add network wireguard_wg0
fi

uci set network.@wireguard_wg0[0]=wireguard_wg0
uci set network.@wireguard_wg0[0].persistent_keepalive='25'
uci set network.@wireguard_wg0[0].public_key="${vpn_endpoint_pubkey}"
uci set network.@wireguard_wg0[0].description="IPv6VPN.ch by ungleich"
uci set network.@wireguard_wg0[0].allowed_ips='::/0'
uci set network.@wireguard_wg0[0].endpoint_host="${vpn_endpoint_host}"
uci set network.@wireguard_wg0[0].endpoint_port='51820'
uci set network.@wireguard_wg0[0].route_allowed_ips='1'

# Firewall configuration
if ! uci show firewall | grep "name='Allow-SSH'"; then
uci add firewall rule
uci set firewall.@rule[-1].name='Allow-SSH'
uci set firewall.@rule[-1].src='wan'
uci set firewall.@rule[-1].dest='lan'
uci set firewall.@rule[-1].proto='tcp'
uci set firewall.@rule[-1].dest_port='22'
uci set firewall.@rule[-1].target='ACCEPT'
fi

if ! uci show firewall | grep "name='Allow-HTTPS'"; then
uci add firewall rule
uci set firewall.@rule[-1].name='Allow-HTTPS'
uci set firewall.@rule[-1].src='wan'
uci set firewall.@rule[-1].dest='lan'
uci set firewall.@rule[-1].proto='tcp'
uci set firewall.@rule[-1].dest_port='443'
uci set firewall.@rule[-1].target='ACCEPT'
fi

if ! uci show firewall | grep "name='Allow-HTTP'"; then
uci add firewall rule
uci set firewall.@rule[-1].name='Allow-HTTP'
uci set firewall.@rule[-1].src='wan'
uci set firewall.@rule[-1].dest='lan'
uci set firewall.@rule[-1].proto='tcp'
uci set firewall.@rule[-1].dest_port='80'
uci set firewall.@rule[-1].target='ACCEPT'
fi

# Add interfaces to the right network zone
uci set firewall.@zone[1].network='wan wan6 wg0'

uci commit

reboot
EOF

echo "Wireguard public key and id: ${id} ${public_key}"
echo ${public_key} > ${vigir_hostname}.public_key

# change to ipv6
vigir_ip=${my_lan_ip}

sleep 15

while ! ping -c5 ${vigir_ip}; do
    echo "Waiting for vigir ${id}"
    sleep 2
done

echo "Wireguard public key and id: ${id} ${public_key}"