import os
import sys
import glob

from utils import *
from utils import _make_path_relative
from utils import xml_indent

import xml.etree.ElementTree as etree
from xml.etree.ElementTree import SubElement

source_pattern = ['*.c', '*.cpp', '*.cxx', '*.s', '*.S', '*.asm']

def OSPath(path):
    import platform

    if type(path) == type('str'):
        if platform.system() == 'Windows':
            return path.replace('/', '\\')
        else:
            return path.replace('\\', '/')
    else:
        if platform.system() == 'Windows':
            return [item.replace('/', '\\') for item in path]
        else:
            return [item.replace('\\', '/') for item in path]

def CollectPaths(paths):
    all_paths = []

    def ParentPaths(path):
        ret = os.path.dirname(path)
        if ret == path or ret == '':
            return []

        return [ret] + ParentPaths(ret)

    for path in paths:
        # path = os.path.abspath(path)
        path = path.replace('\\', '/')
        all_paths = all_paths + [path] + ParentPaths(path)

    all_paths = list(set(all_paths))
    return sorted(all_paths)

'''
Collect all of files under paths
'''
def CollectFiles(paths, pattern):
    files = []
    for path in paths:
        if type(pattern) == type(''):
            files = files + glob.glob(path + '/' + pattern)
        else:
            for item in pattern:
                # print('--> %s' % (path + '/' + item))
                files = files + glob.glob(path + '/' + item)

    return sorted(files)

def CollectAllFilesinPath(path, pattern):
    files = []

    for item in pattern:
        files += glob.glob(path + '/' + item)

    list = os.listdir(path)
    if len(list):
        for item in list:
            if item.startswith('.'):
                continue
            if item == 'bsp':
                continue

            if os.path.isdir(os.path.join(path, item)):
                files = files + CollectAllFilesinPath(os.path.join(path, item), pattern)
    return files

'''
Exclude files from infiles
'''
def ExcludeFiles(infiles, files):
    in_files  = set([OSPath(file) for file in infiles])
    exl_files = set([OSPath(file) for file in files])

    exl_files = in_files - exl_files

    return exl_files

def ExcludePaths(filepath, paths):
    ret = []

    files = os.listdir(filepath)
    for file in files:
        if file.startswith('.'):
            continue

        fullname = os.path.join(filepath, file)

        if os.path.isdir(fullname):
            # print(fullname)
            if not fullname in paths:
                ret = ret + [fullname]
            else:
                ret = ret + ExcludePaths(fullname, paths)

    return ret

def HandleToolOption(tools, env):
    project = ProjectInfo(env)
    BSP_ROOT = os.path.abspath(env['BSP_ROOT'])

    CPPDEFINES = project['CPPDEFINES']
    paths = ['${ProjDirPath}/' + _make_path_relative(BSP_ROOT, os.path.normpath(i)).replace('\\', '/') for i in project['CPPPATH']]

    for tool in tools:
        if tool.get('id').find('c.compile') != 1:
            options = tool.findall('option')
            for option in options:
                if option.get('id').find('c.compiler.include.paths') != -1:
                    # find all of paths in this project
                    include_paths = option.findall('listOptionValue')
                    project_paths = []
                    for item in include_paths:
                        project_paths += [item.get('value')]

                    if len(project_paths) > 0:
                        cproject_paths = set(paths) - set(project_paths)
                    else:
                        cproject_paths = paths

                    # print('c.compiler.include.paths')
                    cproject_paths = sorted(cproject_paths)
                    for item in cproject_paths:
                        SubElement(option, 'listOptionValue', {'builtIn': 'false', 'value': item})

                if option.get('id').find('c.compiler.defs') != -1:
                    defs = option.findall('listOptionValue')
                    project_defs = []
                    for item in defs:
                        project_defs += [item.get('value')]
                    if len(project_defs) > 0:
                        cproject_defs = set(CPPDEFINES) - set(project_defs)
                    else:
                        cproject_defs = CPPDEFINES

                    # print('c.compiler.defs')
                    cproject_defs = sorted(cproject_defs)
                    for item in cproject_defs:
                        SubElement(option, 'listOptionValue', {'builtIn': 'false', 'value': item})

        if tool.get('id').find('c.linker') != -1:
            options = tool.findall('option')
            for option in options:
                if option.get('id').find('c.linker.scriptfile') != -1:
                    linker_script = 'link.lds'
                    items = env['LINKFLAGS'].split(' ')
                    if '-T' in items:
                        linker_script = items[items.index('-T') + 1]
                        linker_script = '${ProjDirPath}/' + linker_script

                    # print('c.linker.scriptfile')
                    listOptionValue = option.find('listOptionValue')
                    if listOptionValue != None:
                        listOptionValue.set('value', linker_script)
                    else:
                        SubElement(option, 'listOptionValue', {'builtIn': 'false', 'value': linker_script})

                if option.get('id').find('c.linker.nostart') != -1:
                    if env['LINKFLAGS'].find('-nostartfiles') != -1:
                        option.set('value', 'true')
                    else:
                        option.set('value', 'false')

    return

def HandleRTTRoot(env):
    bsp_root = env['BSP_ROOT']
    rtt_root = env['RTT_ROOT']

    if not rtt_root.startswith(bsp_root):
        to_SubElement = True
        # print('handle virtual root')

        # always use '/' path separator
        rtt_root = rtt_root.replace('\\', '/')

        project = etree.parse('.project')
        root = project.getroot()

        linkedResources = root.find('linkedResources')
        if linkedResources == None:
            # add linkedResources
            linkedResources = SubElement(root, 'linkedResources')
            # print('add linkedResources')
        else:
            links = linkedResources.findall('link')
            # search exist 'rt-thread' virtual folder
            for link in links:
                if link.find('name').text == 'rt-thread':
                    # handle location
                    to_SubElement = False
                    location = link.find('location')
                    location.text = rtt_root

        if to_SubElement:
            # print('to subelement for virtual folder')
            link = SubElement(linkedResources, 'link')
            name = SubElement(link, 'name')
            name.text = 'rt-thread'
            type = SubElement(link, 'type')
            type.text = '2'
            location = SubElement(link, 'location')
            location.text = rtt_root

        out = open('.project', 'w')
        out.write('<?xml version="1.0" encoding="UTF-8"?>\n')
        xml_indent(root)
        out.write(etree.tostring(root, encoding='utf-8'))
        out.close()

    return

def TargetEclipse(env):
    global source_pattern

    print('Update eclipse setting...')

    if not os.path.exists('.cproject'):
        print('no eclipse CDT project found!')
        return

    HandleRTTRoot(env)

    project = ProjectInfo(env)

    all_paths = [OSPath(path) for path in CollectPaths(project['DIRS'])]
    # print(all_paths)
    bsp_root = os.path.abspath(env['BSP_ROOT'])

    exclude_paths = ExcludePaths(bsp_root, all_paths)
    paths = exclude_paths
    exclude_paths = []
    for path in paths:
        # add bsp and libcpu folder and not collect source files (too more files)
        if path.endswith('rt-thread\\bsp') or path.endswith('rt-thread\\libcpu'):
            exclude_paths += [path]
            continue

        set = CollectAllFilesinPath(path, source_pattern)
        if len(set):
            exclude_paths += [path]

    exclude_paths = [_make_path_relative(bsp_root, path).replace('\\', '/') for path in exclude_paths]
    env['ExPaths'] = exclude_paths

    all_files = CollectFiles(all_paths, source_pattern)
    src_files = project['FILES']

    exclude_files = ExcludeFiles(all_files, src_files)
    exclude_files = [_make_path_relative(bsp_root, file).replace('\\', '/') for file in exclude_files]
    env['ExFiles'] = exclude_files

    cproject = etree.parse('.cproject')

    root = cproject.getroot()
    cconfigurations = root.findall('storageModule/cconfiguration')
    for cconfiguration in cconfigurations:
        tools = cconfiguration.findall('storageModule/configuration/folderInfo/toolChain/tool')
        HandleToolOption(tools, env)

        sourceEntries = cconfiguration.find('storageModule/configuration/sourceEntries')
        entry = sourceEntries.find('entry')
        if entry != None:
            sourceEntries.remove(entry)

        excluding = exclude_paths + exclude_files
        excluding = sorted(excluding)
        value = ''
        for item in excluding:
            if value == '':
                value = item
            else:
                value += '|' + item
        excluding = value

        SubElement(sourceEntries, 'entry', {'excluding': excluding, 'flags': 'VALUE_WORKSPACE_PATH|RESOLVED', 'kind':'sourcePath', 'name':""})

    # write back to .cproject
    out = open('.cproject', 'w')
    out.write('<?xml version="1.0" encoding="UTF-8" standalone="no"?>\n')
    out.write('<?fileVersion 4.0.0?>')
    xml_indent(root)
    out.write(etree.tostring(root, encoding='utf-8'))
    out.close()

    print('done!')

    return