X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=include%2Fmodules%2Fcap.h;h=6dcb9f3bc4d037d5a9055cfe7472824e8adcb190;hb=79892a727e323dcc4bce7e9c0cf3c99c5fe61706;hp=e05263ad3cb018f607bbaaf860264d6bf9ec56b3;hpb=912fd7a922beaeecdf602fa9d70964aee2a6bb63;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/include/modules/cap.h b/include/modules/cap.h
index e05263ad3..6dcb9f3bc 100644
--- a/include/modules/cap.h
+++ b/include/modules/cap.h
@@ -28,7 +28,14 @@ namespace Cap
 	static const unsigned int MAX_VALUE_LENGTH = 100;
 
 	typedef intptr_t Ext;
-	typedef LocalIntExt ExtItem;
+	class ExtItem : public LocalIntExt
+	{
+	 public:
+		ExtItem(Module* mod);
+		std::string serialize(SerializeFormat format, const Extensible* container, void* item) const CXX11_OVERRIDE;
+		void unserialize(SerializeFormat format, Extensible* container, const std::string& value) CXX11_OVERRIDE;
+	};
+
 	class Capability;
 
 	enum Protocol
@@ -55,6 +62,11 @@ namespace Cap
 		 * @param add If true, the capability is being added, otherwise its being removed
 		 */
 		virtual void OnCapAddDel(Capability* cap, bool add) = 0;
+
+		/** Called whenever the value of a cap changes.
+		 * @param cap Capability whose value changed
+		 */
+		virtual void OnCapValueChange(Capability* cap) { }
 	};
 
 	class Manager : public DataProvider
@@ -82,6 +94,11 @@ namespace Cap
 		 * @return Capability object pointer if found, NULL otherwise
 		 */
 		virtual Capability* Find(const std::string& name) const = 0;
+
+		/** Notify manager when a value of a cap changed
+		 * @param cap Cap whose value changed
+		 */
+		virtual void NotifyValueChange(Capability* cap) = 0;
 	};
 
 	/** Represents a client capability.
@@ -135,6 +152,16 @@ namespace Cap
 
 		friend class ManagerImpl;
 
+	 protected:
+		/** Notify the manager that the value of the capability changed.
+		 * Must be called if the value of the cap changes for any reason.
+		 */
+		void NotifyValueChange()
+		{
+			if (IsRegistered())
+				manager->NotifyValueChange(this);
+		}
+
 	 public:
 		/** Constructor, initializes the capability.
 		 * Caps are active by default.
@@ -170,7 +197,7 @@ namespace Cap
 			if (!IsRegistered())
 				return false;
 			Ext caps = extitem->get(user);
-			return (caps & GetMask());
+			return ((caps & GetMask()) != 0);
 		}
 
 		/** Turn the capability on/off for a user. If the cap is not registered this method has no effect.
@@ -258,4 +285,51 @@ namespace Cap
 			return NULL;
 		}
 	};
+
+	/** Reference to a cap. The cap may be provided by another module.
+	 */
+	class Reference
+	{
+		dynamic_reference_nocheck<Capability> ref;
+
+	 public:
+		/** Constructor, initializes the capability reference
+		 * @param mod Module creating this object
+		 * @param Name Raw name of the cap as used in the protocol (CAP LS, etc.)
+		 */
+		Reference(Module* mod, const std::string& Name)
+			: ref(mod, "cap/" + Name)
+		{
+		}
+
+		/** Check whether a user has the referenced capability turned on.
+		 * @param user User to check
+		 * @return True if the user is using the referenced capability, false otherwise
+		 */
+		bool get(LocalUser* user)
+		{
+			if (ref)
+				return ref->get(user);
+			return false;
+		}
+	};
+
+	class MessageBase : public ClientProtocol::Message
+	{
+	 public:
+		MessageBase(const std::string& subcmd)
+			: ClientProtocol::Message("CAP", ServerInstance->Config->ServerName)
+		{
+			PushParamPlaceholder();
+			PushParam(subcmd);
+		}
+
+		void SetUser(LocalUser* user)
+		{
+			if (user->registered & REG_NICK)
+				ReplaceParamRef(0, user->nick);
+			else
+				ReplaceParam(0, "*");
+		}
+	};
 }