#!/bin/bash

WG_NUM=$1
WG_CONF="/etc/wireguard"
PRIV_KEY_FILE="wg${WG_NUM}_privatekey"
PUB_KEY_FILE="wg${WG_NUM}_publickey"
CONF_FILE="wg${WG_NUM}.conf"

if [ -z "$1" ]; then
    echo "Wireguard interface number not provided."
    exit 1
fi

if ! [[ "$1" =~ ^-?[0-9]+([.][0-9]+)?$ ]]; then
    echo "Wireguard interface number has to be a number."
    exit 2
fi

wg genkey | tee wg"$WG_NUM"_privatekey | wg pubkey > wg"$WG_NUM"_publickey

PRIVATE_KEY=$(cat $PRIV_KEY_FILE)

sudo mkdir -p $WG_CONF
sudo mv ./$PRIV_KEY_FILE $WG_CONF
sudo mv ./$PUB_KEY_FILE $WG_CONF

read -p "Enter IP (as peer) (e.g. 192.168.180.2/24): " ADDRESS
read -p "Enter DNS server (e.g. 8.8.8.8): " DNS

CONF_CONTENT="[Interface]
PrivateKey = $PRIVATE_KEY
Address = $ADDRESS
DNS = $DNS
"

echo "$CONF_CONTENT" > "$CONF_FILE"
sudo mv "$CONF_FILE" "$WG_CONF"

echo "Config saved to: $WG_CONF/$CONF_FILE"