From 464a4bf527b9da30985a6576f702429621a849a6 Mon Sep 17 00:00:00 2001
From: jacqueline <me@jacqueline.id.au>
Date: Fri, 21 Apr 2023 18:42:49 +1000
Subject: [PATCH] Add a RAII-friendly DI wrapper for driver instances

---
 src/drivers/driver_cache.cpp         | 43 ++++++++++++++++++++++
 src/drivers/include/driver_cache.hpp | 54 ++++++++++++++++++++++++++++
 2 files changed, 97 insertions(+)
 create mode 100644 src/drivers/driver_cache.cpp
 create mode 100644 src/drivers/include/driver_cache.hpp

diff --git a/src/drivers/driver_cache.cpp b/src/drivers/driver_cache.cpp
new file mode 100644
index 00000000..650e6f16
--- /dev/null
+++ b/src/drivers/driver_cache.cpp
@@ -0,0 +1,43 @@
+#include "driver_cache.hpp"
+
+#include <memory>
+#include <mutex>
+
+#include "display.hpp"
+#include "display_init.hpp"
+#include "storage.hpp"
+#include "touchwheel.hpp"
+
+namespace drivers {
+
+DriverCache::DriverCache() : gpios_(std::make_unique<GpioExpander>()) {}
+DriverCache::~DriverCache() {}
+
+auto DriverCache::AcquireGpios() -> GpioExpander* {
+  return gpios_.get();
+}
+
+auto DriverCache::AcquireDac() -> std::shared_ptr<AudioDac> {
+  return Acquire(dac_, [&]() -> AudioDac* {
+    return AudioDac::create(AcquireGpios()).value_or(nullptr);
+  });
+}
+
+auto DriverCache::AcquireDisplay() -> std::shared_ptr<Display> {
+  return Acquire(display_, [&]() -> Display* {
+    return Display::create(AcquireGpios(), displays::kST7735R);
+  });
+}
+
+auto DriverCache::AcquireStorage() -> std::shared_ptr<SdStorage> {
+  return Acquire(storage_, [&]() -> SdStorage* {
+    return SdStorage::create(AcquireGpios()).value_or(nullptr);
+  });
+}
+
+auto DriverCache::AcquireTouchWheel() -> std::shared_ptr<TouchWheel> {
+  return Acquire(touchwheel_,
+                 [&]() -> TouchWheel* { return new TouchWheel(); });
+}
+
+}  // namespace drivers
diff --git a/src/drivers/include/driver_cache.hpp b/src/drivers/include/driver_cache.hpp
new file mode 100644
index 00000000..c56ebc3f
--- /dev/null
+++ b/src/drivers/include/driver_cache.hpp
@@ -0,0 +1,54 @@
+#pragma once
+
+#include <memory>
+#include <mutex>
+
+#include "dac.hpp"
+#include "display.hpp"
+#include "gpio_expander.hpp"
+#include "storage.hpp"
+#include "touchwheel.hpp"
+
+namespace drivers {
+
+class DriverCache {
+ private:
+  std::unique_ptr<GpioExpander> gpios_;
+  std::weak_ptr<AudioDac> dac_;
+  std::weak_ptr<Display> display_;
+  std::weak_ptr<SdStorage> storage_;
+  std::weak_ptr<TouchWheel> touchwheel_;
+  // TODO(jacqueline): Haptics, samd
+
+  std::mutex mutex_;
+
+  template <typename T, typename F>
+  auto Acquire(std::weak_ptr<T> ptr, F factory) -> std::shared_ptr<T> {
+    std::shared_ptr<T> acquired = ptr.lock();
+    if (acquired) {
+      return acquired;
+    }
+
+    std::lock_guard<std::mutex> lock(mutex_);
+
+    acquired = ptr.lock();
+    if (acquired) {
+      return acquired;
+    }
+    acquired.reset(factory());
+    ptr = acquired;
+    return acquired;
+  }
+
+ public:
+  DriverCache();
+  ~DriverCache();
+
+  auto AcquireGpios() -> GpioExpander*;
+  auto AcquireDac() -> std::shared_ptr<AudioDac>;
+  auto AcquireDisplay() -> std::shared_ptr<Display>;
+  auto AcquireStorage() -> std::shared_ptr<SdStorage>;
+  auto AcquireTouchWheel() -> std::shared_ptr<TouchWheel>;
+};
+
+}  // namespace drivers