1 files changed, 72 insertions, 0 deletions
diff --git a/asl/status.cpp b/asl/status.cpp
index c081f33..eaee9ae 100644
--- a/asl/status.cpp
+++ b/asl/status.cpp
@@ -2,6 +2,7 @@
 #include "asl/allocator.hpp"
 #include "asl/string.hpp"
 #include "asl/atomic.hpp"
+#include "asl/format.hpp"
 
 // @Todo Use custom allocator
 using Allocator = asl::DefaultAllocator;
@@ -10,16 +11,56 @@ static Allocator g_allocator{};
 namespace
 {
 
+// @Todo Make a proper string builder, replace the tests's StringSink too
+class StringSink : public asl::Writer
+{
+    isize_t m_current_len{};
+    char*   m_data{};
+    
+public:
+    ~StringSink() override
+    {
+        reset();
+    }
+    
+    void write(asl::span<const asl::byte> str) override
+    {
+        m_data = reinterpret_cast<char*>(asl::GlobalHeap::realloc(
+            m_data,
+            asl::layout::array<char>(m_current_len),
+            asl::layout::array<char>(m_current_len + str.size())));
+        
+        asl::memcpy(m_data + m_current_len, str.data(), str.size()); // NOLINT
+        
+        m_current_len += str.size();
+    }
+
+    constexpr asl::string_view str() const { return {m_data, m_current_len}; }
+
+    void reset()
+    {
+        if (m_data != nullptr)
+        {
+            m_current_len = 0;
+            asl::GlobalHeap::dealloc(m_data, asl::layout::array<char>(m_current_len));
+            m_data = nullptr;
+        }
+    }
+};
+
 struct StatusInternal
 {
     asl::string<Allocator> msg;
     asl::status_code code;
     asl::atomic<int32_t> ref_count;
 
+    // @Todo Once we have string builder, move the string instead
+
     constexpr StatusInternal(asl::string_view msg_, asl::status_code code_)
         : msg{msg_, g_allocator}
         , code{code_}
     {
+        ASL_ASSERT(code != asl::status_code::ok);
         atomic_store(&ref_count, 1);
     }
 };
@@ -30,6 +71,13 @@ asl::status::status(status_code code, string_view msg)
     : m_payload{alloc_new<StatusInternal>(g_allocator, msg, code)}
 {}
 
+asl::status::status(status_code code, string_view fmt, span<format_internals::type_erased_arg> args)
+{
+    StringSink sink;
+    format_internals::format(&sink, fmt, args);
+    m_payload = alloc_new<StatusInternal>(g_allocator, sink.str(), code);
+}
+
 asl::status_code asl::status::code_internal() const
 {
     ASL_ASSERT(!is_inline());
@@ -60,3 +108,27 @@ void asl::status::unref()
     }
 }
 
+void asl::AslFormat(asl::Formatter& f, const asl::status& s)
+{
+    string_view status_str{};
+    
+    switch (s.code())
+    {
+        case status_code::ok: status_str = "ok"_sv; break;
+        case status_code::unknown: status_str = "unknown"_sv; break;
+        case status_code::internal: status_str = "internal"_sv; break;
+        case status_code::runtime: status_str = "runtime"_sv; break;
+        case status_code::invalid_argument: status_str = "invalid_argument"_sv; break;
+        default: status_str = "<unknown>"_sv; break;
+    }
+
+    if (s.is_inline())
+    {
+        format(f.writer(), "[{}]", status_str);
+    }
+    else
+    {
+        format(f.writer(), "[{}: {}]", status_str, s.message_internal());
+    }
+}
+