refactor: move @babel/helper-get-function-arity to ts (babel#12414)

* refactor: move @babel/helper-get-function-arity to ts * Review comments * chore: add flow interface
nicolo-ribaudo · Dec 2, 2020 · c13853b · c13853b
1 parent a26f4ef
commit c13853b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/lib/babel-packages.js.flow b/lib/babel-packages.js.flow
@@ -43,3 +43,9 @@ declare module "@babel/helper-optimise-call-expression" {
     optional: boolean
   ): BabelNodeCallExpression | BabelNodeOptionalCallExpression;
 }
+
+declare module "@babel/helper-get-function-arity" {
+  declare export default function helperGetFunctionArity(
+    node: BabelNodeFunction
+  ): number;
+}
diff --git a/...el-helper-get-function-arity/src/index.js → ...el-helper-get-function-arity/src/index.ts b/...el-helper-get-function-arity/src/index.js → ...el-helper-get-function-arity/src/index.ts
@@ -1,7 +1,7 @@
 import * as t from "@babel/types";
 
-export default function (node): number {
-  const params: Array<Object> = node.params;
+export default function (node: t.Function): number {
+  const params = node.params;
   for (let i = 0; i < params.length; i++) {
     const param = params[i];
     if (t.isAssignmentPattern(param) || t.isRestElement(param)) {