1 files changed, 68 insertions, 16 deletions
diff --git a/src/IsCharacterInSet.cpp b/src/IsCharacterInSet.cpp
index 82625e9..1b2882f 100644
--- a/src/IsCharacterInSet.cpp
+++ b/src/IsCharacterInSet.cpp
@@ -9,27 +9,79 @@
 
 #include "IsCharacterInSet.hpp"
 
+#include <set>
+
 namespace Uri {
 
-    bool IsCharacterInSet(
-        char c,
-        std::initializer_list< char > characterSet
-    ) {
+    /**
+     * This contains the private properties of the CharacterSet class.
+     */
+    struct CharacterSet::Impl {
+        /**
+         * This holds the characters in the set.
+         */
+        std::set< char > charactersInSet;
+    };
+
+    CharacterSet::~CharacterSet() = default;
+    CharacterSet::CharacterSet(const CharacterSet& other)
+        : impl_(new Impl(*other.impl_))
+    {
+    }
+    CharacterSet::CharacterSet(CharacterSet&& other) = default;
+    CharacterSet& CharacterSet::operator=(const CharacterSet& other) {
+        if (this != &other) {
+            *impl_ = *other.impl_;
+        }
+        return *this;
+    }
+    CharacterSet& CharacterSet::operator=(CharacterSet&& other) = default;
+
+    CharacterSet::CharacterSet()
+        : impl_(new Impl)
+    {
+    }
+
+    CharacterSet::CharacterSet(char c)
+        : impl_(new Impl)
+    {
+        (void)impl_->charactersInSet.insert(c);
+    }
+
+    CharacterSet::CharacterSet(char first, char last)
+        : impl_(new Impl)
+    {
+        for (char c = first; c < last + 1; ++c) {
+            (void)impl_->charactersInSet.insert(c);
+        }
+    }
+
+    CharacterSet::CharacterSet(
+        std::initializer_list< const CharacterSet > characterSets
+    )
+        : impl_(new Impl)
+    {
         for (
-            auto charInSet = characterSet.begin();
-            charInSet != characterSet.end();
-            ++charInSet
+            auto characterSet = characterSets.begin();
+            characterSet != characterSets.end();
+            ++characterSet
         ) {
-            const auto first = *charInSet++;
-            const auto last = *charInSet;
-            if (
-                (c >= first)
-                && (c <= last)
-            ) {
-                return true;
-            }
+            impl_->charactersInSet.insert(
+                characterSet->impl_->charactersInSet.begin(),
+                characterSet->impl_->charactersInSet.end()
+            );
         }
-        return false;
+    }
+
+    bool CharacterSet::Contains(char c) const {
+        return impl_->charactersInSet.find(c) != impl_->charactersInSet.end();
+    }
+
+    bool IsCharacterInSet(
+        char c,
+        const CharacterSet& characterSet
+    ) {
+        return characterSet.Contains(c);
     }
 
 }