feat: support decorator for class. not support class property/method …

…in current

src/evaluate.ts

@@ -1199,6 +1199,22 @@ const visitors: EvaluateMap = {
     const ClassConstructor = evaluate(
       path.createChild(path.node.body, path.scope.createChild("class"))
     );
+
+    // support class decorators
+    const classDecorators = (path.node.decorators || [])
+      .map(node => evaluate(path.createChild(node)))
+      .reverse(); // revers decorators
+
+    // TODO: support class property decorator
+    // support class method decorators
+    // const propertyDecorators = path.node.body.body.filter(
+    //   node => node.decorators && node.decorators.length
+    // );
+
+    for (const decorator of classDecorators) {
+      decorator(ClassConstructor);
+    }
+
     path.scope.const(path.node.id.name, ClassConstructor);
   },
   ClassBody(path) {
@@ -1343,6 +1359,9 @@ const visitors: EvaluateMap = {
   ClassExpression(path) {
     //
   },
+  Decorator(path) {
+    return evaluate(path.createChild(path.node.expression));
+  },
   Super(path) {
     const { ctx } = path;
     const { SuperClass, ClassConstructor, ClassEntity } = ctx;

src/type.ts

@@ -100,6 +100,7 @@ export interface INodeTypeMap {
   ExportSpecifier: t.ExportSpecifier;
   SpreadProperty: t.SpreadProperty;
   DoExpression: t.DoExpression;
+  Decorator: t.Decorator;
 }
 
 export type EvaluateMap = {

test/experimental/decorators/decorator.test.ts

@@ -0,0 +1,44 @@
+import test from "ava";
+import vm from "../../../src/vm";
+
+test("DoExpression", t => {
+  const sandbox: any = vm.createContext({});
+
+  const Person: any = vm.runInContext(
+    `
+function testable(target){
+	target.testable = true;
+}
+
+@testable
+class Person{
+}
+
+module.exports = Person;
+    `,
+    sandbox
+  );
+  t.deepEqual(Person.testable, true);
+});
+
+test("DoExpression", t => {
+  const sandbox: any = vm.createContext({});
+
+  const Person: any = vm.runInContext(
+    `
+function testable(target){
+	target.prototype.testable = true;
+}
+
+@testable
+class Person{
+}
+
+module.exports = Person;
+    `,
+    sandbox
+  );
+  const person = new Person();
+  t.deepEqual(Person.testable, undefined);
+  t.deepEqual(person.testable, true);
+});