import version from './version' import { EventEmitter } from 'events' import { stripSlashes, createDebug } from '@feathersjs/commons' import { HOOKS, hooks, middleware } from '@feathersjs/hooks' import { eventHook, eventMixin } from './events' import { hookMixin } from './hooks' import { wrapService, getServiceOptions, protectedMethods } from './service' import { FeathersApplication, ServiceMixin, Service, ServiceOptions, ServiceInterface, Application, FeathersService, ApplicationHookOptions } from './declarations' import { enableHooks } from './hooks' const debug = createDebug('@feathersjs/feathers') export class Feathers extends EventEmitter implements FeathersApplication { services: Services = {} as Services settings: Settings = {} as Settings mixins: ServiceMixin>[] = [hookMixin, eventMixin] version: string = version _isSetup = false protected registerHooks: (this: any, allHooks: any) => any constructor() { super() this.registerHooks = enableHooks(this) this.registerHooks({ around: [eventHook] }) } get(name: L): Settings[L] { return this.settings[name] } set(name: L, value: Settings[L]) { this.settings[name] = value return this } configure(callback: (this: this, app: this) => void) { callback.call(this, this) return this } defaultService(location: string): ServiceInterface { throw new Error(`Can not find service '${location}'`) } service( location: L ): FeathersService { const path = (stripSlashes(location) || '/') as L const current = this.services[path] if (typeof current === 'undefined') { this.use(path, this.defaultService(path) as any) return this.service(path) } return current as any } protected _setup() { this._isSetup = true return Object.keys(this.services) .reduce( (current, path) => current.then(() => { const service: any = this.service(path as any) if (typeof service.setup === 'function') { debug(`Setting up service for \`${path}\``) return service.setup(this, path) } }), Promise.resolve() ) .then(() => this) } get setup() { return this._setup } set setup(value) { this._setup = (value as any)[HOOKS] ? value : hooks( value, middleware().params('server').props({ app: this }) ) } protected _teardown() { this._isSetup = false return Object.keys(this.services) .reduce( (current, path) => current.then(() => { const service: any = this.service(path as any) if (typeof service.teardown === 'function') { debug(`Tearing down service for \`${path}\``) return service.teardown(this, path) } }), Promise.resolve() ) .then(() => this) } get teardown() { return this._teardown } set teardown(value) { this._teardown = (value as any)[HOOKS] ? value : hooks( value, middleware().params('server').props({ app: this }) ) } use( path: L, service: keyof any extends keyof Services ? ServiceInterface | Application : Services[L], options?: ServiceOptions ): this { if (typeof path !== 'string') { throw new Error(`'${path}' is not a valid service path.`) } const location = (stripSlashes(path) || '/') as L const subApp = service as Application const isSubApp = typeof subApp.service === 'function' && subApp.services if (isSubApp) { Object.keys(subApp.services).forEach((subPath) => this.use(`${location}/${subPath}` as any, subApp.service(subPath) as any) ) return this } const protoService = wrapService(location, service, options as ServiceOptions) const serviceOptions = getServiceOptions(protoService) for (const name of protectedMethods) { if (serviceOptions.methods.includes(name)) { throw new Error(`'${name}' on service '${location}' is not allowed as a custom method name`) } } debug(`Registering new service at \`${location}\``) // Add all the mixins this.mixins.forEach((fn) => fn.call(this, protoService, location, serviceOptions)) this.services[location] = protoService // If we ran setup already, set this service up explicitly, this will not `await` if (this._isSetup && typeof protoService.setup === 'function') { debug(`Setting up service for \`${location}\``) protoService.setup(this, location) } return this } async unuse( location: L ): Promise> { const path = (stripSlashes(location) || '/') as L const service = this.services[path] as Service if (service && typeof service.teardown === 'function') { await service.teardown(this as any, path) } delete this.services[path] return service as any } hooks(hookMap: ApplicationHookOptions) { const untypedMap = hookMap as any if (untypedMap.before || untypedMap.after || untypedMap.error || untypedMap.around) { // regular hooks for all service methods this.registerHooks(untypedMap) } else if (untypedMap.setup || untypedMap.teardown) { // .setup and .teardown application hooks hooks(this, untypedMap) } else { // Other registration formats are just `around` hooks this.registerHooks({ around: untypedMap }) } return this } }