import os
import json
import redis
import argparse
import subprocess
from anthropic import Anthropic
from key_manager import KeyManager

# Configurações de Caminho
BASE_DIR = "/home/deploy/projetos"
INSTRUCOES_DIR = "/home/deploy/instrucoes"

class SwarmAgent:
    def __init__(self, task_data):
        self.data = task_data
        self.repo_name = task_data.get('repository')
        self.repo_url = f"http://10.99.0.3:3000/{self.repo_name}.git"
        self.local_path = os.path.join(BASE_DIR, self.repo_name.split('/')[-1])
        self.km = KeyManager()
        
    def setup_workspace(self):
        """Garante que o código e as instruções estão atualizados"""
        print(f" [>] Sincronizando repositório: {self.repo_name}")
        os.makedirs(BASE_DIR, exist_ok=True)
        
        if not os.path.exists(self.local_path):
            subprocess.run(["git", "clone", self.repo_url, self.local_path])
        else:
            subprocess.run(["git", "-C", self.local_path, "pull"])
            
        # Sincroniza instruções globais
        subprocess.run(["git", "-C", INSTRUCOES_DIR, "pull"])

    def get_system_prompt(self):
        """Carrega a doutrina e as instruções de cada VPS"""
        with open(f"{INSTRUCOES_DIR}/global/doutrina_agentes.md", "r") as f:
            doutrina = f.read()
        with open(f"{INSTRUCOES_DIR}/global/claude.md", "r") as f:
            especifico = f.read()
            
        ip_vps = subprocess.getoutput("hostname -I").split()[0]
        
        return f"""
        {doutrina}
        {especifico}
        
        CONTEXTO ATUAL:
        - Você está na VPS com IP: {ip_vps}
        - Projeto: {self.repo_name}
        - Tarefa (Issue): {json.dumps(self.data.get('data', {}).get('issue', {}))}
        """

    def post_comment(self, message):
        """Posta um comentário de feedback na Issue do Gitea usando urllib (zero-dependency)"""
        issue_id = self.data.get('data', {}).get('issue', {}).get('number')
        if not issue_id:
            return

        url = f"http://10.99.0.3:3000/api/v1/repos/{self.repo_name}/issues/{issue_id}/comments"
        
        import base64
        from urllib import request, parse
        
        # Auth Basic
        auth_str = 'ruicesar:h$tg@g5aga$ra1E3$C-yHW$-BA@DF2@Grfa!3#'
        auth_bytes = auth_str.encode('ascii')
        base64_auth = base64.b64encode(auth_bytes).decode('ascii')
        
        data = json.dumps({"body": message}).encode('utf-8')
        req = request.Request(url, data=data, method='POST')
        req.add_header('Content-Type', 'application/json')
        req.add_header('Authorization', f'Basic {base64_auth}')
        
        try:
            with request.urlopen(req) as response:
                if response.status == 201:
                    print(f" [✓] Comentário postado na Issue #{issue_id}")
                else:
                    print(f" [X] Erro ao postar comentário: {response.status}")
        except Exception as e:
            print(f" [E] Falha na comunicação com Gitea via urllib: {e}")

    def execute(self):
        self.setup_workspace()
        
        api_key = self.km.get_key('anthropic')
        if not api_key:
            return

        client = Anthropic(api_key=api_key)
        prompt = self.get_system_prompt()

        print(" [!] IA Pensando na solução...")
        
        # Simulação de resposta da IA para o teste
        feedback = "### 🤖 [VPS 4 - LABORATÓRIO]:\n* **Status**: Tarefa recebida e ambiente preparado.\n* **Ação**: Sincronização de arquivos concluída.\n* **Próximo**: Iniciando correção do erro de regressão de versão."
        
        self.post_comment(feedback)

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--data", help="Dados do Webhook JSON")
    args = parser.parse_args()
    
    if args.data:
        agent = SwarmAgent(json.loads(args.data))
        agent.execute()