#!/usr/bin/env python3

import os,sys,subprocess,os.path

import assets.tools
from assets.tools import exec, patch

#
# main functions
#


def deps_win32():
    exec('rustup target add i686-pc-windows-msvc')


def fetch():
    exec('wget -q -O version https://gitlab.com/librewolf-community/browser/source/-/raw/main/version')
    exec('wget -q -O source_release https://gitlab.com/librewolf-community/browser/source/-/raw/main/release')
    exec('wget -q -O librewolf-$(cat version)-$(cat source_release).source.tar.gz.sha256sum https://gitlab.com/librewolf-community/browser/source/-/jobs/artifacts/main/raw/librewolf-$(cat version)-$(cat source_release).source.tar.gz.sha256sum?job=Build')
    exec('wget -q -O librewolf-$(cat version)-$(cat source_release).source.tar.gz https://gitlab.com/librewolf-community/browser/source/-/jobs/artifacts/main/raw/librewolf-$(cat version)-$(cat source_release).source.tar.gz?job=Build')
    exec('sha256sum -c librewolf-$(cat version)-$(cat source_release).source.tar.gz.sha256sum')
    exec('cat librewolf-$(cat version)-$(cat source_release).source.tar.gz.sha256sum')


    
def build(debug=False):
    
    exec('rm -rf librewolf-$(cat version)-$(cat source_release)')
    exec('tar xf librewolf-$(cat version)-$(cat source_release).source.tar.gz')
    
    with open('version','r') as file:
        version = file.read().rstrip()
        with open('source_release','r') as file:
            source_release = file.read().rstrip()
            os.chdir('librewolf-{}-{}'.format(version,source_release))

            # patches
            if debug:
                exec('cp -v ../assets/mozconfig.windows.debug mozconfig')
            else:
                exec('cp -v ../assets/mozconfig.windows mozconfig')

                # two patches for windows only, currently
                patch('../assets/disable-verify-mar.patch')
                patch('../assets/package-manifest.patch')

                # perform the build and package
                exec('MACH_USE_SYSTEM_PYTHON=1 ./mach build')
                exec('MACH_USE_SYSTEM_PYTHON=1 ./mach package')
                os.chdir('..')



def artifacts():

    # Trying to fix issue #146 -> https://gitlab.com/librewolf-community/browser/windows/-/issues/146
    # (keep this False for now)
    _with_app_name = False
    
    with open('version','r') as file1:
        version = file1.read().rstrip()
        source_release = ''
        with open('source_release','r') as file3:
            source_release = file3.read().rstrip()
        buildzip_filename = 'firefox-{}-{}.en-US.win64.zip'.format(version,source_release)
        if _with_app_name:
            buildzip_filename = 'librewolf-{}-{}.en-US.win64.zip'.format(version,source_release)
        exec('cp -v librewolf-{}-{}/obj-x86_64-pc-mingw32/dist/{} .'.format(version,source_release,buildzip_filename))
        exec('cp -v librewolf-{}-{}/obj-x86_64-pc-mingw32/dist/install/sea/firefox-{}-{}.en-US.win64.installer.exe .'.format(version,source_release,version,source_release))
        exec('rm -rf work && mkdir work')
        os.chdir('work')
        exec('unzip -q ../{}'.format(buildzip_filename))
        if not _with_app_name:
            exec('mv firefox librewolf')
        os.chdir('librewolf')
        if not _with_app_name:
            exec('mv firefox.exe librewolf.exe')
        os.chdir('..')
        os.chdir('..')

        # let's get 'release'.
        with open('release','r') as file2:
            release = file2.read().rstrip()
            if release == '0' :
                full_version = '{}'.format(version)
            else:
                full_version = '{}.{}'.format(version,release)

            # let's copy in the .ico icon.
            exec('cp -v assets/librewolf.ico work/librewolf')

            # Let's make the portable zip first.
            do_zip = True
            if do_zip:
                os.chdir('work')
                exec('rm -rf librewolf-{}'.format(version))
                os.makedirs('librewolf-{}/Profiles/Default'.format(version), exist_ok=True)
                os.makedirs('librewolf-{}/LibreWolf'.format(version), exist_ok=True)
                exec('cp -r librewolf/* librewolf-{}/LibreWolf'.format(version))
                exec('wget -q -O librewolf-{}/librewolf-portable.exe https://gitlab.com/librewolf-community/browser/windows/uploads/64b929c39999d00efb56419f963e1b22/librewolf-portable.exe'.format(version))
                zipname = 'librewolf-{}.en-US.win64.zip'.format(full_version)
                exec("rm -f ../{}".format(zipname))
                exec("zip -qr9 ../{} librewolf-{}".format(zipname,version))            
                os.chdir('..')

            # With that out of the way, we need to create the main nsis setup.
            os.chdir('work')
            exec("mkdir x86-ansi")
            exec("wget -q -O ./x86-ansi/nsProcess.dll https://shorsh.de/upload/we7v/nsProcess.dll")
            setupname = 'librewolf-{}.en-US.win64-setup.exe'.format(full_version)
            exec('sed \"s/pkg_version/{}/g\" < ../assets/setup.nsi > tmp.nsi'.format(full_version))
            exec('cp -v ../assets/librewolf.ico .')
            exec('cp -v ../assets/banner.bmp .')
            exec('makensis-3.01.exe -V1 tmp.nsi')
            exec('rm -rf tmp.nsi librewolf.ico banner.bmp x86-ansi')
            exec("mv {} ..".format(setupname))
            os.chdir('..')

            # Latest addition: better portable app
            os.chdir('work')

            exec('rm -rf librewolf-{}'.format(version))
            os.makedirs('librewolf-{}/Profiles/Default'.format(version), exist_ok=True)
            os.makedirs('librewolf-{}/LibreWolf'.format(version), exist_ok=True)
            exec('cp -r librewolf/* librewolf-{}/LibreWolf'.format(version))
            # on gitlab: https://gitlab.com/ltGuillaume
            exec('git clone https://github.com/ltGuillaume/LibreWolf-Portable')
            exec('cp -v LibreWolf-Portable/LibreWolf-Portable.* LibreWolf-Portable/*.exe librewolf-{}/'.format(version))
            os.chdir('librewolf-{}'.format(version))
            # installed from: https://www.autohotkey.com/
            exec('echo \\"c:/Program Files/AutoHotkey/Compiler/Ahk2Exe.exe\\" /in LibreWolf-Portable.ahk /icon LibreWolf-Portable.ico > tmp.bat')
            exec('cmd /c tmp.bat')
            exec('rm -f tmp.bat')
            # let's remove the ahk and icon and embedded executables
            exec('rm -f LibreWolf-Portable.ahk LibreWolf-Portable.ico dejsonlz4.exe jsonlz4.exe')
            os.chdir('..')

            pa_zipname = 'librewolf-{}.en-US.win64-portable.zip'.format(full_version)
            exec("rm -f ../{}".format(pa_zipname))
            exec("zip -qr9 ../{} librewolf-{}".format(pa_zipname,version))            
            
            os.chdir('..')




# Utility function to upload() function.
def do_upload(filename,token):
    exec('echo _ >> upload.txt')
    exec('curl --request POST --header \"PRIVATE-TOKEN: {}\" --form \"file=@{}\" \"https://gitlab.com/api/v4/projects/13852981/uploads\" >> upload.txt'.format(token,filename),False)
    exec('echo _ >> upload.txt')


def upload(token):

    with open('version','r') as file1:
        version = file1.read().rstrip()
        with open('release','r') as file2:
            release = file2.read().rstrip()
            if release == '0' :
                full_version = '{}'.format(version)
            else:
                full_version = '{}.{}'.format(version,release)

            # Files we need to upload..
            if do_zip:
                zip_filename = 'librewolf-{}.en-US.win64.zip'.format(full_version)
            setup_filename = 'librewolf-{}.en-US.win64-setup.exe'.format(full_version)
            pazip_filename = 'librewolf-{}.en-US.win64-portable.zip'.format(full_version)
            if do_zip:
                exec('sha256sum {} {} {} > sha256sums.txt'.format(setup_filename,zip_filename,pazip_filename))
            else:
                exec('sha256sum {} {} > sha256sums.txt'.format(setup_filename,pazip_filename))
            exec('rm -f upload.txt')
            do_upload(setup_filename,token)
            if do_zip:
                do_upload(zip_filename,token)
            do_upload(pazip_filename,token)
            do_upload('sha256sums.txt',token)

            
#
# parse commandline for commands
#

help_msg = '''
Use: ./mk.py <command> ...

commands:
  fetch
  build
  build-debug
  artifacts
  upload <token>

'''

done_something = False

in_upload=False
for arg in sys.argv:
    if in_upload:
        upload(arg)
        done_something=True
    elif arg == 'fetch':
        fetch()
        done_something = True
    elif arg == 'build':
        build()
        done_something = True
    elif arg == 'build-debug':
        build(True)
        done_something = True
    elif arg == 'artifacts':
        artifacts()
        done_something = True
    elif arg == 'upload':
        in_upload = True
    else:
        if arg == sys.argv[0]:
            pass
        else:
            print(help_msg)
            sys.exit(1)


        
if done_something:
    sys.exit(0)
    
print(help_msg)
sys.exit(1)