@anthonylzq/simba.js 5.2.0 → 6.2.0

package/lib/src/functions/api/database.js

@@ -5,9 +5,10 @@ const writeFile = require('../../utils/writeFile')
 
 /**
  * @param {Object} args
- * @param {String} projectName
+ * @param {String} args.projectName
+ * @param {Boolean} args.fastify
  */
-const mongoF = async ({ projectName }) => {
+const mongoF = async ({ projectName, fastify }) => {
   const createFoldersCommand = `mkdir ${projectName}/src/database/mongo \
 ${projectName}/src/database/mongo/models \
 ${projectName}/src/database/mongo/queries`
@@ -22,9 +23,126 @@ ${projectName}/src/database/mongo/queries`
       file: `${projectName}/src/database/index.ts`
     },
     mongo: {
+      connection: {
+        content: fastify
+          ? `import { connect, connection } from 'mongoose'
+import { FastifyLoggerInstance } from 'fastify'
+
+const ENVIRONMENTS_WITHOUT_RECONNECTION = ['ci', 'local']
+const dbConnection = async (
+  logger: FastifyLoggerInstance
+): Promise<{
+  connect: () => Promise<typeof import('mongoose')>
+  disconnect: () => Promise<void>
+}> => {
+  const connectionConfig = {
+    keepAlive: true,
+    useNewUrlParser: true,
+    useUnifiedTopology: true
+  }
+
+  connection.on('connected', () => {
+    logger.info('Mongo connection established.')
+  })
+  connection.on('reconnected', () => {
+    logger.info('Mongo connection reestablished')
+  })
+  connection.on('disconnected', () => {
+    if (
+      !ENVIRONMENTS_WITHOUT_RECONNECTION.includes(
+        process.env.NODE_ENV as string
+      )
+    ) {
+      logger.info(
+        'Mongo connection disconnected. Trying to reconnected to Mongo...'
+      )
+      setTimeout(() => {
+        connect(process.env.MONGO_URI as string, {
+          ...connection,
+          connectTimeoutMS: 3000,
+          socketTimeoutMS: 3000
+        })
+      }, 3000)
+    }
+  })
+  connection.on('close', () => {
+    logger.info('Mongo connection closed')
+  })
+  connection.on('error', (e: Error) => {
+    logger.info('Mongo connection error:')
+    logger.error(e)
+  })
+
+  return {
+    connect: () => connect(process.env.MONGO_URI as string, connectionConfig),
+    disconnect: () => connection.close()
+  }
+}
+
+export { dbConnection }
+`
+          : `import { connect, connection } from 'mongoose'
+import { HttpLogger } from 'express-pino-logger'
+
+const ENVIRONMENTS_WITHOUT_RECONNECTION = ['ci', 'local']
+const dbConnection = async (
+  logger: HttpLogger['logger']
+): Promise<{
+  connect: () => Promise<typeof import('mongoose')>
+  disconnect: () => Promise<void>
+}> => {
+  const connectionConfig = {
+    keepAlive: true,
+    useNewUrlParser: true,
+    useUnifiedTopology: true
+  }
+
+  connection.on('connected', () => {
+    logger.info('Mongo connection established.')
+  })
+  connection.on('reconnected', () => {
+    logger.info('Mongo connection reestablished')
+  })
+  connection.on('disconnected', () => {
+    if (
+      !ENVIRONMENTS_WITHOUT_RECONNECTION.includes(
+        process.env.NODE_ENV as string
+      )
+    ) {
+      logger.info(
+        'Mongo connection disconnected. Trying to reconnected to Mongo...'
+      )
+      setTimeout(() => {
+        connect(process.env.MONGO_URI as string, {
+          ...connection,
+          connectTimeoutMS: 3000,
+          socketTimeoutMS: 3000
+        })
+      }, 3000)
+    }
+  })
+  connection.on('close', () => {
+    logger.info('Mongo connection closed')
+  })
+  connection.on('error', (e: Error) => {
+    logger.info('Mongo connection error:')
+    logger.error(e)
+  })
+
+  return {
+    connect: () => connect(process.env.MONGO_URI as string, connectionConfig),
+    disconnect: () => connection.close()
+  }
+}
+
+export { dbConnection }
+`,
+        file: `${projectName}/src/database/mongo/connection.ts`
+      },
       index: {
         content: `export * from './models'
 export * from './queries'
+export * from './connection'
 `,
         file: `${projectName}/src/database/mongo/index.ts`
       },
@@ -141,21 +259,25 @@ export { store, remove, get, update }
   }
 
   await Promise.all([
-    await writeFile(database.index.file, database.index.content),
-    await writeFile(database.mongo.index.file, database.mongo.index.content),
-    await writeFile(
+    writeFile(database.index.file, database.index.content),
+    writeFile(
+      database.mongo.connection.file,
+      database.mongo.connection.content
+    ),
+    writeFile(database.mongo.index.file, database.mongo.index.content),
+    writeFile(
       database.mongo.models.index.file,
       database.mongo.models.index.content
     ),
-    await writeFile(
+    writeFile(
       database.mongo.models.user.file,
       database.mongo.models.user.content
     ),
-    await writeFile(
+    writeFile(
       database.mongo.queries.index.file,
       database.mongo.queries.index.content
     ),
-    await writeFile(
+    writeFile(
       database.mongo.queries.user.file,
       database.mongo.queries.user.content
     )
@@ -166,13 +288,14 @@ export { store, remove, get, update }
  * @param {Object} args
  * @param {Boolean|undefined} args.mongo
  * @param {String} args.projectName
+ * @param {Boolean} args.fastify
  */
-module.exports = async ({ mongo = true, projectName }) => {
+module.exports = async ({ mongo = true, projectName, fastify }) => {
   const createFoldersCommand = `mkdir ${projectName}/src/database`
 
   if (platform() === 'win32')
     await exec(createFoldersCommand.replaceAll('/', '\\'))
   else await exec(createFoldersCommand)
 
-  if (mongo) await mongoF({ projectName })
+  if (mongo) await mongoF({ projectName, fastify })
 }

package/lib/src/functions/api/index.js

@@ -9,6 +9,7 @@ const types = require('./types')
 const network = require('./network')
 const utils = require('./utils')
 const writeFile = require('../../utils/writeFile')
+const { ENVIRONMENTS_WITH_MONGO_URI } = require('../../utils/constants')
 
 /*
  * Express api:
@@ -163,7 +164,7 @@ DELETE http://localhost:1996/api/user/60e7e3b93b01c1a7aa74cd6b
     },
     '.env': {
       content: `MONGO_URI = ${
-        process.env.LOCAL
+        ENVIRONMENTS_WITH_MONGO_URI.includes(process.env.NODE_ENV)
           ? process.env.MONGO_URI
           : `mongodb://mongo:mongo@mongo:27017/${projectName}`
       }`,
@@ -194,7 +195,8 @@ Server.start()
     // /database
     database({
       mongo,
-      projectName
+      projectName,
+      fastify
     }),
     // /network
     network({

package/lib/src/functions/api/network.js

@@ -75,33 +75,35 @@ export { applyRoutes }
     server: {
       content: `import { Server as HttpServer } from 'http'
 import express from 'express'
-import mongoose from 'mongoose'
 import cors from 'cors'
 import pino, { HttpLogger } from 'express-pino-logger'
 
+import { dbConnection } from 'database'
 import { applyRoutes } from './router'
 
 const PORT = (process.env.PORT as string) || 1996
+const ENVIRONMENTS_WITHOUT_PRETTY_PRINT = ['production', 'ci']
 
 class Server {
   #app: express.Application
-  #connection: mongoose.Connection | undefined
   #log: HttpLogger
   #server: HttpServer | undefined
+  #connection: Awaited<ReturnType<typeof dbConnection>> | undefined
 
   constructor() {
     this.#app = express()
     this.#log = pino({
-      transport:
-        process.env.ENVIRONMENT !== 'production'
-          ? {
-              target: 'pino-pretty',
-              options: {
-                translateTime: 'HH:MM:ss Z',
-                ignore: 'pid,hostname'
-              }
+      transport: !ENVIRONMENTS_WITHOUT_PRETTY_PRINT.includes(
+        process.env.NODE_ENV as string
+      )
+        ? {
+            target: 'pino-pretty',
+            options: {
+              translateTime: 'HH:MM:ss Z',
+              ignore: 'pid,hostname'
             }
-          : undefined
+          }
+        : undefined
     })
     this.#config()
   }
@@ -130,57 +132,26 @@ class Server {
     applyRoutes(this.#app)
   }
 
-  async #mongo(): Promise<void> {
-    this.#connection = mongoose.connection
-    const connection = {
-      keepAlive: true,
-      useNewUrlParser: true,
-      useUnifiedTopology: true
-    }
-    this.#connection.on('connected', () => {
-      this.#log.logger.info('Mongo connection established.')
-    })
-    this.#connection.on('reconnected', () => {
-      this.#log.logger.info('Mongo connection reestablished')
-    })
-    this.#connection.on('disconnected', () => {
-      this.#log.logger.info('Mongo connection disconnected')
-      this.#log.logger.info('Trying to reconnected to Mongo...')
-      setTimeout(() => {
-        mongoose.connect(process.env.MONGO_URI as string, {
-          ...connection,
-          connectTimeoutMS: 3000,
-          socketTimeoutMS: 3000
-        })
-      }, 3000)
-    })
-    this.#connection.on('close', () => {
-      this.#log.logger.info('Mongo connection closed')
-    })
-    this.#connection.on('error', (e: Error) => {
-      this.#log.logger.info('Mongo connection error:')
-      this.#log.logger.error(e)
-    })
-    await mongoose.connect(process.env.MONGO_URI as string, connection)
+  async #dbConnection() {
+    this.#connection = await dbConnection(this.#log.logger)
   }
 
-  public start(): void {
-    this.#server = this.#app.listen(PORT, () => {
-      this.#log.logger.info(\`Server running at port \${PORT}\`)
-    })
-
+  public async start(): Promise<void> {
     try {
-      this.#mongo()
+      await this.#dbConnection()
+      await this.#connection?.connect()
+      this.#server = this.#app.listen(PORT, () => {
+        this.#log.logger.info(\`Server running at port \${PORT}\`)
+      })
     } catch (e) {
       this.#log.logger.error(e)
     }
   }
 
-  public stop(): void {
+  public async stop(): Promise<void> {
     try {
-      if (this.#server) this.#server.close()
-
-      process.exit(0)
+      await this.#connection?.disconnect()
+      this.#server?.close()
     } catch (e) {
       this.#log.logger.error(e)
     }
@@ -955,7 +926,6 @@ export { applyRoutes }
     server: {
       content: `import { createServer, Server as HttpServer } from 'http'
 import express from 'express'
-import mongoose from 'mongoose'
 import cors from 'cors'
 import pino, { HttpLogger } from 'express-pino-logger'
 import { ApolloServer } from 'apollo-server-express'
@@ -965,31 +935,34 @@ import {
   ApolloServerPluginLandingPageGraphQLPlayground
 } from 'apollo-server-core'
 
+import { dbConnection } from 'database'
 import { mergedSchema as schema } from 'graphQL'
 import { applyRoutes } from './router'
 
 const PORT = (process.env.PORT as string) || 1996
+const ENVIRONMENTS_WITHOUT_PRETTY_PRINT = ['production', 'ci']
 
 class Server {
   #app: express.Application
-  #connection: mongoose.Connection | undefined
-  #server: HttpServer
   #log: HttpLogger
+  #server: HttpServer
+  #connection: Awaited<ReturnType<typeof dbConnection>> | undefined
 
   constructor() {
     this.#app = express()
     this.#server = createServer(this.#app)
     this.#log = pino({
-      transport:
-        process.env.ENVIRONMENT !== 'production'
-          ? {
-              target: 'pino-pretty',
-              options: {
-                translateTime: 'HH:MM:ss Z',
-                ignore: 'pid,hostname'
-              }
+      transport: !ENVIRONMENTS_WITHOUT_PRETTY_PRINT.includes(
+        process.env.NODE_ENV as string
+      )
+        ? {
+            target: 'pino-pretty',
+            options: {
+              translateTime: 'HH:MM:ss Z',
+              ignore: 'pid,hostname'
             }
-          : undefined
+          }
+        : undefined
     })
     this.#config()
   }
@@ -1017,38 +990,8 @@ class Server {
     )
   }
 
-  async #mongo(): Promise<void> {
-    this.#connection = mongoose.connection
-    const connection = {
-      keepAlive: true,
-      useNewUrlParser: true,
-      useUnifiedTopology: true
-    }
-    this.#connection.on('connected', () => {
-      this.#log.logger.info('Mongo connection established.')
-    })
-    this.#connection.on('reconnected', () => {
-      this.#log.logger.info('Mongo connection reestablished')
-    })
-    this.#connection.on('disconnected', () => {
-      this.#log.logger.info('Mongo connection disconnected')
-      this.#log.logger.info('Trying to reconnected to Mongo...')
-      setTimeout(() => {
-        mongoose.connect(process.env.MONGO_URI as string, {
-          ...connection,
-          connectTimeoutMS: 3000,
-          socketTimeoutMS: 3000
-        })
-      }, 3000)
-    })
-    this.#connection.on('close', () => {
-      this.#log.logger.info('Mongo connection closed')
-    })
-    this.#connection.on('error', (e: Error) => {
-      this.#log.logger.info('Mongo connection error:')
-      this.#log.logger.error(e)
-    })
-    await mongoose.connect(process.env.MONGO_URI as string, connection)
+  async #dbConnection() {
+    this.#connection = await dbConnection(this.#log.logger)
   }
 
   public async start(): Promise<void> {
@@ -1072,7 +1015,8 @@ class Server {
         path: '/api'
       })
       applyRoutes(this.#app)
-      await this.#mongo()
+      await this.#dbConnection()
+      await this.#connection?.connect()
       this.#server.listen(PORT, () => {
         this.#log.logger.info(\`Server listening at port: \${PORT}\`)
         this.#log.logger.info(
@@ -1084,11 +1028,10 @@ class Server {
     }
   }
 
-  public stop(): void {
+  public async stop(): Promise<void> {
     try {
-      if (this.#server) this.#server.close()
-
-      process.exit(0)
+      await this.#connection?.disconnect()
+      this.#server?.close()
     } catch (e) {
       this.#log.logger.error(e)
     }
@@ -1294,20 +1237,25 @@ export { applyRoutes }
     },
     server: {
       content: `import fastify, { FastifyInstance } from 'fastify'
-import mongoose from 'mongoose'
 
+import { dbConnection } from 'database'
 import { applyRoutes } from './router'
 import { validatorCompiler } from './utils'
 
 const PORT = process.env.PORT ?? 1996
+const ENVIRONMENTS_WITHOUT_PRETTY_PRINT = ['production', 'ci']
 
 class Server {
   #app: FastifyInstance
-  #connection: mongoose.Connection | undefined
+  #connection: Awaited<ReturnType<typeof dbConnection>> | undefined
 
   constructor() {
     this.#app = fastify({
-      logger: { prettyPrint: process.env.NODE_ENV !== 'production' }
+      logger: {
+        prettyPrint: !ENVIRONMENTS_WITHOUT_PRETTY_PRINT.includes(
+          process.env.NODE_ENV as string
+        )
+      }
     })
     this.#config()
   }
@@ -1328,44 +1276,15 @@ class Server {
     applyRoutes(this.#app)
   }
 
-  async #mongo(): Promise<void> {
-    this.#connection = mongoose.connection
-    const connection = {
-      keepAlive: true,
-      useNewUrlParser: true,
-      useUnifiedTopology: true
-    }
-    this.#connection.on('connected', () => {
-      this.#app.log.info('Mongo connection established.')
-    })
-    this.#connection.on('reconnected', () => {
-      this.#app.log.info('Mongo connection reestablished')
-    })
-    this.#connection.on('disconnected', () => {
-      this.#app.log.info('Mongo connection disconnected')
-      this.#app.log.info('Trying to reconnected to Mongo...')
-      setTimeout(() => {
-        mongoose.connect(process.env.MONGO_URI as string, {
-          ...connection,
-          connectTimeoutMS: 3000,
-          socketTimeoutMS: 3000
-        })
-      }, 3000)
-    })
-    this.#connection.on('close', () => {
-      this.#app.log.info('Mongo connection closed')
-    })
-    this.#connection.on('error', (e: Error) => {
-      this.#app.log.info('Mongo connection error:')
-      this.#app.log.error(e)
-    })
-    await mongoose.connect(process.env.MONGO_URI as string, connection)
+  async #dbConnection() {
+    this.#connection = await dbConnection(this.#app.log)
   }
 
   public async start(): Promise<void> {
     try {
+      await this.#dbConnection()
+      await this.#connection?.connect()
       await this.#app.listen(PORT)
-      this.#mongo()
     } catch (e) {
       console.error(e)
     }
@@ -1373,8 +1292,8 @@ class Server {
 
   public async stop(): Promise<void> {
     try {
+      await this.#connection?.disconnect()
       await this.#app.close()
-      process.exit(0)
     } catch (e) {
       console.error(e)
     }
@@ -2266,20 +2185,25 @@ import {
   ApolloServerPluginLandingPageDisabled
 } from 'apollo-server-core'
 import { ApolloServerPlugin } from 'apollo-server-plugin-base'
-import mongoose from 'mongoose'
+import { dbConnection } from 'database'
 
 import { mergedSchema as schema } from 'graphQL'
 import { applyRoutes } from './router'
 
 const PORT = process.env.PORT ?? 1996
+const ENVIRONMENTS_WITHOUT_PRETTY_PRINT = ['production', 'ci']
 
 class Server {
   #app: FastifyInstance
-  #connection: mongoose.Connection | undefined
+  #connection: Awaited<ReturnType<typeof dbConnection>> | undefined
 
   constructor() {
     this.#app = fastify({
-      logger: { prettyPrint: process.env.NODE_ENV !== 'production' }
+      logger: {
+        prettyPrint: !ENVIRONMENTS_WITHOUT_PRETTY_PRINT.includes(
+          process.env.NODE_ENV as string
+        )
+      }
     })
     this.#config()
   }
@@ -2298,38 +2222,8 @@ class Server {
     applyRoutes(this.#app)
   }
 
-  async #mongo(): Promise<void> {
-    this.#connection = mongoose.connection
-    const connection = {
-      keepAlive: true,
-      useNewUrlParser: true,
-      useUnifiedTopology: true
-    }
-    this.#connection.on('connected', () => {
-      this.#app.log.info('Mongo connection established.')
-    })
-    this.#connection.on('reconnected', () => {
-      this.#app.log.info('Mongo connection reestablished')
-    })
-    this.#connection.on('disconnected', () => {
-      this.#app.log.info('Mongo connection disconnected')
-      this.#app.log.info('Trying to reconnected to Mongo...')
-      setTimeout(() => {
-        mongoose.connect(process.env.MONGO_URI as string, {
-          ...connection,
-          connectTimeoutMS: 3000,
-          socketTimeoutMS: 3000
-        })
-      }, 3000)
-    })
-    this.#connection.on('close', () => {
-      this.#app.log.info('Mongo connection closed')
-    })
-    this.#connection.on('error', (e: Error) => {
-      this.#app.log.info('Mongo connection error:')
-      this.#app.log.error(e)
-    })
-    await mongoose.connect(process.env.MONGO_URI as string, connection)
+  async #dbConnection() {
+    this.#connection = await dbConnection(this.#app.log)
   }
 
   #fastifyAppClosePlugin(): ApolloServerPlugin {
@@ -2368,7 +2262,8 @@ class Server {
           path: '/api'
         })
       )
-      await this.#mongo()
+      await this.#dbConnection()
+      await this.#connection?.connect()
       await this.#app.listen(PORT)
       this.#app.log.info(
         \`GraphQL server listening at: http://localhost:\${PORT}\${server.graphqlPath}\`
@@ -2380,8 +2275,8 @@ class Server {
 
   public async stop(): Promise<void> {
     try {
+      await this.#connection?.disconnect()
       await this.#app.close()
-      process.exit(0)
     } catch (e) {
       console.error(e)
     }