#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2025/4/28
# @USER    : Shengji He
# @File    : generate_md_index.py
# @Software: PyCharm
# @Version  : Python-
# @TASK:
import os
import glob


def main():
    # -----------------------------------------------------------
    # get md files
    # -----------------------------------------------------------
    root_folder = r'..\Python-100-Days'

    main_msg = ''
    other_msg = ''

    for directory in os.listdir(root_folder):
        cur_folder = os.path.join(root_folder, directory)
        if not os.path.isdir(cur_folder):
            continue

        md_files = glob.glob(os.path.join(cur_folder, '**', '*.md'), recursive=True)
        if len(md_files) == 0:
            continue
        msg = f'{directory}\n--------------\n.. toctree::\n   :maxdepth: 4\n\n'
        for md_file in md_files:
            md_file_name = os.path.relpath(md_file, root_folder).replace('\\', '/')
            msg += f'   {md_file_name}\n'
        if 'Day' in directory:
            main_msg += msg + '\n\n'
        else:
            other_msg += msg + '\n\n'

    final_msg = '\n\nPython-100-Days\n===============\n\n' + main_msg
    final_msg += '\n\nOthers\n===============\n\n' + other_msg

    # 将 final_msg 保存到 index.rst 文件
    output_file = os.path.join(root_folder, 'modules.rst')
    with open(output_file, 'w', encoding='utf-8') as f:
        f.write(final_msg)


if __name__ == '__main__':
    main()
    print('done')