✨ Macro support

Author: intellild

package.json

@@ -63,7 +63,9 @@
     "@babel/types": "^7.6.3",
     "@favoloso/conventional-changelog-emoji": "^0.10.0",
     "@release-it/conventional-changelog": "^1.1.0",
+    "@types/babel-plugin-macros": "^2.8.5",
     "@types/jest": "^24.0.19",
+    "babel-plugin-macros": "^3.1.0",
     "babel-test": "^0.2.3",
     "conventional-changelog-cli": "^2.0.25",
     "husky": "^4.2.3",
@@ -72,6 +74,9 @@
     "release-it": "13.7.1",
     "typescript": "^3.6.4"
   },
+  "peerDependencies": {
+    "babel-plugin-macros": "^3.1.0"
+  },
   "dependencies": {
     "@babel/helper-plugin-utils": "^7.0.0"
   }

src/macro.ts

@@ -0,0 +1,23 @@
+import { createMacro, MacroHandler } from 'babel-plugin-macros';
+import { transform } from './transform';
+
+const metadataMacro: MacroHandler = ({ references }) => {
+  references.default.forEach(reference => {
+    const decorator = reference.findParent(parent => parent.isDecorator())
+    if (!decorator) {
+      throw new Error("Metadata macro should be used as class decorator");
+    }
+    const classDeclaration = decorator.findParent(parent => parent.isClassDeclaration());
+    if (!classDeclaration) {
+      throw new Error("Metadata macro should be used as class decorator");
+    }
+    if (classDeclaration.isClassDeclaration()) {
+      if (classDeclaration.node.decorators) {
+        classDeclaration.node.decorators = classDeclaration.node.decorators.filter(it => it !== decorator.node);
+      }
+      transform(classDeclaration);
+    }
+  })
+};
+
+export default createMacro(metadataMacro);

src/plugin.ts

@@ -2,6 +2,7 @@ import { PluginObj } from '@babel/core';
 import { declare } from '@babel/helper-plugin-utils';
 import { parameterVisitor } from './parameter/parameterVisitor';
 import { metadataVisitor } from './metadata/metadataVisitor';
+import { transform } from './transform';
 
 export default declare(
   (api: any): PluginObj => {
@@ -20,23 +21,7 @@ export default declare(
            */
           programPath.traverse({
             ClassDeclaration(path) {
-              for (const field of path.get('body').get('body')) {
-                if (
-                  field.type !== 'ClassMethod' &&
-                  field.type !== 'ClassProperty'
-                )
-                  continue;
-
-                parameterVisitor(path, field as any);
-                metadataVisitor(path, field as any);
-              }
-
-              /**
-               * We need to keep binding in order to let babel know where imports
-               * are used as a Value (and not just as a type), so that
-               * `babel-transform-typescript` do not strip the import.
-               */
-              (path.parentPath.scope as any).crawl();
+              transform(path);
             }
           });
         }

src/transform.ts

@@ -0,0 +1,24 @@
+import { NodePath } from '@babel/traverse';
+import { ClassDeclaration } from '@babel/types';
+import { metadataVisitor } from './metadata/metadataVisitor';
+import { parameterVisitor } from './parameter/parameterVisitor';
+
+export function transform(path: NodePath<ClassDeclaration>) {
+  for (const field of path.get('body').get('body')) {
+    if (
+      field.type !== 'ClassMethod' &&
+      field.type !== 'ClassProperty'
+    )
+      continue;
+
+    parameterVisitor(path, field as any);
+    metadataVisitor(path, field as any);
+  }
+
+  /**
+   * We need to keep binding in order to let babel know where imports
+   * are used as a Value (and not just as a type), so that
+   * `babel-transform-typescript` do not strip the import.
+   */
+  (path.parentPath.scope as any).crawl();
+}

yarn.lock

@@ -844,6 +844,13 @@
     "@babel/helper-plugin-utils" "^7.10.4"
     "@babel/plugin-transform-typescript" "^7.10.4"
 
+"@babel/runtime@^7.12.5":
+  version "7.17.2"
+  resolved "https://registry.npmjs.org/@babel/runtime/-/runtime-7.17.2.tgz#66f68591605e59da47523c631416b18508779941"
+  integrity sha512-hzeyJyMA1YGdJTuWU0e/j4wKXrU4OMFvY2MSlaI9B7VQb0r5cxTE3EAIS2Q7Tn2RIcDkRvTA/v2JsAEhxe99uw==
+  dependencies:
+    regenerator-runtime "^0.13.4"
+
 "@babel/runtime@^7.8.4":
   version "7.11.2"
   resolved "https://registry.yarnpkg.com/@babel/runtime/-/runtime-7.11.2.tgz#f549c13c754cc40b87644b9fa9f09a6a95fe0736"
@@ -1209,6 +1216,24 @@
   dependencies:
     defer-to-connect "^2.0.0"
 
+"@types/babel-plugin-macros@^2.8.5":
+  version "2.8.5"
+  resolved "https://registry.npmjs.org/@types/babel-plugin-macros/-/babel-plugin-macros-2.8.5.tgz#04474f9898aa9112afc22fa0e7e53a898fcaba4c"
+  integrity sha512-+NcIm/VBaSb4xaycov9f4Vmk4hMVPrgISoHEk+kalgVK6BlkwDZbXkW9kt1jCXVczTKXldL5RHIacExaWzxAkA==
+  dependencies:
+    "@types/babel__core" "*"
+
+"@types/babel__core@*":
+  version "7.1.18"
+  resolved "https://registry.npmjs.org/@types/babel__core/-/babel__core-7.1.18.tgz#1a29abcc411a9c05e2094c98f9a1b7da6cdf49f8"
+  integrity sha512-S7unDjm/C7z2A2R9NzfKCK1I+BAALDtxEmsJBwlB3EzNfb929ykjL++1CK9LO++EIp2fQrC8O+BwjKvz6UeDyQ==
+  dependencies:
+    "@babel/parser" "^7.1.0"
+    "@babel/types" "^7.0.0"
+    "@types/babel__generator" "*"
+    "@types/babel__template" "*"
+    "@types/babel__traverse" "*"
+
 "@types/babel__core@^7.1.0":
   version "7.1.10"
   resolved "https://registry.yarnpkg.com/@types/babel__core/-/babel__core-7.1.10.tgz#ca58fc195dd9734e77e57c6f2df565623636ab40"
@@ -1590,6 +1615,15 @@ babel-plugin-jest-hoist@^24.9.0:
   dependencies:
     "@types/babel__traverse" "^7.0.6"
 
+babel-plugin-macros@^3.1.0:
+  version "3.1.0"
+  resolved "https://registry.npmjs.org/babel-plugin-macros/-/babel-plugin-macros-3.1.0.tgz#9ef6dc74deb934b4db344dc973ee851d148c50c1"
+  integrity sha512-Cg7TFGpIr01vOQNODXOOaGz2NpCU5gl8x1qJFbb6hbZxR7XrcE2vtbAsTAbJ7/xwJtUuJEw8K8Zr/AE0LHlesg==
+  dependencies:
+    "@babel/runtime" "^7.12.5"
+    cosmiconfig "^7.0.0"
+    resolve "^1.19.0"
+
 babel-preset-jest@^24.9.0:
   version "24.9.0"
   resolved "https://registry.yarnpkg.com/babel-preset-jest/-/babel-preset-jest-24.9.0.tgz#192b521e2217fb1d1f67cf73f70c336650ad3cdc"
@@ -3485,6 +3519,13 @@ is-ci@2.0.0, is-ci@^2.0.0:
   dependencies:
     ci-info "^2.0.0"
 
+is-core-module@^2.8.1:
+  version "2.8.1"
+  resolved "https://registry.npmjs.org/is-core-module/-/is-core-module-2.8.1.tgz#f59fdfca701d5879d0a6b100a40aa1560ce27211"
+  integrity sha512-SdNCUs284hr40hFTFP6l0IfZ/RSrMXF3qgoRHd3/79unUTvrFO/JoXwkGm+5J/Oe3E/b5GsnG330uUNgRpu1PA==
+  dependencies:
+    has "^1.0.3"
+
 is-data-descriptor@^0.1.4:
   version "0.1.4"
   resolved "https://registry.yarnpkg.com/is-data-descriptor/-/is-data-descriptor-0.1.4.tgz#0b5ee648388e2c860282e793f1856fec3f301b56"
@@ -5145,6 +5186,11 @@ path-parse@^1.0.6:
   resolved "https://registry.yarnpkg.com/path-parse/-/path-parse-1.0.6.tgz#d62dbb5679405d72c4737ec58600e9ddcf06d24c"
   integrity sha512-GSmOT2EbHrINBf9SR7CDELwlJ8AENk3Qn7OikK4nFYAu3Ote2+JYNVvkpAEQm3/TLNEJFD/xZJjzyxg3KBWOzw==
 
+path-parse@^1.0.7:
+  version "1.0.7"
+  resolved "https://registry.npmjs.org/path-parse/-/path-parse-1.0.7.tgz#fbc114b60ca42b30d9daf5858e4bd68bbedb6735"
+  integrity sha512-LDJzPVEEEPR+y48z93A0Ed0yXb8pAByGWo/k5YYdYgpY2/2EsOsksJrq7lOHxryrVOn1ejG6oAp8ahvOIQD8sw==
+
 path-type@^1.0.0:
   version "1.1.0"
   resolved "https://registry.yarnpkg.com/path-type/-/path-type-1.1.0.tgz#59c44f7ee491da704da415da5a4070ba4f8fe441"
@@ -5706,6 +5752,15 @@ resolve@^1.1.6, resolve@^1.10.0, resolve@^1.3.2:
   dependencies:
     path-parse "^1.0.6"
 
+resolve@^1.19.0:
+  version "1.22.0"
+  resolved "https://registry.npmjs.org/resolve/-/resolve-1.22.0.tgz#5e0b8c67c15df57a89bdbabe603a002f21731198"
+  integrity sha512-Hhtrw0nLeSrFQ7phPp4OOcVjLPIeMnRlr5mcnVuMe7M/7eBn98A3hmFRLoFo3DLZkivSYwhRUJTyPyWAk56WLw==
+  dependencies:
+    is-core-module "^2.8.1"
+    path-parse "^1.0.7"
+    supports-preserve-symlinks-flag "^1.0.0"
+
 responselike@^1.0.2:
   version "1.0.2"
   resolved "https://registry.yarnpkg.com/responselike/-/responselike-1.0.2.tgz#918720ef3b631c5642be068f15ade5a46f4ba1e7"
@@ -6229,6 +6284,11 @@ supports-color@^6.1.0:
   dependencies:
     has-flag "^3.0.0"
 
+supports-preserve-symlinks-flag@^1.0.0:
+  version "1.0.0"
+  resolved "https://registry.npmjs.org/supports-preserve-symlinks-flag/-/supports-preserve-symlinks-flag-1.0.0.tgz#6eda4bd344a3c94aea376d4cc31bc77311039e09"
+  integrity sha512-ot0WnXS9fgdkgIcePe6RHNk1WA8+muPa6cSjeR3V8K27q9BB1rTE3R1p7Hv0z1ZyAc8s6Vvv8DIyWf681MAt0w==
+
 symbol-tree@^3.2.2:
   version "3.2.4"
   resolved "https://registry.yarnpkg.com/symbol-tree/-/symbol-tree-3.2.4.tgz#430637d248ba77e078883951fb9aa0eed7c63fa2"