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.

116 lines
2.7 KiB

4 years ago
'use strict'
const assert = require('assert')
const registeredPlugins = Symbol.for('registered-plugin')
const {
kReply,
kRequest
} = require('./symbols.js')
function getMeta (fn) {
return fn[Symbol.for('plugin-meta')]
}
function getPluginName (func) {
// let's see if this is a file, and in that case use that
// this is common for plugins
const cache = require.cache
const keys = Object.keys(cache)
for (var i = 0; i < keys.length; i++) {
if (cache[keys[i]].exports === func) {
return keys[i]
}
}
// if not maybe it's a named function, so use that
if (func.name) {
return func.name
}
return null
}
function getFuncPreview (func) {
// takes the first two lines of the function if nothing else works
return func.toString().split('\n').slice(0, 2).map(s => s.trim()).join(' -- ')
}
function getDisplayName (fn) {
return fn[Symbol.for('fastify.display-name')]
}
function shouldSkipOverride (fn) {
return !!fn[Symbol.for('skip-override')]
}
function checkDependencies (fn) {
const meta = getMeta(fn)
if (!meta) return
const dependencies = meta.dependencies
if (!dependencies) return
assert(Array.isArray(dependencies), 'The dependencies should be an array of strings')
dependencies.forEach(dependency => {
assert(
this[registeredPlugins].indexOf(dependency) > -1,
`The dependency '${dependency}' of plugin '${meta.name}' is not registered`
)
})
}
function checkDecorators (fn) {
const meta = getMeta(fn)
if (!meta) return
const decorators = meta.decorators
if (!decorators) return
if (decorators.fastify) _checkDecorators.call(this, 'Fastify', decorators.fastify)
if (decorators.reply) _checkDecorators.call(this[kReply], 'Reply', decorators.reply)
if (decorators.request) _checkDecorators.call(this[kRequest], 'Request', decorators.request)
}
function _checkDecorators (instance, decorators) {
assert(Array.isArray(decorators), 'The decorators should be an array of strings')
decorators.forEach(decorator => {
assert(
instance === 'Fastify' ? decorator in this : decorator in this.prototype,
`The decorator '${decorator}' is not present in ${instance}`
)
})
}
function registerPluginName (fn) {
const meta = getMeta(fn)
if (!meta) return
const name = meta.name
if (!name) return
this[registeredPlugins].push(name)
}
function registerPlugin (fn) {
registerPluginName.call(this, fn)
checkDecorators.call(this, fn)
checkDependencies.call(this, fn)
return shouldSkipOverride(fn)
}
module.exports = {
getPluginName,
getFuncPreview,
registeredPlugins,
getDisplayName,
registerPlugin
}
module.exports[Symbol.for('internals')] = {
shouldSkipOverride,
getMeta,
checkDecorators,
checkDependencies
}