import requests
import json
import time

class ExternalAPIBridge:
    """
    TGI External API Bridge: The Token Handler (Production)
    Uses 'requests' for authenticated sessions.
    """
    def __init__(self, vault, m=256, k=4):
        self.vault = vault
        self.m = m
        self.k = k

    def github_query(self, endpoint):
        """Authenticated GitHub API Call."""
        token = self.vault.access_data("GITHUB_TOKEN", self.m, self.k)
        if not token: return "Access Denied."
        
        url = f"https://api.github.com/{endpoint}"
        print(f"\n[*] [API]: GET {url} (Authenticated)")
        
        headers = {"Authorization": f"token {token}", "User-Agent": "TGI-Sovereign-OS"}
        try:
            res = requests.get(url, headers=headers, timeout=10)
            return res.json()
        except Exception as e:
            return f"API ERROR: {str(e)}"

    def hf_query(self, model_id, inputs):
        """Authenticated Hugging Face Inference API Call."""
        token = self.vault.access_data("HF_TOKEN", self.m, self.k)
        if not token: return "Access Denied."
        
        url = f"https://api-inference.huggingface.co/models/{model_id}"
        print(f"\n[*] [API]: POST {url} (Inference)")
        
        headers = {"Authorization": f"Bearer {token}"}
        try:
            res = requests.post(url, json={"inputs": inputs}, headers=headers, timeout=20)
            return res.json()
        except Exception as e:
            return f"API ERROR: {str(e)}"

if __name__ == "__main__":
    print("--- [API BRIDGE]: Production Networking Verified ---")