mikopbxbot/bot/services/mikopbx_service.py

import aiohttp
import logging
import json
from typing import Optional, Dict, Any
from config import (
    MIKOPBX_HOST, 
    MIKOPBX_API_TOKEN,
    MIKOPBX_ADMIN_LOGIN, 
    MIKOPBX_ADMIN_PASSWORD
)

logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)

class MikoPBXService:
    def __init__(self):
        self.host = MIKOPBX_HOST.rstrip('/')
        self.api_token = MIKOPBX_API_TOKEN
        self.admin_login = MIKOPBX_ADMIN_LOGIN
        self.admin_password = MIKOPBX_ADMIN_PASSWORD
        self.session = None

    async def _get_session(self):
        if self.session is None:
            self.session = aiohttp.ClientSession()
        return self.session

    def _get_auth_headers(self) -> Dict[str, str]:
        headers = {
            "Content-Type": "application/json",
            "Accept": "application/json"
        }
        if self.api_token:
            headers["Authorization"] = f"Bearer {self.api_token}"
            logger.info(">>> Using Bearer Token authentication")
        elif self.admin_login and self.admin_password:
            import base64
            credentials = f"{self.admin_login}:{self.admin_password}"
            encoded = base64.b64encode(credentials.encode()).decode()
            headers["Authorization"] = f"Basic {encoded}"
            logger.info(">>> Using Basic Auth (fallback)")
        return headers

    async def _make_request(self, method: str, endpoint: str, json_data: Optional[Dict] = None, params: Optional[Dict] = None):
        session = await self._get_session()
        url = f"{self.host}{endpoint}"
        headers = self._get_auth_headers()

        logger.info(f"\n{'='*70}")
        logger.info(f">>> REQUEST: {method.upper()} {url}")
        if params:
            logger.info(f">>> PARAMS: {params}")
        if json_data:
            logger.info(f">>> BODY:\n{json.dumps(json_data, indent=2, ensure_ascii=False)}")
        logger.info(f">>> HEADERS: { {k: v for k, v in headers.items() if k != 'Authorization'} }")

        try:
            if method.upper() == "GET":
                async with session.get(url, params=params, headers=headers) as resp:
                    text = await resp.text()
                    logger.info(f"<<< RESPONSE [{resp.status}]:\n{text[:3000]}")
                    try:
                        return await resp.json()
                    except:
                        return {"result": False, "raw": text}
            elif method.upper() == "DELETE":
                async with session.delete(url, headers=headers) as resp:
                    text = await resp.text()
                    logger.info(f"<<< RESPONSE [{resp.status}]:\n{text[:3000]}")
                    try:
                        return await resp.json()
                    except:
                        return {"result": False, "raw": text}
            else:
                async with session.post(url, json=json_data, headers=headers) as resp:
                    text = await resp.text()
                    logger.info(f"<<< RESPONSE [{resp.status}]:\n{text[:3000]}")
                    try:
                        return await resp.json()
                    except:
                        return {"result": False, "raw": text}
        except Exception as e:
            logger.error(f"!!! REQUEST ERROR: {e}")
            return {"result": False, "messages": [str(e)]}

    # ==================== EXTENSIONS ====================

    async def get_extension_template(self) -> Optional[Dict]:
        """Получить шаблон нового сотрудника"""
        logger.info(">>> Trying to get extension template...")
        
        # Пробуем v3
        response = await self._make_request(
            "GET", 
            "/pbxcore/api/v3/extensions", 
            params={"id": ""}
        )
        
        data = response.get("data")
        
        if isinstance(data, dict):
            logger.info(">>> Template received as dict")
            return data
        elif isinstance(data, list) and len(data) > 0:
            logger.info(">>> Template received as list, using first item as template")
            return data[0].copy() if isinstance(data[0], dict) else None
        else:
            logger.warning(">>> Could not get template from API. Will use minimal structure.")
            return None

    async def create_extension(self, number: str, secret: str = None, username: str = "", email: str = "") -> Dict[str, Any]:
        logger.info(f"\n>>> Creating extension {number}...")

        template = await self.get_extension_template()
        
        if not template:
            logger.warning(">>> Using minimal fallback template")
            template = {
                "type": "SIP",
                "show_in_phonebook": "1",
                "is_general_user_number": "1",
            }

        # Заполняем данными
        template["number"] = number
        if username:
            template["user_username"] = username
        if email:
            template["user_email"] = email
        if secret:
            template["sip_secret"] = secret

        logger.info(f">>> Final creation payload prepared")

        response = await self._make_request(
            "POST",
            "/pbxcore/api/v3/extensions",
            json_data=template
        )
        
        if response.get("result"):
            return {
                "success": True,
                "number": number,
                "data": response.get("data", {})
            }
        else:
            error_msg = response.get("messages") or response.get("error") or ["Unknown error"]
            return {
                "success": False,
                "error": error_msg
            }

    async def get_extension(self, number: str) -> Optional[Dict]:
        response = await self._make_request("GET", f"/pbxcore/api/v3/extensions/{number}")
        return response.get("data") if response.get("result") else None

    async def set_extension_secret(self, number: str, new_secret: str) -> Dict[str, Any]:
        current = await self.get_extension(number)
        if not current:
            return {"success": False, "error": "Extension not found"}
        current["sip_secret"] = new_secret
        response = await self._make_request("PUT", f"/pbxcore/api/v3/extensions/{number}", json_data=current)
        return {"success": response.get("result", False), "messages": response.get("messages", [])}

    async def delete_extension(self, number: str) -> Dict[str, Any]:
        response = await self._make_request("DELETE", f"/pbxcore/api/v3/extensions/{number}")
        return {"success": response.get("result", False), "messages": response.get("messages", [])}

    async def close(self):
        if self.session:
            await self.session.close()