# Wireguard functions for Wire-Py
import os
import shutil
import subprocess
import zipfile
from datetime import datetime
from tkinter import filedialog, ttk
import tkinter as tk
from pathlib import Path
from subprocess import check_call

font_color = '#4011a7'
path_to_file = Path('/etc/wire_py/wg_py')
_u = Path.read_text(Path('/tmp/_u'))


def msg_window():
    msg = tk.Toplevel()
    msg.resizable(width=False, height=False)
    msg.x_width = 340
    msg.y_height = 140
    msg.title('Import error!')
    msg.monitor_center_x = msg.winfo_screenwidth() / 2 - (msg.x_width / 2)
    msg.monitor_center_y = msg.winfo_screenheight() / 2 - (msg.y_height / 2)
    msg.geometry('%dx%d+%d+%d' % (msg.x_width, msg.y_height, msg.monitor_center_x, msg.monitor_center_y))
    msg.columnconfigure(0, weight=1)
    msg.configure(pady=20)
    msg.warning = tk.PhotoImage(file=r'/usr/share/icons/wp-icons/64/error.png')
    msg.i_warning = tk.Label(msg, image=msg.warning)
    msg.i_warning.grid(column=0, row=0)
    label = tk.Label(msg, text='Oh... no valid Wireguard File!\nPlease select a valid Wireguard File')
    label.config(font=('Ubuntu', 11), padx=15, pady=15)
    label.grid(column=1, row=0)
    button = tk.Button(msg, text='OK', command=msg.destroy)
    button.config(padx=15, pady=5)
    button.grid(column=0, columnspan=2, row=1)
    wg_icon_2 = tk.PhotoImage(file=r'/usr/share/icons/wp-icons/48/wg_msg.png')
    msg.iconphoto(True, wg_icon_2)
    msg.winfo_toplevel()


def msg_exp_successful_window():
    msg = tk.Toplevel()
    msg.resizable(width=False, height=False)
    msg.x_width = 340
    msg.y_height = 140
    msg.title('Import error!')
    msg.monitor_center_x = msg.winfo_screenwidth() / 2 - (msg.x_width / 2)
    msg.monitor_center_y = msg.winfo_screenheight() / 2 - (msg.y_height / 2)
    msg.geometry('%dx%d+%d+%d' % (msg.x_width, msg.y_height, msg.monitor_center_x, msg.monitor_center_y))
    msg.columnconfigure(0, weight=1)
    msg.configure(pady=20)
    msg.warning = tk.PhotoImage(file=r'/usr/share/icons/wp-icons/64/info.png')
    msg.i_warning = tk.Label(msg, image=msg.warning)
    msg.i_warning.grid(column=0, row=0)
    label = tk.Label(msg, text='Your zip file is in home directory')
    label.config(font=('Ubuntu', 11), padx=15, pady=15)
    label.grid(column=1, row=0)
    button = tk.Button(msg, text='OK', command=msg.destroy)
    button.config(padx=15, pady=5)
    button.grid(column=0, columnspan=2, row=1)
    wg_icon_2 = tk.PhotoImage(file=r'/usr/share/icons/wp-icons/48/wg_vpn.png')
    msg.iconphoto(True, wg_icon_2)
    msg.winfo_toplevel()


def msg_exp_fail_window():
    msg = tk.Toplevel()
    msg.resizable(width=False, height=False)
    msg.x_width = 340
    msg.y_height = 140
    msg.title('Export error!')
    msg.monitor_center_x = msg.winfo_screenwidth() / 2 - (msg.x_width / 2)
    msg.monitor_center_y = msg.winfo_screenheight() / 2 - (msg.y_height / 2)
    msg.geometry('%dx%d+%d+%d' % (msg.x_width, msg.y_height, msg.monitor_center_x, msg.monitor_center_y))
    msg.columnconfigure(0, weight=1)
    msg.configure(pady=20)
    msg.warning = tk.PhotoImage(file=r'/usr/share/icons/wp-icons/64/error.png')
    msg.i_warning = tk.Label(msg, image=msg.warning)
    msg.i_warning.grid(column=0, row=0)
    label = tk.Label(msg, text='Export failed! Please try again')
    label.config(font=('Ubuntu', 11), padx=15, pady=15)
    label.grid(column=1, row=0)
    button = tk.Button(msg, text='OK', command=msg.destroy)
    button.config(padx=15, pady=5)
    button.grid(column=0, columnspan=2, row=1)
    wg_icon_2 = tk.PhotoImage(file=r'/usr/share/icons/wp-icons/48/wg_msg.png')
    msg.iconphoto(True, wg_icon_2)
    msg.winfo_toplevel()


class GreenLabel:
    def __init__(self):
        self.StrVar = None
        self.lb_tunnel = None

    def green_show_label(self):
        self.lb_tunnel = tk.Label(self, textvariable=self.StrVar, fg='green')
        self.lb_tunnel.config(font=('Ubuntu', 11, 'bold'))
        self.lb_tunnel.grid(column=2, padx=10, row=0)
        self.columnconfigure(2, weight=1)
        self.rowconfigure(0, weight=1)

    def columnconfigure(self, param, weight):
        pass

    def rowconfigure(self, param, weight):
        pass


class StartStopBTN:
    def __init__(self):
        self.lb_frame_btn_lbox = None
        self.wg_switch = None
        self.btn_stst = None
        self.wg_vpn_start = tk.PhotoImage(file=r'/usr/share/icons/wp-icons/48/wg_vpn-start.png')
        self.wg_vpn_stop = tk.PhotoImage(file=r'/usr/share/icons/wp-icons/48/wg_vpn-stop.png')

    def button_stop(self):
        self.btn_stst = tk.Button(self.lb_frame_btn_lbox, image=self.wg_vpn_stop, bd=0, command=self.wg_switch)
        self.btn_stst.grid(column=0, row=0, padx=15, pady=8)

    def button_start(self):
        self.btn_stst = tk.Button(self.lb_frame_btn_lbox, image=self.wg_vpn_start, bd=0, command=self.wg_switch)
        self.btn_stst.grid(column=0, row=0, padx=15, pady=8)


class ConToDict:
    @classmethod
    def covert_to_dict(cls, file):
        dictlist = []
        for lines in file.readlines():
            line_plit = lines.split()
            dictlist = dictlist + line_plit
        dictlist.remove('[Interface]')
        dictlist.remove('[Peer]')
        for items in dictlist:
            if items == '=':
                dictlist.remove(items)
        for _ in dictlist:  # Here is the beginning (Loop) of convert List to Dictionary
            a = [dictlist[0], dictlist[1]]
            b = [dictlist[2], dictlist[3]]
            c = [dictlist[4], dictlist[5]]
            d = [dictlist[6], dictlist[7]]
            e = [dictlist[8], dictlist[9]]
            f = [dictlist[10], dictlist[11]]
            g = [dictlist[12], dictlist[13]]
            h = [dictlist[14], dictlist[15]]
            new_list = [a, b, c, d, e, f, g, h]
            final_dict = {}
            for elements in new_list:
                final_dict[elements[0]] = elements[1]

            # end... result a Dictionary
        address = final_dict['Address']
        dns = final_dict['DNS']
        if ',' in dns:
            dns = dns[:-1]
        endpoint = final_dict['Endpoint']
        if 'PresharedKey' in final_dict:
            pre_key = final_dict['PresharedKey']
        else:
            pre_key = final_dict['PreSharedKey']
        return address, dns, endpoint, pre_key


class TunnelActiv:
    @staticmethod
    def active():  # Shows the active tunnel
        active = os.popen('nmcli con show --active | grep -iPo "(.*)(wireguard)"').read().split()
        if not active:
            active = ''
        else:
            active = active[0]

        return active


class ShowAddress:
    def __init__(self):
        self.lb_frame2 = None
        self.lb_frame = None
        self.endpoint = None
        self.dns = None
        self.address = None
        self.enp = None
        self.DNS = None
        self.add = None

    def init_and_report(self, data=None):
        # Address Label
        self.add = tk.StringVar()
        self.add.set('Address:    ' + data[0])
        self.DNS = tk.StringVar()
        self.DNS.set('       DNS:    ' + data[1])
        self.enp = tk.StringVar()
        self.enp.set('Endpoint:    ' + data[2])

    def label_empty(self):
        self.add.set('')
        self.DNS.set('')
        self.enp.set('')

    def show_data(self):
        # Address Label
        self.address = tk.Label(self.lb_frame, textvariable=self.add, fg='blue')
        self.address.grid(column=0, row=4, sticky='w', padx=10, pady=6)
        self.address.config(font=('Ubuntu', 9))
        # DNS Label
        self.dns = tk.Label(self.lb_frame, textvariable=self.DNS, fg='blue')
        self.dns.grid(column=0, row=6, sticky='w', padx=10, pady=6)
        self.dns.config(font=('Ubuntu', 9))
        # Endpoint Label
        self.endpoint = tk.Label(self.lb_frame2, textvariable=self.enp, fg='blue')
        self.endpoint.grid(column=0, row=7, sticky='w', padx=10, pady=10)
        self.endpoint.config(font=('Ubuntu', 9))


class ListTunnels:
    @staticmethod
    def tl_list():
        wg_s = os.popen('nmcli con show | grep -iPo "(.*)(wireguard)"').read().split()
        tl = wg_s[::3]  # tl = Tunnel list # Show of 4.Element in list
        return tl


class ImportTunnel:

    def __init__(self):

        self.select_tunnel = None
        self.wg_switch = None
        self.btn_stst = None
        self.lb_tunnel = None
        self.StrVar = None
        self.a = None
        self.l_box = None

    def wg_import_select(self):
        try:
            filepath = filedialog.askopenfilename(initialdir=str(_u), title='Select Wireguard config File',
                                                  filetypes=[('WG config files', '*.conf')], )

            with open(filepath, 'r') as file:
                read = file.read()
            path_split = filepath.split('/')
            path_split1 = path_split[-1]
            self.a = TunnelActiv.active()
            if 'PrivateKey = ' in read and 'PublicKey = ' in read:
                with open(filepath, 'r') as file:
                    key = ConToDict.covert_to_dict(file)
                    pre_key = key[3]
                    print(type(pre_key))
                    if len(pre_key) != 0:
                        with open('/etc/wire_py/.keys', 'r') as readfile:
                            if pre_key in readfile:
                                print('Tunnel already available! Please use another file for import')
                            else:
                                with open('/etc/wire_py/.keys', 'a') as keyfile:
                                    keyfile.write('\r\n' + pre_key)
                                    if len(path_split1) > 17:
                                        p1 = shutil.copy(filepath, Path('/etc/wire_py/'))
                                        path_split = path_split1[len(path_split1) - 17:]
                                        os.rename(p1, Path('/etc/wire_py') / str(path_split))
                                        new_conf = '/etc/wire_py/' + path_split
                                        if self.a != '':
                                            check_call(['nmcli', 'connection', 'down', TunnelActiv.active()])
                                            ShowAddress.label_empty(self)

                                        subprocess.check_output(['nmcli', 'connection', 'import', 'type',
                                                                 'wireguard', 'file', new_conf], text=True)

                                    else:
                                        shutil.copy(filepath, Path('/etc/wire_py/'))
                                        if self.a != '':
                                            check_call(['nmcli', 'connection', 'down', TunnelActiv.active()])
                                            ShowAddress.label_empty(self)

                                        subprocess.check_output(['nmcli', 'connection', 'import', 'type',
                                                                 'wireguard', 'file', filepath], text=True)

                                    self.StrVar.set('')
                                    self.a = TunnelActiv.active()
                                    self.l_box.insert(0, self.a)
                                    self.l_box.update()
                                    self.StrVar = tk.StringVar()
                                    self.StrVar.set(self.a)
                                    GreenLabel.green_show_label(self)
                                    StartStopBTN.button_stop(self)
                                    wg_read = Path('/etc/wire_py') / str(self.a + '.conf')
                                    with open(wg_read, 'r') as file_for_key:
                                        data = ConToDict.covert_to_dict(file_for_key)
                                    # Address Label
                                    ShowAddress.init_and_report(self, data)
                                    ShowAddress.show_data(self)
                                    check_call(['nmcli', 'con', 'mod', self.a, 'connection.autoconnect', 'no'])
                                    os.chmod(str(wg_read), 0o600)

            if 'PrivateKey = ' not in read:
                msg_window()
        except EOFError:
            pass
        except TypeError:
            pass
        except FileNotFoundError:
            pass
        except subprocess.CalledProcessError:
            print('Tunnel exist!')


class FileHandle:

    def __init__(self):

        self.wg_autostart = None
        self.autoconnect = None
        self.auto_con = None
        self.autoconnect_var = None
        self.tl = None
        self.selected_option = None
        self.l_box = None

    def box_set(self):
        try:
            select_tunnel = self.l_box.curselection()
            select_tl = self.l_box.get(select_tunnel[0])
            if self.selected_option.get() == 0:
                Path.unlink(path_to_file)
                tl = ListTunnels.tl_list()
                if len(tl) == 0:
                    self.wg_autostart.configure(state='disabled')
            if self.selected_option.get() >= 1:
                Path.write_text(path_to_file, select_tl)
        except IndexError:
            self.selected_option.set(1)

        OnOff.on_off(self)


class OnOff:
    def __init__(self):
        self.wg_autostart = None
        self.selected_option = None
        self.auto_con = None
        self.autoconnect = None
        self.autoconnect_var = None
        self.lb_frame_buttons = None

    def on_off(self):
        if Path.exists(path_to_file):
            self.selected_option.set(1)
            self.autoconnect_var.set('')
            if not Path.is_dir(Path('/etc/wire_py')):
                Path.mkdir(Path('/etc/wire_py'))
            self.auto_con = Path.read_text(path_to_file)
        else:
            self.wg_autostart.configure(state='disabled')
            self.auto_con = 'no Autoconnect'
        self.autoconnect_var.set('')
        self.autoconnect_var = tk.StringVar()
        self.autoconnect_var.set(self.auto_con)
        self.autoconnect = tk.Label(self, textvariable=self.autoconnect_var, fg='blue', padx=5)
        self.autoconnect.config(font=('Ubuntu', 11))
        self.autoconnect.grid(column=0, row=4, sticky='ne', pady=20)


class ExportTunnels:
    @staticmethod
    def wg_export():
        _u1 = str(_u[6:])
        now_time = datetime.now()
        now_datetime = now_time.strftime('wg-exp-' + '%m-%d-%Y' + '-' + '%H:%M')
        tl = ListTunnels.tl_list()
        try:
            if len(tl) != 0:
                wg_tar = str(_u) + '/' + now_datetime
                p_to_conf = Path('/etc/wire_py/')
                shutil.make_archive(wg_tar, 'zip', p_to_conf)
                os.chown(wg_tar + '.zip', 1000, 1000)
                with zipfile.ZipFile((wg_tar + '.zip'), 'r') as zf:
                    if len(zf.namelist()) != 0:
                        msg_exp_successful_window()
                    else:
                        msg_exp_fail_window()

        except TypeError:
            pass