kstd/vector: implement single-element erase

1 files changed, 16 insertions, 0 deletions

diff --git a/libs/kstd/include/kstd/vector b/libs/kstd/include/kstd/vector
index 0d4aac8..74aefa9 100644
--- a/libs/kstd/include/kstd/vector
+++ b/libs/kstd/include/kstd/vector
@@ -690,6 +690,22 @@ namespace kstd
       return begin() + prefix_size;
     }
 
+    constexpr auto erase(const_iterator position) -> iterator
+    {
+      if (position == end())
+      {
+        os::panic("[kstd:vector] Attempted to erase end()!");
+      }
+
+      auto prefix_size = std::ranges::distance(cbegin(), position);
+
+      std::ranges::move(begin() + prefix_size + 1, end(), begin() + prefix_size);
+      std::allocator_traits<allocator_type>::destroy(m_allocator, end() - 1);
+      --m_size;
+
+      return begin() + prefix_size;
+    }
+
     //! Append a given element to this vector via copy construction.
     constexpr auto push_back(value_type const & value) -> void
     {