100 lines
2.5 KiB
Python
100 lines
2.5 KiB
Python
"""
|
|
/admin endpoints.
|
|
"""
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlmodel import select
|
|
|
|
from ..config import Config
|
|
from ..db import Connection, TagValue, User, UserCreate
|
|
from ._common import Responses, get_current_config, get_current_user
|
|
|
|
router = APIRouter(prefix="/admin", tags=["admin"])
|
|
|
|
|
|
@router.put(
|
|
"/install/config",
|
|
responses={
|
|
status.HTTP_200_OK: Responses.OK_NONE,
|
|
status.HTTP_409_CONFLICT: Responses.ENTRY_EXISTS,
|
|
},
|
|
)
|
|
async def initial_configure(
|
|
config: Config,
|
|
current_config: Config | None = Depends(Config.load),
|
|
):
|
|
"""
|
|
PUT ./install/config: Configure `kiwi-vpn`.
|
|
|
|
Status:
|
|
|
|
- 409: `kiwi-vpn` already installed
|
|
"""
|
|
|
|
# fail if already configured
|
|
if current_config is not None:
|
|
raise HTTPException(status_code=status.HTTP_409_CONFLICT)
|
|
|
|
# create config file, connect to database
|
|
config.save()
|
|
Connection.connect(config.db.uri)
|
|
|
|
|
|
@router.put(
|
|
"/install/admin",
|
|
responses={
|
|
status.HTTP_201_CREATED: Responses.OK_NONE,
|
|
status.HTTP_400_BAD_REQUEST: Responses.NOT_INSTALLED,
|
|
status.HTTP_409_CONFLICT: Responses.ENTRY_EXISTS,
|
|
},
|
|
status_code=status.HTTP_201_CREATED,
|
|
)
|
|
async def create_initial_admin(
|
|
admin_user: UserCreate,
|
|
_: Config = Depends(get_current_config),
|
|
):
|
|
"""
|
|
PUT ./install/admin: Create the first administrative user.
|
|
|
|
Status:
|
|
|
|
- 409: not the first user
|
|
"""
|
|
|
|
# fail if any user exists
|
|
with Connection.session as db:
|
|
if db.exec(select(User).limit(1)).first() is not None:
|
|
raise HTTPException(status_code=status.HTTP_409_CONFLICT)
|
|
|
|
# create an administrative user
|
|
if (new_user := User.create(user=admin_user)) is None:
|
|
raise HTTPException(status_code=status.HTTP_409_CONFLICT)
|
|
|
|
new_user.add_tags([TagValue.admin])
|
|
new_user.update()
|
|
|
|
|
|
@router.put(
|
|
"/config",
|
|
responses={
|
|
status.HTTP_200_OK: Responses.OK_NONE,
|
|
status.HTTP_400_BAD_REQUEST: Responses.NOT_INSTALLED,
|
|
status.HTTP_401_UNAUTHORIZED: Responses.NEEDS_USER,
|
|
status.HTTP_403_FORBIDDEN: Responses.NEEDS_PERMISSION,
|
|
},
|
|
)
|
|
async def set_config(
|
|
config: Config,
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
"""
|
|
PUT ./config: Edit `kiwi-vpn` main config.
|
|
"""
|
|
|
|
# check permissions
|
|
if not current_user.is_admin:
|
|
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN)
|
|
|
|
# update config file, reconnect to database
|
|
config.save()
|
|
Connection.connect(config.db.uri)
|