#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
#
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
#
# 文件名称 : make_ycmd_extra_conf.py
# 文件描述 : 创建用于ycmd工作路径的文件
# 编写人员 : HHL
# 修改人员 :
# 编修描述 : V0.01，时间2017.09.11
#
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
#
__version__  = '0.01'
__status__   = 'beta'
#
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
#

'''
    程序的主要功能是获取目录下的带有*.h文件路径，将这些路劲按照一定
的格式加入到<.ycmd_extra.conf.py>这个文件内。
    程序目录使用一个文本文件读取
'''
import os
import sys
import os.path

ycmd_str_head = '''#!/usr/bin/env python
# -*- coding: UTF-8 -*-
#
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
#
# 文件名称 : ycm_extra_conf.py
# 文件描述 : ycmd工作路径配置
# 编写人员 : HHL
# 修改人员 :
# 编修描述 : V0.01，时间2017.09.11
#
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
#
__version__  = '0.01'
__status__   = 'beta'
#
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#
#
# 以下是作者原文
# https://github.com/Valloric/ycmd
#
# This file is NOT licensed under the GPLv3, which is the license for the rest
# of YouCompleteMe.
#
# Here's the license text for this file:
#
# This is free and unencumbered software released into the public domain.
#
# Anyone is free to copy, modify, publish, use, compile, sell, or
# distribute this software, either in source code form or as a compiled
# binary, for any purpose, commercial or non-commercial, and by any
# means.
#
# In jurisdictions that recognize copyright laws, the author or authors
# of this software dedicate any and all copyright interest in the
# software to the public domain. We make this dedication for the benefit
# of the public at large and to the detriment of our heirs and
# successors. We intend this dedication to be an overt act of
# relinquishment in perpetuity of all present and future rights to this
# software under copyright law.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
# IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
# OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
# ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
# OTHER DEALINGS IN THE SOFTWARE.
#
# For more information, please refer to <http://unlicense.org/>

import os
import ycm_core

# These are the compilation flags that will be used in case there's no
# compilation database set (by default, one is not set).
# CHANGE THIS LIST OF FLAGS. YES, THIS IS THE DROID YOU HAVE BEEN LOOKING FOR.
flags = [
'-Wall',
'-Wextra',
'-Werror',
'-Wc++98-compat',
'-Wno-long-long',
'-Wno-variadic-macros',
'-fexceptions',
'-DNDEBUG',
'-std=c99',
# ...and the same thing goes for the magic -x option which specifies the
# language that the files to be compiled are written in. This is mostly
# relevant for c++ headers.
# For a C project, you would set this to 'c' instead of 'c++'.
'-x', #I don't know if I need remove -x
'c',
"-I",
".",
#'-isystem',
#'../llvm/include',
#'-isystem',
#'../llvm/tools/clang/include',
#'-I./driver',
#'-I./debug',
#'-I./include',
# system includes
#'-isystem',
#'/usr/include',
'-isystem',
'/usr/lib/gcc/arm-none-eabi/4.8/include',
'-isystem',
'/usr/lib/gcc/arm-none-eabi/4.8/include-fixed',
'-isystem',
'/usr/lib/gcc/arm-none-eabi/4.8/../../../arm-none-eabi/include',

# auto write by make_ycmd_extra_con.py
USER_WRITE
]
# Set this to the absolute path to the folder (NOT the file!) containing the
# compile_commands.json file to use that instead of 'flags'. See here for
# more details: http://clang.llvm.org/docs/JSONCompilationDatabase.html
#
# You can get CMake to generate this file for you by adding:
#   set( CMAKE_EXPORT_COMPILE_COMMANDS 1 )
# to your CMakeLists.txt file.
#
# Most projects will NOT need to set this to anything; you can just change the
# 'flags' list of compilation flags. Notice that YCM itself uses that approach.
compilation_database_folder = ''

if os.path.exists( compilation_database_folder ):
  database = ycm_core.CompilationDatabase( compilation_database_folder )
else:
  database = None

SOURCE_EXTENSIONS = [ '.cpp', '.cxx', '.cc', '.c', '.m', '.mm' ]

def DirectoryOfThisScript():
  return os.path.dirname( os.path.abspath( __file__ ) )


def MakeRelativePathsInFlagsAbsolute( flags, working_directory ):
  if not working_directory:
    return list( flags )
  new_flags = []
  make_next_absolute = False
  path_flags = [ '-isystem', '-I', '-iquote', '--sysroot=' ]
  for flag in flags:
    new_flag = flag

    if make_next_absolute:
      make_next_absolute = False
      if not flag.startswith( '/' ):
        new_flag = os.path.join( working_directory, flag )

    for path_flag in path_flags:
      if flag == path_flag:
        make_next_absolute = True
        break

      if flag.startswith( path_flag ):
        path = flag[ len( path_flag ): ]
        new_flag = path_flag + os.path.join( working_directory, path )
        break

    if new_flag:
      new_flags.append( new_flag )
  return new_flags


def IsHeaderFile( filename ):
  extension = os.path.splitext( filename )[ 1 ]
  return extension in [ '.h', '.hxx', '.hpp', '.hh' ]


def GetCompilationInfoForFile( filename ):
  # The compilation_commands.json file generated by CMake does not have entries
  # for header files. So we do our best by asking the db for flags for a
  # corresponding source file, if any. If one exists, the flags for that file
  # should be good enough.
  if IsHeaderFile( filename ):
    basename = os.path.splitext( filename )[ 0 ]
    for extension in SOURCE_EXTENSIONS:
      replacement_file = basename + extension
      if os.path.exists( replacement_file ):
        compilation_info = database.GetCompilationInfoForFile(
          replacement_file )
        if compilation_info.compiler_flags_:
          return compilation_info
    return None
  return database.GetCompilationInfoForFile( filename )


def FlagsForFile( filename, **kwargs ):
  if database:
    # Bear in mind that compilation_info.compiler_flags_ does NOT return a
    # python list, but a "list-like" StringVec object
    compilation_info = GetCompilationInfoForFile( filename )
    if not compilation_info:
      return None

    final_flags = MakeRelativePathsInFlagsAbsolute(
      compilation_info.compiler_flags_,
      compilation_info.compiler_working_dir_ )

    # NOTE: This is just for YouCompleteMe; it's highly likely that your project
    # does NOT need to remove the stdlib flag. DO NOT USE THIS IN YOUR
    # ycm_extra_conf IF YOU'RE NOT 100% SURE YOU NEED IT.
    #try:
    #  final_flags.remove( '-stdlib=libc++' )
    #except ValueError:
    #  pass
  else:
    relative_to = DirectoryOfThisScript()
    final_flags = MakeRelativePathsInFlagsAbsolute( flags, relative_to )

  return { 'flags': final_flags }
'''

# 默认需要查找的类型
global_default_search_type = ('h',)
# 默认的输出文件名称
global_default_out_put_file_name = '_ycm_extra_conf.py'
# 默认的输入参数文件名称
global_default_in_param_file_name = 'ycmd_include_path.conf'

# 需要检查的路径
global_work_dir_list = []
# 输出数据
global_out_data_list = []


# 遍历指定的目录，找出含有相应文件的目录
def walk_type_file_dir(in_dir):

    # 申明下全局变量神马的
    global global_default_search_type
    # 定义局部变量
    include_py_dir = []

    try:
        for t_root, t_dir, t_file in os.walk(in_dir):
            for tf in t_file:
                t = tf.split(".")
                # 判断当前目录下是否有我们需要的文件
                if t[-1] in global_default_search_type:
                    include_py_dir.append(t_root)
                    break
    except Exception as e :
        print(e)

    #print(include_py_dir)
    return(include_py_dir)

# 程序入口
def make_ycmd_extra_conf():

    global ycmd_str_head
    global global_out_data_list
    global global_work_dir_list
    global global_default_out_put_file_name
    global global_default_in_param_file_name

    #先获取当前路径
    global_work_dir_list.append(os.getcwd())

    # 获取配置文件
    try:
        fs = sys.argv[1]
    except:
        fs = global_default_in_param_file_name

    # 输出路径配置文件名称
    print('路径配置文件名称:')
    print(fs)

    # 取得文件中的路径
    with open(fs, 'r') as f:
        fl = f.readlines()
        for l in fl:
            l = l.replace('\n', '').replace('\r', '')
            if os.path.isdir(l):
                global_work_dir_list.append(l)

    # 输出工作路径
    print('需要查找的路径: ')
    for wdl in global_work_dir_list:
        print(wdl)

    # 遍历输入目录，找出含有预定文件的目录并记录到 out_data_list
    for t_dir in global_work_dir_list:
        global_out_data_list += walk_type_file_dir(t_dir)

    print('共查找到目录 %d 个' %(len(global_out_data_list)))

    # 路劲名称改组为ycmd flags的格式
    t_flag_str = ''
    for t_fg in global_out_data_list:
        t_flag_str += "\'-I" + t_fg + "\',\n"

    #去掉最后一行尾部的换行符‘\n’
    t_flag_str = t_flag_str[: -1]

    # 用t_flag_str替换ycmd_str_head中的"USER_WRITE"
    t_ycmd_str_head = ycmd_str_head.replace('USER_WRITE', t_flag_str)

    # 写入到文件
    with open(global_default_out_put_file_name, 'w') as t_yf:
        t_yf = open(global_default_out_put_file_name, 'w')
        t_yf.write(t_ycmd_str_head)
        t_yf.close()

        print('写入成功,文件名: %s' %global_default_out_put_file_name)

# main程序入口
if __name__ == '__main__':
    make_ycmd_extra_conf()