import os
import sys
import dataclasses
from lex_c import lex_token
from lex_c import lex
import lex_c
from node_declear import dist_node_type_struct
from node_declear import dist_node_type_union
from node_declear import dist_node_type_enum
from node_declear import dist_node_type_typedef

_NodeTypeTable=[
  "file","vdecl","fdef"
]


@dataclasses.dataclass
class node:
  name:str
  type:str="base"
  token_list:list[lex_token]=dataclasses.field(default_factory=list)

# 文件节点
@dataclasses.dataclass
class node_file(node):
  type:str="file"
  body:list=dataclasses.field(default_factory=list)

# 变量定义节点
@dataclasses.dataclass
class node_variable_def(node):
  type:str="variable_def"
  vvalue=None
  vtype:str="unknown"
  vattr:list[str]=dataclasses.field(default_factory=list)

# 结构体声明节点
@dataclasses.dataclass
class node_struct_decl(node):
  type:str="struct_decl"

# 结构体定义节点
@dataclasses.dataclass
class node_struct_def(node):
  type:str="struct_def"
  body:list[node_variable_def]=dataclasses.field(default_factory=list)

# 联合体声明节点
@dataclasses.dataclass
class node_union_decl(node):
  type:str="union_decl"

# 联合体定义节点
@dataclasses.dataclass
class node_union_def(node):
  type:str="union_def"
  body:list[node_variable_def]=dataclasses.field(default_factory=list)

# 枚举声明节点
@dataclasses.dataclass
class node_enum_decl(node):
  type:str="enum_decl"

# 枚举定义节点
@dataclasses.dataclass
class node_enum_def(node):
  type:str="enum_def"
  body:list[dict]=dataclasses.field(default_factory=list)

# 函数声明节点
@dataclasses.dataclass
class node_func_decl(node):
  type:str="func_decl"
  rettype:str="unknown"
  retattr:list[str]=dataclasses.field(default_factory=list)
  para:list[node_variable_def]=dataclasses.field(default_factory=list)

#typedef 节点
@dataclasses.dataclass
class node_typedef(node):
  type:str="typedef"
  attr:list[str]=dataclasses.field(default_factory=list)
  body:node=None

# 函数定义节点
@dataclasses.dataclass
class node_func_def(node):
  type:str="func_def"
  rettype:str="unknown"
  retattr:list[str]=dataclasses.field(default_factory=list)
  para:list[node_variable_def]=dataclasses.field(default_factory=list)
  body:list[node]=dataclasses.field(default_factory=list)


# 找到闭合的括号
def find_close(token_list:list[lex_token],token:tuple[int,int]):
  if token_list[0].token!=token[0]:
    return 0
  num=0
  for index,item in enumerate(token_list):
    if(item.token==token[0]):
      num+=1
    elif(item.token==token[1]):
      num-=1
    if(num==0):
      return index
  raise Exception(f"没有找到闭合的符号 {token[1]}")

# 找到一个完整的语句
def find_sentence(token_list:list[lex_token]):
  bracket_flag=False
  index=0
  while index<len(token_list):
    if(token_list[index].token==lex_c.TOKEN("(")):
      bracket_index=find_close(token_list[index:],(lex_c.TOKEN("("),lex_c.TOKEN(")")))
      if(bracket_index>0):
        bracket_flag=True
        index+=bracket_index
    elif(token_list[index].token==lex_c.TOKEN("{")):
      bracket_index=find_close(token_list[index:],(lex_c.TOKEN("{"),lex_c.TOKEN("}")))
      if(bracket_index>0):
        index+=bracket_index
        if(bracket_flag==True):
          return token_list[:index+1]
    elif(token_list[index].token==lex_c.TOKEN(";")):
      return token_list[:index+1]
    index+=1
  raise Exception(f"没有找到完整的语句")



# 判断一个语句的类型
def dist_node_type(token_list:list[lex_token]):
  if(token_list[0].token==lex_c.TOKEN_EXTERN):
    token_list=token_list[1:]
  if(token_list[-1].token==lex_c.TOKEN(";")):
    token_list=token_list[:-1]
  if(token_list[0].token==lex_c.TOKEN_STRUCT):
    return dist_node_type_struct(token_list=token_list)
  if(token_list[0].token==lex_c.TOKEN_UNION):
    return dist_node_type_union(token_list=token_list)
  if(token_list[0].token==lex_c.TOKEN_ENUM):
    return dist_node_type_enum(token_list=token_list)
  if(token_list[0].token==lex_c.TOKEN_TYPEDEF):
    return dist_node_type_typedef(token_list=token_list)

  raise Exception(f"无法处理的token类型 {token_list[0]}")


if __name__ == "__main__":
  file_name="main.c"
  with open(file_name,mode='rb') as f:
    token_list=lex(f.read())
  file=node_file(name=file_name,token_list=token_list,body=[])
  while len(token_list)>0:
    sentence=find_sentence(token_list)
    node_d=dist_node_type(sentence)
    file.body.append(node_d)
    print('找到一个语句：')
    for item in sentence:
      print(f"\t{item}")
    token_list=token_list[len(sentence):]