From 0a3c097339d3448f8dcab9aa3c0a9b1345e5cebd Mon Sep 17 00:00:00 2001
From: SoniEx2 <endermoneymod@gmail.com>
Date: Tue, 23 Mar 2021 12:08:50 -0300
Subject: Fix bug preventing multiple impl trait in a row

---
 Cargo.toml                  |  2 +-
 src/lib.rs                  |  1 +
 tests/09-multiple-traits.rs | 26 ++++++++++++++++++++++++++
 tests/progress.rs           |  1 +
 4 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 tests/09-multiple-traits.rs

diff --git a/Cargo.toml b/Cargo.toml
index 7a1091f..dfc22ee 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "impl_trait"
-version = "0.1.0"
+version = "0.1.1"
 authors = ["SoniEx2 <endermoneymod@gmail.com>"]
 edition = "2018"
 description = "Allows impl trait inside inherent impl."
diff --git a/src/lib.rs b/src/lib.rs
index 061345d..d8e5a15 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -294,6 +294,7 @@ pub fn impl_trait(item: TokenStream) -> TokenStream {
                         trait_span = Some(tt.span());
                         in_generic = true;
                         in_path = true;
+                        in_impl = false;
                         continue;
                     }
                 },
diff --git a/tests/09-multiple-traits.rs b/tests/09-multiple-traits.rs
new file mode 100644
index 0000000..d75ab31
--- /dev/null
+++ b/tests/09-multiple-traits.rs
@@ -0,0 +1,26 @@
+// Checks that multiple `impl trait` in a row are valid.
+
+use impl_trait::impl_trait;
+
+struct Foo;
+trait Bar {
+}
+trait Baz {
+}
+
+impl_trait! {
+    impl Foo {
+        impl trait Bar {
+        }
+        impl trait Baz {
+        }
+    }
+}
+
+fn static_assert_1<T: Bar>(_t: T) {}
+fn static_assert_2<T: Baz>(_t: T) {}
+
+fn main() {
+    static_assert_1(Foo);
+    static_assert_2(Foo);
+}
diff --git a/tests/progress.rs b/tests/progress.rs
index aa42baa..22110b0 100644
--- a/tests/progress.rs
+++ b/tests/progress.rs
@@ -9,6 +9,7 @@ fn tests() {
     t.pass("tests/06-impl-trait.rs");
     t.pass("tests/07-docs.rs");
     t.pass("tests/08-inherent-docs.rs");
+    t.pass("tests/09-multiple-traits.rs");
     t.pass("tests/98-readme.rs");
     t.pass("tests/99-goal.rs");
 }
-- 
cgit 1.4.1