1 files changed, 168 insertions, 0 deletions
diff --git a/Source/WebCore/html/URLSearchParams.cpp b/Source/WebCore/html/URLSearchParams.cpp
new file mode 100644
index 000000000..ddd873cda
--- /dev/null
+++ b/Source/WebCore/html/URLSearchParams.cpp
@@ -0,0 +1,168 @@
+/*
+ * Copyright (C) 2016 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
+ * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "config.h"
+#include "URLSearchParams.h"
+
+#include "DOMURL.h"
+#include "ExceptionCode.h"
+#include "URLParser.h"
+
+namespace WebCore {
+
+URLSearchParams::URLSearchParams(const String& init, DOMURL* associatedURL)
+    : m_associatedURL(associatedURL)
+    , m_pairs(init.startsWith('?') ? URLParser::parseURLEncodedForm(StringView(init).substring(1)) : URLParser::parseURLEncodedForm(init))
+{
+}
+
+URLSearchParams::URLSearchParams(const Vector<WTF::KeyValuePair<String, String>>& pairs)
+    : m_pairs(pairs)
+{
+}
+
+ExceptionOr<Ref<URLSearchParams>> URLSearchParams::create(Variant<Vector<Vector<String>>, Vector<WTF::KeyValuePair<String, String>>, String>&& variant)
+{
+    auto visitor = WTF::makeVisitor([&](const Vector<Vector<String>>& vector) -> ExceptionOr<Ref<URLSearchParams>> {
+        Vector<WTF::KeyValuePair<String, String>> pairs;
+        for (const auto& pair : vector) {
+            if (pair.size() != 2)
+                return Exception { TypeError };
+            pairs.append({pair[0], pair[1]});
+        }
+        return adoptRef(*new URLSearchParams(WTFMove(pairs)));
+    }, [&](const Vector<WTF::KeyValuePair<String, String>>& pairs) {
+        return adoptRef(*new URLSearchParams(pairs));
+    }, [&](const String& string) {
+        return adoptRef(*new URLSearchParams(string, nullptr));
+    });
+    return WTF::visit(visitor, variant);
+}
+
+String URLSearchParams::get(const String& name) const
+{
+    for (const auto& pair : m_pairs) {
+        if (pair.key == name)
+            return pair.value;
+    }
+    return String();
+}
+
+bool URLSearchParams::has(const String& name) const
+{
+    for (const auto& pair : m_pairs) {
+        if (pair.key == name)
+            return true;
+    }
+    return false;
+}
+
+void URLSearchParams::sort()
+{
+    std::stable_sort(m_pairs.begin(), m_pairs.end(), [] (const auto& a, const auto& b) {
+        return WTF::codePointCompareLessThan(a.key, b.key);
+    });
+    updateURL();
+}
+
+void URLSearchParams::set(const String& name, const String& value)
+{
+    for (auto& pair : m_pairs) {
+        if (pair.key != name)
+            continue;
+        if (pair.value != value)
+            pair.value = value;
+        bool skippedFirstMatch = false;
+        m_pairs.removeAllMatching([&] (const auto& pair) {
+            if (pair.key == name) {
+                if (skippedFirstMatch)
+                    return true;
+                skippedFirstMatch = true;
+            }
+            return false;
+        });
+        updateURL();
+        return;
+    }
+    m_pairs.append({name, value});
+    updateURL();
+}
+
+void URLSearchParams::append(const String& name, const String& value)
+{
+    m_pairs.append({name, value});
+    updateURL();
+}
+
+Vector<String> URLSearchParams::getAll(const String& name) const
+{
+    Vector<String> values;
+    values.reserveInitialCapacity(m_pairs.size());
+    for (const auto& pair : m_pairs) {
+        if (pair.key == name)
+            values.uncheckedAppend(pair.value);
+    }
+    return values;
+}
+
+void URLSearchParams::remove(const String& name)
+{
+    if (m_pairs.removeAllMatching([&] (const auto& pair) { return pair.key == name; }))
+        updateURL();
+}
+
+String URLSearchParams::toString() const
+{
+    return URLParser::serialize(m_pairs);
+}
+
+void URLSearchParams::updateURL()
+{
+    if (m_associatedURL)
+        m_associatedURL->setQuery(URLParser::serialize(m_pairs));
+}
+
+void URLSearchParams::updateFromAssociatedURL()
+{
+    ASSERT(m_associatedURL);
+    String search = m_associatedURL->search();
+    m_pairs = search.startsWith('?') ? URLParser::parseURLEncodedForm(StringView(search).substring(1)) : URLParser::parseURLEncodedForm(search);
+}
+
+std::optional<WTF::KeyValuePair<String, String>> URLSearchParams::Iterator::next()
+{
+    auto& pairs = m_target->pairs();
+    if (m_index >= pairs.size())
+        return std::nullopt;
+
+    auto& pair = pairs[m_index++];
+    return WTF::KeyValuePair<String, String> { pair.key, pair.value };
+}
+
+URLSearchParams::Iterator::Iterator(URLSearchParams& params)
+    : m_target(params)
+{
+}
+    
+}