feat: add auth directive to surveys query

src/main/config/apollo-server.ts

@@ -1,5 +1,6 @@
 import typeDefs from '@/main/graphql/type-defs'
 import resolvers from '@/main/graphql/resolvers'
+import schemaDirectives from '@/main/graphql/directives'
 
 import { ApolloServer } from 'apollo-server-express'
 import { Express } from 'express'
@@ -28,6 +29,8 @@ export default (app: Express): void => {
   const server = new ApolloServer({
     resolvers,
     typeDefs,
+    schemaDirectives,
+    context: ({ req }) => ({ req }),
     plugins: [{
       requestDidStart: () => ({
         willSendResponse: ({ response, errors }) => handleErrors(response, errors)

src/main/graphql/directives/auth-directive.ts

@@ -0,0 +1,22 @@
+import { makeAuthMiddleware } from '@/main/factories'
+
+import { SchemaDirectiveVisitor, ForbiddenError } from 'apollo-server-express'
+import { GraphQLField, defaultFieldResolver } from 'graphql'
+
+export class AuthDirective extends SchemaDirectiveVisitor {
+  visitFieldDefinition (field: GraphQLField<any, any>): any {
+    const { resolve = defaultFieldResolver } = field
+    field.resolve = async (parent, args, context, info) => {
+      const request = {
+        accessToken: context?.req?.headers?.['x-access-token']
+      }
+      const httpResponse = await makeAuthMiddleware().handle(request)
+      if (httpResponse.statusCode === 200) {
+        Object.assign(context?.req, httpResponse.body)
+        return resolve.call(this, parent, args, context, info)
+      } else {
+        throw new ForbiddenError(httpResponse.body.message)
+      }
+    }
+  }
+}

src/main/graphql/directives/index.ts

@@ -0,0 +1,5 @@
+import { AuthDirective } from './auth-directive'
+
+export default {
+  auth: AuthDirective
+}

src/main/graphql/type-defs/base.ts

@@ -2,6 +2,8 @@ import { gql } from 'apollo-server-express'
 
 export default gql`
   scalar DateTime
+
+  directive @auth on FIELD_DEFINITION
   
   type Query {
     _: String

src/main/graphql/type-defs/survey.ts

@@ -2,7 +2,7 @@ import { gql } from 'apollo-server-express'
 
 export default gql`
   extend type Query {
-    surveys: [Survey!]!
+    surveys: [Survey!]! @auth
   }
 
   type Survey {