feat: add url_search_params to ada

anonrig · anonrig · commit 62d73ff35f6d · 2023-06-26T16:34:44.000-04:00
diff --git a/include/ada.h b/include/ada.h
@@ -23,6 +23,8 @@
 #include "ada/url_components.h"
 #include "ada/url_aggregator.h"
 #include "ada/url_aggregator-inl.h"
+#include "ada/url_search_params.h"
+#include "ada/url_search_params-inl.h"
 
 // Public API
 #include "ada/ada_version.h"
diff --git a/include/ada/url_search_params-inl.h b/include/ada/url_search_params-inl.h
@@ -0,0 +1,63 @@
+/**
+ * @file url_search_params-inl.h
+ * @brief Inline declarations for the URL Search Params
+ */
+#ifndef ADA_URL_SEARCH_PARAMS_INL_H
+#define ADA_URL_SEARCH_PARAMS_INL_H
+
+#include "ada.h"
+#include "ada/url_search_params.h"
+
+#include <algorithm>
+#include <optional>
+#include <string>
+#include <string_view>
+#include <vector>
+
+namespace ada {
+
+inline void url_search_params::append(const std::string_view key,
+                                      const std::string_view value) {
+  params.emplace_back(std::string(key), std::string(value));
+}
+
+inline size_t url_search_params::get_size() const noexcept {
+  return params.size();
+}
+
+inline std::optional<std::string_view> url_search_params::get(
+    const std::string_view key) {
+  auto entry = std::find_if(params.begin(), params.end(), [&key](auto param) {
+    return std::get<0>(param) == key;
+  });
+
+  if (entry == params.end()) {
+    return std::nullopt;
+  }
+
+  return std::get<1>(*entry);
+}
+
+inline std::vector<std::string> url_search_params::get_all(
+    const std::string_view key) {
+  std::vector<std::string> out{};
+
+  for (auto& param : params) {
+    if (std::get<0>(param) == key) {
+      out.emplace_back(std::get<1>(param));
+    }
+  }
+
+  return out;
+}
+
+inline bool url_search_params::has(const std::string_view key) noexcept {
+  auto entry = std::find_if(params.begin(), params.end(), [&key](auto param) {
+    return std::get<0>(param) == key;
+  });
+  return entry != params.end();
+}
+
+}  // namespace ada
+
+#endif  // ADA_URL_SEARCH_PARAMS_INL_H
diff --git a/include/ada/url_search_params.h b/include/ada/url_search_params.h
@@ -0,0 +1,72 @@
+/**
+ * @file url_search_params.h
+ * @brief Declaration for the URL Search Params
+ */
+#ifndef ADA_URL_SEARCH_PARAMS_H
+#define ADA_URL_SEARCH_PARAMS_H
+
+namespace ada {
+
+#include <optional>
+#include <string>
+#include <string_view>
+#include <vector>
+
+/**
+ * @see https://url.spec.whatwg.org/#interface-urlsearchparams
+ */
+struct url_search_params {
+  url_search_params() = default;
+  url_search_params(const url_search_params &u) = default;
+  url_search_params(url_search_params &&u) noexcept = default;
+  url_search_params &operator=(url_search_params &&u) noexcept = default;
+  url_search_params &operator=(const url_search_params &u) = default;
+  ~url_search_params() = default;
+
+  [[nodiscard]] inline size_t get_size() const noexcept;
+
+  /**
+   * @see https://url.spec.whatwg.org/#dom-urlsearchparams-append
+   */
+  inline void append(std::string_view key, std::string_view value);
+
+  /**
+   * @see https://url.spec.whatwg.org/#dom-urlsearchparams-delete
+   */
+  inline void remove(std::string_view key);
+  inline void remove(std::string_view key, std::string_view value);
+
+  /**
+   * @see https://url.spec.whatwg.org/#dom-urlsearchparams-get
+   */
+  inline std::optional<std::string_view> get(std::string_view key);
+
+  /**
+   * @see https://url.spec.whatwg.org/#dom-urlsearchparams-getall
+   */
+  inline std::vector<std::string> get_all(std::string_view key);
+
+  /**
+   * @see https://url.spec.whatwg.org/#dom-urlsearchparams-has
+   */
+  inline bool has(std::string_view key) noexcept;
+
+  /**
+   * @see https://url.spec.whatwg.org/#dom-urlsearchparams-set
+   */
+  inline void set(std::string_view key, std::string_view value);
+
+  /**
+   * @see https://url.spec.whatwg.org/#dom-urlsearchparams-sort
+   */
+  inline void sort() const noexcept;
+
+  inline std::string to_string();
+
+ private:
+  uint32_t size{0};
+  std::vector<std::tuple<std::string, std::string>> params{};
+};  // url_search_params
+
+}  // namespace ada
+#endif
diff --git a/src/ada.cpp b/src/ada.cpp
@@ -11,3 +11,4 @@
 #include "url_components.cpp"
 #include "url_aggregator.cpp"
 #include "ada_c.cpp"
+#include "url_search_params.cpp"
diff --git a/src/url_search_params.cpp b/src/url_search_params.cpp
@@ -0,0 +1,62 @@
+#include "ada.h"
+#include "ada/url_search_params.h"
+
+#include <optional>
+#include <string>
+#include <string_view>
+#include <vector>
+
+namespace ada {
+
+inline void url_search_params::remove(const std::string_view key) {
+  params.erase(
+      std::remove_if(params.begin(), params.end(),
+                     [&key](auto param) { return std::get<0>(param) == key; }),
+      params.end());
+}
+
+inline void url_search_params::remove(const std::string_view key,
+                                      std::string_view value) {
+  params.erase(std::remove_if(params.begin(), params.end(),
+                              [&key, &value](auto param) {
+                                return std::get<0>(param) == key &&
+                                       std::get<1>(param) == value;
+                              }),
+               params.end());
+}
+
+inline void url_search_params::sort() const noexcept {
+  // TODO: Implement this
+}
+
+inline void url_search_params::set(const std::string_view key,
+                                   const std::string_view value) {
+  params.erase(
+      std::remove_if(params.begin(), params.end(),
+                     [&key](auto param) { return std::get<0>(param) == key; }),
+      params.end());
+
+  params.emplace_back(std::string(key), std::string(value));
+}
+
+inline std::string url_search_params::to_string() {
+  if (params.empty()) {
+    return "";
+  }
+  std::string out = "?";
+  for (size_t i = 0; i < params.size(); i++) {
+    auto param = params[i];
+    auto key = std::get<0>(param);
+    auto value = std::get<1>(param);
+
+    if (i != 0) {
+      out += "&";
+    }
+    out.append(key);
+    out += "=";
+    out.append(value);
+  }
+  return out;
+}
+
+}  // namespace ada
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
@@ -15,25 +15,29 @@ add_executable(url_components url_components.cpp)
 add_executable(basic_tests basic_tests.cpp)
 add_executable(from_file_tests from_file_tests.cpp)
 add_executable(ada_c ada_c.cpp)
+add_executable(url_search_params url_search_params.cpp)
 
 target_link_libraries(wpt_tests PRIVATE simdjson GTest::gtest_main)
 target_link_libraries(url_components PRIVATE simdjson GTest::gtest_main)
 target_link_libraries(basic_tests PRIVATE simdjson GTest::gtest_main)
 target_link_libraries(from_file_tests PRIVATE simdjson GTest::gtest_main)
 target_link_libraries(ada_c PRIVATE simdjson GTest::gtest_main)
+target_link_libraries(url_search_params PRIVATE simdjson GTest::gtest_main)
 
 gtest_discover_tests(wpt_tests)
 gtest_discover_tests(url_components)
 gtest_discover_tests(basic_tests)
 gtest_discover_tests(from_file_tests)
 gtest_discover_tests(ada_c)
+gtest_discover_tests(url_search_params)
 
 add_cpp_test(basic_fuzzer)
 
 if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU")
   if (CMAKE_CXX_COMPILER_VERSION VERSION_LESS 9)
     target_link_libraries(wpt_tests PUBLIC stdc++fs)
     target_link_libraries(url_components PUBLIC stdc++fs)
+    target_link_libraries(url_search_params PUBLIC stdc++fs)
   endif()
 endif()
 
@@ -43,4 +47,5 @@ if(MSVC OR MINGW)
   target_compile_definitions(basic_fuzzer PRIVATE _CRT_SECURE_NO_WARNINGS)
   target_compile_definitions(from_file_tests PRIVATE _CRT_SECURE_NO_WARNINGS)
   target_compile_definitions(basic_tests PRIVATE _CRT_SECURE_NO_WARNINGS)
+  target_compile_definitions(url_search_params PRIVATE _CRT_SECURE_NO_WARNINGS)
 endif()
diff --git a/tests/url_components.cpp b/tests/url_components.cpp
@@ -2,7 +2,6 @@
 #include <filesystem>
 #include <iostream>
 #include <memory>
-#include <map>
 #include <set>
 
 #include "simdjson.h"
diff --git a/tests/url_search_params.cpp b/tests/url_search_params.cpp
@@ -0,0 +1,13 @@
+#include "ada.h"
+#include "gtest/gtest.h"
+
+TEST(url_search_params, append) {
+  auto search_params = ada::url_search_params();
+  search_params.append("key", "value");
+  ASSERT_EQ(search_params.get_size(), 1);
+  ASSERT_TRUE(search_params.has("key"));
+  search_params.append("key", "value2");
+  ASSERT_EQ(search_params.get_size(), 2);
+  ASSERT_EQ(search_params.get_all("key").size(), 2);
+  SUCCEED();
+}
