From 289c533aadb4ee5d6a7ef2a740c4d38cd7b6cc59 Mon Sep 17 00:00:00 2001
From: Cameron Martin <cameron_martin@shelter.org.uk>
Date: Tue, 27 Nov 2018 17:27:21 +0000
Subject: [PATCH] [@babel/traverse]: Fixed type of path when using enter & exit
 visitor.

---
 types/babel__traverse/babel__traverse-tests.ts | 8 ++++++--
 types/babel__traverse/index.d.ts               | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/types/babel__traverse/babel__traverse-tests.ts b/types/babel__traverse/babel__traverse-tests.ts
index 5b4fbbef5f..974831dbca 100644
--- a/types/babel__traverse/babel__traverse-tests.ts
+++ b/types/babel__traverse/babel__traverse-tests.ts
@@ -4,10 +4,14 @@ import * as t from "@babel/types";
 // Examples from: https://github.com/thejameskyle/babel-handbook/blob/master/translations/en/plugin-handbook.md
 const MyVisitor: Visitor = {
     Identifier: {
-        enter() {
+        enter(path) {
+            // $ExpectType NodePath<Identifier>
+            path;
             console.log("Entered!");
         },
-        exit() {
+        exit(path) {
+            // $ExpectType NodePath<Identifier>
+            path;
             console.log("Exited!");
         }
     }
diff --git a/types/babel__traverse/index.d.ts b/types/babel__traverse/index.d.ts
index 465a60464f..740f5f4429 100644
--- a/types/babel__traverse/index.d.ts
+++ b/types/babel__traverse/index.d.ts
@@ -142,7 +142,7 @@ export type Visitor<S = Node> = VisitNodeObject<Node> & {
     [P in Node["type"]]?: VisitNode<S, Extract<Node, { type: P; }>>;
 };
 
-export type VisitNode<T, P> = VisitNodeFunction<T, P> | VisitNodeObject<T>;
+export type VisitNode<T, P> = VisitNodeFunction<T, P> | VisitNodeObject<P>;
 
 export type VisitNodeFunction<T, P> = (this: T, path: NodePath<P>, state: any) => void;