import os
import re
import time

# 用来给输出文件命名
out_name = "chinese"
# 项目路径
dir_path = "./src"
# 存放文件的数组
file_list = []
# 其他文件数组
other_file_list = []
# 定义一个包含中文的正则表达式
zh_pattern = re.compile(u'[\u4e00-\u9fa5]')
# 获取当前时间戳
now = int(time.time())

# 递归获取遍历的文件
def traverse(path):
  for file in os.listdir(path):
    file_path = os.path.join(path, file)
    if os.path.isdir(file_path):
      traverse(file_path)
    # 只统计js文件 后端如果使用的话可以自己选择文件类型
    elif os.path.isfile(file_path) and file.endswith(".js"):
      file_list.append(file_path)
    else:
      other_file_list.append(file_path)

# 获取中文
def getChineseFunc():
  with open(f"{out_name}_{now}.txt", "w", encoding="utf-8") as f:
    for path in file_list:
      # 打开文件并逐行读取内容
      with open(path, "r+", encoding="utf-8") as file:
        lines = file.readlines()
        for i in range(len(lines)):
          # 如果是中文列表里的内容
          if zh_pattern.search(lines[i]):
            # 如果是 /* // {/* * 开头的就pass，说明是注释
            if lines[i].strip().startswith("/*") or lines[i].strip().startswith("//") or lines[i].strip().startswith("{/*") or lines[i].strip().startswith("*"):
              pass
            else:
              # 将结果写入文件
              f.write(lines[i].strip() + "\n")

traverse(dir_path)
getChineseFunc()
print("获取项目中文成功")