#!/bin/bash

############################################################################################################ 
############################################################################################################ 

# Angemeldeten Benutzer ermitteln
a_user=$(whoami)
# wg_workdir Arbeitsverzeichnis wg
wg_wdir="/home/$a_user/.config/wg_nmcli/"
# Pfad wg Icons
wg_pic="/home/$a_user/.icons/"
# Wireguard tunnel auslesen mit grep nur Wireguard zeigen 
# mit sed alles nach dem ersten Leerzeichen bis Zeilenende weggeschneiden und .tunnel.txt schreiben
nmcli connection show | grep -iPo "(.*)(wireguard)" | sed 's/  .*//' > "${wg_wdir}.tunnel.txt"
# Variable der verfügbaren Tunnelanzeige für Yad
tunnel=$(nmcli connection show | grep -iPo "(.*)(wireguard)" | sed 's/  .*//') 
# Variable der aktiven Tunnelanzeige für Yad 
active=$(nmcli connection show --active | grep -iPo "(.*)(wireguard)" | sed 's/  .*//')

############################################################################################################ 
############################################################################################################ 
# funktion wg_notify

wg_notify(){
    
    a_user=$(whoami)
    active=$(nmcli connection show --active | grep -iPo "(.*)(wireguard)" | sed 's/  .*//')
    wg_pic="/home/$a_user/.icons/" 
    notify-send --icon="${wg_pic}wg-vpn.png" "$active powered"
    # autoconnect wird hier abgeschalten damit auch neu importierte nach einem Systemneustart nicht aktiv sind
    nmcli con mod "$active" connection.autoconnect no
    # ab hier beginnt yad notify
    #Pipe erstellen
    PIPE="$HOME/.pipe.tmp"
    rm "$PIPE" > /dev/null 2>&1
    mkfifo "$PIPE"
    exec 3<> "$PIPE"
    #Yad Dialog erstellen
    yad --notification --listen --command=wire_g --icon-size=32 --no-middle <&3 & 
    #Menüeinträge definieren
    echo "menu:\
    $active stop!/sbin/wg_stop| " >&3
    #Icon des Menübuttons definieren
    echo "icon:${wg_pic}wg-vpn.png" >&3
    #Name des Menüs definieren
    echo "tooltip:$active activ" >&3

}
export -f wg_notify

# function end
############################################################################################################ 
############################################################################################################ 
# funktion wireguard import 

import_wg(){
    a_user=$(whoami)
    wg_pic="/home/$a_user/.icons/"
    wg_wdir="/home/$a_user/.config/wg_nmcli/"
    wgconf=$(yad --file --separator=" \n" \
                 --button="OK" --button="Cancel" \
                 --width=1200 --height=800 \
                 --window-icon="${wg_pic}wg-import.png" \
                 --title="Wireguard .conf Select file")
    if [ "$?" = "0" ]
        then
            active=$(nmcli connection show --active | grep -iPo "(.*)(wireguard)" | sed 's/  .*//')
            if [ -n "$active" ]
                then  
                    wg_stop
                    nmcli connection import type wireguard file "$wgconf" > "${wg_wdir}.tmp.txt"
                    wg_notify
            else
                    nmcli connection import type wireguard file "$wgconf" > "${wg_wdir}.tmp.txt"
                    wg_notify
            fi
    fi }
export -f import_wg

# function end
############################################################################################################ 
############################################################################################################ 
# funktion wireguard remove

remove(){ 
    a_user=$(whoami)
    wg_pic="/home/$a_user/.icons/"
    wg_wdir="/home/$a_user/.config/wg_nmcli/"
    tunnel=$(nmcli connection show | grep -iPo "(.*)(wireguard)" | sed 's/  .*//')
    nmcli connection show --active | grep -iPo "(.*)(wireguard)" | sed 's/  .*//' &> "${wg_wdir}.wg_noactive.txt"
    active=$(cat "${wg_wdir}.wg_noactive.txt")

    selection=$(yad --no-click --list --height=300 --width=60 \
                    --title="Wire-G Trash" --center \
                    --fixed --buttons-layout=center \
                    --button="OK" --button="Cancel" \
                    --separator="" --window-icon="${wg_pic}wg-trash.png" \
                    --borders=8 --column 'Delete' "$tunnel")
    if [ "$?" = "0" ]
        then
            nmcli connection delete "$selection" &> "${wg_wdir}.tmp.txt" 
            success=$(grep -i 'erfolgreich gelöscht\|successfully deleted' "${wg_wdir}.tmp.txt")
            if [ "$?" = "0" ]
                then
                    yad --window-icon="${wg_pic}wg-trash.png" \
                        --image-on-top --image="${wg_pic}wg-info.png" \
                        --text="Your tunnel $selection was successfully deleted.\n" \
                        --text-align=center --button="OK" --title "Wire-G Trash" \
                        --fixed --center --buttons-layout=center --borders=8 --center
            else 
                    yad --window-icon="${wg_pic}wg-trash.png" \
                        --image-on-top --image="${wg_pic}wg-info.png" \
                        --text="Oh something went wrong.\nPlease delete tunnel in the network manager." \
                        --text-align=center --button="OK" --title "Wire-G Trash" \
                        --fixed --center --buttons-layout=center --borders=8 --center 
            fi 
            if [ "$selection" = "$active" ]
                then
                    pkill wire_g
                    pkill yad
            fi
     fi } 
export -f remove
 
# function end
############################################################################################################ 
############################################################################################################ 
#funktion for active tunnel a stop 
up_or_down(){
    active=$(nmcli connection show --active | grep -iPo "(.*)(wireguard)" | sed 's/  .*//')
    if [ -z "$active" ]
        then
            yad --image-on-top --image="${wg_pic}wg-info.png" \
            --text="There is no active tunnel that you could stop.\n" \
            --text-align=center --button="OK" --title "Wire-G" \
            --fixed --center --buttons-layout=center --borders=8 --center
    else
            wg_stop

    fi }
export -f up_or_down

# funktion end
############################################################################################################ 
############################################################################################################         
# function button "more" 

more(){ 
    a_user=$(whoami)
    wg_pic="/home/$a_user/.icons/"

    selection=$(yad --text "<span color='#626ff1'><b>Wire-G Author: Désiré Werner Menrath</b></span> 
<span color='#626ff1'><b>E-Mail: polunga40@unity-mail.de</b></span> 
<span color='#626ff1'><b>At all tunnels, the auto start is disabled"!"</b></span> 
<span color='#626ff1'><b>Use without warranty"!"</b></span>
                                                             " \
                    --height=100 --width=80 \
                    --title="Wire-G" --center \
                    --fixed --borders=8 --close-on-unfocus \
                    --buttons-layout=center \
                    --button=Import:"bash -c import_wg" \
                    --button="Wire-G Stop":up_or_down \
                    --button="Cancel":1 \
                    --separator="" --window-icon="${wg_pic}wg-vpn.png")
}
export -f more

# function end
############################################################################################################ 
############################################################################################################
# funktion wireguard enable select Tunnel

connect(){

    success=$(grep -i 'erfolgreich aktiviert\|successfully activated' "${wg_wdir}.tmp.txt")
    if [ "$?" = "0" ]
        then
            nmcli con mod "$selection" connection.autoconnect no 
            wg_notify    
    fi }

# function end
############################################################################################################ 
############################################################################################################                                                                                                                                 # --text "<b>Aktiver Tunnel:</b> <span color='#0fad0a'><b>$active</b></span> # mit <b>gewünschtes wort</b> wird fett geschrieben. 
# Farbe für Variable $active festgelegt mit fetter Schrift. (bei --text zwischen " ")

selection=$(yad --no-click --list \
                --close-on-unfocus --center \
                --height=300 --width=60 --title="Wire-G" \
                --fixed --buttons-layout=center \
                --button="OK" --button="Cancel" \
                --button=More:"bash -c more" --button=Remove:"bash -c remove" --separator="" \
                --window-icon="${wg_pic}wg-vpn.png" \
                --borders=8 --image-on-top --image="${wg_pic}wg-active.png" \
                --text "<b>Activ Tunnel: </b> <span color='#0fad0a'><b>$active</b></span> 
<b>----------------------------------------------------------------------------------</b>  
<b>If no tunnels are listed,</b> 
<b>so they have to import their tunnel beforehand.</b>" \
                --column 'Selection' "$tunnel")   
if [ "$?" = "0" ]
    then 

        if [ -n "$active" ]
            then
                wg_stop
                nmcli connection up "$selection" &> "${wg_wdir}.tmp.txt"
                connect
        else
                nmcli connection up "$selection" &> "${wg_wdir}.tmp.txt"
                connect
        fi
fi

############################################  END  ##########################################################