Improve logging and shutdown

Sources/App/configure.swift

@@ -51,17 +51,15 @@ public func configure(_ app: Application) async throws {
     log("[\(df.string(from: Date()))] Server started")
 }
 
-public func shutdown() {
+public func shutdown() async {
-    print("[\(df.string(from: Date()))] Server shutdown")
+    // Gracefully shut down by closing potentially open socket
-    asyncScheduler.schedule {
+    await deviceManager.removeDeviceConnection()
-        // Gracefully shut down by closing potentially open socket
+    do {
-        await deviceManager.removeDeviceConnection()
+        try await asyncScheduler.shutdownGracefully()
-        do {
+    } catch {
-            try await asyncScheduler.shutdownGracefully()
+        print("Failed to shut down MultiThreadedEventLoopGroup: \(error)")
-        } catch {
-            print("Failed to shut down MultiThreadedEventLoopGroup: \(error)")
-        }
     }
+    print("[\(df.string(from: Date()))] Server shutdown")
 }
 
 private func loadKeys(at url: URL) throws -> (deviceKey: Data, remoteKey: Data) {

Sources/App/entrypoint.swift

@@ -23,21 +23,29 @@ private extension Vapor.Application {
 @main
 enum Entrypoint {
     static func main() async throws {
-        var env = try Environment.detect()
+        var env = Environment.production
         try LoggingSystem.bootstrap(from: &env)
         
         let app = Application(env)
-        defer {
+        func cleanup() async {
-            shutdown()
+            await shutdown()
             app.shutdown()
         }
         
         do {
             try await configure(app)
+            await cleanup()
         } catch {
             app.logger.report(error: error)
+            await cleanup()
+            throw error
+        }
+        do {
+            try await app.runFromAsyncMainEntrypoint()
+            await cleanup()
+        } catch {
+            await cleanup()
             throw error
         }
-        try await app.runFromAsyncMainEntrypoint()
     }
 }