aboutsummaryrefslogtreecommitdiff
path: root/include/cru/common/SelfResolvable.hpp
blob: 94f3ae87bc8fef4a5afee33afec7ad0be44bcde8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
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