Merge pull request #6 from PoCInnovation/feat/api-integration

Feat/api integration

Author: lg-epitech

.gitignore

@@ -9,3 +9,5 @@
 /backend/venv
 
 __pycache__/
+
+.vscode/

backend/app/core/config.py

@@ -4,6 +4,7 @@
 from dotenv import load_dotenv
 from os import getenv
 from sqlmodel import create_engine, SQLModel
+from app.telemetry.monitor import SystemMonitor
 
 load_dotenv()
 
@@ -12,7 +13,8 @@
 db_pass = getenv("POSTGRES_PASSWORD", "distribox_password")
 db_port = getenv("POSTGRES_PORT", "5432")
 
-database_url = f"postgresql+psycopg2://{db_user}:{db_pass}@localhost:{db_port}/{db_name}"
+database_url = f"postgresql+psycopg2://{db_user}:{
+    db_pass}@localhost:{db_port}/{db_name}"
 engine = create_engine(database_url, echo=True)
 SQLModel.metadata.create_all(engine)
 
@@ -24,7 +26,11 @@ class QEMUConfig:
     def get_connection(cls):
         if cls.qemu_conn is None or cls.qemu_conn.isAlive() == 0:
             try:
-                qemu_conn = libvirt.open("qemu:///system")
+                cls.qemu_conn = libvirt.open("qemu:///system")
             except libvirt.libvirtError:
                 raise
-        return qemu_conn
+        return cls.qemu_conn
+
+
+system_monitor = SystemMonitor(interval=3,
+                               get_connection=QEMUConfig.get_connection)

backend/app/core/xml_builder.py

@@ -41,11 +41,14 @@ def build_xml(vm_read: VmRead):
     etree.SubElement(channel, "target", type="virtio",
                      name="org.qemu.guest_agent.0")
 
-    # disk_seed = etree.SubElement(devices, "disk", type="file", device="cdrom")
-    # etree.SubElement(disk_seed, "driver", name="qemu", type="raw")
-    # etree.SubElement(disk_seed, "source", file="/var/lib/distribox/images/seed.iso")
-    # etree.SubElement(disk_seed, "target", dev="hdb", bus="ide")
-    # etree.SubElement(disk_seed, "readonly")
+    disk_seed = etree.SubElement(devices, "disk", type="file", device="cdrom")
+    etree.SubElement(disk_seed, "driver", name="qemu", type="raw")
+    etree.SubElement(
+        disk_seed,
+        "source",
+        file="/var/lib/distribox/images/seed.iso")
+    etree.SubElement(disk_seed, "target", dev="hdb", bus="ide")
+    etree.SubElement(disk_seed, "readonly")
 
     iface = etree.SubElement(devices, "interface", type="network")
     etree.SubElement(iface, "source", network="default")

backend/app/main.py

@@ -2,7 +2,7 @@
 from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from sqlmodel import Session, select
-from app.routes import vm, auth
+from app.routes import vm, image, host, auth
 from app.orm.user import UserORM
 from app.utils.auth import hash_password
 from app.core.config import engine
@@ -63,3 +63,5 @@ async def general_exception_handler(_, exc: Exception):
 
 app.include_router(auth.router, prefix="/auth", tags=["auth"])
 app.include_router(vm.router, prefix="/vms", tags=["vms"])
+app.include_router(image.router, prefix="/images", tags=["images"])
+app.include_router(host.router, prefix="/host", tags=["host"])

backend/app/models/host.py

@@ -0,0 +1,20 @@
+from pydantic import BaseModel
+from app.models.resources import MemoryInfoBase, DiskInfoBase, CPUInfoBase
+
+
+class DiskInfoHost(DiskInfoBase):
+    distribox_used: float
+
+
+class MemoryInfoHost(MemoryInfoBase):
+    pass
+
+
+class CPUInfoHost(CPUInfoBase):
+    pass
+
+
+class HostInfoBase(BaseModel):
+    disk: DiskInfoHost
+    mem: MemoryInfoHost
+    cpu: CPUInfoHost

backend/app/models/image.py

@@ -0,0 +1,11 @@
+from pydantic import BaseModel
+
+
+class ImageBase(BaseModel):
+    name: str
+    virtual_size: float
+    actual_size: float
+
+
+class ImageRead(ImageBase):
+    pass

backend/app/models/resources.py

@@ -0,0 +1,24 @@
+from pydantic import BaseModel
+
+
+class ResourceStatsBase(BaseModel):
+    total: float
+    used: float
+    available: float
+    percent_used: float
+
+
+class DiskInfoBase(ResourceStatsBase):
+    pass
+
+
+class MemoryInfoBase(ResourceStatsBase):
+    pass
+
+
+class CPUInfoBase(BaseModel):
+    percent_used_total: float
+    percent_used_per_cpu: list[float]
+    percent_used_per_vm: list
+    percent_used_total_vms: float
+    cpu_count: int

backend/app/models/vm.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel
-from typing import Optional
+from uuid import UUID
 
 
 class VmBase(BaseModel):
@@ -10,9 +10,13 @@ class VmBase(BaseModel):
 
 
 class VmRead(VmBase):
-    id: str
+    id: UUID
     state: str
 
 
 class VmCreate(VmBase):
     pass
+
+
+class PasswordCreated(BaseModel):
+    password: str

backend/app/routes/host.py

@@ -0,0 +1,12 @@
+from fastapi import APIRouter, status
+from app.services.host_service import HostService
+from app.models.host import HostInfoBase
+
+
+router = APIRouter()
+
+
+@router.get("/info", status_code=status.HTTP_200_OK,
+            response_model=HostInfoBase)
+def get_host_info():
+    return HostService.get_host_info()

backend/app/routes/image.py

@@ -0,0 +1,15 @@
+from fastapi import APIRouter, status
+from app.services.image_service import ImageService
+from app.models.image import ImageRead
+
+
+router = APIRouter()
+
+
+@router.get("/", status_code=status.HTTP_200_OK,
+            response_model=list[ImageRead])
+def get_distribox_image_list():
+    try:
+        return ImageService.get_distribox_image_list()
+    except Exception:
+        raise