# -*- coding: UTF-8 -*-
from setuphelpers import *
import json
from cryptography.fernet import Fernet
import os
from jinja2 import Environment, FileSystemLoader

uninstallkey = []

variables = {
    'sogo_server': 'mail.server.local'
}

# Read local variables file if available
if isfile(makepath(programfiles32,'wapt','private','symetric.txt')) and isfile(makepath(programfiles32,'wapt','private','variables.txt')):
    print('Reading local encrypted variables file')
    f = Fernet(open(makepath(programfiles32,'wapt','private','symetric.txt'),'r').read())
    variables.update(json.loads(f.decrypt(open(makepath(programfiles32,'wapt','private','variables.txt'),'r').read())))

install_dir = programfiles
if iswin64():
    install_dir = programfiles32
connector_version = '60.0.0'
integrator_version = '60.0.0'

def install():
    for extension in ['connector','integrator']:
        if not isdir(makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-' + extension)):
            mkdirs(makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-' + extension))

    # connector can be extracted as is
    run('"%s" x -y -o"%s" sogo-connector-%s.xpi' % (makepath(programfiles,'7-Zip','7z.exe'), makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-connector'), connector_version))

    # Integrator must be customized, so first extract it
    run('"%s" x -y -o"%s" sogo-integrator-%s-sogo-demo.xpi' % (makepath(programfiles,'7-Zip','7z.exe'), makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-integrator'), integrator_version))
    # And replace the extensions.rdf content
    jinja2 = Environment(
        loader=FileSystemLoader(os.path.dirname(os.path.abspath(__file__))),
        trim_blocks=True
    )
    open(makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-integrator','chrome','content','extensions.rdf'),'w').write(
        jinja2.get_template('extensions.rdf.j2').render(
            sogo_server = variables['sogo_server']
        )
    )
    # Also remove custom settings
    os.unlink(makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-integrator','defaults','preferences','site.js'))


def uninstall():
    print('Removing SOGo extensions for Thunderbird')
    for extension in ['connector','integrator']:
        remove_tree(makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-' + extension))

def audit():
    try:
        run('type "%s" | find /I "em:version=" | find "%s"' % (makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-connector','install.rdf'), connector_version))
    except:
        print('SOGO connector version check failed')
        return "ERROR"
    try:
        run('type "%s" | find /I "em:version=" | find "%s"' % (makepath(install_dir,'Mozilla Thunderbird','distribution','bundles','sogo-integrator','install.rdf'), integrator_version))
    except:
        print('SOGO integrator version check failed')
        return "ERROR"
    return "OK"