c3mm.py

# Cossacks 3 (Text) Mod Manager version 1.3.0
# Require python 3.5.x to work : https://python.org/downloads
# Get all update at http://www.cossacks3.com/forum/index.php?threads/cossacks-3-text-mod-manager.17753

"""
# === News in 1.3.0 === #
    Re-writing the script, simplify and normalize data use and adding some comentary
    Now mods and configs files are stored in .json files
    Now mods are check before install
    Now give just the name of the mod without extention (mod in place of mod.json)
    Patching file function are modified to save file in case of an unexpected error during patching
    If the program failed to uninstall a mod, it will restore all corupted files
    New command uninstall_all
    New command giveall, who print all installed mod
    Now logs are saved in "c3mm/logs.txt"

    Warning : Don't modify a file (already modded with c3mm) by yourself! I know isn't convenient for modder, so I'm working on a Cossacks3 (Text) Mod Creator.
"""


"""
Structur uses
Mod  := {'name':name, 'version':version, 'info':info, 'file':{name => list of (str old, str new)}}
Data := {'mod':{name => mod mod}, 'file':{name => list of str mod}}
"""

import json, sys, os, time

CONFIG_PATH = "c3mm/"
MOD_PATH    = "mod/"
COPY_PATH   = "c3mm/copy/"
BCKUP_PATH  = "c3mm/bckup/"
INFO_PATH   = "c3mm/c3mm.json"
LOG_PATH    = "c3mm/log.txt"
HELP = """
Cossacks3 (text) Mod Manager Version 1.3.0.
Type python c3mm.py action [parameter]
    action:
        install modname   -> intall a mod find by name
        uninstall modname -> unintall a mod find by name
        help              -> show this message
        uninstall_all     -> uninstall all mod
"""

PATH = [CONFIG_PATH, MOD_PATH, COPY_PATH, BCKUP_PATH]
FILE = [INFO_PATH, LOG_PATH]
for path in PATH:
    try:
        os.mkdir(path)
    except:
        pass
for file in FILE:
    try:
        open(file, 'r')
    except:
        open(file, 'w')

class Alert:
    linfo = []
    lerror = []

    def info(i):
        Alert.linfo.append(i)
        print(i)

    def error(e, data):
        Alert.lerror.append(e)
        print(e)
        quit(data)

    def input(m):
        return input(m)

    def list(l):
        for i in l:
            print(i)

def loadjson(path):
    return json.loads(open(path, 'r').read())

def savejson(path, data):
    open(path, "w").write(json.dumps(data))

def mkPath(path):
    rpath = ""
    for dir in path.split('/')[:-1]:
        rpath += dir
        try:
            os.mkdir(rpath)
        except:
            pass
        rpath += '/'

def copyFile(file, path):
    mkPath(path)
    open(path, "w").write(open(file, 'r').read())

def checkMod(mod):
    """
    checkMod function, use to check if a mod is valid
    Mod mod     -> mod to check
    Bool return -> is the mod valid?
    """
    for file in mod['file']:
        Alert.info("Check existence of '" + file + "'")
        try:
            f = open(file, 'r').read()
            Alert.info("'" + file + "' exist.")
        except:
            Alert.info("'" + file + "' don't exist.")
            return False

        for old in mod['file'][file]:
            Alert.info("Check exsitence of '" + old + "' in '" + file + "'")
            if not old in f:
                Alert.info("Not found.")
                return False
            else:
                Alert.info("Found.")
    return True

def apply(mod):
    """
    apply function, use to modify files
    Mod mod -> mod that contain all modifications
    None return
    """
    mfile = {}
    for file in mod['file']:
        Alert.info("Patching '" + file + '"')
        f = open(file, 'r').read()

        #Calculate modifications
        for old in mod['file'][file]:
            new = mod['file'][file][old]
            f = f.replace(old, new)

        mfile[file] = f

    #Apply mod
    for file in mfile:
        open(file, 'w').write(mfile[file])

def install(mod, data):
    """
    install function, use to add a mod.
    Mod mod   -> mod to install
    Data data -> c3mm main data
    None return
    """
    if not checkMod(mod):
        Alert.error("'" + mod['name'] + "' mod version " + mod['version'] + " isn't valid", data)

    Alert.info("Install '" + mod['name'] + "' version " + mod['version'])

    #Add mod to list of installed mod
    data['mod'][mod['name']] = mod

    #Adding mod in list of file modification
    for file in mod['file']:
        if not file in data['file']:
            data['file'][file] = [mod['name']]
            #backup file
            copyFile(file, BCKUP_PATH + file)
        else:
            data['file'][file].append(mod['name'])

    apply(mod)

    for file in mod['file']:
        #make a save of the file
        copyFile(file, COPY_PATH + file)

def uninstall(mod, data):
    """
    uninstall function, use to delete a mod.
    Mod mod   -> mod to uninstall
    Data data -> c3mm main data
    None return
    """
    uninstallmod = mkuninstallmod(mod)

    if not checkMod(uninstallmod):
        Alert.info("'" + mod['name'] + "' mod uninstaller isn't valid.")
        Alert.info("Do you want restore all files concerned by this mod?")
        Alert.info("List of mod will uninstalled in this case:")
        modlist = []
        for file in mod['file']:
            for mod in data['file'][file]:
                if not mod in modlist:
                    modlist.append(mod)
                    Alert.info(mod)
        choice = Alert.chocie("Restore all file? yes/no:")
        if choice == 'no':
            quit(data)

        filelist = [file for file in mod['file']]
        for file in filelist:
            Alert("Restore '" + file + "'")
            copyFile(BCKUP_PATH + file, file)
            data['mod'][mod]['file'].remove(file)

        for mod in modlist():
            uninstall(mod, data)



    #Delete mod to list of installed mod
    del data['mod'][mod['name']]

    #Delete mod of list of file modification
    for file in data['file']:
        if mod['name'] in data['file'][file]:
            data['file'][file].remove(mod['name'])

    Alert.info("Uninstall '" + mod['name'] + "' version " + mod['version'])

    apply(uninstallmod)

def mkuninstallmod(mod):
    """
    mkuninstallmod function, use to make an uninstalator from a mod
    Mod mod    -> mod to uninstall
    Mod return -> uninstalator mod
    """
    uninstallmod = {}
    uninstallmod['file'] = {}

    for file in mod['file']:
        uninstallmod['file'][file] = {}
        for old in mod['file'][file]:
            new = mod['file'][file][old]
            #For each modification of mod, inverse new and old
            uninstallmod['file'][file][new] = old

    return uninstallmod

def checkStatu(data):
    """
    checkStatu funtion, check if a modded file has been modified since last launch of this program
    Data data   -> main data
    Bool return ->
    """
    badfile = []
    #Check all modded file
    for file in data['file']:
        if open(file, 'r').read() != open(COPY_PATH + file, 'r').read():
            Alert.info("File '" + file + "' has been modified since last launch of c3mm.")
            badfile.append(file)

    badmod = []
    #Uninstall all mod concerned by these files
    for file in badfile:
        for mod in data['file'][file]:
            if not mod in badmod:
                badmod.append(mod)

    for mod in badmod:
        _mod = data['mod'][mod]
        for file in badfile:
            if file in _mod['file']:
                del _mod['file'][file]
        uninstall(_mod, data)

    return len(badmod) == 0

def quit(data):

    dfile = []
    for file in data['file']:
        if data['file'][file] == []:
            dfile.append(file)

    for file in dfile:
        del data['file'][file]

    savejson(INFO_PATH, data)

    logs = time.strftime('%D %H:%M\n',time.localtime())
    for info in Alert.linfo:
        logs += info + '\n'
    for error in Alert.lerror:
        logs += error + '\n'

    open(LOG_PATH, "a").write(logs)
    exit()

def main(argc, argv):

    try:
        data = loadjson(INFO_PATH)
    except:
        open(INFO_PATH, "w").write('{"mod":{}, "file":{}}')
        data = loadjson(INFO_PATH)

    if not checkStatu(data):
        quit(data)

    elif argc == 1 or argv[1] == "help":
        print(HELP)

    elif argv[1] == 'install':

        if argc < 3:
            Alert.error("You must give a mod name", data)

        mod = loadjson(MOD_PATH + argv[2] + '.json')
        modname = mod['name']

        if modname in data['mod']:
            if mod["version"] == data["mod"][modname]["version"]:
                Alert.error("Mod '" + modname + "' version " + mod["version"] + " is already install.", data)
            else:
                Alert.info("Mod '" + modname + "' version " + data["mod"][modname]["version"] + " is already install. Do you want change it for version " + mod["version"] + "?")
                choice = Alert.input("yes/no:")
                if choice == "no":
                    quit()

                uninstall(data['mod'][modname], data)

        install(mod, data)

    elif argv[1] == 'uninstall':

        if argc < 3:
            Alert.error("You must give a mod name", data)

        mod = loadjson(MOD_PATH + argv[2] + '.json')
        modname = mod['name']
        uninstall(data['mod'][modname], data)

    elif argv[1] == 'uninstall_all':

        Alert.info("Restore all files")
        for file in data['file']:
            Alert.info("Restore '" + file + "'")
            copyFile(BCKUP_PATH + file, file)

        data['file'] = {}
        data['mod'] = {}

    elif argv[1] == 'giveall':
        Alert.list([mod + " version " + data['mod'][mod]['version'] for mod in data['mod']])

    quit(data)

if __name__ == '__main__':
    main(len(sys.argv), sys.argv)