43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
from datetime import datetime, timezone
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
from fastapi import APIRouter, HTTPException
|
||
|
|
|
||
|
|
from app.config import settings
|
||
|
|
from app.models import ServiceActionResponse, ServiceListResponse, ServiceState
|
||
|
|
from app.services.systemd_manager import SystemdManager
|
||
|
|
|
||
|
|
router = APIRouter(prefix='/services', tags=['services'])
|
||
|
|
manager = SystemdManager()
|
||
|
|
|
||
|
|
|
||
|
|
@router.get('', response_model=ServiceListResponse)
|
||
|
|
def list_services() -> ServiceListResponse:
|
||
|
|
items = manager.list_services(settings.monitored_services)
|
||
|
|
return ServiceListResponse(count=len(items), time=datetime.now(timezone.utc), items=items)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get('/{name}', response_model=ServiceState)
|
||
|
|
def get_service(name: str) -> ServiceState:
|
||
|
|
items = manager.list_services([name])
|
||
|
|
if not items:
|
||
|
|
raise HTTPException(status_code=404, detail='Service not found')
|
||
|
|
return items[0]
|
||
|
|
|
||
|
|
|
||
|
|
@router.post('/{name}/{action}', response_model=ServiceActionResponse)
|
||
|
|
def service_action(name: str, action: Literal['start', 'stop', 'restart']) -> ServiceActionResponse:
|
||
|
|
if action == 'start':
|
||
|
|
manager.start(name)
|
||
|
|
elif action == 'stop':
|
||
|
|
manager.stop(name)
|
||
|
|
elif action == 'restart':
|
||
|
|
manager.restart(name)
|
||
|
|
return ServiceActionResponse(
|
||
|
|
status='accepted',
|
||
|
|
action=action,
|
||
|
|
service=name,
|
||
|
|
time=datetime.now(timezone.utc),
|
||
|
|
message=f'{action} requested',
|
||
|
|
)
|