refactor(provider): allow more customization

Author: RomainLanz

providers/transmit_provider.ts

@@ -65,10 +65,10 @@ export default class TransmitProvider {
       return ctx.response.noContent()
     })
 
-    if (config.routeHandlerDomain) {
-      registerRoute.domain(config.routeHandlerDomain)
-      subscribeRoute.domain(config.routeHandlerDomain)
-      unsubscribeRoute.domain(config.routeHandlerDomain)
+    if (config.routeHandlerModifier) {
+      config.routeHandlerModifier(registerRoute)
+      config.routeHandlerModifier(subscribeRoute)
+      config.routeHandlerModifier(unsubscribeRoute)
     }
   }
 

src/types/main.ts

@@ -8,6 +8,7 @@
  */
 
 import type { TransportFactory } from '@rlanz/bus/types/main'
+import type { Route } from '@adonisjs/core/http'
 
 /**
  * A Duration can be a number in milliseconds or a string formatted as a duration
@@ -26,5 +27,5 @@ export type Broadcastable = { [key: string]: Broadcastable } | string | number |
 export interface TransmitConfig {
   pingInterval?: Duration | false
   transport: null | { driver: TransportFactory; channel?: string }
-  routeHandlerDomain?: string
+  routeHandlerModifier?: (route: Route) => void
 }

tests/transmit_provider.spec.ts

@@ -26,11 +26,11 @@ test.group('Provider', () => {
     assert.equal((router.routes[2] as Route).getPattern(), '__transmit/unsubscribe')
   })
 
-  test('allow to customize the domain of the routes', async ({ assert }) => {
+  test('allow to customize the route registration', async ({ assert }) => {
     const app = await setupApp({
       transmit: {
         transport: null,
-        routeHandlerDomain: 'example.com',
+        routeHandlerModifier: (route) => route.domain('example.com'),
       },
     })