import json
import argparse
from docx import Document
import os
import glob

def json_to_docx(input_paths, output_path, show_name):
    document = Document()
    
    for input_path in input_paths:
        with open(input_path, 'r', encoding='utf-8') as f:
            data = json.load(f)
            for entry in data:
                if show_name and 'name' in entry:
                    paragraph = document.add_paragraph()
                    paragraph.add_run(entry['name']).bold = True
                    paragraph.add_run('\n' + entry['message'])
                else:
                    paragraph = document.add_paragraph(entry['message'])
    
    document.save(output_path)

def get_json_files_from_directory(directory):
    # 获取指定目录下所有的 JSON 文件
    return glob.glob(os.path.join(directory, '*.json'))

def main():
    parser = argparse.ArgumentParser(description='Convert JSON messages to DOCX format.')
    parser.add_argument('input', nargs='+', help='Input JSON file paths or directory.')
    parser.add_argument('-o', '--output', help='Output DOCX file path.', default=None)
    parser.add_argument('-n', '--noname', action='store_true', help='Do not include name field in output.')

    args = parser.parse_args()

    input_files = []
    
    for input_path in args.input:
        if os.path.isdir(input_path):
            # 如果是目录，则获取该目录下的所有 JSON 文件
            input_files.extend(get_json_files_from_directory(input_path))
        elif os.path.isfile(input_path) and input_path.endswith('.json'):
            # 如果是单个文件，直接添加到列表
            input_files.append(input_path)

    if not input_files:
        print("No JSON files found in the specified paths.")
        return

    if args.output is None:
        # 如果没有指定输出路径，使用第一个输入路径并将扩展名改为 .docx
        output_path = os.path.splitext(input_files[0])[0] + '.docx'
    else:
        output_path = args.output

    json_to_docx(input_files, output_path, not args.noname)
    print(f'DOCX file saved to: {output_path}')

if __name__ == '__main__':
    main()