X-Git-Url: https://git.sesse.net/?p=ccbs;a=blobdiff_plain;f=bigscreen%2Fwidestring.cpp;fp=bigscreen%2Fwidestring.cpp;h=c31f4fcb29981442173179bee2c32de8d1f810f5;hp=0000000000000000000000000000000000000000;hb=141adc9f5acb9ed1e98a206abb0d2e9cfdce6d2f;hpb=8c70d386605dbacdb807bd213c5b153761055086

diff --git a/bigscreen/widestring.cpp b/bigscreen/widestring.cpp
new file mode 100644
index 0000000..c31f4fc
--- /dev/null
+++ b/bigscreen/widestring.cpp
@@ -0,0 +1,54 @@
+#include <iconv.h>
+#include <endian.h>
+#include <exception>
+#include <pqxx/util>
+#include "widestring.h"
+
+static iconv_t ucs4_iconv;
+static bool iconv_initialized = false;
+
+void widestring::operator= (const char *from)
+{
+	if (!iconv_initialized) {
+#if __BYTE_ORDER == __LITTLE_ENDIAN
+		ucs4_iconv = iconv_open("ucs-4le", "utf-8");
+#else
+		ucs4_iconv = iconv_open("ucs-4be", "utf-8");
+#endif
+
+		iconv_initialized = true;
+	}
+	
+	unsigned bytes = std::strlen(from);
+	char *from_buf = strdup(from);
+	wchar_t *to_buf = new wchar_t[bytes + 1];
+
+	char *inptr = from_buf, *outptr = reinterpret_cast<char *> (to_buf);
+
+	size_t in_left = bytes;
+	size_t out_left = bytes * sizeof(wchar_t);
+
+	size_t ret = iconv(ucs4_iconv, NULL, NULL, &outptr, &out_left);
+	if (ret == (size_t)(-1)) {
+		throw std::runtime_error("Error in iconv during initialization");
+	}
+
+	ret = iconv(ucs4_iconv, &inptr, &in_left, &outptr, &out_left);
+	if (ret == (size_t)(-1)) {
+		perror("iconv");
+		throw std::runtime_error("Error in iconv during conversion");
+	}
+
+	erase(begin(), end());
+	std::copy(to_buf, reinterpret_cast<wchar_t *> (outptr), std::back_inserter(*this));
+
+	free(from_buf);
+	delete[] to_buf;
+}
+
+template<>
+void pqxx::from_string<widestring>(const char *from, widestring &to)
+{
+	to = from;
+}
+