CNIRevelator/src/updater.py

# -*- coding: utf8 -*-
"""
********************************************************************************
*                             CNIRevelator                                     *
*                                                                              *
*  Desc:       Application launcher updating system                            *
*                                                                              *
*  Copyright © 2018-2019 Adrien Bourmault (neox95)                             *
*                                                                              *
*  This file is part of CNIRevelator.                                          *
*                                                                              *
*  CNIRevelator is free software: you can redistribute it and/or modify        *
*  it under the terms of the GNU General Public License as published by        *
*  the Free Software Foundation, either version 3 of the License, or           *
*  any later version.                                                          *
*                                                                              *
*  CNIRevelator is distributed in the hope that it will be useful,             *
*  but WITHOUT ANY WARRANTY*without even the implied warranty of               *
*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               *
*  GNU General Public License for more details.                                *
*                                                                              *
*  You should have received a copy of the GNU General Public License           *
* along with CNIRevelator. If not, see <https:*www.gnu.org/licenses/>.         *
********************************************************************************
"""

from win32com.client import Dispatch
from tkinter.messagebox import *
from tkinter import *
import pythoncom
import sys
import time
import os
import shutil
import zipfile
import hashlib
import subprocess
import psutil
import datetime

import critical     # critical.py
import github       # github.py
import ihm          # ihm.py
import logger       # logger.py
import globs        # globs.py
import downloader   # downloader.py
import lang         # lang.py

UPDATE_IS_MADE = False
UPATH = ' '

launcherWindow = ihm.launcherWindowCur

def createShortcut(path, target='', wDir='', icon=''):
    """
    Creates a shortcut for a program or an internet link
    """
    ext = path[-3:]
    if ext == 'url':
        shortcut = file(path, 'w')
        shortcut.write('[InternetShortcut]\n')
        shortcut.write('URL=%s' % target)
        shortcut.close()
    else:
        shell = Dispatch('WScript.Shell')
        shortcut = shell.CreateShortCut(shell.SpecialFolders("Desktop") + r"\{}".format(path))
        shortcut.Targetpath = target
        shortcut.WorkingDirectory = wDir
        if icon == '':
            pass
        else:
            shortcut.IconLocation = icon
        shortcut.save()

def spawnProcess(args, cd):
    """
    Creates a new independant process. Used to launch a new version after update
    """
    subprocess.Popen(args, close_fds=True, cwd=cd, creationflags=subprocess.DETACHED_PROCESS)

def exitProcess(arg):
    """
    Forcefully quits a process. Used to help deletion of an old version or to quit properly
    """
    # Quit totally without remain in memory
    for process in psutil.process_iter():
        if process.pid == os.getpid():
            process.terminate()
    sys.exit(arg)
    
def setUpdateChannel(choice):
    """
    Sets the new update channel and forces new update at next launch
    """
    if choice == "Beta":
        with open(globs.CNIRUrlConfig, 'w') as (configFile):
            configFile.write("{}\n0\n0".format(globs.CNIRBetaURL))
            # Force new update
            try:
                os.remove(globs.CNIRLastUpdate)
            except:
                pass
    elif choice == "Stable":
        with open(globs.CNIRUrlConfig, 'w') as (configFile):
            configFile.write("{}\n0\n0".format(globs.CNIRDefaultURL))
            # Force new update
            try:
                os.remove(globs.CNIRLastUpdate)
            except:
                pass
                
def getUpdateChannel():
    """
    Returns the current update channel
    """
    with open(globs.CNIRUrlConfig, 'r') as (configFile):
        url = configFile.read()
    if not "master" in url:
        return "Beta"
    else:
        return "Stable"
        
def getLatestVersion(credentials):
    """
    Returns the latest version of the software
    """
    
    finalver, finalurl, finalchecksum = [None]*3

    # Global Handlers
    logfile = logger.logCur
     

    # First retrieving the urls !
    while True:
        try:
            # Open the config file
            logfile.printdbg('Reading urlconf.ig')
            with open(globs.CNIRUrlConfig, 'r') as (configFile):
                try:
                    # Reading it
                    reading = configFile.read()
                    # Parsing it
                    urlparsed = reading.split("\n")
                    break

                except Exception as e:
                    raise IOError(str(e))

        except FileNotFoundError:
            logfile.printdbg('Recreate urlconf.ig')
            # Recreating the url file
            try:
                os.mkdir(globs.CNIRFolder + '\\config')
            except:
                pass
            with open(globs.CNIRUrlConfig, 'w') as (configFile):
                configFile.write("{}\n0\n0".format(globs.CNIRDefaultURL))

    # Getting the list of versions of the software
    logfile.printdbg('Retrieving the software versions')
    try:
        os.mkdir(globs.CNIRFolder + '\\downloads')
    except:
        pass
    getTheVersions = downloader.newdownload(credentials, urlparsed[0], globs.CNIRVerStock, "the version repository").download()
    
    logfile.printdbg('Parsing the software versions')
    with open(globs.CNIRVerStock) as versionsFile:
        versionsTab = versionsFile.read().split("\n")[1].split("||")
        logfile.printdbg('Versions retrieved : {}'.format(versionsTab))
    # Choose the newer
    finalver = globs.version.copy()
    for entry in versionsTab:
        if not entry:
            continue
        verstr, url, checksum = entry.split("|")
        # Calculating sum considering we can have 99 sub versions
        ver = verstr.split(".")
        ver = [int(i) for i in ver]
        finalsum = finalver[2] + finalver[1]*100 + finalver[0]*100*100
        sum = ver[2] + ver[1]*100 + ver[0]*100*100
        # Make a statement
        if sum >= finalsum:
            finalver = ver.copy()
            finalurl = url
            finalchecksum = checksum
        else:
            finalurl = url
            finalchecksum = None
    
    return (finalver, finalurl, finalchecksum)


def tessInstall(PATH, credentials):
    # Global Handlers
    logfile = logger.logCur
     
    
    # Verifying that Tesseract is installed
    if not os.path.exists(PATH + '\\Tesseract-OCR5\\'):
        finalver, finalurl, finalchecksum = getLatestVersion(credentials)
        
        if finalurl == None:
            logfile.printerr('Unable to get the Tesseract url')
            return False
        
        tesseracturl = finalurl.replace("CNIRevelator.zip", "tesseract_5.zip")
        
        # WE ASSUME THAT THE MAIN FILE IS CNIRevelator.zip AND THAT THE TESSERACT PACKAGE IS tesseract_5.zip
        logfile.printdbg('Preparing download of Tesseract OCR 4...')
        getTesseract = downloader.newdownload(credentials, tesseracturl, PATH + '\\downloads\\TsrtPackage.zip', "Tesseract 5 OCR Module").download()
        
        try:
            # CHECKSUM
            BUF_SIZE = 65536  # lets read stuff in 64kb chunks!
            
            sha1 = hashlib.sha1()
            
            with open(globs.CNIRFolder + '\\downloads\\TsrtPackage.zip', 'rb') as f:
                while True:
                    data = f.read(BUF_SIZE)
                    if not data:
                        break
                    sha1.update(data)
                    
            check = sha1.hexdigest()
            logfile.printdbg("SHA1: {0}".format(check))
            
            if not check == globs.CNIRTesserHash:
                logfile.printerr("Checksum error")
                return False
            
            # Unzip Tesseract   
            logfile.printdbg("Unzipping the package")
            launcherWindow.printmsg(lang.all[globs.CNIRlang]["Installing the updates"])
            zip_ref = zipfile.ZipFile(PATH + '\\downloads\\TsrtPackage.zip', 'r')
            zip_ref.extractall(PATH)
            zip_ref.close()            
            # Cleanup
            try:
                os.remove(UPATH + '\\downloads\\TsrtPackage.zip')
            except:
                pass                
            return True

        except:
            return False
    else:
        return True

## Main Batch Function
def batch(credentials):
    # Global Handlers
    logfile = logger.logCur
     

    # Get the latest version of CNIRevelator
    finalver, finalurl, finalchecksum = getLatestVersion(credentials)
    
    if finalver == globs.version:
        logfile.printdbg('The software is already the newer version')
        return True

    logfile.printdbg('Preparing download for the new version')

    getTheUpdate = downloader.newdownload(credentials, finalurl, globs.CNIRFolder + '\\downloads\\CNIPackage.zip', "CNIRevelator {}.{}.{}".format(finalver[0], finalver[1], finalver[2])).download()
    
    launcherWindow.printmsg(lang.all[globs.CNIRlang]["Verifying download..."])
    
    # CHECKSUM
    BUF_SIZE = 65536  # lets read stuff in 64kb chunks!
    
    sha1 = hashlib.sha1()
    
    with open(globs.CNIRFolder + '\\downloads\\CNIPackage.zip', 'rb') as f:
        while True:
            data = f.read(BUF_SIZE)
            if not data:
                break
            sha1.update(data)
            
    check = sha1.hexdigest()
    logfile.printdbg("SHA1: {0}".format(check))
    
    if not check == finalchecksum:
        logfile.printerr("Checksum error")
        return False

    # And now prepare install
    global UPATH
    UPATH = globs.CNIRFolder + '\\..\\CNIRevelator' + "{}.{}.{}".format(finalver[0], finalver[1], finalver[2])
    logfile.printdbg("Make place")
    launcherWindow.printmsg(lang.all[globs.CNIRlang]["Preparing installation..."])
    # Cleanup
    try:
        shutil.rmtree(UPATH + 'temp')
    except Exception as e:
        logfile.printdbg('Unable to cleanup : ' +str(e))
    try:
        shutil.rmtree(UPATH)
    except Exception as e:
        logfile.printdbg('Unable to cleanup : ' +str(e))
    # Unzip    
    logfile.printdbg("Unzipping the package")
    launcherWindow.printmsg(lang.all[globs.CNIRlang]["Installing the updates"])
    zip_ref = zipfile.ZipFile(globs.CNIRFolder + '\\downloads\\CNIPackage.zip', 'r')
    zip_ref.extractall(UPATH + "temp")
    zip_ref.close()
    
    # Move to the right place
    shutil.copytree(UPATH + 'temp\\CNIRevelator', UPATH)
    shutil.rmtree(UPATH + 'temp')
    logfile.printdbg('Extracted :' + UPATH + '\\CNIRevelator.exe')    

    # Make a shortcut
    # hide main window
    pythoncom.CoInitialize()
    root = Tk()
    root.withdraw()
    res = askquestion(lang.all[globs.CNIRlang]["Shortcut creation"], lang.all[globs.CNIRlang]["Would you like to create/update the shortcut for CNIRevelator on your desktop ?"])
    if res == "yes":
        createShortcut("CNIRevelator.lnk", UPATH + '\\CNIRevelator.exe', UPATH)
    root.destroy()

    launcherWindow.printmsg(lang.all[globs.CNIRlang]["Success !"])
    
    # Cleanup
    try:
        os.remove(globs.CNIRFolder + '\\downloads\\CNIPackage.zip')
    except:
        pass
    # Time to quit
    launcherWindow.printmsg(lang.all[globs.CNIRlang]["Launching the new version..."])
    global UPDATE_IS_MADE
    UPDATE_IS_MADE = True
    return True

## Main Function
def umain():
    
    # Global Handlers
    logfile = logger.logCur
    
    # Cleaner for the old version if detected
    if len(sys.argv) > 2 and str(sys.argv[1]) == "DELETE":
        globs.CNIRNewVersion = True
        launcherWindow.printmsg(lang.all[globs.CNIRlang]["Deleting old version"])
        logfile.printdbg("Old install detected : {}".format(sys.argv[2]))
        while os.path.exists(str(sys.argv[2])):
            try:
                shutil.rmtree(str(sys.argv[2]))
            except Exception as e:
                logfile.printerr(str(e))
                logfile.printdbg('Trying stop the process !')
                launcherWindow.printmsg('Fail :{}'.format(e))
                try:
                    for process in psutil.process_iter():
                        if process.name() == 'CNIRevelator.exe':
                            logfile.printdbg('Process found. Command line: {}'.format(process.cmdline()))
                            if process.pid == os.getpid():
                                logfile.printdbg("Don't touch us ! {} = {}".format(process.pid, os.getpid()))
                            else:
                                logfile.printdbg('Terminating process !')
                                process.terminate()
                                shutil.rmtree(str(sys.argv[2]))
                                break
                except Exception as e:
                    logfile.printerr(str(e))
                    launcherWindow.printmsg('Fail :{}'.format(e))
        launcherWindow.printmsg(lang.all[globs.CNIRlang]['Starting...'])
    
    # check we want open a file
    elif len(sys.argv) > 1 and str(sys.argv[1]) != "DELETE":
        globs.CNIROpenFile = True
        logfile.printdbg("Command line received : {}".format(sys.argv))
    
    credentials = downloader.newcredentials()
    
    if not credentials.valid:
        if credentials.login == "nointernet":
            logfile.printerr("No Internet Error. No effective update !")
            launcherWindow.printmsg(lang.all[globs.CNIRlang]["No Internet Error. No effective update !"])
        else:
            logfile.printerr("Credentials Error. No effective update !")
            launcherWindow.printmsg(lang.all[globs.CNIRlang]["Credentials Error. No effective update !"])
        time.sleep(2)
        launcherWindow.exit()
        return 0
        
    github.credentials = credentials
    
    # Check if update is needed
    currentDate = datetime.datetime.now()
    
    if os.path.isfile(globs.CNIRLastUpdate):
        with open(globs.CNIRLastUpdate, 'r') as (configFile):
            try:
                # Reading it
                lastUpdate = datetime.datetime.strptime(configFile.read(),"%d/%m/%Y")
            except Exception as e:
                critical.crashCNIR(False)
                time.sleep(3)
                launcherWindow.exit()
                return 1
    else:
        lastUpdate = datetime.datetime(1970,1,1)

    if not globs.CNIRNewVersion and os.path.exists(globs.CNIRFolder + '\\Tesseract-OCR5\\') and (currentDate - lastUpdate).days < 7:
        logfile.printdbg("No need to update, {} days remaining".format(7 - (currentDate - lastUpdate).days))
        launcherWindow.exit()
        return 0
    
    # Update batch
    try:
        try:
            # EXECUTING THE UPDATE BATCH
            success = batch(credentials)
        except Exception as e:
            critical.crashCNIR(False)
            launcherWindow.printmsg('ERROR : ' + str(e))
            time.sleep(3)
            launcherWindow.exit()
            return 1

        if success:
            logfile.printdbg("Software is up-to-date !")
            launcherWindow.printmsg('Software is up-to-date !')
            # Recreating the url file
            lastUpdate = currentDate
            try:
                os.mkdir(globs.CNIRFolder + '\\config')
            except:
                pass
    
            with open(globs.CNIRLastUpdate, 'w') as (configFile):
                try:
                    # Writing it
                    configFile.write("{}/{}/{}".format(currentDate.day, currentDate.month, currentDate.year))
                except Exception as e:
                    critical.crashCNIR(False)
                    time.sleep(3)
                    launcherWindow.exit()
                    return                    
            if UPDATE_IS_MADE:
                launcherWindow.exit()
                return 0
            
        else:
            logfile.printerr("An error occured. No effective update !")
            launcherWindow.printmsg(lang.all[globs.CNIRlang]['An error occured. No effective update !'])
            time.sleep(2)
            launcherWindow.exit()
            return 0         
            
    except:
        critical.crashCNIR(False)
        launcherWindow.exit()
        return 2
        
    try:
        try:
            # INSTALLING TESSERACT OCR
            success = tessInstall(globs.CNIRFolder, credentials)
        except Exception as e:
            critical.crashCNIR(False)
            launcherWindow.printmsg('ERROR : ' + str(e))
            time.sleep(3)
            launcherWindow.exit()
            return 1

        if success:
            logfile.printdbg("Software is up-to-date !")
            launcherWindow.printmsg(lang.all[globs.CNIRlang]['Software is up-to-date !'])
        else:
            logfile.printerr("An error occured. No effective update !")
            launcherWindow.printmsg(lang.all[globs.CNIRlang]['An error occured. No effective update !'])
            time.sleep(2)
            launcherWindow.exit()
            return 0

    except:
        critical.crashCNIR(False)
        launcherWindow.exit()
        return 2

    time.sleep(2)
    launcherWindow.exit()
    return 0