import os
import sys
import json




'''
更新日志：

2024.8.9
添加对 -imacros 选项中宏定义的读取
2024.8.19
读取到的宏定义删除多余的 $ 符号

'''



def get_out_path(sub:str):
  with open('ohos_config.json') as f:
    j=json.loads(f.read())
    return os.path.join(j["out_path"],sub)


# 找到指定后缀的文件
def find_type(path:str,fix:str):
    dlist=os.listdir(path)
    file_list=[]
    for i in dlist:
        ps=os.path.join(path, i)
        if os.path.isdir(ps):
            file_list+=find_type(ps,fix)
        else:
            if(ps[-len(fix):]==fix):
                file_list.append(ps)
    return file_list



# 输出目录列表
def list_dirs(path:str,file=None):
    dlist=os.listdir(path)
    file_list=[]
    for i in dlist:
        ps=os.path.join(path, i)
        if os.path.isdir(ps) and not os.path.islink(ps):
            if(file is None):
              file_list.append(ps)
            else:
              print(ps,file=file)
            file_list+=list_dirs(ps,file)
    file_list.sort()
    return file_list







# 从配置文件中读取宏定义
def read_defines(values:list):
  flage=False
  def_list=[]
  for item in values:
    if(item=='-imacros'):
      flage=True
      continue
    if(flage==True):
      flage=False
      # print(item)
      if(os.path.exists(item)):
        with open(item) as f:
          data=f.readlines()
          for defs in data:
            split=defs.split()
            if(split[0]=="#define"):
              def_list.append(split[1]+'='+split[2])
  return def_list




def unescape(line:str):
  table=[("\\\"","\""),("\\<","<"),("\\>",">"),("\\$","$"),("\\(","("),("\\)",")")]
  for item in table:
    line=line.replace(item[0],item[1])
  # 删除多余的 $ 符号
  line=line.replace('$','')
  return line

# 找到include和define
def find_inc_def(file_list:list):
  defines=[]
  include_dirs=[]
  ldflags=[]
  current_path=os.path.abspath('.')
  for item in file_list:
    with open(item) as f:
      for line in f.readlines():
        index=line.find("=")
        if(index>0):
          line_sp=[line[:index],line[index+1:]]
        else:
          line_sp=[line]
        if(len(line_sp)>=2):
          head=line_sp[0].strip()
          # defines 中可能含有空格 这里单独处理
          if(head=="defines"):
            values=line_sp[-1].strip().replace('-D','\n').split('\n')
            for d in values:
              defines.append(unescape(d.strip()))
          if(head=="cflags") or head=="include_dirs":
            # print(f"head:{head}")
            values=line_sp[-1].strip().split()
            # 读取宏文件里定义的 define
            _def_list=read_defines(values)
            defines+=_def_list
            for d in values:
              # cflags 中也可能包含define
              if(d[:2]=='-D'):
                defines.append(unescape(d[2:]))
              elif(d[0:2]=="-I"):
                if(d[0:4]=="-I.."):
                  dir=os.path.normpath(d[9:])
                  dir=os.path.join(current_path,dir)
                else:
                  dir=os.path.join(get_out_path(''),d[2:])
                if(os.path.exists(dir)):
                  include_dirs.append(dir)
              # else:
              #   print(d)
          elif(head=='ldflags'):
            values=line_sp[-1].strip().split()
            values_dict={"file:":item,"values":values}
            ldflags.append(values_dict)
          # else:
          #   print(f"unknow:{head}")
  defines=list(set(defines))
  include_dirs=list(set(include_dirs))
  # ldflags=list(set(ldflags))
  defines.sort()
  # inc的先后顺序很重要，排序之后可能造成语法解析有问题
  # include_dirs.sort()
  return defines,include_dirs,ldflags




# 更新配置文件
def flush_cpp_setting(defines:list,include_dirs:list):
  set_file=".vscode/c_cpp_properties.json"
  with open(set_file,encoding="utf-8") as f:
    j=json.loads(f.read())
  j["configurations"][0]["includePath"]=include_dirs
  j["configurations"][0]["defines"]=defines
  d=json.dumps(j,sort_keys=True, indent=2, separators=(',', ': '))
  with open(set_file,mode='w+',encoding="utf-8") as f:
    f.write(d)




# 输出编译的文件
def find_build_files():
  _tmp_file="ninja_commands.txt"
  _cmd=f"{get_out_path('')}../../../prebuilts/build-tools/linux-x86/bin/ninja -C {get_out_path('')} -t commands > {_tmp_file}"
  with os.popen(_cmd) as f:
    ret=f.readlines()
    for item in ret:
      print(item)

  c_file_list=[]
  with open(_tmp_file) as f:
    for item in f.readlines():
      item_sp=item.split()
      for item_x in item_sp:
        item_x_fix=item_x.split('.')[-1]
        if(item_x_fix in ['c','cpp','C','CPP','S','s']):
          f_path=item_x[9:]
          if f_path not in c_file_list:
            c_file_list.append(f_path)
  os.remove(_tmp_file)
  c_file_list.sort()
  return c_file_list



if __name__ == "__main__":

  # 读取 .ninja 文件来设置define和include
  file_list=find_type(get_out_path("obj"),".ninja")
  defines,include_dirs,ldflags=find_inc_def(file_list)
  flush_cpp_setting(defines,include_dirs)

  with open("ldflags_list"+".json",mode="w+",encoding="utf-8") as g:
    g.write(json.dumps(ldflags,sort_keys=True, indent=2, separators=(',', ': ')))


  # 打印编译的源文件
  # for item in find_build_files():
  #   print(item)

  # with open("dirs_list.txt",mode='w+') as f:
  #   dirs_list=list_dirs('.',file=f)
  #   for item in dirs_list:
  #     print(item,file=f)


  # print("defines:")
  # for item in defines:
  #   print(f"\t{item}")
  # print("include_dirs:")
  # for item in include_dirs:
  #   print(f"\t{item}")
  # print("ldflags:")
  # for item in ldflags:
  #   print(f"\t{item}")