|
1 | 1 | const Koa = require('koa'); |
2 | 2 | const Router = require('koa-router'); |
3 | 3 | const session = require('koa-session'); |
4 | | -const cors = require('@koa/cors'); |
5 | 4 | const jsonBody = require('koa-json-body'); |
6 | 5 | const { get } = require('lodash'); |
7 | 6 | const { Queue } = require('bullmq'); |
@@ -41,6 +40,49 @@ const queue = new Queue('generator', { connection: bullRedisConnection }); |
41 | 40 |
|
42 | 41 | const cancelSignalRedis = new Redis(REDIS_CONNECTION_STRING); // New connection to make sure that pub/sub will work correctly. |
43 | 42 |
|
| 43 | +const createCorsMiddleware = (options = {}) => { |
| 44 | + const { |
| 45 | + allowHeaders = [ |
| 46 | + 'Accept', |
| 47 | + 'Accept-Language', |
| 48 | + 'Content-Language', |
| 49 | + 'Content-Type', |
| 50 | + 'Authorization', |
| 51 | + 'X-Requested-With', |
| 52 | + ], |
| 53 | + allowMethods = ['GET', 'HEAD', 'POST', 'PUT', 'DELETE', 'OPTIONS'], |
| 54 | + credentials = false, |
| 55 | + } = options; |
| 56 | + |
| 57 | + return async (ctx, next) => { |
| 58 | + const requestOrigin = ctx.get('Origin'); |
| 59 | + if (requestOrigin) { |
| 60 | + ctx.set('Access-Control-Allow-Origin', requestOrigin); |
| 61 | + } else if (!credentials) { |
| 62 | + ctx.set('Access-Control-Allow-Origin', '*'); |
| 63 | + } |
| 64 | + |
| 65 | + if (credentials) { |
| 66 | + ctx.set('Access-Control-Allow-Credentials', 'true'); |
| 67 | + } |
| 68 | + |
| 69 | + ctx.set('Access-Control-Allow-Methods', allowMethods.join(', ')); |
| 70 | + const requestedHeaders = ctx.get('Access-Control-Request-Headers'); |
| 71 | + if (requestedHeaders) { |
| 72 | + ctx.set('Access-Control-Allow-Headers', requestedHeaders); |
| 73 | + } else { |
| 74 | + ctx.set('Access-Control-Allow-Headers', allowHeaders.join(', ')); |
| 75 | + } |
| 76 | + |
| 77 | + if (ctx.method === 'OPTIONS') { |
| 78 | + ctx.status = 204; |
| 79 | + return; |
| 80 | + } |
| 81 | + |
| 82 | + await next(); |
| 83 | + }; |
| 84 | +}; |
| 85 | + |
44 | 86 | async function generatePoster(buildId, props) { |
45 | 87 | const { id } = await addPoster({ buildId, props }); |
46 | 88 |
|
@@ -284,6 +326,7 @@ async function main() { |
284 | 326 |
|
285 | 327 | router.post('/login', async (ctx) => { |
286 | 328 | const authResponse = await authEndpoints.authorize(ctx.request, ctx.response, ctx.session); |
| 329 | + console.log(JSON.stringify(authResponse.body)); |
287 | 330 | ctx.session = null; |
288 | 331 | if (authResponse.modifiedSession) { |
289 | 332 | ctx.session = authResponse.modifiedSession; |
@@ -324,11 +367,7 @@ async function main() { |
324 | 367 | app |
325 | 368 | .use(errorHandler) |
326 | 369 | .use(unAuthorizedRouter.routes()) |
327 | | - .use( |
328 | | - cors({ |
329 | | - credentials: true, |
330 | | - }), |
331 | | - ) |
| 370 | + .use(createCorsMiddleware({ credentials: true })) |
332 | 371 | .use(authMiddleware) |
333 | 372 | .use(jsonBody({ fallback: true, limit: '10mb' })) |
334 | 373 | .use(router.routes()) |
|
0 commit comments