feat(handler): add logger and graphiql to config options

Author: schettn

packages/pylon/src/app/handler/pylon-handler.ts

@@ -1,18 +1,20 @@
-import {createSchema, createYoga} from 'graphql-yoga'
 import {GraphQLScalarType, Kind} from 'graphql'
 import {
   DateTimeISOResolver,
   GraphQLVoid,
   JSONObjectResolver,
   JSONResolver
 } from 'graphql-scalars'
+import {createSchema, createYoga} from 'graphql-yoga'
 
-import {useSentry} from '../envelop/use-sentry'
-import {Context} from '../../context'
-import {resolversToGraphQLResolvers} from '../../define-pylon'
-import {PylonConfig} from '../..'
 import {readFileSync} from 'fs'
+import {logger} from 'hono/logger'
 import path from 'path'
+import {app, PylonConfig} from '../..'
+import {Context} from '../../context'
+import {resolversToGraphQLResolvers} from '../../define-pylon'
+import {useSentry} from '../envelop/use-sentry'
+import {graphqlViewerHandler} from './graphql-viewer-handler'
 
 interface PylonHandlerOptions {
   graphql: {
@@ -117,6 +119,14 @@ export const handler = (options: PylonHandlerOptions) => {
     schema
   })
 
+  // Load middlewares based on config
+  if (config?.logger !== false) {
+    app.use('*', logger())
+  }
+  if (config?.graphiql !== false) {
+    app.get('/viewer', graphqlViewerHandler)
+  }
+
   const handler = async (c: Context): Promise<Response> => {
     let executionContext: Context['executionCtx'] | {} = {}
 

packages/pylon/src/app/index.ts

@@ -1,10 +1,7 @@
 import {Hono} from 'hono'
-import {logger} from 'hono/logger'
 import {sentry} from '@hono/sentry'
 
 import {asyncContext, Env} from '../context'
-import {graphqlViewerHandler} from './handler/graphql-viewer-handler'
-
 export const app = new Hono<Env>()
 
 app.use('*', sentry())
@@ -21,12 +18,8 @@ app.use('*', async (c, next) => {
   })
 })
 
-app.use('*', logger())
-
 app.use((c, next) => {
   // @ts-ignore
   c.req.id = crypto.randomUUID()
   return next()
 })
-
-app.get('/viewer', graphqlViewerHandler)

packages/pylon/src/index.ts

@@ -18,7 +18,13 @@ export {getEnv} from './get-env.js'
 export {createDecorator} from './create-decorator.js'
 export {createPubSub as experimentalCreatePubSub} from 'graphql-yoga'
 
-export type PylonConfig = Pick<YogaServerOptions<Context, Context>, 'plugins'>
+export type PylonConfig = Pick<
+  YogaServerOptions<Context, Context>,
+  'plugins'
+> & {
+  logger?: boolean
+  graphiql?: boolean
+}
 
 export type ID = string & {readonly brand?: unique symbol}
 export type Int = number & {readonly brand?: unique symbol}