#!/usr/bin/env python3
"""
Generate status badges for README.md
"""

import json
import requests
from datetime import datetime

def generate_badge_url(label, message, color="brightgreen"):
    """Generate a shields.io badge URL"""
    return f"https://img.shields.io/badge/{label}-{message}-{color}"

def generate_workflow_badge(repo_url, workflow_name, branch="main"):
    """Generate workflow status badge"""
    # For Gitea, you might need to adjust this based on your instance
    badge_url = f"{repo_url}/actions/workflows/{workflow_name}/badge.svg?branch={branch}"
    return badge_url

def main():
    """Generate badges for the project"""
    repo_url = "https://git.b4l.co.th/B4L/Northern-Thailand-Ping-River-Monitor"
    
    badges = {
        "CI/CD": generate_workflow_badge(repo_url, "ci.yml"),
        "Security": generate_workflow_badge(repo_url, "security.yml"),
        "Documentation": generate_workflow_badge(repo_url, "docs.yml"),
        "Python": generate_badge_url("Python", "3.9%2B", "blue"),
        "FastAPI": generate_badge_url("FastAPI", "0.104%2B", "green"),
        "Docker": generate_badge_url("Docker", "Ready", "blue"),
        "License": generate_badge_url("License", "MIT", "green"),
        "Version": generate_badge_url("Version", "v3.1.1", "blue"),
    }
    
    print("# Status Badges")
    print()
    print("Add these badges to your README.md:")
    print()
    
    for name, url in badges.items():
        print(f"[![{name}]({url})]({repo_url})")
    
    print()
    print("# Markdown Format")
    print()
    
    badge_line = " ".join([f"[![{name}]({url})]({repo_url})" for name, url in badges.items()])
    print(badge_line)

if __name__ == "__main__":
    main()