From d941d2917fa4649a655ea5d2c2b524ea55de548f Mon Sep 17 00:00:00 2001
From: Richard Zak <richard.j.zak@gmail.com>
Date: Sun, 3 Mar 2024 18:01:30 -0500
Subject: [PATCH] feat: implement `Extend` for `FlagSet`, closes #21

Signed-off-by: Richard Zak <richard.j.zak@gmail.com>
---
 src/lib.rs | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 3751852..d9680eb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -772,6 +772,35 @@ impl<F: Flags, R: Into<FlagSet<F>>> RemAssign<R> for FlagSet<F> {
     }
 }
 
+impl<F: Flags, R: Into<FlagSet<F>>> Extend<R> for FlagSet<F> {
+    /// Add values by iterating over some collection.
+    ///
+    /// ```
+    /// use flagset::{FlagSet, flags};
+    ///
+    /// flags! {
+    ///     pub enum Flag: u8 {
+    ///         Foo = 1,
+    ///         Bar = 2,
+    ///         Baz = 4
+    ///     }
+    /// }
+    ///
+    /// let flag_vec = vec![Flag::Bar, Flag::Baz];
+    /// let mut some_extended_flags = FlagSet::from(Flag::Foo);
+    /// some_extended_flags.extend(flag_vec);
+    /// assert_eq!(some_extended_flags, Flag::Foo | Flag::Bar | Flag::Baz);
+    /// ```
+    fn extend<T>(&mut self, iter: T)
+    where
+        T: IntoIterator<Item = R>,
+    {
+        for item in iter {
+            *self |= item;
+        }
+    }
+}
+
 impl<F: Flags> FlagSet<F> {
     /// Creates a new set from bits; returning `Err(InvalidBits)` on invalid/unknown bits.
     ///