"""
调用cloc代码行数统计工具统计本项目代码量的脚本

https://github.com/AlDanial/cloc
"""

import pathlib
from os import popen

# 配置项
Git_Ignore_File = "../.gitignore"  # gitignore文件路径，用以排除不需要列目录的文件
Project_Root_Path = "../"  # 项目根目录路径
# 配置项完


def cloc(gitignore_file: str = ".gitignore") -> None:
    """
    使用cloc统计项目代码行数 \n
    :param gitignore_file: gitignore文件路径
    :return: None
    """

    ignored_dir = ""
    gitignore_file_p = pathlib.Path(gitignore_file)
    with gitignore_file_p.open("r", encoding="UTF-8") as f:
        for dir_name in f.readlines():
            if not dir_name.startswith("#"):
                dir_name = dir_name.replace("/", "").replace("\n", ",")
                ignored_dir += dir_name

    # 调用cloc，并排除gitignore中的目录，需要提前将cloc添加到系统环境变量
    cmd = f"cloc --exclude-dir {ignored_dir} {Project_Root_Path}"

    with popen(cmd) as p:
        cmd_result = p.read()
        # 如果cmd执行正常退出则p.close()返回None，失败则返回状态码
        if p.close():
            print("cloc调用失败，请检查")
        else:
            # 根据cloc返回结果，连续两个换行符后面的内容是需要的信息
            cloc_result = cmd_result.split("\n\n", 1)[1]
            print(cloc_result)
    # TODO 将 self.cloc_result 写入 Markdown 文件


if __name__ == "__main__":
    cloc(Git_Ignore_File)