#!/usr/bin/env python3
import sys
import re
from pathlib import Path

SAFE_SKILL_ID_PATTERN = re.compile(r"^[A-Za-z0-9._-]+$")


def is_safe_skill_id(skill_id):
    return bool(SAFE_SKILL_ID_PATTERN.fullmatch(skill_id or ""))


def filter_safe_skill_ids(skill_ids):
    return [skill_id for skill_id in skill_ids if is_safe_skill_id(skill_id)]


def format_skills_for_batch(skill_ids):
    safe_skill_ids = filter_safe_skill_ids(skill_ids)
    if not safe_skill_ids:
        return ""
    # Use newline separator for robustness (avoiding Windows command line length limits)
    return "\n".join(safe_skill_ids) + "\n"


def get_bundle_skills(bundle_queries, bundles_path=None):
    if bundles_path is None:
        bundles_path = Path(__file__).parent.parent.parent / "docs" / "users" / "bundles.md"
    else:
        bundles_path = Path(bundles_path)
    
    if not bundles_path.exists():
        print(f"Error: {bundles_path} not found", file=sys.stderr)
        return []

    content = bundles_path.read_text(encoding="utf-8")
    
    # Split by bundle headers
    sections = re.split(r'\n### ', content)
    
    selected_skills = set()
    
    for query in bundle_queries:
        query = query.lower().strip('"\'')
        found = False
        for section in sections:
            header_line = section.split('\n')[0].lower()
            if query in header_line:
                found = True
                # Extract skill names from bullet points: - [`skill-name`](../../skills/skill-name/)
                skills = re.findall(r'- \[`([^`]+)`\]', section)
                selected_skills.update(filter_safe_skill_ids(skills))
        
        if not found:
            # If query not found in any header, check if it's a skill name itself
            # (Just in case the user passed a skill name instead of a bundle)
            if is_safe_skill_id(query):
                selected_skills.add(query)

    return sorted(list(selected_skills))


if __name__ == "__main__":
    if len(sys.argv) < 2:
        # Default to Essentials if no query
        queries = ["essentials"]
    else:
        queries = sys.argv[1:]
    
    skills = get_bundle_skills(queries)
    if skills:
        sys.stdout.write(format_skills_for_batch(skills))