init
This commit is contained in:
176
bot/services/database.py
Normal file
176
bot/services/database.py
Normal file
@@ -0,0 +1,176 @@
|
||||
import aiosqlite
|
||||
import logging
|
||||
from typing import List, Dict, Optional
|
||||
from config import DB_PATH
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class Database:
|
||||
def __init__(self):
|
||||
self.db_path = DB_PATH
|
||||
|
||||
async def init_db(self):
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute("""
|
||||
CREATE TABLE IF NOT EXISTS users (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
telegram_id INTEGER UNIQUE NOT NULL,
|
||||
telegram_username TEXT,
|
||||
full_name TEXT,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
""")
|
||||
|
||||
await db.execute("""
|
||||
CREATE TABLE IF NOT EXISTS sip_accounts (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
user_id INTEGER NOT NULL,
|
||||
extension_number TEXT UNIQUE NOT NULL,
|
||||
sip_secret TEXT,
|
||||
username TEXT,
|
||||
email TEXT,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
FOREIGN KEY (user_id) REFERENCES users (id)
|
||||
)
|
||||
""")
|
||||
await db.commit()
|
||||
logger.info("Database initialized")
|
||||
|
||||
async def add_user(self, telegram_id: int, username: str = None, full_name: str = None) -> int:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute(
|
||||
"INSERT OR IGNORE INTO users (telegram_id, telegram_username, full_name) VALUES (?, ?, ?)",
|
||||
(telegram_id, username, full_name)
|
||||
)
|
||||
await db.commit()
|
||||
if cursor.lastrowid == 0:
|
||||
# User already exists, get their id
|
||||
cursor = await db.execute(
|
||||
"SELECT id FROM users WHERE telegram_id = ?",
|
||||
(telegram_id,)
|
||||
)
|
||||
row = await cursor.fetchone()
|
||||
return row[0] if row else None
|
||||
return cursor.lastrowid
|
||||
|
||||
async def get_user(self, telegram_id: int) -> Optional[Dict]:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute(
|
||||
"SELECT * FROM users WHERE telegram_id = ?",
|
||||
(telegram_id,)
|
||||
)
|
||||
row = await cursor.fetchone()
|
||||
if row:
|
||||
return {
|
||||
"id": row[0],
|
||||
"telegram_id": row[1],
|
||||
"telegram_username": row[2],
|
||||
"full_name": row[3],
|
||||
"created_at": row[4]
|
||||
}
|
||||
return None
|
||||
|
||||
async def add_sip_account(self, user_id: int, extension_number: str, sip_secret: str = None,
|
||||
username: str = None, email: str = None) -> int:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute(
|
||||
"""INSERT INTO sip_accounts
|
||||
(user_id, extension_number, sip_secret, username, email)
|
||||
VALUES (?, ?, ?, ?, ?)""",
|
||||
(user_id, extension_number, sip_secret, username, email)
|
||||
)
|
||||
await db.commit()
|
||||
return cursor.lastrowid
|
||||
|
||||
async def get_user_sip_accounts(self, telegram_id: int) -> List[Dict]:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute("""
|
||||
SELECT sa.* FROM sip_accounts sa
|
||||
JOIN users u ON sa.user_id = u.id
|
||||
WHERE u.telegram_id = ?
|
||||
ORDER BY sa.created_at DESC
|
||||
""", (telegram_id,))
|
||||
rows = await cursor.fetchall()
|
||||
return [
|
||||
{
|
||||
"id": row[0],
|
||||
"user_id": row[1],
|
||||
"extension_number": row[2],
|
||||
"sip_secret": row[3],
|
||||
"username": row[4],
|
||||
"email": row[5],
|
||||
"created_at": row[6]
|
||||
}
|
||||
for row in rows
|
||||
]
|
||||
|
||||
async def get_all_sip_accounts(self) -> List[Dict]:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute("""
|
||||
SELECT sa.*, u.telegram_id, u.telegram_username, u.full_name
|
||||
FROM sip_accounts sa
|
||||
JOIN users u ON sa.user_id = u.id
|
||||
ORDER BY sa.created_at DESC
|
||||
""")
|
||||
rows = await cursor.fetchall()
|
||||
return [
|
||||
{
|
||||
"id": row[0],
|
||||
"user_id": row[1],
|
||||
"extension_number": row[2],
|
||||
"sip_secret": row[3],
|
||||
"username": row[4],
|
||||
"email": row[5],
|
||||
"created_at": row[6],
|
||||
"telegram_id": row[7],
|
||||
"telegram_username": row[8],
|
||||
"full_name": row[9]
|
||||
}
|
||||
for row in rows
|
||||
]
|
||||
|
||||
async def update_sip_secret(self, extension_number: str, new_secret: str) -> bool:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute(
|
||||
"UPDATE sip_accounts SET sip_secret = ? WHERE extension_number = ?",
|
||||
(new_secret, extension_number)
|
||||
)
|
||||
await db.commit()
|
||||
return cursor.rowcount > 0
|
||||
|
||||
async def update_extension_number(self, old_number: str, new_number: str) -> bool:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute(
|
||||
"UPDATE sip_accounts SET extension_number = ? WHERE extension_number = ?",
|
||||
(new_number, old_number)
|
||||
)
|
||||
await db.commit()
|
||||
return cursor.rowcount > 0
|
||||
|
||||
async def delete_sip_account(self, extension_number: str) -> bool:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute(
|
||||
"DELETE FROM sip_accounts WHERE extension_number = ?",
|
||||
(extension_number,)
|
||||
)
|
||||
await db.commit()
|
||||
return cursor.rowcount > 0
|
||||
|
||||
async def get_sip_account_by_number(self, number: str) -> Optional[Dict]:
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute(
|
||||
"SELECT * FROM sip_accounts WHERE extension_number = ?",
|
||||
(number,)
|
||||
)
|
||||
row = await cursor.fetchone()
|
||||
if row:
|
||||
return {
|
||||
"id": row[0],
|
||||
"user_id": row[1],
|
||||
"extension_number": row[2],
|
||||
"sip_secret": row[3],
|
||||
"username": row[4],
|
||||
"email": row[5],
|
||||
"created_at": row[6]
|
||||
}
|
||||
return None
|
||||
268
bot/services/mikopbx_service.py
Normal file
268
bot/services/mikopbx_service.py
Normal file
@@ -0,0 +1,268 @@
|
||||
import aiohttp
|
||||
import logging
|
||||
from typing import Optional, Dict, Any
|
||||
from config import MIKOPBX_HOST, MIKOPBX_ADMIN_LOGIN, MIKOPBX_ADMIN_PASSWORD
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class MikoPBXService:
|
||||
def __init__(self):
|
||||
self.host = MIKOPBX_HOST.rstrip('/')
|
||||
self.login = MIKOPBX_ADMIN_LOGIN
|
||||
self.password = MIKOPBX_ADMIN_PASSWORD
|
||||
self.session = None
|
||||
self.auth_cookies = None
|
||||
|
||||
async def _get_session(self):
|
||||
if self.session is None:
|
||||
self.session = aiohttp.ClientSession()
|
||||
return self.session
|
||||
|
||||
async def login(self) -> bool:
|
||||
"""Login to MikoPBX admin panel to get session cookie"""
|
||||
session = await self._get_session()
|
||||
try:
|
||||
url = f"{self.host}/admin-cabinet/session/start"
|
||||
data = {
|
||||
"login": self.login,
|
||||
"password": self.password
|
||||
}
|
||||
async with session.post(url, data=data) as resp:
|
||||
if resp.status == 200:
|
||||
self.auth_cookies = resp.cookies
|
||||
logger.info("Successfully logged into MikoPBX")
|
||||
return True
|
||||
else:
|
||||
logger.error(f"Login failed: {resp.status}")
|
||||
return False
|
||||
except Exception as e:
|
||||
logger.error(f"Login error: {e}")
|
||||
return False
|
||||
|
||||
async def _make_request(self, method: str, endpoint: str, data: Optional[Dict] = None, params: Optional[Dict] = None):
|
||||
"""Make authenticated request to MikoPBX API"""
|
||||
session = await self._get_session()
|
||||
|
||||
if not self.auth_cookies:
|
||||
await self.login()
|
||||
|
||||
url = f"{self.host}{endpoint}"
|
||||
headers = {
|
||||
"X-Requested-With": "XMLHttpRequest",
|
||||
"Content-Type": "application/x-www-form-urlencoded; charset=UTF-8"
|
||||
}
|
||||
|
||||
cookies = {}
|
||||
if self.auth_cookies:
|
||||
for key, cookie in self.auth_cookies.items():
|
||||
cookies[key] = cookie.value
|
||||
|
||||
try:
|
||||
if method.upper() == "GET":
|
||||
async with session.get(url, params=params, cookies=cookies, headers=headers) as resp:
|
||||
return await resp.json()
|
||||
else:
|
||||
async with session.post(url, data=data, cookies=cookies, headers=headers) as resp:
|
||||
return await resp.json()
|
||||
except Exception as e:
|
||||
logger.error(f"API request error: {e}")
|
||||
return {"result": False, "messages": [str(e)]}
|
||||
|
||||
async def get_extension_template(self) -> Optional[Dict]:
|
||||
"""Get empty template for new extension"""
|
||||
response = await self._make_request(
|
||||
"GET",
|
||||
"/pbxcore/api/extensions/getRecord",
|
||||
params={"id": ""}
|
||||
)
|
||||
if response.get("result"):
|
||||
return response.get("data")
|
||||
return None
|
||||
|
||||
async def create_extension(self, number: str, secret: str = None, username: str = "", email: str = "") -> Dict[str, Any]:
|
||||
"""Create new SIP extension with optional custom secret"""
|
||||
template = await self.get_extension_template()
|
||||
if not template:
|
||||
return {"success": False, "error": "Failed to get template"}
|
||||
|
||||
# Fill template with our data
|
||||
template["number"] = number
|
||||
if username:
|
||||
template["user_username"] = username
|
||||
if email:
|
||||
template["user_email"] = email
|
||||
|
||||
# Ensure SIP type and basic settings
|
||||
template["type"] = "SIP"
|
||||
template["show_in_phonebook"] = "1"
|
||||
template["is_general_user_number"] = "1"
|
||||
|
||||
# Set custom secret if provided
|
||||
if secret:
|
||||
template["sip_secret"] = secret
|
||||
|
||||
response = await self._make_request(
|
||||
"POST",
|
||||
"/pbxcore/api/extensions/saveRecord",
|
||||
data=template
|
||||
)
|
||||
|
||||
if response.get("result"):
|
||||
created_data = response.get("data", {})
|
||||
return {
|
||||
"success": True,
|
||||
"number": number,
|
||||
"mikopbx_id": created_data.get("id") or created_data.get("sip_uniqid"),
|
||||
"data": created_data
|
||||
}
|
||||
else:
|
||||
return {
|
||||
"success": False,
|
||||
"error": response.get("messages", ["Unknown error"])
|
||||
}
|
||||
|
||||
async def update_extension(self, extension_id: str, data: Dict) -> Dict[str, Any]:
|
||||
"""Update existing extension"""
|
||||
# First get current record
|
||||
current = await self._make_request(
|
||||
"GET",
|
||||
"/pbxcore/api/extensions/getRecord",
|
||||
params={"id": extension_id}
|
||||
)
|
||||
|
||||
if not current.get("result"):
|
||||
return {"success": False, "error": "Extension not found"}
|
||||
|
||||
record = current.get("data", {})
|
||||
record.update(data)
|
||||
|
||||
response = await self._make_request(
|
||||
"POST",
|
||||
"/pbxcore/api/extensions/saveRecord",
|
||||
data=record
|
||||
)
|
||||
|
||||
return {
|
||||
"success": response.get("result", False),
|
||||
"messages": response.get("messages", [])
|
||||
}
|
||||
|
||||
async def change_extension_number(self, old_number: str, new_number: str) -> Dict[str, Any]:
|
||||
"""Change extension number (delete old + create new with same secret)"""
|
||||
# Get current record by number (we'll search via getList if available, else use template approach)
|
||||
# For simplicity we get template and update the number
|
||||
template = await self.get_extension_template()
|
||||
if not template:
|
||||
return {"success": False, "error": "Failed to get template"}
|
||||
|
||||
# We need the existing record. Since there's no direct "get by number" we use a workaround:
|
||||
# Try to find by attempting getRecord with number as ID (MikoPBX sometimes accepts it)
|
||||
current = await self._make_request(
|
||||
"GET",
|
||||
"/pbxcore/api/extensions/getRecord",
|
||||
params={"id": old_number}
|
||||
)
|
||||
|
||||
if current.get("result"):
|
||||
record = current.get("data", {})
|
||||
old_secret = record.get("sip_secret", "")
|
||||
|
||||
# Delete old extension first
|
||||
delete_response = await self._make_request(
|
||||
"POST",
|
||||
"/pbxcore/api/extensions/deleteRecord",
|
||||
data={"id": record.get("id", old_number)}
|
||||
)
|
||||
|
||||
# Create new one with same settings but new number
|
||||
record["number"] = new_number
|
||||
record["id"] = "" # new record
|
||||
record.pop("sip_uniqid", None)
|
||||
|
||||
create_response = await self._make_request(
|
||||
"POST",
|
||||
"/pbxcore/api/extensions/saveRecord",
|
||||
data=record
|
||||
)
|
||||
|
||||
if create_response.get("result"):
|
||||
return {
|
||||
"success": True,
|
||||
"old_number": old_number,
|
||||
"new_number": new_number,
|
||||
"secret": old_secret
|
||||
}
|
||||
|
||||
return {"success": False, "error": "Could not change number"}
|
||||
|
||||
async def set_extension_secret(self, number: str, new_secret: str) -> Dict[str, Any]:
|
||||
"""Update secret for existing extension"""
|
||||
current = await self._make_request(
|
||||
"GET",
|
||||
"/pbxcore/api/extensions/getRecord",
|
||||
params={"id": number}
|
||||
)
|
||||
|
||||
if not current.get("result"):
|
||||
return {"success": False, "error": "Extension not found"}
|
||||
|
||||
record = current.get("data", {})
|
||||
record["sip_secret"] = new_secret
|
||||
|
||||
response = await self._make_request(
|
||||
"POST",
|
||||
"/pbxcore/api/extensions/saveRecord",
|
||||
data=record
|
||||
)
|
||||
|
||||
return {
|
||||
"success": response.get("result", False),
|
||||
"messages": response.get("messages", [])
|
||||
}
|
||||
|
||||
async def get_all_extensions(self) -> list:
|
||||
"""Get list of all extensions (if supported by MikoPBX)"""
|
||||
# Note: MikoPBX may have /pbxcore/api/extensions/getList endpoint
|
||||
response = await self._make_request(
|
||||
"GET",
|
||||
"/pbxcore/api/extensions/getList"
|
||||
)
|
||||
|
||||
if response.get("result"):
|
||||
return response.get("data", [])
|
||||
return []
|
||||
|
||||
async def delete_extension(self, number: str) -> Dict[str, Any]:
|
||||
"""Delete extension"""
|
||||
current = await self._make_request(
|
||||
"GET",
|
||||
"/pbxcore/api/extensions/getRecord",
|
||||
params={"id": number}
|
||||
)
|
||||
|
||||
if not current.get("result"):
|
||||
return {"success": False, "error": "Extension not found"}
|
||||
|
||||
record = current.get("data", {})
|
||||
record_id = record.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 get_extension_by_number(self, number: str) -> Optional[Dict]:
|
||||
"""Try to find extension by number"""
|
||||
# This is a simplified approach - in real scenario we would use getList
|
||||
# For now return None, or implement getList
|
||||
return None
|
||||
|
||||
async def close(self):
|
||||
if self.session:
|
||||
await self.session.close()
|
||||
Reference in New Issue
Block a user