url.hpp

/*
 * Copyright (c) 2025, NVIDIA CORPORATION.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#pragma once
 
#include <optional>
#include <string>
 
#include <curl/curl.h>
 
namespace kvikio::detail {
class CurlUrlHandle {
 private:
  CURLU* _handle{nullptr};
 
 public:
  CurlUrlHandle();
 
  ~CurlUrlHandle() noexcept;
 
  CurlUrlHandle(CurlUrlHandle const&)            = delete;
  CurlUrlHandle& operator=(CurlUrlHandle const&) = delete;
 
  CurlUrlHandle(CurlUrlHandle&& other) noexcept;
  CurlUrlHandle& operator=(CurlUrlHandle&& other) noexcept;
 
  CURLU* get() const;
};
 
class UrlParser {
 public:
  struct UrlComponents {
    std::optional<std::string> scheme;
 
    std::optional<std::string> host;
 
    std::optional<std::string> port;
 
    std::optional<std::string> path;
 
    std::optional<std::string> query;
 
    std::optional<std::string> fragment;
  };
 
  static UrlComponents parse(std::string const& url,
                             std::optional<unsigned int> bitmask_url_flags       = std::nullopt,
                             std::optional<unsigned int> bitmask_component_flags = std::nullopt);
 
  static std::optional<std::string> extract_component(
    CurlUrlHandle const& handle,
    CURLUPart part,
    std::optional<unsigned int> bitmask_component_flags = std::nullopt,
    std::optional<CURLUcode> allowed_err_code           = std::nullopt);
 
  static std::optional<std::string> extract_component(
    std::string const& url,
    CURLUPart part,
    std::optional<unsigned int> bitmask_url_flags       = std::nullopt,
    std::optional<unsigned int> bitmask_component_flags = std::nullopt,
    std::optional<CURLUcode> allowed_err_code           = std::nullopt);
};
}  // namespace kvikio::detail