feat(#49): api and websocket servers optional

barthofu · Sep 4, 2022 · 0138d52 · 0138d52
1 parent 70900da
commit 0138d52
Show file tree

Hide file tree

Showing 6 changed files with 41 additions and 17 deletions.
diff --git a/src/config/api.ts b/src/config/api.ts
@@ -1,4 +1,5 @@
 export const apiConfig: APIConfigType = {
 
+    enabled: false,
     port: process.env['API_PORT'] ? parseInt(process.env['API_PORT']) : 4000,
 }
diff --git a/src/config/index.ts b/src/config/index.ts
@@ -2,4 +2,5 @@ export * from './general'
 export * from './database'
 export * from './logs'
 export * from './stats'
-export * from './api'
+export * from './api'
+export * from './websocket'
diff --git a/src/config/websocket.ts b/src/config/websocket.ts
@@ -0,0 +1,4 @@
+export const websocketConfig: WebsocketConfigType = {
+
+    enabled: false,
+}
diff --git a/src/main.ts b/src/main.ts
@@ -10,7 +10,7 @@ import { initDataTable, waitForDependency } from '@utils/functions'
 import { Server } from '@api/server'
 
 import { clientConfig } from './client'
-import { generalConfig } from '@config'
+import { apiConfig, generalConfig, websocketConfig } from '@config'
 import { NoBotTokenError } from '@errors'
 
 async function run() {
@@ -43,18 +43,22 @@ async function run() {
     await client.login(process.env.BOT_TOKEN)
 
     // start the api server
-    const server = await waitForDependency(Server)
-    await server.start()
+    if (apiConfig.enabled) {
+        const server = await waitForDependency(Server)
+        await server.start()
+    }
 
     // connect to the dashboard websocket
-    const webSocket = await waitForDependency(WebSocket)
-    await webSocket.init(client.user?.id || null)
+    if (websocketConfig.enabled) {
+        const webSocket = await waitForDependency(WebSocket)
+        await webSocket.init(client.user?.id || null)
+    }
 
     // upload images to imgur if configured
     if (process.env.IMGUR_CLIENT_ID && generalConfig.automaticUploadImagesToImgur) {
         const imagesUpload = await waitForDependency(ImagesUpload)
         await imagesUpload.syncWithDatabase()
-    }    
+    }
 }
 
 run()
diff --git a/src/services/Logger.ts b/src/services/Logger.ts
@@ -327,22 +327,30 @@ export class Logger {
         this.console('info', chalk.green(`${symbol} ${numberAlign(scheduledJobs)} ${chalk.bold('scheduled jobs')} loaded`), true)
 
         // connected
-        this.console('info', chalk.gray(boxen(
-            ` API Server listening on port ${chalk.bold(apiConfig.port)} `,
-            {
-                padding: 0,
-                margin: 1,
-                borderStyle: 'round',
-                dimBorder: true
-            }
-        )), true)
+        if (apiConfig.enabled) {
+
+            this.console('info', chalk.gray(boxen(
+                ` API Server listening on port ${chalk.bold(apiConfig.port)} `,
+                {
+                    padding: 0,
+                    margin: {
+                        top: 1,
+                        bottom: 0,
+                        left: 1,
+                        right: 1
+                    },
+                    borderStyle: 'round',
+                    dimBorder: true
+                }
+            )), true)
+        }
 
         this.console('info', chalk.hex('7289DA')(boxen(
             ` ${this.client.user ? `${chalk.bold(this.client.user.tag)}` : 'Bot'} is ${chalk.green('connected')}! `,
             {
                 padding: 0,
                 margin: {
-                    top: 0,
+                    top: 1,
                     bottom: 1,
                     left: 1 * 3,
                     right: 1 * 3

diff --git a/src/utils/types/configs.d.ts b/src/utils/types/configs.d.ts
@@ -87,5 +87,11 @@ type StatsConfigType = {
 
 type APIConfigType = {
 
+    enabled: boolean
     port: number
+}
+
+type WebsocketConfigType = {
+
+    enabled: boolean
 }