94 lines
2.8 KiB
C++
94 lines
2.8 KiB
C++
// Copyright (c) 2023, AgiBot Inc.
|
||
// All rights reserved.
|
||
|
||
#pragma once
|
||
|
||
#include <optional>
|
||
#include <regex>
|
||
#include <sstream>
|
||
#include <string>
|
||
#include <string_view>
|
||
|
||
namespace aimrt::common::util {
|
||
|
||
/**
|
||
* @brief url元素
|
||
*
|
||
* @tparam StringType
|
||
*/
|
||
template <class StringType = std::string_view>
|
||
requires(std::is_same_v<StringType, std::string_view> ||
|
||
std::is_same_v<StringType, std::string>)
|
||
struct Url {
|
||
StringType protocol; // 协议
|
||
StringType host; // host
|
||
StringType service; // 端口
|
||
StringType path; // 路径
|
||
StringType query; // 参数
|
||
StringType fragment; // 额外信息
|
||
};
|
||
|
||
/**
|
||
* @brief url解析
|
||
* @note url结构：[protocol://][host][:service][path][?query][#fragment]
|
||
* @param url_str url字符串
|
||
* @return std::optional<UrlView> url结构，nullopt则代表解析失败
|
||
*/
|
||
template <class StringType = std::string_view>
|
||
std::optional<Url<StringType>> ParseUrl(std::string_view url_str) {
|
||
std::regex url_regex(
|
||
R"(^(([^:\/?#]+)://)?(([^\/?#]*))?([^?#]*)(\?([^#]*))?(#(.*))?)",
|
||
std::regex::ECMAScript);
|
||
std::match_results<std::string_view::const_iterator> url_match_result;
|
||
|
||
if (!std::regex_match(url_str.begin(), url_str.end(), url_match_result, url_regex))
|
||
return std::nullopt;
|
||
|
||
Url<StringType> url;
|
||
if (url_match_result[2].matched)
|
||
url.protocol =
|
||
StringType(url_match_result[2].first, url_match_result[2].second);
|
||
if (url_match_result[4].matched) {
|
||
StringType auth(url_match_result[4].first, url_match_result[4].second);
|
||
size_t pos = auth.find_first_of(':');
|
||
if (pos != StringType::npos) {
|
||
url.host = auth.substr(0, pos);
|
||
url.service = auth.substr(pos + 1);
|
||
} else {
|
||
url.host = auth;
|
||
}
|
||
}
|
||
if (url_match_result[5].matched)
|
||
url.path = StringType(url_match_result[5].first, url_match_result[5].second);
|
||
if (url_match_result[7].matched)
|
||
url.query = StringType(url_match_result[7].first, url_match_result[7].second);
|
||
if (url_match_result[9].matched)
|
||
url.fragment = StringType(url_match_result[9].first, url_match_result[9].second);
|
||
|
||
return std::optional<Url<StringType>>{url};
|
||
}
|
||
|
||
/**
|
||
* @brief url拼接
|
||
* @note url结构：[protocol://][host][:service][path][?query][#fragment]
|
||
* @param url url结构
|
||
* @return std::string url字符串
|
||
*/
|
||
template <class StringType = std::string_view>
|
||
std::string JoinUrl(const Url<StringType>& url) {
|
||
std::stringstream ss;
|
||
|
||
if (!url.protocol.empty()) ss << url.protocol << "://";
|
||
if (!url.host.empty()) ss << url.host;
|
||
if (!url.service.empty()) ss << ":" << url.service;
|
||
if (!url.path.empty()) {
|
||
if (url.path[0] != '/') ss << '/';
|
||
ss << url.path;
|
||
}
|
||
if (!url.query.empty()) ss << '?' << url.query;
|
||
if (!url.fragment.empty()) ss << '#' << url.fragment;
|
||
|
||
return ss.str();
|
||
}
|
||
|
||
} // namespace aimrt::common::util
|