hiro/apollo-server
1
0
Fork 0
mirror of https://github.com/vale981/apollo-server synced 2025-03-05 09:41:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fastify) Implement fastify upload middleware

Browse source
This commit is contained in:
Remy Korrelboom 2019-01-02 14:17:14 +01:00
parent 59873843a8
commit 05158d3e0d
1 changed files with 50 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
packages/apollo-server-fastify/src/ApolloServer.ts
View file

@ -1,14 +1,15 @@
import { renderPlaygroundPage } from '@apollographql/graphql-playground-html';
import { Accepts } from 'accepts';
import {
ApolloServerBase,
ApolloServerBase, FileUploadOptions, formatApolloErrors,
PlaygroundRenderPageOptions,
processFileUploads,
processFileUploads
} from 'apollo-server-core';
import { FastifyInstance, FastifyReply, FastifyRequest } from 'fastify';
import { IncomingMessage, OutgoingMessage, Server } from 'http';
import { graphqlFastify } from './fastifyApollo';
const kMultipart = Symbol('multipart');
const fastJson = require('fast-json-stringify');
export interface ServerRegistration {
@ -27,6 +28,33 @@ const stringifyHealthCheck = fastJson({
},
});
const fileUploadMiddleware = (
uploadsConfig: FileUploadOptions,
server: ApolloServerBase,
) => (
req: FastifyRequest<IncomingMessage>,
reply: FastifyReply<OutgoingMessage>,
done: (err: Error | null, body?: any) => void
) => {
if ((req.req as any)[kMultipart] && typeof processFileUploads === 'function') {
processFileUploads(req.req, reply.res, uploadsConfig)
.then(body => {
req.body = body;
done(null);
})
.catch(error => {
if (error.status && error.expose) reply.status(error.status);
throw formatApolloErrors([error], {
formatter: server.requestOptions.formatError,
debug: server.requestOptions.debug,
});
});
} else {
done(null);
}
};
export class ApolloServer extends ApolloServerBase {
protected supportsSubscriptions(): boolean {
return true;
@ -84,23 +112,8 @@ export class ApolloServer extends ApolloServerBase {
reply.send();
});
if (typeof processFileUploads === 'function' && this.uploadsConfig) {
instance.addContentTypeParser(
'multipart',
async (request: IncomingMessage) =>
// This extra function guarding is being mandated by TypeScript.
// It certainly shouldn't be possible for this parse to even
// be present unless `processFileUploads` was a function when
// the handler was added (initially at server startup).
typeof processFileUploads === 'function' &&
processFileUploads(request, this.uploadsConfig),
);
}
instance.route({
method: ['GET', 'POST'],
url: '/',
beforeHandler: (
const beforeHandlers = [
(
req: FastifyRequest<IncomingMessage>,
reply: FastifyReply<OutgoingMessage>,
done: () => void,
@ -134,7 +147,24 @@ export class ApolloServer extends ApolloServerBase {
}
}
done();
},
}
];
if (typeof processFileUploads === 'function' && this.uploadsConfig) {
instance.addContentTypeParser(
'multipart',
(request: IncomingMessage, done: (err: Error | null, body?: any) => void) => {
(request as any)[kMultipart] = true;
done(null);
}
);
beforeHandlers.push(fileUploadMiddleware(this.uploadsConfig, this));
}
instance.route({
method: ['GET', 'POST'],
url: '/',
beforeHandler: beforeHandlers,
handler: await graphqlFastify(this.graphQLServerOptions.bind(this)),
});
},