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/x-www-form-urlencoded; charset=UTF-8",
            "Accept": "application/json",
            "X-Requested-With": "XMLHttpRequest"
        }
        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, 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 data:
            logger.info(f">>> BODY: {data}")

        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[:2500]}")
                    try:
                        return await resp.json()
                    except:
                        return {"result": False, "raw": text}
            else:
                async with session.post(url, data=data, headers=headers) as resp:
                    text = await resp.text()
                    logger.info(f"<<< RESPONSE [{resp.status}]:\n{text[:2500]}")
                    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 (legacy but working) ====================

    async def get_extension_template(self) -> Optional[Dict]:
        """Получаем шаблон через старый, но рабочий эндпоинт"""
        logger.info(">>> Getting extension template via legacy endpoint...")
        response = await self._make_request(
            "GET", 
            "/pbxcore/api/extensions/getRecord", 
            params={"id": ""}
        )
        if response.get("result"):
            logger.info(">>> Template received successfully")
            return response.get("data")
        logger.error(f">>> Failed to get template: {response.get('messages')}")
        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:
            return {"success": False, "error": "Failed to get template from MikoPBX"}

        # Заполняем шаблон
        template["number"] = number
        if username:
            template["user_username"] = username
        if email:
            template["user_email"] = email
        template["type"] = "SIP"
        template["show_in_phonebook"] = "1"
        template["is_general_user_number"] = "1"

        if secret:
            template["sip_secret"] = secret

        logger.info(f">>> Sending saveRecord request...")

        response = await self._make_request(
            "POST",
            "/pbxcore/api/extensions/saveRecord",
            data=template
        )
        
        if response.get("result"):
            logger.info(">>> Extension created successfully!")
            return {
                "success": True,
                "number": number,
                "data": response.get("data", {})
            }
        else:
            error = response.get("messages", ["Unknown error"])
            logger.error(f">>> Creation failed: {error}")
            return {
                "success": False,
                "error": error
            }

    async def get_extension(self, number: str) -> Optional[Dict]:
        response = await self._make_request(
            "GET", 
            "/pbxcore/api/extensions/getRecord", 
            params={"id": 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(
            "POST", 
            "/pbxcore/api/extensions/saveRecord", 
            data=current
        )
        return {"success": response.get("result", False), "messages": response.get("messages", [])}

    async def delete_extension(self, number: str) -> Dict[str, Any]:
        current = await self.get_extension(number)
        if not current:
            return {"success": False, "error": "Extension not found"}
        record_id = current.get("id") or number
        response = await self._make_request(
            "POST", 
            "/pbxcore/api/extensions/deleteRecord", 
            data={"id": record_id}
        )
        return {"success": response.get("result", False), "messages": response.get("messages", [])}

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