239 lines
7.9 KiB
C++
239 lines
7.9 KiB
C++
#ifndef CCString_H
|
||
#define CCString_H
|
||
|
||
#include <sstream>
|
||
#include <cstdio>
|
||
#include <string>
|
||
#include "cstdarg"
|
||
#include "CCByteArray.h"
|
||
#include <locale>
|
||
#include <codecvt>
|
||
#include <sstream>
|
||
|
||
#ifdef __linux__
|
||
#include <iconv.h>
|
||
#endif
|
||
|
||
#if __OHOS__
|
||
#pragma warning(push)
|
||
#pragma warning(disable: -Wformat-security)
|
||
// 或者对于 GCC/Clang
|
||
#pragma GCC diagnostic push
|
||
#pragma GCC diagnostic ignored "-Wformat-security"
|
||
#endif
|
||
|
||
using CCStream = std::ostringstream;
|
||
using CCSStream = std::stringstream;
|
||
using WString = std::wstring;
|
||
namespace CTL {
|
||
class CCWString;
|
||
|
||
/**
|
||
* @brief 字符串类，继承自std::string，提供额外的功能
|
||
*/
|
||
class String : public std::string{
|
||
template<typename T>
|
||
static void format_one_arg(std::string& result, T&& value) {
|
||
size_t pos = result.find("{}");
|
||
if (pos != std::string::npos) {
|
||
std::string replacement;
|
||
if constexpr (std::is_same_v<std::decay_t<T>, const char*> ||
|
||
std::is_same_v<std::decay_t<T>, char*>) {
|
||
replacement = value ? value : "null";
|
||
}
|
||
else if constexpr (std::is_arithmetic_v<std::decay_t<T>>) {
|
||
replacement = std::to_string(value);
|
||
}
|
||
else {
|
||
replacement = std::to_string(value); // 假设对象有 to_string 方法
|
||
}
|
||
result.replace(pos, 2, replacement);
|
||
}
|
||
}
|
||
static std::string format_braces_template(std::string fmt_str) {
|
||
return fmt_str;
|
||
}
|
||
template<typename T, typename... Rest>
|
||
static std::string format_braces_template(std::string fmt_str, T&& first, Rest&&... rest) {
|
||
format_one_arg(fmt_str, std::forward<T>(first));
|
||
if constexpr (sizeof...(rest) > 0) {
|
||
return format_braces_template(fmt_str, std::forward<Rest>(rest)...);
|
||
}
|
||
return fmt_str;
|
||
}
|
||
template<typename T>
|
||
static std::string format_braces_template(std::string fmt_str, T&& first) {
|
||
format_one_arg(fmt_str, std::forward<T>(first));
|
||
return fmt_str;
|
||
}
|
||
public:
|
||
// 默认构造函数
|
||
String() = default;
|
||
template<typename... Args>
|
||
String(const char* fmt, Args... args) {
|
||
const auto str = format(fmt,args...);
|
||
this->assign(str);
|
||
}
|
||
String(const std::initializer_list<std::string> list){
|
||
for (auto& i : list) {
|
||
this->append(i);
|
||
}
|
||
}
|
||
explicit String(const CCSStream& stream);
|
||
// 继承std::string的构造函数
|
||
using std::string::string;
|
||
// 转换字符串为字节
|
||
[[nodiscard]] Byte* toBytes() const;
|
||
[[nodiscard]] Byte* toBytes();
|
||
// 从basic_string构造String对象
|
||
String(const basic_string &basicString);
|
||
// 格式化字符串
|
||
String Format(const char * standardCharsets) const;
|
||
String Format(const char * standardCharsets);
|
||
// 指定大小构造字符串
|
||
String(size_t Size);
|
||
// 字符串转换为整数
|
||
int to_int() const;
|
||
float to_float() const;
|
||
double to_double() const;
|
||
// 字符串转换为整数（非const版本）
|
||
int to_int();
|
||
float to_float();
|
||
double to_double();
|
||
/**
|
||
* @brief 格式化字符串的静态方法
|
||
* @param fmt 格式化字符串的格式
|
||
* @param args 可变参数，与格式化字符串匹配
|
||
* @return 返回格式化后的String对象
|
||
*/
|
||
template<typename... Args>
|
||
static std::string format(const char* fmt, Args... args) {
|
||
std::string fmt_str = fmt;
|
||
if (fmt_str.find("{}") != std::string::npos) {
|
||
return format_braces_template(fmt_str, args...);
|
||
}
|
||
const int length = std::snprintf(nullptr, 0, fmt, args...);
|
||
if (length <= 0) {
|
||
return fmt_str; // 格式化失败，返回原始格式字符串
|
||
}
|
||
|
||
// 分配足够的空间来存储格式化后的字符串
|
||
std::string result(length, '\0');
|
||
std::snprintf(&result[0], length + 1, fmt, args...);
|
||
return result;
|
||
}
|
||
// 转换String对象为WString对象
|
||
[[nodiscard]] WString to_CCWString() const {
|
||
#ifdef _WIN32
|
||
std::wstring_convert<std::codecvt_utf8<wchar_t>> converter;
|
||
return converter.from_bytes(data());
|
||
#elif __linux__
|
||
iconv_t cd = iconv_open("UTF-8", "WCHAR_T");
|
||
if (cd == (iconv_t)-1) {
|
||
return L"";
|
||
}
|
||
size_t inBytes = size() * sizeof(wchar_t);
|
||
size_t outBytes = inBytes * 2;
|
||
char* inbuf = (char*)data();
|
||
char* outbuf = new char[outBytes];
|
||
char* outbufEnd = outbuf + outBytes;
|
||
size_t convertedBytes = iconv(cd, &inbuf, &inBytes, &outbuf, &outBytes);
|
||
iconv_close(cd);
|
||
if (convertedBytes == (size_t)-1) {
|
||
delete[] outbuf;
|
||
return L"";
|
||
}
|
||
auto* wcBuf = reinterpret_cast<wchar_t*>(outbuf);
|
||
size_t wcLength = (outbufEnd - outbuf) / sizeof(wchar_t);
|
||
WString result(wcBuf, wcLength); // 正确调用wchar_t数组构造
|
||
delete[] outbuf;
|
||
return result;
|
||
#endif
|
||
}
|
||
String& operator<<(const char* str) {
|
||
this->append(str);
|
||
return *this;
|
||
}
|
||
String& operator<<(const int value) {
|
||
this->append(std::to_string(value));
|
||
return *this;
|
||
}
|
||
String& operator<<(const float value) {
|
||
this->append(std::to_string(value));
|
||
return *this;
|
||
}
|
||
String& operator<<(const double value) {
|
||
this->append(std::to_string(value));
|
||
return *this;
|
||
}
|
||
String& operator<<(const String& other) {
|
||
this->append(other);
|
||
return *this;
|
||
}
|
||
};
|
||
|
||
/**
|
||
* @brief 宽字符串类，继承自std::wstring，提供额外的功能
|
||
*/
|
||
class CCWString:public WString
|
||
{
|
||
public:
|
||
// 默认构造函数
|
||
CCWString() = default;
|
||
// 从std::wstring构造CCWString对象
|
||
CCWString(const std::wstring& wstr) {
|
||
assign(wstr);
|
||
}
|
||
// 转换CCWString对象为String对象
|
||
[[nodiscard]] String to_CCString() const {
|
||
#ifdef _WIN32
|
||
std::wstring_convert<std::codecvt_utf8<wchar_t>> converter;
|
||
return converter.to_bytes(data());
|
||
#elif __linux__
|
||
iconv_t cd = iconv_open("UTF-8", "WCHAR_T"); // 修正转换方向
|
||
if (cd == (iconv_t)-1) return "";
|
||
|
||
size_t inBytes = size() * sizeof(wchar_t);
|
||
char* inbuf = (char*)data();
|
||
size_t outBytes = inBytes * 4; // 扩大输出缓冲区
|
||
char* outbuf = new char[outBytes];
|
||
char* outptr = outbuf;
|
||
|
||
if (iconv(cd, &inbuf, &inBytes, &outptr, &outBytes) == (size_t)-1) {
|
||
delete[] outbuf;
|
||
iconv_close(cd);
|
||
return "";
|
||
}
|
||
iconv_close(cd);
|
||
|
||
// 使用转换后的char数组构造String
|
||
String result(outbuf, outptr - outbuf);
|
||
delete[] outbuf;
|
||
return result;
|
||
#endif
|
||
}
|
||
};
|
||
|
||
}
|
||
|
||
inline CTL::String& operator<<(const char* lhs, CTL::String& rhs) {
|
||
rhs.append(lhs);
|
||
return rhs;
|
||
}
|
||
|
||
// 为CTL::String类型提供哈希函数
|
||
template <>
|
||
struct std::hash<CTL::String> {
|
||
std::size_t operator()(const CTL::String& s) const noexcept {
|
||
return std::hash<std::string>()(s);
|
||
}
|
||
};
|
||
|
||
#if __OHOS__
|
||
#pragma warning(pop)
|
||
// 或者对于 GCC/Clang
|
||
#pragma GCC diagnostic pop
|
||
#endif
|
||
|
||
#endif
|