From 08d27356a6a6d97ed692ff65b9f78d0d0f0aa723 Mon Sep 17 00:00:00 2001
From: Vincent <magicdroidx@gmail.com>
Date: Sat, 29 Jun 2024 21:59:42 +0100
Subject: [PATCH] feat: implement SharedPtr

---
 include/bedrock/core/memory.h | 102 ++++++++++++++++++++++++++++++++--
 1 file changed, 96 insertions(+), 6 deletions(-)

diff --git a/include/bedrock/core/memory.h b/include/bedrock/core/memory.h
index 6f484a4e5..bacf63bf5 100644
--- a/include/bedrock/core/memory.h
+++ b/include/bedrock/core/memory.h
@@ -21,21 +21,111 @@ class OwnerPtr;
 
 template <typename T>
 struct SharedCounter {
-    T *ptr;
-    std::atomic<int> share_count;
-    std::atomic<int> weak_count;
+    T *ptr{nullptr};
+    std::atomic<int> share_count{1};
+    std::atomic<int> weak_count{0};
 };
 
 template <typename T>
 class SharedPtr {
 public:
-    SharedCounter<T> *pc;
+    constexpr SharedPtr() noexcept = default;
+    constexpr SharedPtr(nullptr_t) noexcept {}  // NOLINT(*-explicit-constructor)
+    explicit SharedPtr(T *ptr)
+    {
+        if (ptr) {
+            pc_ = new SharedCounter<T>();
+            pc_->ptr = ptr;
+        }
+    }
+
+    ~SharedPtr() noexcept
+    {
+        if (pc_) {
+            if (--pc_->share_count <= 0) {
+                delete pc_->ptr;
+                pc_->ptr = nullptr;
+
+                if (pc_->weak_count <= 0) {
+                    delete pc_;
+                }
+            }
+        }
+    }
+
+    // copy constructor
+    SharedPtr(const SharedPtr &other) noexcept
+    {
+        if (other.pc_) {
+            other.pc_->share_count++;
+        }
+        pc_ = other.pc_;
+    }
+
+    // move constructor
+    SharedPtr(SharedPtr &&other) noexcept
+    {
+        pc_ = other.pc_;
+        other.pc_ = nullptr;
+    }
+
+    // copy assignment operator
+    SharedPtr &operator=(const SharedPtr &other) noexcept
+    {
+        SharedPtr(other).swap(*this);
+        return *this;
+    }
+
+    // move assignment operator
+    SharedPtr &operator=(SharedPtr &&other) noexcept
+    {
+        SharedPtr(std::move(other)).swap(*this);
+        return *this;
+    }
+
+    void reset() noexcept
+    {
+        SharedPtr().swap(*this);
+    }
+
+    void reset(T *ptr) noexcept
+    {
+        SharedPtr(ptr).swap(*this);
+    }
+
+    T *operator->() const noexcept
+    {
+        return get();
+    }
+
+    T &operator*() const noexcept
+    {
+        return *get();
+    }
+
+    explicit operator bool() const
+    {
+        return get() != nullptr;
+    }
+
+private:
+    void swap(SharedPtr &other) noexcept
+    {
+        std::swap(pc_, other.pc_);
+    }
+
+    T *get() const noexcept
+    {
+        return pc_ ? pc_->ptr : nullptr;
+    }
+
+    SharedCounter<T> *pc_{nullptr};
 };
 
 template <typename T>
 class WeakPtr {
-public:
-    SharedCounter<T> *pc;
+private:
+    SharedCounter<T> *pc_;
 };
 
 template <typename T>