Create a string_view like thing for utf8 strings

utf8_string_view provides length and character iteration over a std::string_view containing utf8 data.

1 files changed, 65 insertions, 0 deletions

diff --git a/lib/unicode.h b/lib/unicode.h
index dde1d3d..2945650 100644
--- a/lib/unicode.h
+++ b/lib/unicode.h
@@ -14,4 +14,69 @@ uint32_t get_codepoint(const char *);
 
 #ifdef __cplusplus
 }
+
+#	include <string_view>
+#	include <algorithm>
+
+struct utf8_string_view {
+	struct iter {
+		constexpr explicit iter(const char * p) : pos {p} { }
+
+		[[nodiscard]] auto
+		operator!=(const iter & other) const
+		{
+			return pos != other.pos;
+		}
+
+		auto &
+		operator++()
+		{
+			pos = next_char(pos);
+			return *this;
+		}
+
+		[[nodiscard]] auto
+		operator*() const
+		{
+			return get_codepoint(pos);
+		}
+
+	private:
+		const char * pos;
+	};
+
+	// cppcheck-suppress noExplicitConstructor; NOLINTNEXTLINE(hicpp-explicit-conversions)
+	template<typename... Args> constexpr utf8_string_view(Args &&... args) : str {std::forward<Args>(args)...} { }
+
+	[[nodiscard]] auto
+	begin() const
+	{
+		return iter {str.cbegin()};
+	}
+
+	[[nodiscard]] auto
+	end() const
+	{
+		return iter {str.cend()};
+	}
+
+	[[nodiscard]] size_t length() const;
+
+private:
+	std::string_view str;
+};
+template<> struct std::iterator_traits<utf8_string_view::iter> {
+	using difference_type = size_t;
+	using value_type = uint32_t;
+	using pointer = void;
+	using reference = void;
+	using iterator_category = std::forward_iterator_tag;
+};
+
+[[nodiscard]] inline size_t
+utf8_string_view::length() const
+{
+	return std::distance(begin(), end());
+}
+
 #endif