Initial commit

skills/git-pr/scripts/assign_milestone.py

@@ -0,0 +1,146 @@
+#!/usr/bin/env python3
+"""
+Lister milestones et gérer assignation
+Usage: assign_milestone.py <pr_number> [--milestone <name>]
+Output: Milestone assigné ou "ignored" (stdout)
+"""
+
+import argparse
+import json
+import subprocess
+import sys
+import re
+from milestone_cache import MilestoneCache
+
+
+def get_repo_info():
+    """Récupère owner/repo depuis git remote"""
+    try:
+        result = subprocess.run(
+            ["git", "remote", "get-url", "origin"],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        url = result.stdout.strip()
+        # Extraire owner/repo depuis URL GitHub
+        match = re.search(r'github\.com[:/](.+/.+?)(?:\.git)?$', url)
+        if match:
+            return match.group(1)
+        raise ValueError(f"Format URL invalide: {url}")
+    except Exception as e:
+        print(f"❌ Erreur récupération repo: {e}", file=sys.stderr)
+        sys.exit(1)
+
+
+def get_open_milestones(repo):
+    """Récupère les milestones ouverts via gh API"""
+    try:
+        result = subprocess.run(
+            ["gh", "api", f"repos/{repo}/milestones", "--jq",
+             "[.[] | select(.state == \"open\") | {number, title}]"],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        return json.loads(result.stdout)
+    except subprocess.CalledProcessError as e:
+        print(f"❌ Erreur récupération milestones: {e.stderr}", file=sys.stderr)
+        return []
+    except json.JSONDecodeError as e:
+        print(f"❌ Erreur parsing JSON milestones: {e}", file=sys.stderr)
+        return []
+
+
+def get_open_milestones_cached(repo):
+    """Récupère milestones avec cache"""
+    cache = MilestoneCache()
+    if not cache.cache.get("milestones"):
+        milestones = get_open_milestones(repo)
+        cache.refresh_from_api(milestones)
+        return milestones
+    return cache.cache["milestones"]
+
+
+def find_milestone(repo, query):
+    """Cherche milestone par query (exact ou partiel)"""
+    cache = MilestoneCache()
+    result = cache.find(query)
+    if result:
+        return result
+    normalized = cache.normalize_semver(query)
+    result = cache.find(normalized)
+    if result:
+        return result
+    milestones = get_open_milestones(repo)
+    cache.refresh_from_api(milestones)
+    result = cache.find(normalized)
+    if result:
+        return result
+    return cache.create(normalized)
+
+
+def assign_milestone(pr_number, milestone_title):
+    """Assigne un milestone à la PR"""
+    try:
+        subprocess.run(
+            ["gh", "pr", "edit", str(pr_number), "--milestone", milestone_title],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        return True
+    except subprocess.CalledProcessError as e:
+        print(f"❌ Erreur assignation milestone: {e.stderr}", file=sys.stderr)
+        return False
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Assigne un milestone à une PR")
+    parser.add_argument("pr_number", type=int, help="Numéro de la PR")
+    parser.add_argument("--milestone", help="Nom du milestone à assigner")
+    args = parser.parse_args()
+
+    repo = get_repo_info()
+    milestones = get_open_milestones_cached(repo)
+
+    if not milestones:
+        print("ℹ️  Aucun milestone ouvert - ignoré")
+        print("ignored")
+        sys.exit(0)
+
+    # Si --milestone fourni, utiliser find_milestone pour recherche intelligente
+    if args.milestone:
+        try:
+            milestone = find_milestone(repo, args.milestone)
+            if assign_milestone(args.pr_number, milestone['title']):
+                print(f"✅ Milestone '{milestone['title']}' assigné")
+                print(milestone['title'])
+                sys.exit(0)
+            else:
+                sys.exit(1)
+        except Exception as e:
+            print(f"❌ Erreur: {e}", file=sys.stderr)
+            sys.exit(1)
+
+    # Sinon, retourner JSON pour AskUserQuestion
+    # Suggérer le premier milestone (généralement le plus récent)
+    milestones_with_suggestion = [
+        {
+            "number": m["number"],
+            "title": m["title"],
+            "is_suggested": i == 0
+        }
+        for i, m in enumerate(milestones)
+    ]
+
+    output = {
+        "milestones": milestones_with_suggestion,
+        "needs_user_input": True
+    }
+
+    print(json.dumps(output, indent=2))
+
+
+if __name__ == "__main__":
+    main()