"""Роуты для работы с инстансами VM/LXC.""" import logging from typing import List from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException from sqlalchemy.orm import Session, joinedload from .. import models, proxmox_client as pve, schemas from ..config import settings from ..database import SessionLocal, get_db from ..deps import get_current_user logger = logging.getLogger(__name__) router = APIRouter(prefix="/instances", tags=["instances"]) def _ensure_owner(instance: models.Instance, user: models.User) -> None: """Проверяет владельца инстанса.""" if user.role == models.Role.admin: return if instance.owner_id != user.id: raise HTTPException(status_code=404, detail="VPS не найден") def _sanitize_name(raw: str) -> str: """Оставляет в имени только латиницу, цифры и дефис.""" cleaned = "".join(c for c in raw.strip() if c.isalnum() or c == "-").lower() return cleaned or "vps" @router.get("", response_model=List[schemas.InstanceOut]) def list_my_instances(db: Session = Depends(get_db), user: models.User = Depends(get_current_user)): """Возвращает список инстансов пользователя или всех инстансов для администратора.""" query = db.query(models.Instance).options(joinedload(models.Instance.template)) query = query.filter(models.Instance.status != models.InstanceStatus.deleted) if user.role != models.Role.admin: query = query.filter(models.Instance.owner_id == user.id) return query.order_by(models.Instance.created_at.desc()).all() def _provision(instance_id: int, template_id: int, node: str) -> None: """Создаёт VPS в фоновой задаче и сохраняет результат в базе.""" db = SessionLocal() try: instance = db.query(models.Instance).filter(models.Instance.id == instance_id).first() if not instance: logger.warning("provision: instance %s не найден", instance_id) return template = db.query(models.Template).filter(models.Template.id == template_id).first() if not template: instance.status = models.InstanceStatus.error db.commit() return try: clean_name = _sanitize_name(instance.name) if template.guest_type == models.GuestType.vm: if not template.source_vmid: raise ValueError("У VM-шаблона не указан source_vmid") pve.clone_vm(template.source_vmid, instance.vmid, clean_name, node) pve.resize_vm(instance.vmid, template.cores, template.memory_mb, node) pve.resize_disk(instance.vmid, template.disk_gb, node) if instance.ciuser and instance.root_password: pve.configure_cloud_init(instance.vmid, instance.ciuser, instance.root_password, node) pve.guest_action("vm", instance.vmid, "start", node) else: # LXC создаётся ТОЛЬКО из архивного шаблона (vztmpl). # Создание из существующего LXC по VMID не поддерживается. if not template.source_template: raise ValueError("У LXC-шаблона не указан source_template") # create_lxc делает всё сам: ostemplate → passwd → resize → start. password = pve.create_lxc( new_vmid=instance.vmid, name=clean_name, template_volid=template.source_template, cores=template.cores, memory_mb=template.memory_mb, disk_gb=template.disk_gb, node=node, ) instance.root_password = password instance.status = models.InstanceStatus.running logger.info("Инстанс %s (vmid=%s) создан и запущен", instance.id, instance.vmid) except Exception as exc: logger.exception("Ошибка создания инстанса %s", instance_id) instance.status = models.InstanceStatus.error instance.root_password = f"ERROR: {type(exc).__name__}: {exc}" db.commit() finally: db.close() @router.post("", response_model=schemas.InstanceOut) def create_instance( payload: schemas.InstanceCreate, background_tasks: BackgroundTasks, db: Session = Depends(get_db), user: models.User = Depends(get_current_user), ): """Регистрирует один инстанс и запускает его создание в фоне.""" db.query(models.User).filter(models.User.id == user.id).with_for_update().one() pending = ( db.query(models.Instance) .filter( models.Instance.owner_id == user.id, models.Instance.status == models.InstanceStatus.creating, ) .first() ) if pending: raise HTTPException(status_code=409, detail="Предыдущий VPS ещё создаётся. Дождитесь завершения операции.") template = ( db.query(models.Template) .filter(models.Template.id == payload.template_id, models.Template.is_active == True) # noqa: E712 .first() ) if not template: raise HTTPException(status_code=404, detail="Шаблон не найден") try: vmid = pve.get_next_vmid() except Exception as exc: logger.error("Proxmox недоступен при запросе nextid: %s", exc) raise HTTPException(status_code=502, detail=f"Proxmox недоступен: {exc}") active = ( db.query(models.Instance) .filter(models.Instance.vmid == vmid, models.Instance.status != models.InstanceStatus.deleted) .first() ) if active: raise HTTPException(status_code=409, detail=f"VMID {vmid} уже используется в панели") instance = models.Instance( name=_sanitize_name(payload.name), vmid=vmid, node=settings.pve_node, guest_type=template.guest_type, status=models.InstanceStatus.creating, owner_id=user.id, template_id=template.id, ciuser=payload.ciuser or None, root_password=payload.cipassword or None, ) db.add(instance) db.commit() db.refresh(instance) background_tasks.add_task(_provision, instance.id, template.id, settings.pve_node) return instance @router.get("/{instance_id}", response_model=schemas.InstanceOut) def get_instance(instance_id: int, db: Session = Depends(get_db), user: models.User = Depends(get_current_user)): """Возвращает один инстанс.""" instance = db.query(models.Instance).filter(models.Instance.id == instance_id).first() if not instance: raise HTTPException(status_code=404, detail="VPS не найден") _ensure_owner(instance, user) return instance @router.get("/{instance_id}/status") def get_instance_status(instance_id: int, db: Session = Depends(get_db), user: models.User = Depends(get_current_user)): """Возвращает фактический статус инстанса из Proxmox.""" instance = db.query(models.Instance).filter(models.Instance.id == instance_id).first() if not instance: raise HTTPException(status_code=404, detail="VPS не найден") _ensure_owner(instance, user) try: return pve.get_status(instance.guest_type.value, instance.vmid, instance.node) except Exception as exc: logger.warning("get_status: vmid=%s: %s", instance.vmid, exc) raise HTTPException(status_code=502, detail=f"Proxmox недоступен: {exc}") @router.get("/{instance_id}/ip", response_model=schemas.InstanceIPOut) def get_instance_ip(instance_id: int, db: Session = Depends(get_db), user: models.User = Depends(get_current_user)): """Возвращает IP-адрес инстанса.""" instance = db.query(models.Instance).filter(models.Instance.id == instance_id).first() if not instance: raise HTTPException(status_code=404, detail="VPS не найден") _ensure_owner(instance, user) try: return schemas.InstanceIPOut(ip=pve.get_instance_ip(instance.vmid, instance.node)) except Exception as exc: logger.warning("get_instance_ip: vmid=%s: %s", instance.vmid, exc) return schemas.InstanceIPOut(ip=f"ошибка: {exc}") @router.get("/{instance_id}/live") def get_instance_live(instance_id: int, db: Session = Depends(get_db), user: models.User = Depends(get_current_user)): """Возвращает live-показатели CPU, RAM и uptime.""" instance = db.query(models.Instance).filter(models.Instance.id == instance_id).first() if not instance: raise HTTPException(status_code=404, detail="VPS не найден") _ensure_owner(instance, user) try: return pve.get_live_stats(instance.vmid, instance.node) except Exception as exc: logger.warning("get_instance_live: vmid=%s: %s", instance.vmid, exc) return {"cpu": 0, "mem_used": 0, "mem_total": 0, "uptime": 0, "status": "error", "error": str(exc)} @router.post("/{instance_id}/action") def instance_action(instance_id: int, payload: schemas.InstanceAction, db: Session = Depends(get_db), user: models.User = Depends(get_current_user)): """Выполняет действие start, stop, shutdown или reboot.""" instance = db.query(models.Instance).filter(models.Instance.id == instance_id).first() if not instance: raise HTTPException(status_code=404, detail="VPS не найден") _ensure_owner(instance, user) try: pve.guest_action(instance.guest_type.value, instance.vmid, payload.action, instance.node) except ValueError as exc: raise HTTPException(status_code=400, detail=str(exc)) except Exception as exc: logger.warning("instance_action: vmid=%s action=%s: %s", instance.vmid, payload.action, exc) raise HTTPException(status_code=502, detail=f"Ошибка: {exc}") if payload.action == "start": instance.status = models.InstanceStatus.running elif payload.action in ("stop", "shutdown"): instance.status = models.InstanceStatus.stopped db.commit() return {"ok": True} @router.delete("/{instance_id}") def delete_instance(instance_id: int, db: Session = Depends(get_db), user: models.User = Depends(get_current_user)): """Останавливает работающую VM/LXC и удаляет её из Proxmox.""" instance = db.query(models.Instance).filter(models.Instance.id == instance_id).first() if not instance: raise HTTPException(status_code=404, detail="VPS не найден") _ensure_owner(instance, user) if instance.status == models.InstanceStatus.deleting: raise HTTPException(status_code=409, detail="Инстанс уже удаляется") instance.status = models.InstanceStatus.deleting db.commit() try: pve.delete_guest(instance.guest_type.value, instance.vmid, instance.node) except Exception as exc: logger.exception("Ошибка удаления instance_id=%s", instance_id) instance.status = models.InstanceStatus.error db.commit() raise HTTPException(status_code=502, detail=f"Ошибка удаления: {exc}") instance.status = models.InstanceStatus.deleted db.commit() return {"ok": True}