feat: add option to disable bot auto-start

pubkeyapp · Feb 20, 2024 · cd72751 · cd72751
1 parent 5949645
commit cd72751
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 0 deletions.
diff --git a/.env.example b/.env.example
@@ -16,6 +16,8 @@ AUTH_SOLANA_ADMIN_IDS=
 AUTH_SOLANA_ENABLED=true
 # Enable Bull UI
 BULL_ADMIN=admin:6c9107073a49d1e6129bfba07d494050c182d7630d3c86aca94ffa43cf1533f2
+# Enable Bot auto-start
+#BOT_AUTO_START=true
 # Set Cloak keychain and master key (generate with `yarn cloak generate`)
 CLOAK_MASTER_KEY=
 CLOAK_KEYCHAIN=

diff --git a/libs/api/bot/data-access/src/lib/api-bot-manager.service.ts b/libs/api/bot/data-access/src/lib/api-bot-manager.service.ts
@@ -21,6 +21,10 @@ export class ApiBotManagerService implements OnModuleInit {
   constructor(private readonly core: ApiCoreService, private readonly botMember: ApiBotMemberService) {}
 
   async onModuleInit() {
+    if (!this.core.config.botAutoStart) {
+      this.logger.verbose(`Bot auto start is disabled`)
+      return
+    }
     const bots = await this.core.data.bot.findMany({ where: { status: BotStatus.Active } })
     for (const bot of bots) {
       this.logger.verbose(`Starting bot ${bot.name}`)

diff --git a/libs/api/core/data-access/src/lib/config/api-core-config.service.ts b/libs/api/core/data-access/src/lib/config/api-core-config.service.ts
@@ -102,6 +102,10 @@ export class ApiCoreConfigService {
     return this.service.get<string>('apiUrl') as string
   }
 
+  get botAutoStart(): boolean {
+    return this.service.get<boolean>('botAutoStart') ?? false
+  }
+
   get cookieDomains(): string[] {
     return this.service.get<string[]>('cookieDomains') ?? []
   }

diff --git a/libs/api/core/data-access/src/lib/config/configuration.ts b/libs/api/core/data-access/src/lib/config/configuration.ts
@@ -33,6 +33,8 @@ export interface ApiCoreConfig {
   authSolanaAdminIds: string[]
   authSolanaLinkEnabled: boolean
   authSolanaLoginEnabled: boolean
+  // Bot
+  botAutoStart: boolean
   // Cookies
   cookieDomains: string[]
   cookieName: string
@@ -69,6 +71,7 @@ export function configuration(): ApiCoreConfig {
     authSolanaAdminIds: getFromEnvironment('AUTH_SOLANA_ADMIN_IDS'),
     authSolanaLinkEnabled: process.env['AUTH_SOLANA_LINK_ENABLED'] === 'true',
     authSolanaLoginEnabled: process.env['AUTH_SOLANA_LOGIN_ENABLED'] === 'true',
+    botAutoStart: process.env['BOT_AUTO_START'] === 'true',
     cookieDomains,
     cookieName: '__session',
     cookieSecure: process.env['COOKIE_SECURE'] === 'true',

diff --git a/libs/api/core/data-access/src/lib/config/validation-schema.ts b/libs/api/core/data-access/src/lib/config/validation-schema.ts
@@ -15,6 +15,8 @@ export const validationSchema = Joi.object({
   AUTH_SOLANA_ADMIN_IDS: Joi.string(),
   AUTH_SOLANA_LINK_ENABLED: Joi.boolean().default(true),
   AUTH_SOLANA_LOGIN_ENABLED: Joi.boolean().default(false),
+  // Bot
+  BOT_AUTO_START: Joi.boolean().default(true),
   CLOAK_MASTER_KEY: Joi.string().required().error(new Error(`CLOAK_MASTER_KEY is required.`)),
   CLOAK_KEYCHAIN: Joi.string().required().error(new Error(`CLOAK_KEYCHAIN is required.`)),
   COOKIE_NAME: Joi.string().default('__session'),