#!/usr/bin/env python2
# -*- coding:utf-8 -*-
# SPDX-License-Identifier: GPL-2.0+
#
# Copyright (C) 2021-2023 ArtInChip Technology Co., Ltd

import xml.etree.ElementTree as etree
from xml.etree.ElementTree import SubElement
from building import *
from utils import *
from utils import _make_path_relative
from utils import xml_indent
import platform
import shutil
import glob
import rtconfig

def TargetEclipse(env, sdk=False, prj_name=None):
    print('Update eclipse setting...')

    aic_root = os.path.normpath(os.environ["AIC_ROOT"])
    prj_out_dir = os.environ["PRJ_OUT_DIR"]
    if sdk:
        prj_eclipse_dir = os.path.join(aic_root, prj_out_dir + "../project_eclipse_sdk")
    else:
        prj_eclipse_dir = os.path.join(aic_root, prj_out_dir + "../project_eclipse")
    prj_eclipse_dir = os.path.normpath(prj_eclipse_dir)
    template_dir = os.path.join(aic_root, "tools/scripts/template/eclipse")
    template_dir = os.path.normpath(template_dir)
    proj = ProjectInfo(env)

    # (1) generate '.project'
    # (1.1) read xml from template
    project = etree.parse(os.path.join(template_dir, 'template.project'))
    l_root = project.getroot()
    l_linkedResources = l_root.find('linkedResources')
    if l_linkedResources != None:
        l_root.remove(l_linkedResources)
    if not sdk:
        l_linkedResources = SubElement(l_root, 'linkedResources')

    # (1.2) modify project name
    l_name = l_root.find("./name")
    l_name.text = rtconfig.SOC

    if sdk:
        print('Copy .c .h file...')
        # (1.3) create 'dir'
        #if os.path.exists(prj_eclipse_dir):
        #    shutil.rmtree(prj_eclipse_dir)
        tmp_dirs = []
        tmp_h_dirs = []

        # (1.4) copy '.c file'
        for f in proj['FILES']:
            des_f = os.path.relpath(f, aic_root)
            des_f = os.path.join(prj_eclipse_dir, des_f)
            des_d = os.path.dirname(des_f)
            if des_d not in tmp_dirs:
                tmp_dirs.append(des_d)
                try:
                    os.makedirs(des_d)
                except:
                    pass
            shutil.copy(f, des_f)

        # (1.5) copy '.h file'
        for f in proj['HEADERS']:
            src_d = os.path.dirname(f)
            des_f = os.path.relpath(f, aic_root)
            des_f = os.path.join(prj_eclipse_dir, des_f)
            des_d = os.path.dirname(des_f)
            if des_d not in tmp_dirs:
                tmp_dirs.append(des_d)
                try:
                    os.makedirs(des_d)
                except:
                    pass
            if des_d not in tmp_h_dirs:
                tmp_h_dirs.append(des_d)
                for f in glob.iglob(src_d + '/*.h'):
                    des_f = os.path.basename(f)
                    des_f = os.path.join(des_d, des_f)
                    shutil.copy(f, des_f)
    else:
        # (1.3) create 'dir' link
        for d in proj['DIRS']:
            d = os.path.relpath(d, aic_root)
            d = d.replace('\\', '/')
            l_link = SubElement(l_linkedResources, 'link')
            l_name = SubElement(l_link, 'name')
            l_name.text = d
            l_type = SubElement(l_link, 'type')
            l_type.text = '2'
            l_location = SubElement(l_link, 'locationURI')
            l_location.text = 'virtual:/virtual'

        # (1.4) create '.c file' link
        for f in proj['FILES']:
            f = os.path.relpath(f, aic_root)
            f = f.replace('\\', '/')
            l_link = SubElement(l_linkedResources, 'link')
            l_name = SubElement(l_link, 'name')
            l_name.text = f
            l_type = SubElement(l_link, 'type')
            l_type.text = '1'
            l_location = SubElement(l_link, 'locationURI')
            l_location.text = 'PARENT-3-PROJECT_LOC/' + f

        # (1.5) create '.h file' link
        for f in proj['HEADERS']:
            f = os.path.relpath(f, aic_root)
            f = f.replace('\\', '/')
            l_link = SubElement(l_linkedResources, 'link')
            l_name = SubElement(l_link, 'name')
            l_name.text = f
            l_type = SubElement(l_link, 'type')
            l_type.text = '1'
            l_location = SubElement(l_link, 'locationURI')
            l_location.text = 'PARENT-3-PROJECT_LOC/' + f

    # (1.6) write to '.project'
    if not os.path.exists(prj_eclipse_dir):
        os.makedirs(prj_eclipse_dir)
    out = open(os.path.join(prj_eclipse_dir, '.project'), 'w')
    out.write('<?xml version="1.0" encoding="UTF-8"?>\n')
    xml_indent(l_root)
    out.write(etree.tostring(l_root, encoding='utf-8').decode('utf-8'))
    out.close()
    #print(etree.tostring(l_root, encoding='utf-8').decode('utf-8'))

    # (2) generate '.cproject'
    # (2.1) read xml from template
    cproject = etree.parse(os.path.join(template_dir, 'template.cproject'))
    l_root = cproject.getroot()

    # (2.2) 'debug' configuration
    l_configuration = l_root.find(".//configuration[@buildProperties='org.eclipse.cdt.build.core.buildArtefactType=org.eclipse.cdt.build.core.buildArtefactType.exe,org.eclipse.cdt.build.core.buildType=org.eclipse.cdt.build.core.buildType.debug']")

    # (2.2.1) 'target.other' configuration
    l_target = l_configuration.find(".//option[@superClass='ilg.gnumcueclipse.managedbuild.cross.riscv.option.target.other']")
    l_target.attrib['value'] = rtconfig.DEVICE

    # (2.2.2) 'assembler.defs' configuration
    l_defs = l_configuration.find(".//option[@superClass='ilg.gnumcueclipse.managedbuild.cross.riscv.option.assembler.defs']")
    # remove old Option
    for l_opt in l_defs.findall('listOptionValue'):
        l_defs.remove(l_opt)
    for d in proj['CPPDEFINES']:
        SubElement(l_defs, 'listOptionValue', builtIn="false", value=d)
    flags = env['ASFLAGS'].split()
    for flg in flags:
        if flg.startswith('-D'):
            SubElement(l_defs, 'listOptionValue', builtIn="false", value=flg.replace('-D',''))

    # (2.2.3) 'assembler.include.paths' configuration
    l_path = l_configuration.find(".//option[@superClass='ilg.gnumcueclipse.managedbuild.cross.riscv.option.assembler.include.paths']")
    # remove old Option
    for l_opt in l_path.findall('listOptionValue'):
        l_path.remove(l_opt)
    for d in proj['CPPPATH']:
        d = os.path.relpath(d, aic_root)
        if sdk:
            d = '"' + "${ProjDirPath}\\" + d + '"'
        else:
            d = '"' + "${ProjDirPath}\..\..\..\\" + d + '"'
        SubElement(l_path, 'listOptionValue', builtIn="false", value=d)

    # (2.2.4) 'c.compiler.defs' configuration
    l_defs = l_configuration.find(".//option[@superClass='ilg.gnumcueclipse.managedbuild.cross.riscv.option.c.compiler.defs']")
    # remove old Option
    for l_opt in l_defs.findall('listOptionValue'):
        l_defs.remove(l_opt)
    for d in proj['CPPDEFINES']:
        SubElement(l_defs, 'listOptionValue', builtIn="false", value=d)

    # (2.2.5) 'c.compiler.include.paths' configuration
    l_path = l_configuration.find(".//option[@superClass='ilg.gnumcueclipse.managedbuild.cross.riscv.option.c.compiler.include.paths']")
    # remove old Option
    for l_opt in l_path.findall('listOptionValue'):
        l_path.remove(l_opt)
    for d in proj['CPPPATH']:
        d = os.path.relpath(d, aic_root)
        if sdk:
            d = '"' + "${ProjDirPath}\\" + d + '"'
        else:
            d = '"' + "${ProjDirPath}\..\..\..\\" + d + '"'
        SubElement(l_path, 'listOptionValue', builtIn="false", value=d)

    # (2.2.6) 'c.linker.scriptfile' configuration
    l_path = l_configuration.find(".//option[@superClass='ilg.gnumcueclipse.managedbuild.cross.riscv.option.c.linker.scriptfile']")
    # remove old Option
    for l_opt in l_path.findall('listOptionValue'):
        l_path.remove(l_opt)
    flags = env['LINKFLAGS'].split()
    if '-T' in flags:
        index = flags.index('-T')
        src_f = flags[index + 1]
        rel_f = os.path.relpath(src_f, aic_root)
        if sdk:
            print('Copy .ld file...')
            f = '"' + "${ProjDirPath}\\" + rel_f + '"'
            # copy .ld file
            des_f = os.path.join(prj_eclipse_dir, rel_f)
            des_d = os.path.dirname(des_f)
            try:
                os.makedirs(des_d)
            except:
                pass
            shutil.copy(src_f, des_f)
        else:
            f = '"' + "${ProjDirPath}\..\..\..\\" + rel_f + '"'
        SubElement(l_path, 'listOptionValue', builtIn="false", value=f)

    # (2.2.7) 'c.linker.libs' configuration
    l_path = l_configuration.find(".//option[@superClass='ilg.gnumcueclipse.managedbuild.cross.riscv.option.c.linker.libs']")
    # remove old Option
    for l_opt in l_path.findall('listOptionValue'):
        l_path.remove(l_opt)
    if env.has_key('LIBS'):
        libs = env['LIBS']
    else:
        libs = []
    flags = env['LINKFLAGS'].split()
    for flg in flags:
        if flg.startswith('-l'):
            libs.append(flg.replace('-l', ''))
    libs = set(libs)
    for lib in libs:
        SubElement(l_path, 'listOptionValue', builtIn="false", value=lib)

    # (2.2.8) 'c.linker.libs' configuration
    l_path = l_configuration.find(".//option[@superClass='ilg.gnumcueclipse.managedbuild.cross.riscv.option.c.linker.paths']")
    # remove old Option
    for l_opt in l_path.findall('listOptionValue'):
        l_path.remove(l_opt)
    if env.has_key('LIBPATH'):
        libpath = env['LIBPATH']
        for src_d in libpath:
            rel_d = os.path.relpath(src_d, aic_root)
            if sdk:
                print('Copy lib file...')
                base_d = 'libs'
                d = '"' + "${ProjDirPath}\\" + base_d  + '"'
                # copy lib file
                des_d = os.path.join(prj_eclipse_dir, base_d)
                try:
                    os.makedirs(des_d)
                except:
                    pass
                for f in glob.iglob(src_d + '/*'):
                    if f.endswith(".a"):
                        des_f = os.path.basename(f)
                        des_f = os.path.join(des_d, des_f)
                        shutil.copy(f, des_f)
            else:
                d = '"' + "${ProjDirPath}\..\..\..\\" + rel_d + '"'
            SubElement(l_path, 'listOptionValue', builtIn="false", value=d)

    # (2.2.9) post-build command 'make image'
    prj_chip = os.environ["PRJ_CHIP"]
    prj_board = os.environ["PRJ_BOARD"]
    post_cmd = ''
    if sdk:
        # copy  'target/chip/board/pack/*'
        print('Copy pack file...')
        src_d = os.path.join(aic_root, 'target', prj_chip, prj_board, 'pack')
        des_d = os.path.join(prj_eclipse_dir, 'target', prj_chip, prj_board, 'pack')
        try:
            os.makedirs(des_d)
        except:
            pass
        for f in glob.iglob(src_d + '/*'):
            des_f = os.path.basename(f)
            des_f = os.path.join(des_d, des_f)
            shutil.copy(f, des_f)
        # copy  'bsp/artinchip/sys/chip/*.pbp'
        print('Copy .pbp file...')
        src_d = os.path.join(aic_root, 'bsp/artinchip/sys', prj_chip)
        des_d = os.path.join(prj_eclipse_dir, 'bsp/artinchip/sys', prj_chip)
        try:
            os.makedirs(des_d)
        except:
            pass
        for f in glob.iglob(src_d + '/*.pbp'):
            des_f = os.path.basename(f)
            des_f = os.path.join(des_d, des_f)
            shutil.copy(f, des_f)
        # copy script
        print('Copy scripts file...')
        des_d = os.path.join(prj_eclipse_dir, 'tools/scripts')
        try:
            os.makedirs(des_d)
        except:
            pass
        src_d = os.path.join(aic_root, 'tools/scripts/')
        for f in glob.iglob(src_d + '/*.py'):
            des_f = os.path.basename(f)
            des_f = os.path.join(des_d, des_f)
            shutil.copy(f, des_f)
        for f in glob.iglob(src_d + '/*.exe'):
            des_f = os.path.basename(f)
            des_f = os.path.join(des_d, des_f)
            shutil.copy(f, des_f)
        for f in glob.iglob(src_d + '/*.dll'):
            des_f = os.path.basename(f)
            des_f = os.path.join(des_d, des_f)
            shutil.copy(f, des_f)
        # copy bin
        print('Copy bin file...')
        src_d = os.path.join(aic_root, 'tools/env/tools/bin')
        des_d = os.path.join(prj_eclipse_dir, 'tools/bin')
        src_d = os.path.normpath(src_d);
        des_d = os.path.normpath(des_d);
        if not os.path.exists(des_d):
            if platform.system() == 'Linux':
                shutil.copytree(src_d, des_d)
            elif platform.system() == 'Windows':
                shutil.copytree('\\\\?\\' + src_d, '\\\\?\\' + des_d)
        # copy Python39
        print('Copy Python39 file...')
        src_d = os.path.join(aic_root, 'tools/env/tools/Python39')
        des_d = os.path.join(prj_eclipse_dir, 'tools/Python39')
        src_d = os.path.normpath(src_d);
        des_d = os.path.normpath(des_d);
        if not os.path.exists(des_d):
            if platform.system() == 'Linux':
                shutil.copytree(src_d, des_d)
            elif platform.system() == 'Windows':
                shutil.copytree('\\\\?\\' + src_d, '\\\\?\\' + des_d)
        # copy toolchain
        print('Copy toolchain file...')
        src_d = os.path.join(aic_root, 'toolchain')
        des_d = os.path.join(prj_eclipse_dir, 'toolchain')
        src_d = os.path.normpath(src_d);
        des_d = os.path.normpath(des_d);
        if not os.path.exists(des_d):
            if platform.system() == 'Linux':
                shutil.copytree(src_d, des_d)
            elif platform.system() == 'Windows':
                shutil.copytree('\\\\?\\' + src_d, '\\\\?\\' + des_d)
        # copy img src dir
        print('Copy image file...')
        dirs = []
        for i in range(2):
            if os.environ.has_key("img{}_srcdir".format(i)):
                dirs.append(os.environ["img{}_srcdir".format(i)])
        for src_d in dirs:
            des_d = os.path.relpath(src_d, aic_root)
            des_d = os.path.join(prj_eclipse_dir, des_d)
            src_d = os.path.normpath(src_d);
            des_d = os.path.normpath(des_d);
            if not os.path.exists(des_d):
                if platform.system() == 'Linux':
                    shutil.copytree(src_d, des_d)
                elif platform.system() == 'Windows':
                    shutil.copytree('\\\\?\\' + src_d, '\\\\?\\' + des_d)
        # post-build
        if os.environ.has_key("eclipse_sdk_post_build"):
            post_cmd = os.environ["eclipse_sdk_post_build"]
    else:
        # post-build
        if os.environ.has_key("eclipse_post_build"):
            post_cmd = os.environ["eclipse_post_build"]
    l_configuration.attrib['postbuildStep'] = post_cmd

    # (2.2.10) pre-build command
    prj_chip = os.environ["PRJ_CHIP"]
    prj_board = os.environ["PRJ_BOARD"]
    pre_cmd = ''
    if sdk:
        # pre-build
        if os.environ.has_key("eclipse_sdk_pre_build"):
            pre_cmd = os.environ["eclipse_sdk_pre_build"]
    else:
        # pre-build
        if os.environ.has_key("eclipse_pre_build"):
            pre_cmd = os.environ["eclipse_pre_build"]
    l_configuration.attrib['prebuildStep'] = pre_cmd

    # (2.2.11) user data directory
    prj_chip = os.environ["PRJ_CHIP"]
    prj_board = os.environ["PRJ_BOARD"]
    if sdk and os.environ.has_key("aic_fs_image_dir"):
        print('Copy image data directory...')
        aic_fs_image_dir = os.environ["aic_fs_image_dir"]
        src_d = os.path.join(aic_root, aic_fs_image_dir)
        des_d = os.path.join(prj_eclipse_dir, aic_fs_image_dir)
        src_d = os.path.normpath(src_d);
        des_d = os.path.normpath(des_d);
        if not os.path.exists(des_d):
            if platform.system() == 'Linux':
                shutil.copytree(src_d, des_d)
            elif platform.system() == 'Windows':
                shutil.copytree('\\\\?\\' + src_d, '\\\\?\\' + des_d)

    # (2.2.12) 'excluding'
    l_path = l_configuration.find(".//sourceEntries")
    if sdk:
        l_entry = l_path.find("entry")
        l_entry.attrib['excluding'] = "toolchain|tools"
    else:
        l_configuration.remove(l_path)

    # (2.3) write to '.cproject'
    out = open(os.path.join(prj_eclipse_dir, '.cproject'), 'w')
    out.write('<?xml version="1.0" encoding="UTF-8" standalone="no"?>\n')
    out.write('<?fileVersion 4.0.0?>')
    xml_indent(l_root)
    out.write(etree.tostring(l_root, encoding='utf-8').decode('utf-8'))
    out.close()
    #print(etree.tostring(l_root, encoding='utf-8').decode('utf-8'))

    # (3) generate '.settings/language.settings.xml'
    d = os.path.join(prj_eclipse_dir, ".settings")
    if not os.path.exists(d):
        os.makedirs(d)
    des = os.path.join(d, "language.settings.xml")
    src = os.path.join(template_dir, 'template.language.settings.xml')
    shutil.copy(src, des)

    # (4) generate '.settings/org.eclipse.cdt.core.prefs'
    d = os.path.join(prj_eclipse_dir, ".settings")
    if not os.path.exists(d):
        os.makedirs(d)
    src_f = os.path.join(template_dir, 'template.org.eclipse.cdt.core.prefs')
    des_f = os.path.join(d, "org.eclipse.cdt.core.prefs")
    with open(src_f, 'r') as f:
        template_prefs_str = f.read()
    if sdk:
        template_prefs_str = template_prefs_str.replace(r'\\..\\..\\..', '')
        template_prefs_str = template_prefs_str.replace(r'tools\\env\\tools\\bin', r'tools\\bin')
    with open(des_f, 'w') as f:
        f.write(template_prefs_str)

    print('done!')
    exit(0)