implement == and <=> operator in shared_ptr

1 files changed, 24 insertions, 5 deletions

diff --git a/libs/kstd/include/kstd/bits/shared_ptr.hpp b/libs/kstd/include/kstd/bits/shared_ptr.hpp
index 674807d..8930095 100644
--- a/libs/kstd/include/kstd/bits/shared_ptr.hpp
+++ b/libs/kstd/include/kstd/bits/shared_ptr.hpp
@@ -546,11 +546,6 @@ namespace kstd
       return ptr.pointer == nullptr;
     }
 
-    /**
-     * @brief Defaulted three-way comparator operator.
-     */
-    [[nodiscard]] auto operator<=>(shared_ptr const & other) const = default;
-
   private:
     /**
      * @brief If the candidate type inherits from enable_shared_from_this, assigns the internal weak pointer to this
@@ -624,6 +619,30 @@ namespace kstd
   {
     return shared_ptr<T>(new T(std::forward<Args>(args)...));
   }
+
+  /**
+   * @brief Equality operator for shared_ptr. Two shared_ptr instances are equal if they point to the same object
+   * @tparam T, U Types of the managed objects of the shared_ptr instances being compared.
+   * @param lhs, rhs The shared_ptr instances to compare.
+   * @return true if lhs and rhs point to the same object, false otherwise.
+   */
+  template<typename T, typename U>
+  [[nodiscard]] auto inline operator==(shared_ptr<T> const & lhs, shared_ptr<U> const & rhs) -> bool
+  {
+    return lhs.get() == rhs.get();
+  }
+
+  /**
+   * @brief Three-way comparison operator for shared_ptr. Compares the stored pointers of lhs and rhs using operator<=>.
+   * @tparam T, U Types of the managed objects of the shared_ptr instances being compared.
+   * @param lhs, rhs The shared_ptr instances to compare.
+   * @return The result of comparing the stored pointers of lhs and rhs using operator<=>
+   */
+  template<typename T, typename U>
+  [[nodiscard]] auto inline operator<=>(shared_ptr<T> const & lhs, shared_ptr<U> const & rhs)
+  {
+    return lhs.get() <=> rhs.get();
+  }
 }  // namespace kstd
 
 #endif
\ No newline at end of file