# -*- coding: utf-8 -*-
from setuphelpers import *
import base64
import json

"""
Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()

"""
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
bin_contains = "postgresql-"

def install():
    # Declaring local variables
    package_version = control.get_software_version()
    bin_name = glob.glob("*%s*.exe" % bin_contains)[0]
    app_uninstallkey = "PostgreSQL %s" % package_version.split('.')[0]
    silent_args = "--unattendedmodeui none --mode unattended --enable-components server"

    # Removing old version
    for soft in installed_softwares("PostgreSQL %s" % package_version):
        if Version(soft["version"]) < Version(package_version):
            print("Uninstalling old version: %s" % soft["version"])
            run(uninstall_cmd(soft["key"]))


    # Installing the software
    print("Installing: %s" % bin_name)
    install_exe_if_needed(
        bin_name,
        silentflags=silent_args,
        key=app_uninstallkey,
        min_version=bin_name.replace('postgresql-','').replace('-windows-x64.exe',''),         # version displayed in registry contains "-"
        timeout=600,
    )

    # Adding quiet uninstall string to registry
    uninstallstring = installed_softwares(app_uninstallkey)[0]['uninstall_string']
    register_uninstall(app_uninstallkey,quiet_uninstall_string=f'{uninstallstring} --mode unattended')


                                

# -*- coding: utf-8 -*-
from setuphelpers import *
import os

r"""
Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()

"""

def update_package():
    package_updated = False
    proxies = get_proxies_from_wapt_console()
    if not proxies:
        proxies = get_proxies()
    app_name = control.name
    url = "https://www.enterprisedb.com/downloads/postgres-postgresql-downloads"

    # Send a GET request to the URL
    response = requests.get(url)
    # Check if the request was successful
    if response.status_code == 200:
        # Find the <tbody> tag
        tbody = bs_find(
                url,
                "tbody",
                user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:91.0) Gecko/20100101 Firefox/91.0",
                proxies=proxies,
                features='html.parser',
                )

        if tbody:
            # Find the first line in the tbody
            first_row = tbody.find('tr')
            if first_row:
                # Find all columns in first row
                columns = first_row.find_all('td')
                # Check if there are at least 5 columns
                if len(columns) >= 5:
                    # Check the contents of the first column
                    version = str(columns[0].text.strip())
                    print("Contenu de la première colonne :", version)
                    # Retrieve the 5th column
                    fifth_column = columns[4]
                    # Find an <a> tag with a href starting with "https"
                    download_url = fifth_column.find('a', href=lambda href: href and href.startswith("https"))
                    if download_url:
                        # Show found link
                        print("Lien trouvé dans la 5ème colonne :", download_url['href'])
                        # We prefer keeping the original binary name because it contains the version displayed in Windows registry
                        bin_name = requests.head(download_url['href']).headers['Location'].split('/')[-1]
                        wget(download_url['href'], bin_name)
                        control.set_software_version(version)
                        control.save_control_to_wapt()
                        package_updated = True
                        print("Software version is up-to-date (%s)" % Version(version))
                    else:
                        print("Aucun lien commençant par 'https' n'a été trouvé dans la 5ème colonne.")
                else:
                    print("Il n'y a pas suffisamment de colonnes dans la première ligne.")
            else:
                print("La première ligne dans <tbody> n'a pas été trouvée.")
        else:
            print("Aucun <tbody> n'a été trouvé sur la page.")
    else:
        print("Erreur lors de la requête :", response.status_code)

    remove_outdated_binaries(version, filename_contains="postgresql-")
    return package_updated