"""
Connector Manager — Manus-style connector ecosystem
Manages OAuth tokens, connection state, API access
"""
import json
import os
import time
from typing import Dict, List, Optional
import structlog

log = structlog.get_logger()

CONNECTORS_CONFIG = [
    {
        "id": "github",
        "name": "GitHub",
        "icon": "github",
        "color": "#24292e",
        "env_key": "GITHUB_TOKEN",
        "description": "Repos, Issues, PRs, Commits",
        "scopes": ["repo", "issues", "pull_requests"],
        "category": "code",
    },
    {
        "id": "huggingface",
        "name": "HuggingFace",
        "icon": "huggingface",
        "color": "#ff9d00",
        "env_key": "HF_TOKEN",
        "description": "Spaces, Models, Datasets",
        "scopes": ["spaces", "models"],
        "category": "ai",
    },
    {
        "id": "vercel",
        "name": "Vercel",
        "icon": "vercel",
        "color": "#000000",
        "env_key": "VERCEL_TOKEN",
        "description": "Deployments, Domains, Functions",
        "scopes": ["deployments", "projects"],
        "category": "deploy",
    },
    {
        "id": "openai",
        "name": "OpenAI",
        "icon": "openai",
        "color": "#10a37f",
        "env_key": "OPENAI_API_KEY",
        "description": "GPT-4o, Embeddings, DALL-E",
        "scopes": ["chat", "embeddings"],
        "category": "ai",
    },
    {
        "id": "groq",
        "name": "Groq",
        "icon": "groq",
        "color": "#f55036",
        "env_key": "GROQ_API_KEY",
        "description": "Llama 3.3 70B — Ultra Fast",
        "scopes": ["chat"],
        "category": "ai",
    },
    {
        "id": "cerebras",
        "name": "Cerebras",
        "icon": "cerebras",
        "color": "#7c3aed",
        "env_key": "CEREBRAS_API_KEY",
        "description": "Llama 3.1 70B — Long Context",
        "scopes": ["chat"],
        "category": "ai",
    },
    {
        "id": "openrouter",
        "name": "OpenRouter",
        "icon": "openrouter",
        "color": "#6366f1",
        "env_key": "OPENROUTER_API_KEY",
        "description": "Multi-model router, free tier",
        "scopes": ["chat"],
        "category": "ai",
    },
    {
        "id": "anthropic",
        "name": "Anthropic",
        "icon": "anthropic",
        "color": "#d4a27f",
        "env_key": "ANTHROPIC_API_KEY",
        "description": "Claude 3.5 Sonnet",
        "scopes": ["chat"],
        "category": "ai",
    },
    {
        "id": "n8n",
        "name": "n8n",
        "icon": "n8n",
        "color": "#ea4b71",
        "env_key": "N8N_URL",
        "description": "Workflow automation engine",
        "scopes": ["workflows", "executions"],
        "category": "workflow",
    },
    {
        "id": "telegram",
        "name": "Telegram",
        "icon": "telegram",
        "color": "#0088cc",
        "env_key": "TELEGRAM_BOT_TOKEN",
        "description": "Bot API, messages, webhooks",
        "scopes": ["messages", "bots"],
        "category": "messaging",
    },
    {
        "id": "discord",
        "name": "Discord",
        "icon": "discord",
        "color": "#5865f2",
        "env_key": "DISCORD_BOT_TOKEN",
        "description": "Bot, channels, webhooks",
        "scopes": ["messages", "bots"],
        "category": "messaging",
    },
    {
        "id": "slack",
        "name": "Slack",
        "icon": "slack",
        "color": "#4a154b",
        "env_key": "SLACK_BOT_TOKEN",
        "description": "Messages, channels, workflows",
        "scopes": ["messages", "channels"],
        "category": "messaging",
    },
    {
        "id": "cloudflare",
        "name": "Cloudflare",
        "icon": "cloudflare",
        "color": "#f38020",
        "env_key": "CLOUDFLARE_API_TOKEN",
        "description": "Workers, KV, Pages",
        "scopes": ["workers", "kv", "pages"],
        "category": "infra",
    },
]


class ConnectorManager:
    """Manages all platform connectors — connection state, tokens, status."""

    def __init__(self):
        self._configs = {c["id"]: c for c in CONNECTORS_CONFIG}

    def get_all(self) -> List[Dict]:
        """Get all connectors with connection status."""
        result = []
        for cfg in CONNECTORS_CONFIG:
            token = os.environ.get(cfg["env_key"], "")
            result.append({
                **cfg,
                "connected": bool(token),
                "token_preview": f"{token[:8]}..." if token else None,
            })
        return result

    def get_connected(self) -> List[Dict]:
        """Get only connected connectors."""
        return [c for c in self.get_all() if c["connected"]]

    def get_by_category(self, category: str) -> List[Dict]:
        """Get connectors by category."""
        return [c for c in self.get_all() if c["category"] == category]

    def is_connected(self, connector_id: str) -> bool:
        cfg = self._configs.get(connector_id)
        if not cfg:
            return False
        return bool(os.environ.get(cfg["env_key"], ""))

    def get_token(self, connector_id: str) -> Optional[str]:
        cfg = self._configs.get(connector_id)
        if not cfg:
            return None
        return os.environ.get(cfg["env_key"]) or None

    def set_token(self, connector_id: str, token: str):
        """Set connector token at runtime (does not persist across restarts)."""
        cfg = self._configs.get(connector_id)
        if cfg:
            os.environ[cfg["env_key"]] = token
            log.info("Connector token set", connector=connector_id)

    def get_summary(self) -> Dict:
        all_c = self.get_all()
        connected = [c for c in all_c if c["connected"]]
        by_cat = {}
        for c in all_c:
            cat = c["category"]
            if cat not in by_cat:
                by_cat[cat] = {"total": 0, "connected": 0}
            by_cat[cat]["total"] += 1
            if c["connected"]:
                by_cat[cat]["connected"] += 1
        return {
            "total": len(all_c),
            "connected": len(connected),
            "by_category": by_cat,
            "ai_ready": self.is_connected("openai") or self.is_connected("groq")
                        or self.is_connected("openrouter") or self.is_connected("anthropic")
                        or self.is_connected("cerebras"),
        }