X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=include%2Fdynref.h;h=bcfca9647d6cc0aad4ecd1e1e35875013d72efca;hb=c05f81cac83e80c7727594e3929e0709eccca689;hp=a3d2f9966349d72c2207b26d93b6799814f0b955;hpb=3a3ff949670c61a4a8856e1391222e156eb1cd17;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/include/dynref.h b/include/dynref.h
index a3d2f9966..bcfca9647 100644
--- a/include/dynref.h
+++ b/include/dynref.h
@@ -1,8 +1,8 @@
 /*
  * InspIRCd -- Internet Relay Chat Daemon
  *
- *   Copyright (C) 2013 Attila Molnar <attilamolnar@hush.com>
- *   Copyright (C) 2009 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2019-2020 Sadie Powell <sadie@witchery.services>
+ *   Copyright (C) 2013-2015 Attila Molnar <attilamolnar@hush.com>
  *
  * This file is part of InspIRCd.  InspIRCd is free software: you can
  * redistribute it and/or modify it under the terms of the GNU General Public
@@ -24,8 +24,18 @@
 
 class CoreExport dynamic_reference_base : public interfacebase, public insp::intrusive_list_node<dynamic_reference_base>
 {
+ public:
+	class CaptureHook
+	{
+	 public:
+		/** Called when the target of the dynamic_reference has been acquired
+		 */
+		virtual void OnCapture() = 0;
+	};
+
  private:
 	std::string name;
+	CaptureHook* hook;
 	void resolve();
  protected:
 	ServiceProvider* value;
@@ -33,17 +43,23 @@ class CoreExport dynamic_reference_base : public interfacebase, public insp::int
 	ModuleRef creator;
 	dynamic_reference_base(Module* Creator, const std::string& Name);
 	~dynamic_reference_base();
-	inline const std::string& GetProvider() { return name; }
+	inline const std::string& GetProvider() const { return name; }
 	void SetProvider(const std::string& newname);
+
+	/** Set handler to call when the target object becomes available
+	 * @param h Handler to call
+	 */
+	void SetCaptureHook(CaptureHook* h) { hook = h; }
+
 	void check();
-	operator bool() { return (value != NULL); }
+	operator bool() const { return (value != NULL); }
 	static void reset_all();
 };
 
 inline void dynamic_reference_base::check()
 {
 	if (!value)
-		throw ModuleException("Dynamic reference to '" + name + "' failed to resolve");
+		throw ModuleException("Dynamic reference to '" + name + "' failed to resolve. Are you missing a module?");
 }
 
 template<typename T>
@@ -63,6 +79,16 @@ class dynamic_reference : public dynamic_reference_base
 	{
 		return operator->();
 	}
+
+	const T* operator->() const
+	{
+		return static_cast<T*>(value);
+	}
+
+	const T* operator*() const
+	{
+		return operator->();
+	}
 };
 
 template<typename T>
@@ -81,6 +107,16 @@ class dynamic_reference_nocheck : public dynamic_reference_base
 	{
 		return operator->();
 	}
+
+	const T* operator->() const
+	{
+		return static_cast<T*>(value);
+	}
+
+	const T* operator*() const
+	{
+		return operator->();
+	}
 };
 
 class ModeHandler;