# -*- coding: utf-8 -*-
# code: BY MOHAMED_OS


from __future__ import print_function

import tarfile
from datetime import datetime
from os import chdir, chmod, remove, stat, system, uname
from os.path import exists, isfile, join
from re import MULTILINE, findall
from sys import version_info
from time import sleep

if version_info[0] == 3:
    from urllib.error import HTTPError, URLError
    from urllib.request import Request, urlopen, urlretrieve
else:
    from urllib import urlretrieve

    from urllib2 import HTTPError, Request, URLError, urlopen

# colors
C = "\033[0m"     # clear (end)
R = "\033[0;31m"  # red (error)
G = "\033[0;32m"  # green (process)
B = "\033[0;36m"  # blue (choice)
Y = "\033[0;33m"  # yellow (info)


class Setting():
    link = 'https://raw.githubusercontent.com/MOHAMED19OS/Enigma2_Store/main/Channel/'
    page = 'https://github.com/MOHAMED19OS/Enigma2_Store/tree/main/Channel'

    def __init__(self):
        self.date = datetime.now().strftime("%d-%m-%Y %X")
        self.path_abertis = '/etc/astra/scripts/abertis'
        self.path_astra = '/etc/astra/astra.conf'

    def banner(self):
        system('clear')
        print(B, r"""

 .d8888b.  888                                          888
d88P  Y88b 888                                          888
888    888 888                                          888
888        88888b.   8888b.  88888b.  88888b.   .d88b.  888
888        888 "88b     "88b 888 "88b 888 "88b d8P  Y8b 888
888    888 888  888 .d888888 888  888 888  888 88888888 888
Y88b  d88P 888  888 888  888 888  888 888  888 Y8b.     888
 "Y8888P"  888  888 "Y888888 888  888 888  888  "Y8888  888
""", C, end='')
        print("   Install\n".rjust(30))
        print("Written by {}MOHAMED_OS{} {}(͡๏̯͡๏){}\n".format(
            B, C, Y, C).rjust(73), end='')
        print((self.date).rjust(35))
        sleep(2)

    def info(self):
        try:
            req = Request(self.page)
            req.add_header(
                'User-Agent', 'Mozilla/5.0 (X11; Linux x86_64; rv:103.0) Gecko/20100101 Firefox/103.0')
            response = urlopen(req)
            link = response.read().decode('utf-8')
            data_version = findall('tar.gz["\'],["\']path["\']:["\']Channel/channels_backup_user_(.+?).tar.gz["\']', link)[0]
            data_name = findall('tar.gz["\'],["\']path["\']:["\']Channel/(.+?)["\']', link)[0]
            return data_version, data_name
        except HTTPError as e:
            print('HTTP Error code: ', e.code)
        except URLError as e:
            print('URL Error: ', e.reason)

    def image(self):
        return exists('/etc/opkg/opkg.conf')

    def check(self, pkg):
        with open('/var/lib/opkg/status') as file:
            for item in file.readlines():
                if item.startswith('Package:'):
                    if findall(pkg, item[item.index(' '):].strip(), MULTILINE):
                        return True
            file.close()

    def delete(self):
        print('{}(?){} Now It Will be deleted Old Settings And Add The New'.format(B, C))

        for file in ['lamedb', '*list', '*.tv', '*.radio', 'satellites.xml']:
            if file != 'satellites.xml':
                self.path_dir = '/etc/enigma2/'
            else:
                self.path_dir = '/etc/tuxbox/'
            if isfile(join(self.path_dir, file)):
                remove(join(self.path_dir, file))

        sleep(0.8)
        urlretrieve('http://127.0.0.1/web/servicelistreload?mode=0')

    def main(self):
        self.banner()

        if self.image():
            if not self.check('astra-sm'):
                system('clear')
                print("   >>>>   {}Please Wait{} while we Install {}astra-sm{} ...".format(
                    G, C, Y, C))
                system('opkg install astra-sm')

        chdir('/tmp')

        if isfile(self.info()[1]):
            remove(self.info()[1])

        system('clear')
        self.banner()

        print("\n{}Downloading{} And Installing Channel {}Please Wait{} {}......{}".format(
            Y, C, R, C, G, C))

        try:
            urlretrieve("".join([self.link, self.info()[1]]),
                        filename=self.info()[1])

            self.delete()
            sleep(0.8)

            with tarfile.open(self.info()[1]) as tar_ref:
                for member in tar_ref.getmembers():
                    tar_ref.extract(member, "/")
            tar_ref.close()

            if isfile(self.info()[1]):
                remove(self.info()[1])
        except:
            print('No File Found')
            exit()

        print('{}(?){} Reload UserBouquets and LameDB'.format(B, C))
        urlretrieve('http://127.0.0.1/web/servicelistreload?mode=0')

        if build.image():
            with open('/etc/sysctl.conf', 'r+') as f:
                if not findall('net.core.rmem_default', f.read(), MULTILINE):
                    f.writelines("""\n
## added for Abertis ###
net.core.rmem_default = 16777216
net.core.rmem_max = 16777216
net.core.wmem_default = 16777216
net.core.wmem_max = 16777216
net.ipv4.udp_mem = 8388608 12582912 16777216
net.ipv4.tcp_rmem = 4096 87380 8388608
net.ipv4.tcp_wmem = 4096 65536 8388608
net.ipv4.tcp_tw_recycle = 0""")
            f.close()

            for file_dir in [self.path_abertis, self.path_astra]:
                if exists(file_dir):
                    remove(file_dir)

            urlretrieve(
                "".join([self.link, self.path_astra[11:]]), self.path_astra)

            for name in ['aarch64', 'armv7l', 'mips', 'sh4']:
                if uname()[4] == name:
                    urlretrieve(
                        "".join([self.link, join('astra-sm/', name, self.path_abertis[19:])]), self.path_abertis)

            for file_chmod in [self.path_abertis, self.path_astra]:
                if oct(stat(file_chmod).st_mode)[-3:] != '755':
                    chmod(file_chmod, 0o755)

            print('{}(?){} Device will reboot now'.format(B, C))
            sleep(4)
            system('reboot -f')
        else:
            print('{}(?){} Device will restart now'.format(B, C))
            system('systemctl restart enigma2')


if __name__ == '__main__':
    build = Setting()
    build.info()
    build.main()