You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					145 lines
				
				3.7 KiB
			
		
		
			
		
	
	
					145 lines
				
				3.7 KiB
			| 
								 
											3 years ago
										 
									 | 
							
								'use strict'
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								const path = require('path')
							 | 
						||
| 
								 | 
							
								const fastifyStatic = require('fastify-static')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								// URI prefix to separate static assets for swagger UI
							 | 
						||
| 
								 | 
							
								const staticPrefix = '/static'
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function getRedirectPathForTheRootRoute (url) {
							 | 
						||
| 
								 | 
							
								  let redirectPath
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (url.substr(-1) === '/') {
							 | 
						||
| 
								 | 
							
								    redirectPath = `.${staticPrefix}/index.html`
							 | 
						||
| 
								 | 
							
								  } else {
							 | 
						||
| 
								 | 
							
								    const urlPathParts = url.split('/')
							 | 
						||
| 
								 | 
							
								    redirectPath = `./${urlPathParts[urlPathParts.length - 1]}${staticPrefix}/index.html`
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return redirectPath
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function fastifySwagger (fastify, opts, done) {
							 | 
						||
| 
								 | 
							
								  let staticCSP = false
							 | 
						||
| 
								 | 
							
								  if (opts.staticCSP === true) {
							 | 
						||
| 
								 | 
							
								    const csp = require('../static/csp.json')
							 | 
						||
| 
								 | 
							
								    staticCSP = `default-src 'self'; base-uri 'self'; block-all-mixed-content; font-src 'self' https: data:; frame-ancestors 'self'; img-src 'self' data: validator.swagger.io; object-src 'none'; script-src 'self' ${csp.script.join(' ')}; script-src-attr 'none'; style-src 'self' https: ${csp.style.join(' ')}; upgrade-insecure-requests;`
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								  if (typeof opts.staticCSP === 'string') {
							 | 
						||
| 
								 | 
							
								    staticCSP = opts.staticCSP
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								  if (typeof opts.staticCSP === 'object' && opts.staticCSP !== null) {
							 | 
						||
| 
								 | 
							
								    staticCSP = ''
							 | 
						||
| 
								 | 
							
								    Object.keys(opts.staticCSP).forEach(function (key) {
							 | 
						||
| 
								 | 
							
								      const value = Array.isArray(opts.staticCSP[key]) ? opts.staticCSP[key].join(' ') : opts.staticCSP[key]
							 | 
						||
| 
								 | 
							
								      staticCSP += `${key.toLowerCase()} ${value}; `
							 | 
						||
| 
								 | 
							
								    })
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (typeof staticCSP === 'string' || typeof opts.transformStaticCSP === 'function') {
							 | 
						||
| 
								 | 
							
								    fastify.addHook('onSend', function (request, reply, payload, done) {
							 | 
						||
| 
								 | 
							
								      // set static csp when it is passed
							 | 
						||
| 
								 | 
							
								      if (typeof staticCSP === 'string') {
							 | 
						||
| 
								 | 
							
								        reply.header('content-security-policy', staticCSP.trim())
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								      // mutate the header when it is passed
							 | 
						||
| 
								 | 
							
								      const header = reply.getHeader('content-security-policy')
							 | 
						||
| 
								 | 
							
								      if (header && typeof opts.transformStaticCSP === 'function') {
							 | 
						||
| 
								 | 
							
								        reply.header('content-security-policy', opts.transformStaticCSP(header))
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								      done()
							 | 
						||
| 
								 | 
							
								    })
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const hooks = Object.create(null)
							 | 
						||
| 
								 | 
							
								  if (opts.hooks) {
							 | 
						||
| 
								 | 
							
								    const additionalHooks = [
							 | 
						||
| 
								 | 
							
								      'onRequest',
							 | 
						||
| 
								 | 
							
								      'preHandler'
							 | 
						||
| 
								 | 
							
								    ]
							 | 
						||
| 
								 | 
							
								    for (const hook of additionalHooks) {
							 | 
						||
| 
								 | 
							
								      hooks[hook] = opts.hooks[hook]
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.route({
							 | 
						||
| 
								 | 
							
								    url: '/',
							 | 
						||
| 
								 | 
							
								    method: 'GET',
							 | 
						||
| 
								 | 
							
								    schema: { hide: true },
							 | 
						||
| 
								 | 
							
								    ...hooks,
							 | 
						||
| 
								 | 
							
								    handler: (req, reply) => {
							 | 
						||
| 
								 | 
							
								      reply.redirect(getRedirectPathForTheRootRoute(req.raw.url))
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.route({
							 | 
						||
| 
								 | 
							
								    url: '/uiConfig',
							 | 
						||
| 
								 | 
							
								    method: 'GET',
							 | 
						||
| 
								 | 
							
								    schema: { hide: true },
							 | 
						||
| 
								 | 
							
								    ...hooks,
							 | 
						||
| 
								 | 
							
								    handler: (req, reply) => {
							 | 
						||
| 
								 | 
							
								      reply.send(opts.uiConfig)
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.route({
							 | 
						||
| 
								 | 
							
								    url: '/initOAuth',
							 | 
						||
| 
								 | 
							
								    method: 'GET',
							 | 
						||
| 
								 | 
							
								    schema: { hide: true },
							 | 
						||
| 
								 | 
							
								    ...hooks,
							 | 
						||
| 
								 | 
							
								    handler: (req, reply) => {
							 | 
						||
| 
								 | 
							
								      reply.send(opts.initOAuth)
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.route({
							 | 
						||
| 
								 | 
							
								    url: '/json',
							 | 
						||
| 
								 | 
							
								    method: 'GET',
							 | 
						||
| 
								 | 
							
								    schema: { hide: true },
							 | 
						||
| 
								 | 
							
								    ...hooks,
							 | 
						||
| 
								 | 
							
								    handler: function (req, reply) {
							 | 
						||
| 
								 | 
							
								      reply.send(fastify.swagger())
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.route({
							 | 
						||
| 
								 | 
							
								    url: '/yaml',
							 | 
						||
| 
								 | 
							
								    method: 'GET',
							 | 
						||
| 
								 | 
							
								    schema: { hide: true },
							 | 
						||
| 
								 | 
							
								    ...hooks,
							 | 
						||
| 
								 | 
							
								    handler: function (req, reply) {
							 | 
						||
| 
								 | 
							
								      reply
							 | 
						||
| 
								 | 
							
								        .type('application/x-yaml')
							 | 
						||
| 
								 | 
							
								        .send(fastify.swagger({ yaml: true }))
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // serve swagger-ui with the help of fastify-static
							 | 
						||
| 
								 | 
							
								  fastify.register(fastifyStatic, {
							 | 
						||
| 
								 | 
							
								    root: path.join(__dirname, '..', 'static'),
							 | 
						||
| 
								 | 
							
								    prefix: staticPrefix,
							 | 
						||
| 
								 | 
							
								    decorateReply: false
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.register(fastifyStatic, {
							 | 
						||
| 
								 | 
							
								    root: opts.baseDir || path.join(__dirname, '..'),
							 | 
						||
| 
								 | 
							
								    serve: false
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // Handler for external documentation files passed via $ref
							 | 
						||
| 
								 | 
							
								  fastify.route({
							 | 
						||
| 
								 | 
							
								    url: '/*',
							 | 
						||
| 
								 | 
							
								    method: 'GET',
							 | 
						||
| 
								 | 
							
								    schema: { hide: true },
							 | 
						||
| 
								 | 
							
								    ...hooks,
							 | 
						||
| 
								 | 
							
								    handler: function (req, reply) {
							 | 
						||
| 
								 | 
							
								      const file = req.params['*']
							 | 
						||
| 
								 | 
							
								      reply.sendFile(file)
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  done()
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								module.exports = fastifySwagger
							 |