169 lines
4.9 KiB
Python
169 lines
4.9 KiB
Python
import os
|
|
import time
|
|
from http import HTTPStatus
|
|
from shutil import make_archive
|
|
from subprocess import Popen
|
|
from urllib.parse import urlparse
|
|
|
|
from fastapi import APIRouter, Depends, File
|
|
from fastapi.responses import FileResponse
|
|
|
|
from lnbits.core.models import User
|
|
from lnbits.core.models.notifications import NotificationType
|
|
from lnbits.core.services import (
|
|
enqueue_admin_notification,
|
|
get_balance_delta,
|
|
update_cached_settings,
|
|
)
|
|
from lnbits.core.services.notifications import send_email_notification
|
|
from lnbits.core.services.settings import dict_to_settings
|
|
from lnbits.decorators import check_admin, check_super_user
|
|
from lnbits.server import server_restart
|
|
from lnbits.settings import AdminSettings, Settings, UpdateSettings, settings
|
|
from lnbits.tasks import invoice_listeners
|
|
|
|
from .. import core_app_extra
|
|
from ..crud import get_admin_settings, reset_core_settings, update_admin_settings
|
|
|
|
admin_router = APIRouter(tags=["Admin UI"], prefix="/admin")
|
|
file_upload = File(...)
|
|
|
|
|
|
@admin_router.get(
|
|
"/api/v1/audit",
|
|
name="Audit",
|
|
description="show the current balance of the node and the LNbits database",
|
|
dependencies=[Depends(check_admin)],
|
|
)
|
|
async def api_auditor():
|
|
return await get_balance_delta()
|
|
|
|
|
|
@admin_router.get(
|
|
"/api/v1/monitor",
|
|
name="Monitor",
|
|
description="show the current listeners and other monitoring data",
|
|
dependencies=[Depends(check_admin)],
|
|
)
|
|
async def api_monitor():
|
|
return {
|
|
"invoice_listeners": list(invoice_listeners.keys()),
|
|
}
|
|
|
|
|
|
@admin_router.get(
|
|
"/api/v1/testemail",
|
|
name="TestEmail",
|
|
description="send a test email to the admin",
|
|
dependencies=[Depends(check_admin)],
|
|
)
|
|
async def api_test_email():
|
|
return await send_email_notification(
|
|
settings.lnbits_email_notifications_to_emails,
|
|
"This is a LNbits test email.",
|
|
"LNbits Test Email",
|
|
)
|
|
|
|
|
|
@admin_router.get("/api/v1/settings")
|
|
async def api_get_settings(
|
|
user: User = Depends(check_admin),
|
|
) -> AdminSettings | None:
|
|
admin_settings = await get_admin_settings(user.super_user)
|
|
return admin_settings
|
|
|
|
|
|
@admin_router.put(
|
|
"/api/v1/settings",
|
|
status_code=HTTPStatus.OK,
|
|
)
|
|
async def api_update_settings(data: UpdateSettings, user: User = Depends(check_admin)):
|
|
enqueue_admin_notification(
|
|
NotificationType.settings_update, {"username": user.username}
|
|
)
|
|
await update_admin_settings(data)
|
|
admin_settings = await get_admin_settings(user.super_user)
|
|
if not admin_settings:
|
|
raise ValueError("Updated admin settings not found.")
|
|
update_cached_settings(admin_settings.dict())
|
|
core_app_extra.register_new_ratelimiter()
|
|
return {"status": "Success"}
|
|
|
|
|
|
@admin_router.patch(
|
|
"/api/v1/settings",
|
|
status_code=HTTPStatus.OK,
|
|
)
|
|
async def api_update_settings_partial(data: dict, user: User = Depends(check_admin)):
|
|
updatable_settings = dict_to_settings({**settings.dict(), **data})
|
|
return await api_update_settings(updatable_settings, user)
|
|
|
|
|
|
@admin_router.get(
|
|
"/api/v1/settings/default",
|
|
status_code=HTTPStatus.OK,
|
|
dependencies=[Depends(check_admin)],
|
|
)
|
|
async def api_reset_settings(field_name: str):
|
|
default_settings = Settings()
|
|
return {"default_value": getattr(default_settings, field_name)}
|
|
|
|
|
|
@admin_router.delete("/api/v1/settings", status_code=HTTPStatus.OK)
|
|
async def api_delete_settings(user: User = Depends(check_super_user)) -> None:
|
|
enqueue_admin_notification(
|
|
NotificationType.settings_update, {"username": user.username}
|
|
)
|
|
await reset_core_settings()
|
|
server_restart.set()
|
|
|
|
|
|
@admin_router.get(
|
|
"/api/v1/restart",
|
|
status_code=HTTPStatus.OK,
|
|
dependencies=[Depends(check_super_user)],
|
|
)
|
|
async def api_restart_server() -> dict[str, str]:
|
|
server_restart.set()
|
|
return {"status": "Success"}
|
|
|
|
|
|
@admin_router.get(
|
|
"/api/v1/backup",
|
|
status_code=HTTPStatus.OK,
|
|
dependencies=[Depends(check_super_user)],
|
|
response_class=FileResponse,
|
|
)
|
|
async def api_download_backup() -> FileResponse:
|
|
last_filename = "lnbits-backup"
|
|
filename = f"lnbits-backup-{int(time.time())}.zip"
|
|
db_url = settings.lnbits_database_url
|
|
pg_backup_filename = f"{settings.lnbits_data_folder}/lnbits-database.dmp"
|
|
is_pg = db_url and db_url.startswith("postgres://")
|
|
|
|
if is_pg and db_url:
|
|
p = urlparse(db_url)
|
|
command = (
|
|
f"pg_dump --host={p.hostname} "
|
|
f"--dbname={p.path.replace('/', '')} "
|
|
f"--username={p.username} "
|
|
"--no-password "
|
|
"--format=c "
|
|
f"--file={pg_backup_filename}"
|
|
)
|
|
proc = Popen(
|
|
command, shell=True, env={**os.environ, "PGPASSWORD": p.password or ""}
|
|
)
|
|
proc.wait()
|
|
|
|
make_archive(last_filename, "zip", settings.lnbits_data_folder)
|
|
|
|
# cleanup pg_dump file
|
|
if is_pg:
|
|
proc = Popen(f"rm {pg_backup_filename}", shell=True)
|
|
proc.wait()
|
|
|
|
return FileResponse(
|
|
path=f"{last_filename}.zip", filename=filename, media_type="application/zip"
|
|
)
|