feat: add endpoint for downloading logs

Signed-off-by: Lukas Schaefer <lukas@lschaefer.xyz>

Author: lukasdotcom

appinfo/info.xml

@@ -37,6 +37,14 @@ Setup background job workers as described here: https://docs.nextcloud.com/serve
 			<image>nextcloud/context_chat_backend</image>
 			<image-tag>4.3.0</image-tag>
 		</docker-install>
+		<routes>
+			<route>
+				<url>download-logs</url>
+				<verb>GET</verb>
+				<access_level>ADMIN</access_level>
+				<headers_to_exclude>[]</headers_to_exclude>
+			</route>
+		</routes>
 		<environment-variables>
 			<variable>
 				<name>EXTERNAL_DB</name>

context_chat_backend/controller.py

@@ -2,6 +2,9 @@
 # SPDX-FileCopyrightText: 2023 Nextcloud GmbH and Nextcloud contributors
 # SPDX-License-Identifier: AGPL-3.0-or-later
 #
+import zipfile
+
+from starlette.responses import FileResponse
 
 # isort: off
 from .chain.types import ContextException, LLMOutput, ScopeType, SearchResult
@@ -12,6 +15,7 @@
 import logging
 import multiprocessing as mp
 import os
+import tempfile
 import threading
 from collections.abc import Callable
 from contextlib import asynccontextmanager
@@ -492,3 +496,16 @@ def _(query: Query) -> list[SearchResult]:
 		query.scopeType,
 		query.scopeList,
 	))
+
+
+@app.get('/download-logs')
+@enabled_guard(app)
+def download_logs() -> FileResponse:
+	with tempfile.NamedTemporaryFile('wb', delete=False) as tmp:
+		with zipfile.ZipFile(tmp, mode='w', compression=zipfile.ZIP_DEFLATED) as zip_file:
+			files = os.listdir(os.path.join(persistent_storage(), 'logs'))
+			for file in files:
+				file_path = os.path.join(persistent_storage(), 'logs', file)
+				if os.path.isfile(file_path): # Might be a folder (just skip it then)
+					zip_file.write(file_path)
+		return FileResponse(tmp.name, media_type='application/zip', filename='docker_logs.zip')