#!/usr/bin/env python3
from html.parser import HTMLParser
import urllib.request
from html.entities import name2codepoint
import re
import datetime
import argparse
import os
import time
import glob
import sys
import subprocess

# ===========ARGUMENT PARSING==============
parser = argparse.ArgumentParser()
parser.add_argument("--versionNumber", type=str, help="Specify version number to search for")

args = parser.parse_args()
if args.versionNumber:
    ver = args.versionNumber
else:
    parser.print_help()
    parser.exit()

# =============HTML PARSING================
url = "http://jed-centos7/pld/"
with urllib.request.urlopen(url) as response:
    html = response.read()

webpage = html.decode('utf-8')
# print(webpage)

searchPattern = '(LANforgeGUI_(\d+\.\d+\.\d+)_Linux64.tar.bz2)<\/a><\/td><td align="right">(\d+\-\d+\-\d+\ \d+\:\d+)'
searchResults = re.findall(searchPattern, webpage)

webFiles = []

for file in searchResults:
    if ver == file[1]:
        webFiles.append({'filename': file[0], 'timestamp': datetime.datetime.strptime(file[2], "%Y-%m-%d %H:%M")})
if len(webFiles) == 0:
    print("Failed to find webfile with version number %s" % (ver))
    sys.exit(1)

# =========CHECK DIR FOR FILES=============
filePath = "/home/lanforge/Downloads/"
dir = glob.glob(filePath + "LANforgeGUI_%s*" % ver)
dirFiles = []

for file in dir:
    if ver in file:
        fileTime = datetime.datetime.strptime(time.ctime(os.stat(file).st_ctime),
                                              "%a %b %d %H:%M:%S %Y")  # Fri May  8 08:31:43 2020
        dirFiles.append({'filename': file[25:], 'timestamp': fileTime})

if len(dirFiles) == 0:
    print("Unable to find file in {filePath} with version %s" % ver)


# sys.exit(1)


# ============FIND NEWEST FILES============
def findNewestVersion(filesArray):
    newest = filesArray[0]
    if len(filesArray) > 0:
        for file in filesArray:
            if file['timestamp'] > newest['timestamp']:
                newest = file

    return newest


newestWebFile = findNewestVersion(webFiles)
if len(dirFiles) != 0:
    newestDirFile = findNewestVersion(dirFiles)
else:
    newestDirFile = {'filename': 'placeholder', 'timestamp': datetime.datetime.strptime("0", "%H")}

# =======COMPARE WEB AND DIR FILES=========
if newestWebFile['timestamp'] > newestDirFile['timestamp']:
    try:
        if newestDirFile['filename'] != 'placeholder':
            subprocess.call(["rm", "%s%s" % (filePath, newestDirFile['filename'])])
            print("No file found")
            print("Downloading newest %s from %s" % (newestWebFile['filename'], url))
        else:
            print("Found newer version of GUI")
            print("Downloading %s from %s" % (newestWebFile['filename'], url))
        # =====ATTEMPT DOWNLOAD AND INSTALL=========
        subprocess.call(
            ["curl", "-o", "%s%s" % (filePath, newestWebFile['filename']), "%s%s" % (url, newestWebFile['filename'])])
        time.sleep(5)
    except Exception as e:
        print("%s Download failed. Please try again." % e)
        sys.exit(1)
    try:
        print("Attempting to extract files")
        subprocess.call(["tar", "-xf", "%s%s" % (filePath, newestWebFile['filename']), "-C", "/home/lanforge/"])
    except Exception as e:
        print("%s\nExtraction failed. Please try again" % e)
        sys.exit(1)

    # time.sleep(90)
    try:
        if "/home/lanforge/.config/autostart/LANforge-auto.desktop" not in glob.glob(
                "/home/lanforge/.config/autostart/*"):
            print("Copying LANforge-auto.desktop to /home/lanforge/.config/autostart/")
            subprocess.call(["cp", "/home/lanforge/%s/LANforge-auto.desktop" % (
                newestWebFile['filename'][:len(newestWebFile) - 18]), "/home/lanforge/.config/autostart/"])
    except Exception as e:
        print("%s\nCopy failed. Please try again" % e)
        sys.exit(1)

    try:
        print("Attempting to install %s at /home/lanforge" % newestWebFile['filename'])
        os.system("cd /home/lanforge/%s; sudo bash lfgui_install.bash" % (
            newestWebFile['filename'][:len(newestWebFile) - 18]))
    except Exception as e:
        print("%s\nInstallation failed. Please Try again." % e)
        sys.exit(1)
# =========ATTEMPT TO RESTART GUI==========
#	try:
#		print("Killing current GUI process")
#		os.system("if pgrep java; then pgrep java | xargs kill -9 ;fi")
#	except Exception as e:
#		print("%s\nProcess kill failed. Please try again" % e)
#		sys.exit(1)

else:
    print("Current GUI version up to date")
    sys.exit(0)