Added restart and shutdown buttons

frontend/api.py

@@ -4,10 +4,12 @@ import logging
 from dataclasses import dataclass
 from io import BytesIO
 from os import urandom
+from threading import Timer
 from time import time as epoch_time
-from typing import Any, Callable, Dict, Tuple
+from typing import Any, Callable, Dict, Tuple, Union
 
 from flask import g, request, send_file
+from waitress.server import BaseWSGIServer
 
 from backend.custom_exceptions import (AccessUnauthorized, APIKeyExpired,
                                        APIKeyInvalid, InvalidKeyValue,
@@ -45,6 +47,19 @@ from frontend.input_validation import (AllowNewAccountsVariable, ColorVariable,
 # General variables and functions
 #===================
 
+class APIVariables:
+	server_instance: Union[BaseWSGIServer, None] = None
+	restart: bool = False
+
+def shutdown_server() -> None:
+	APIVariables.server_instance.close()
+	APIVariables.server_instance.task_dispatcher.shutdown()
+	APIVariables.server_instance._map.clear()
+	return
+
+shutdown_server_thread = Timer(1.0, shutdown_server)
+shutdown_server_thread.name = "InternalStateHandler"
+
 @dataclass
 class ApiKeyEntry:
 	exp: int
@@ -564,6 +579,27 @@ def api_get_static_reminder(inputs: Dict[str, Any], s_id: int):
 # Admin panel endpoints
 #===================
 
+@admin_api.route(
+	'/shutdown',
+	'Shut down the application',
+	methods=['POST']
+)
+@endpoint_wrapper
+def api_shutdown():
+	shutdown_server_thread.start()
+	return return_api({})
+
+@admin_api.route(
+	'/restart',
+	'Restart the application',
+	methods=['POST']
+)
+@endpoint_wrapper
+def api_restart():
+	APIVariables.restart = True
+	shutdown_server_thread.start()
+	return return_api({})
+
 @api.route(
 	'/settings',
 	'Get the admin settings',

frontend/static/css/admin.css

@@ -145,16 +145,20 @@ h2 {
 
 .add-user-container,
 .database-container {
+	margin-top: 2rem;
 	margin-bottom: 1rem;
 	width: 100%;
 	display: flex;
 	justify-content: center;
 	align-items: center;
+	gap: 2rem;
+	flex-wrap: wrap;
 }
 
 #add-user-button,
-#download-db-button {
-	margin-top: 2rem;
+#download-db-button,
+#restart-button,
+#shutdown-button {
 
 	width: min(15rem, 100%);
 	height: 2rem;
@@ -166,7 +170,9 @@ h2 {
 	box-shadow: var(--default-shadow);
 }
 
-#download-db-button {
+#download-db-button,
+#restart-button,
+#shutdown-button {
 	height: unset;
 }
 

frontend/static/js/admin.js

@@ -9,6 +9,11 @@ const user_inputs = {
 	password: document.querySelector('#new-password-input')
 };
 
+const power_buttons = {
+	restart: document.querySelector('#restart-button'),
+	shutdown: document.querySelector('#shutdown-button')
+};
+
 function checkLogin() {
 	fetch(`${url_prefix}/api/auth/status?api_key=${api_key}`)
 	.then(response => {
@@ -175,6 +180,32 @@ function loadUsers() {
 	});
 };
 
+function restart_app() {
+	power_buttons.restart.innerText = 'Restarting...';
+	fetch(`${url_prefix}/api/admin/restart?api_key=${api_key}`, {
+		method: "POST"
+	})
+	.then(response =>
+		setTimeout(
+			() => window.location.reload(),
+			1000
+		)
+	);
+};
+
+function shutdown_app() {
+	power_buttons.shutdown.innerText = 'Shutting down...';
+	fetch(`${url_prefix}/api/admin/shutdown?api_key=${api_key}`, {
+		method: "POST"
+	})
+	.then(response =>
+		setTimeout(
+			() => window.location.reload(),
+			1000
+		)
+	);
+};
+
 // code run on load
 
 checkLogin();
@@ -187,3 +218,5 @@ document.querySelector('#add-user-button').onclick = e => toggleAddUser();
 document.querySelector('#add-user-form').action = 'javascript:addUser()';
 document.querySelector('#download-db-button').onclick = e => 
 	window.location.href = `${url_prefix}/api/admin/database?api_key=${api_key}`;
+power_buttons.restart.onclick = e => restart_app();
+power_buttons.shutdown.onclick = e => shutdown_app();

frontend/templates/admin.html

@@ -123,6 +123,11 @@
 			<div class="database-container">
 				<button id="download-db-button">Download Database</button>
 			</div>
+			<h2>Power</h2>
+			<div class="database-container">
+				<button id="restart-button">Restart</button>
+				<button id="shutdown-button">Shutdown</button>
+			</div>
 		</div>
 	</main>
 </body>