#!/usr/bin/env python3
"""更新版本号脚本"""
import sys
from pathlib import Path


def get_version():
    """从 VERSION 文件读取版本号"""
    version_file = Path(__file__).parent / "VERSION"
    return version_file.read_text().strip()


def increment_version(version: str) -> str:
    """递增版本号的修订号"""
    parts = version.split(".")
    if len(parts) == 3:
        parts[2] = str(int(parts[2]) + 1)
        return ".".join(parts)
    return version


def update_version_file(version: str):
    """更新 VERSION 文件"""
    version_file = Path(__file__).parent / "VERSION"
    version_file.write_text(version + "\n")


def create_metadata(version: str):
    """创建 metadata.yml 文件"""
    metadata_file = Path(__file__).parent / "metadata.yml"
    content = f"""version: {version}
company_name: EGM
file_description: EGM Lightning Protection Calculator
product_name: Lightening
"""
    metadata_file.write_text(content)
    print(f"Created metadata.yml with version {version}")


def main():
    # 检查是否只获取版本号
    if len(sys.argv) > 1 and sys.argv[1] == "--get":
        print(get_version())
        return
    
    # 获取当前版本并递增
    current_version = get_version()
    new_version = increment_version(current_version)
    
    # 更新所有文件
    update_version_file(new_version)
    create_metadata(new_version)
    
    print(f"Version updated: {current_version} -> {new_version}")


if __name__ == "__main__":
    main()