#!/usr/bin/env python3

import os
import glob
import argparse
import shutil
from pathlib import Path

import damask

def copy_and_patch(patch,orig,msc_root,editor):
    try:
        shutil.copyfile(orig,orig.parent/patch.stem)
    except shutil.SameFileError:
        pass
    damask.execute(f'patch {orig.parent/patch.stem} {patch} -b')
    with open(orig.parent/patch.stem) as f_in:
        content = f_in.read()
    with open(orig.parent/patch.stem,'w') as f_out:
        f_out.write(content.replace('%INSTALLDIR%',msc_root).replace('%EDITOR%',editor))


parser = argparse.ArgumentParser(
                  description='Apply DAMASK modification to MSC.Marc/Mentat',
                  prog = Path(__file__).name,
                  formatter_class=argparse.ArgumentDefaultsHelpFormatter)

parser.add_argument('--editor', dest='editor', metavar='string', default='vi',
                    help='Name of the editor for MSC.Mentat (executable)')
parser.add_argument('--msc-root', dest='msc_root', metavar='string',
                    default=damask.solver._marc._msc_root,
                    help='MSC.Marc/Mentat root directory')
parser.add_argument('--msc-version', dest='msc_version', type=float, metavar='float',
                    default=damask.solver._marc._msc_version,
                    help='MSC.Marc/Mentat version')
parser.add_argument('--damask-root', dest='damask_root', metavar = 'string',
                    default=damask.solver._marc._damask_root,
                    help='DAMASK root directory')

args = parser.parse_args()
msc_root    = Path(args.msc_root).expanduser()
damask_root = Path(args.damask_root).expanduser()
msc_version = int(args.msc_version) if str(args.msc_version).split('.')[1] == '0' else \
              args.msc_version

matches = {'Marc_tools':  [['comp_user','comp_damask_*mp'],
                           ['run_marc','run_damask_*mp'],
                           ['include_linux64','include_linux64']],
           'Mentat_bin':  [['edit_window','edit_window'],
                           ['submit1','submit?'],
                           ['kill1','kill?']],
           'Mentat_menus':[['job_run.ms','job_run.ms']]}


print('patching files...\n')

for directory in glob.glob(str(damask_root/f'install/MarcMentat/{msc_version}/*')):
    for orig, mods in matches[Path(directory).name]:
        product,subfolder = (msc_root/Path(directory)).name.split('_')
        orig = msc_root/f'{product.lower()}{msc_version}/{subfolder}/{orig}'
        for patch in glob.glob(f'{directory}/{mods}.patch'):
            copy_and_patch(Path(patch),orig,msc_root,args.editor)

print('compiling Mentat menu binaries...')

executable = msc_root/f'mentat{msc_version}/bin/mentat'
menu_file  = msc_root/f'mentat{msc_version}/menus/linux64/main.msb'
os.system(f'xvfb-run -a {executable} -compile {menu_file}')