#! /usr/bin/python3
from tkinter import Tk
from sys import exit
from threading import Thread
from auxiliary.gui import NetgearMon
from auxiliary.utils import environment


def start_gui_thread():
    root = Tk()
    NetgearMon(root)
    root.mainloop()


if __name__ == "__main__":
    try:
        environment()
    except Exception:
        error = '''
        Ensure environment variables are set.
        ROUTER_IP, ROUTER_LOGIN, & ROUTER_PWD
        '''
        print(error)
        exit(1)
    print('\nStarting Netgear Mon\n')
    try:
        thread = Thread(target=start_gui_thread)
        thread.daemon = True
        thread.start()
        thread.join()
    except KeyboardInterrupt:
        print('\nExiting Netgear Mon')

from tkinter import Label, Listbox, Button, W, E
from threading import Thread
from os import environ
from time import time, sleep
from .router_scraper import NetgearRouter


class NetgearMon:
    def __init__(self, master):
        colors = {'black': '#000000', 'green': '#66ff33'}
        self.master = master
        self.master.title("Netgear Mon")
        self.master.configure(bg=colors['black'])
        self.master.geometry("535x545")
        self.running = False
        self.th = None
        font = 'Verdana 10 bold'
        heading = 'Monitoring Devices on: '
        heading += environ['ROUTER_IP']
        self.title_label = Label(
            master, fg=colors['green'],
            bg=colors['black'], font=font, width=120)
        self.ip_label = Label(
            master, fg=colors['green'], bg=colors['black'],
            text=heading, font=font, width=60)
        self.explorer = Listbox(
            master, fg=colors['green'], bg=colors['black'],
            width=65, height=18, highlightcolor=colors['green'])
        self.start_button = Button(
            master, fg=colors['green'],
            text="Start", width=5, bg=colors['black'],
            highlightbackground=colors['black'],
            highlightthickness=2, command=self.start)
        self.stop_button = Button(
            master, fg=colors['green'],
            bg=colors['black'], text="Stop", width=5,
            highlightbackground=colors['black'],
            highlightthickness=2, command=self.stop)
        self.status_label = Label(
            master, fg=colors['green'], bg=colors['black'],
            width=30, text="Press Start to monitor...")
        self.title_label.grid(row=0, sticky=E+W, padx=5, pady=5)
        self.ip_label.grid(row=1, sticky=W, padx=0, pady=20)
        self.explorer.grid(row=2, sticky=W, padx=5)
        self.start_button.grid(row=3, sticky=W, padx=0, pady=5)
        self.stop_button.grid(row=3, sticky=W, padx=80, pady=5)
        self.status_label.grid(row=4, sticky=W, padx=0, pady=20)

    def refresh(self, dev_info):
        self.explorer.delete(0, 'end')
        self.explorer.insert('end', *dev_info)

    def start_thread(self):
        self.running = True
        router = NetgearRouter()
        device_info = [entry for entry in router.scrape_router_ui()]
        device_info.insert(0, ' ')
        device_info.insert(0, router.get_date_time())
        self.refresh(device_info)
        poll = time() + 60
        while self.running:
            if time() > poll:
                router = NetgearRouter()
                device_info = [entry for entry in router.scrape_router_ui()]
                device_info.insert(0, ' ')
                device_info.insert(0, router.get_date_time())
                self.refresh(device_info)
                poll = time() + 60

    def start(self):
        self.th = Thread(target=self.start_thread)
        self.th.start()
        self.animation()
        self.set_status_label('Netgear Mon running...')

    def stop(self):
        self.running = False
        self.th.join()
        self.animation()
        self.set_status_label('Netgear Mon stopped...')

    def set_status_label(self, incoming):
        self.status_label.config(text=incoming)

    def animation(self):
        for x in range(120):
            for i in ('/', '--', '|', '\\'):
                self.set_status_label(i)
                self.status_label.update_idletasks()
                sleep(.001)

#! /usr/bin/python3
import urllib.request
from datetime import datetime
from auxiliary.utils import environment


class NetgearRouter:
    def __init__(self):
        self.ip, self.login, self.passwd = environment()

    def _make_request(self):
        r = urllib.request.HTTPPasswordMgrWithDefaultRealm()
        r.add_password(None, self.ip, self.login, self.passwd)
        auth = urllib.request.HTTPBasicAuthHandler(r)
        opener = urllib.request.build_opener(auth)
        urllib.request.install_opener(opener)
        result = opener.open(self.ip)
        return result.read().decode()

    @staticmethod
    def get_date_time():
        return datetime.now().strftime("%Y-%m-%d-%H:%M:%S")

    def scrape_router_ui(self):
        r = self._make_request()
        self.ip += '/RgAttachedDevices.asp'
        r = self._make_request()
        for line in r.split(';'):
            if ('CPE_Info' or 'WLCPE_Info') in line:
                data = line.split('\"')
                if len(data) > 1:
                    yield [i for i in data if i != ','][1:4]

from os import environ


def environment():
    return [
        environ['ROUTER_IP'],
        environ['ROUTER_LOGIN'],
        environ['ROUTER_PASSWD']
    ]