main.py

import os
import json
import hashlib
import shutil
import glob
import constants
import datetime
from git import Repo
from packaging import version

####### Functions
def readDatasFromFile(filePath):
    return json.loads(open(filePath, 'rb').read())

def writeDatasIntoFile(filePath, datas):
    with open(filePath, 'w') as f:
        json.dump(datas, f)

def getMd5Checksum(filePath):
    return hashlib.md5(open(filePath,'rb').read()).hexdigest()

def cleanAllKeys():
    # remove old root keys
    fileKeysPath = glob.glob('/tmp/repo/AANM/keys/*')
    if len(fileKeysPath) > 1:
        oldest_file = min(fileKeysPath, key=os.path.getctime)
        os.remove(oldest_file)

    # remove old addons keys
    allAddonsKeys = {}
    fileAddonsKeysPath = glob.glob('/tmp/repo/AANM/addons/*.bisign')

    for fileKeyPath in fileAddonsKeysPath:
        filename = fileKeyPath.replace('/tmp/repo/AANM/addons/', '')
        filenameSplit = filename.split('.')
        name = filenameSplit[0]
        curVersion = filenameSplit[3] + '.' + filenameSplit[4]

        if name in allAddonsKeys :
            if version.parse(allAddonsKeys[name]['version']) > version.parse(curVersion) :
                os.remove(fileKeyPath)
            else:
                os.remove(allAddonsKeys[name]['filePath'])

        allAddonsKeys[name] = {
            'version' : curVersion,
            'filePath' : fileKeyPath
        }
    
    # remove old optionals addons keys
    allOptionalsAddonsPath = glob.glob('/tmp/repo/AANM/optionals/*')

    for optionalAddonPath in allOptionalsAddonsPath:
        fileList = glob.glob(optionalAddonPath + '/addons/*.bisign')
        if len(fileList) > 1:
            oldest_file = min(fileList, key=os.path.getctime)
            os.remove(oldest_file)

def updatemod(datas, aceZipChecksum):
    # git clone
    cloned_repo = Repo.clone_from('https:// ' + constants.GITHUB_USERNAME + ':' + constants.GITHUB_TOKEN + '@github.com/Vindicta-Team/Automated-Ace-No-Medical', '/tmp/repo')

    if os.path.isdir('/tmp/repo/AANM'):
        shutil.rmtree('/tmp/repo/AANM')
    
    shutil.copytree(aceModPath, '/tmp/repo/AANM', False, None, shutil.copy2, False, True)

    # remove medical files
    medicalFileList = glob.glob('/tmp/repo/AANM/addons/ace_medical*')
    for medicalFilePath in medicalFileList:
        os.remove(medicalFilePath)
    
    # remove other files requiring medical
    fireFileList = glob.glob('/tmp/repo/AANM/addons/ace_fire*')
    for fireFilePath in fireFileList:
        os.remove(fireFilePath)
    fireFileList = glob.glob('/tmp/repo/AANM/addons/ace_killtracker*')
    for fireFilePath in fireFileList:
        os.remove(fireFilePath)
        
    cleanAllKeys()
    
    # copy mod files
    shutil.copyfile('/app/aanm-files/mod.cpp', '/tmp/repo/AANM/mod.cpp')
    shutil.copyfile('/app/aanm-files/meta.cpp', '/tmp/repo/AANM/meta.cpp')

    # comit files
    cloned_repo.git.add('*')
    cloned_repo.index.commit("update-" + datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S"))
    cloned_repo.remotes.origin.push()

    # if update done update 
    datas['ace'] = aceZipChecksum
    writeDatasIntoFile(constants.JSON_FILE_DB, datas)
    print('\n\n AANM update done')


####### Init logic

# clean up and update mod
if os.path.isfile('/tmp/ace.zip'):
    os.remove('/tmp/ace.zip')

if os.path.isdir('/tmp/repo'):
    shutil.rmtree('/tmp/repo')

aceModPath = '/app/mods/steamapps/workshop/content/'+constants.ARMA_APPID+'/'+constants.ACE_APPID
if not os.path.exists(aceModPath):
    os.makedirs(aceModPath)

# get last ace version
os.system("cd /home/steam/steamcmd && ./steamcmd.sh +login " + constants.STEAM_USERNAME + " '" + constants.STEAM_PASSWORD + "' +force_install_dir " + constants.INSTALL_DIR + " +workshop_download_item " + constants.ARMA_APPID + " " + constants.ACE_APPID + " validate +quit")

# Zip and checksum it
result = shutil.make_archive('/tmp/ace', 'zip', aceModPath)
aceZipChecksum = getMd5Checksum('/tmp/ace.zip')

if os.path.exists(constants.JSON_FILE_DB):
    # retrieve data
    datas = readDatasFromFile(constants.JSON_FILE_DB)
    # update if needed
    if datas['ace'] != aceZipChecksum:
        updatemod(datas, aceZipChecksum)
else:
    datas = json.loads('{"ace":"123"}')
    updatemod(datas, aceZipChecksum)

print('\n\n AANM check finished \n\n')