From 128abc453eb67ad759d81817160ee78a10b66fe6 Mon Sep 17 00:00:00 2001
From: lisyarus <lisyarus@gmail.com>
Date: Sun, 6 Aug 2023 12:55:31 +0300
Subject: [PATCH] Add util::at helper

---
 libs/util/include/psemek/util/at.hpp | 53 ++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 libs/util/include/psemek/util/at.hpp

diff --git a/libs/util/include/psemek/util/at.hpp b/libs/util/include/psemek/util/at.hpp
new file mode 100644
index 00000000..fdf7dec5
--- /dev/null
+++ b/libs/util/include/psemek/util/at.hpp
@@ -0,0 +1,53 @@
+#pragma once
+
+#include <psemek/util/exception.hpp>
+#include <psemek/util/to_string.hpp>
+
+#include <vector>
+
+namespace psemek::util
+{
+
+	struct key_error_base
+		: exception
+	{
+		using exception::exception;
+	};
+
+	template <typename Key>
+	struct key_error
+		: key_error_base
+	{
+		key_error(Key const & key, boost::stacktrace::stacktrace stacktrace = {})
+			: key_error_base(util::to_string("no such key: ", key), std::move(stacktrace))
+			, key_(key)
+		{}
+
+		Key const & key() const
+		{
+			return key_;
+		}
+
+	private:
+		Key key_;
+	};
+
+	template <typename Container, typename Key>
+	auto & at(Container && container, Key const & key)
+	{
+		if (auto it = container.find(key); it != container.end())
+			return it->second;
+
+		throw key_error<Key>(key);
+	}
+
+	template <typename T, typename Key>
+	auto & at(std::vector<T> && container, Key const & key)
+	{
+		if (key < container.size())
+			return container[key];
+
+		throw key_error<Key>(key);
+	}
+
+}