...

1 files changed, 51 insertions, 0 deletions

diff --git a/include/cru/common/SelfResolvable.hpp b/include/cru/common/SelfResolvable.hpp
new file mode 100644
index 00000000..94f3ae87
--- /dev/null
+++ b/include/cru/common/SelfResolvable.hpp
@@ -0,0 +1,51 @@
+#pragma once
+#include "PreConfig.hpp"
+
+#include <memory>
+#include <type_traits>
+
+namespace cru {
+template <typename T>
+class SelfResolvable;
+
+template <typename T>
+class ObjectResolver {
+  friend SelfResolvable<T>;
+
+ private:
+  ObjectResolver(const std::shared_ptr<T*>& resolver) : resolver_(resolver) {}
+
+ public:
+  ObjectResolver(const ObjectResolver&) = default;
+  ObjectResolver& operator=(const ObjectResolver&) = default;
+  ObjectResolver(ObjectResolver&&) = default;
+  ObjectResolver& operator=(ObjectResolver&&) = default;
+  ~ObjectResolver() = default;
+
+  T* Resolve() const {
+    // resolver_ is null only when this has been moved.
+    // You shouldn't resolve a moved resolver. So assert it.
+    Expects(resolver_);
+    return *resolver_;
+  }
+
+ private:
+  std::shared_ptr<T*> resolver_;
+};
+
+template <typename T>
+class SelfResolvable {
+ public:
+  SelfResolvable() : resolver_(new T*(static_cast<T*>(this))) {}
+  SelfResolvable(const SelfResolvable&) = delete;
+  SelfResolvable& operator=(const SelfResolvable&) = delete;
+  SelfResolvable(SelfResolvable&&) = delete;
+  SelfResolvable& operator=(SelfResolvable&&) = delete;
+  virtual ~SelfResolvable() { (*resolver_) = nullptr; }
+
+  ObjectResolver<T> CreateResolver() { return ObjectResolver<T>(resolver_); }
+
+ private:
+  std::shared_ptr<T*> resolver_;
+};
+}  // namespace cru