From c7752ab79b8d01f90e6d471e6da7674e3bda2ffe Mon Sep 17 00:00:00 2001
From: Arash <arashi.sheyda@gmail.com>
Date: Fri, 8 Mar 2024 20:15:02 -0700
Subject: [PATCH] fix: improve logs

---
 src/runtime/server/plugins/neo4j.ts | 11 +----------
 src/runtime/server/utils/driver.ts  | 20 ++++++++++++++++++--
 2 files changed, 19 insertions(+), 12 deletions(-)

diff --git a/src/runtime/server/plugins/neo4j.ts b/src/runtime/server/plugins/neo4j.ts
index 44ec702..d90edec 100644
--- a/src/runtime/server/plugins/neo4j.ts
+++ b/src/runtime/server/plugins/neo4j.ts
@@ -3,8 +3,6 @@
  */
 import type { NitroApp } from 'nitropack'
 
-import { consola } from 'consola'
-import { colors } from 'consola/utils'
 import { useDriver } from '../utils/driver'
 
 type NitroAppPlugin = (nitro: NitroApp) => void
@@ -14,12 +12,5 @@ function defineNitroPlugin(def: NitroAppPlugin): NitroAppPlugin {
 }
 
 export default defineNitroPlugin(async () => {
-  const driver = useDriver()
-  if (process.dev) {
-    const info = await driver.getServerInfo()
-    consola.box({
-      title: colors.bold(colors.blue(' Neo4j Server Info ')),
-      message: JSON.stringify(info, null, 2),
-    })
-  }
+  useDriver()
 })
diff --git a/src/runtime/server/utils/driver.ts b/src/runtime/server/utils/driver.ts
index a9b13ed..cfaedd1 100644
--- a/src/runtime/server/utils/driver.ts
+++ b/src/runtime/server/utils/driver.ts
@@ -13,10 +13,26 @@ export function useDriver() {
   if (!_driver) {
     try {
       _driver = neo4j.driver(config.uri, neo4j.auth.basic(config.auth.username, config.auth.password))
-      consola.success(`Neo4j driver has been ${colors.green('connected.')}`)
+      if (process.dev) {
+        (async () => {
+          try {
+            const info = await _driver.getServerInfo()
+            consola.box({
+              title: colors.bold(colors.green(' Neo4j connection successful. ')),
+              message: JSON.stringify(info, null, 2),
+            })
+          }
+          catch (error) {
+            consola.error('Failed to get server info:', error)
+          }
+        })()
+      }
+      else {
+        consola.success(`Neo4j connection ${colors.green('successful.')}`)
+      }
     }
     catch (error) {
-      consola.error('Failed to create driver:', error)
+      consola.error('Failed to create Neo4j driver:', error)
     }
   }
   return _driver