javascript - 如何使用类装饰器将装饰器应用于所有类方法

Question

我正在使用实验性的 typescript 装饰器来管理 express 中的访问控制。

class AccountController extends Controller {
  
  login(req: Request, res: Response) {
    const { email, password } = req.body;
    const token = await this.model.login(email, password);
    return res.json({
      token
    });
  }

  @hasRole('ADMIN')
  list(req: Request, res: Response) {
    res.json({
      data: await this.model.findAll()
    });
  }
}

hasRole方法装饰器工作正常，我很满意。 Controller实现 REST 方法:

class Controller {
  list(req: Request, res: Response) { // impl }
  get(req: Request, res: Response) { // impl } 
  create(req: Request, res: Response) { // impl }
  update(req: Request, res: Response) { // impl }
  delete(req: Request, res: Response) { // impl }
}

问题是，我必须将相同的装饰器应用于大多数其他 Controller ，而且我发现它非常重复。例如， StockController应该只允许访问 MERCHANT Angular 色，我必须执行以下操作:

class StockController extends Controller {
  @hasRole('MERCHANT')
  list(req: Request, res: Response) { 
    return super.list(req, res);
  }
  @hasRole('MERCHANT')
  get(req: Request, res: Response) { 
    return super.get(req, res);
  } 
  @hasRole('MERCHANT')
  create(req: Request, res: Response) { 
    return super.create(req, res);
  }
  @hasRole('MERCHANT')
  update(req: Request, res: Response) { 
    return super.update(req, res);
  }
  @hasRole('MERCHANT')
  delete(req: Request, res: Response) { 
    return super.delete(req, res);
  }
}

这种方法不仅枯燥重复而且不安全，因为如果我在 Controller 中添加一个方法并且不小心忘记将方法添加到子 Controller ，它们将允许不需要的访问。
我想用类装饰器来处理这个问题，并使用类似以下的东西:

@requireRole('MERCHANT')
class StockController extends Controller {}

但是，从我在 doc 中看到的:

The class decorator is applied to the constructor of the class and can be used to observe, modify, or replace a class definition.

据我了解，我无法在类装饰器中实现“方法 Hook ”。有什么建议？
供您引用， hasRole装饰器如下所示:

export function hasRole(role: string) {
  return function(target: Object, propertyKey: string, descriptor: PropertyDescriptor) {
    const originalMethod = descriptor.value;
    descriptor.value = function(req: Request, res: Response) {
      const user = res.locals.user;
      if (user && user.hasRole(role)) {
        originalMethod.apply(this, [req, res]);
      } else {
        res.status(403).json({});
      }
    }
  }
}

Answer 1

这可以通过覆盖类方法来实现

function AttachToAllClassDecorator<T>(someParam: string) {
    return function(target: new (...params: any[]) => T) {
        for (const key of Object.getOwnPropertyNames(target.prototype)) {
            // maybe blacklist methods here
            let descriptor = Object.getOwnPropertyDescriptor(target.prototype, key);
            if (descriptor) {
                descriptor = someDecorator(someParam)(key, descriptor);
                Object.defineProperty(target.prototype, key, descriptor);
            }
        }
    }
}

基本上遍历所有方法(可能会在它周围添加一些逻辑以将某些方法列入白名单/黑名单)并使用包含方法装饰器的新方法覆盖。
这是方法装饰器的基本示例。

function someDecorator(someParam: string): (methodName: string, descriptor: PropertyDescriptor) => PropertyDescriptor {
    return (methodName: string, descriptor: PropertyDescriptor): PropertyDescriptor => {
        let method = descriptor.value;

        descriptor.value = function(...args: any[]) {
            console.warn(`Here for descriptor ${methodName} with param ${someParam}`);

            return method.apply(this, args);
        }

        return descriptor;
    }
}

TS Playground