76 lines
1.8 KiB
TypeScript
76 lines
1.8 KiB
TypeScript
import { ActionName } from './action';
|
|
import { requireModule } from './utils';
|
|
import { HandlerType } from './resourcer';
|
|
|
|
export type MiddlewareType = string | string[] | HandlerType | HandlerType[] | MiddlewareOptions | MiddlewareOptions[];
|
|
|
|
export interface MiddlewareOptions {
|
|
/**
|
|
* actions 白名单，默认有 list、get、create、update、delete
|
|
*/
|
|
only?: Array<ActionName>;
|
|
|
|
/**
|
|
* actions 黑名单，默认有 list、get、create、update、delete
|
|
*/
|
|
except?: Array<ActionName>;
|
|
|
|
handler?: HandlerType | Function;
|
|
|
|
[key: string]: any;
|
|
}
|
|
|
|
export class Middleware {
|
|
protected options: MiddlewareOptions;
|
|
|
|
constructor(options: MiddlewareOptions | Function) {
|
|
options = requireModule(options);
|
|
if (typeof options === 'function') {
|
|
this.options = { handler: options };
|
|
} else {
|
|
this.options = options;
|
|
}
|
|
}
|
|
|
|
getHandler() {
|
|
const handler = requireModule(this.options.handler);
|
|
if (typeof handler !== 'function') {
|
|
throw new Error('Handler must be a function!');
|
|
}
|
|
return handler;
|
|
}
|
|
|
|
canAccess(name: ActionName) {
|
|
const { only = [], except = [] } = this.options;
|
|
if (only.length > 0) {
|
|
return only.includes(name);
|
|
}
|
|
if (except.length > 0) {
|
|
return !except.includes(name);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static toInstanceArray(middlewares: any): Middleware[] {
|
|
if (!middlewares) {
|
|
return [];
|
|
}
|
|
if (!Array.isArray(middlewares)) {
|
|
middlewares = [middlewares];
|
|
}
|
|
return middlewares.map((middleware) => {
|
|
if (middleware instanceof Middleware) {
|
|
return middleware;
|
|
}
|
|
if (typeof middleware === 'object') {
|
|
return new Middleware(middleware);
|
|
}
|
|
if (typeof middleware === 'function') {
|
|
return new Middleware({ handler: middleware });
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
export default Middleware;
|