anope/include/base.h

/*
 *
 * Copyright (C) 2008-2011 Adam <Adam@anope.org>
 * Copyright (C) 2008-2012 Anope Team <team@anope.org>
 *
 * Please read COPYING and README for further details.
 */

#ifndef BASE_H
#define BASE_H

#include "services.h"

/** The base class that most classes in Anope inherit from
 */
class CoreExport Base
{
	/* References to this base class */
	std::set<dynamic_reference_base *> References;
 public:
	Base();
	virtual ~Base();
	void AddReference(dynamic_reference_base *r);
	void DelReference(dynamic_reference_base *r);
};

class dynamic_reference_base
{
 protected:
	bool invalid;
 public:
	dynamic_reference_base() : invalid(false) { }
	virtual ~dynamic_reference_base() { }
	inline void Invalidate() { this->invalid = true; }
};

template<typename T>
class dynamic_reference : public dynamic_reference_base
{
 protected:
	T *ref;
 public:
	dynamic_reference(T *obj) : ref(obj)
	{
		if (ref)
			ref->AddReference(this);
	}

	dynamic_reference(const dynamic_reference<T> &obj) : ref(obj.ref)
	{
		if (ref)
			ref->AddReference(this);
	}

	virtual ~dynamic_reference()
	{
		if (this->invalid)
		{
			this->invalid = false;
			this->ref = NULL;
		}
		else if (this->operator bool())
			ref->DelReference(this);
	}

	virtual operator bool()
	{
		if (this->invalid)
		{
			this->invalid = false;
			this->ref = NULL;
		}
		return this->ref != NULL;
	}

	virtual inline operator T*()
	{
		if (this->operator bool())
			return this->ref;
		return NULL;
	}

	virtual inline T *operator->()
	{
		if (this->operator bool())
			return this->ref;
		return NULL;
	}

	virtual inline void operator=(T *newref)
	{
		if (this->invalid)
		{
			this->invalid = false;
			this->ref = NULL;
		}
		else if (this->operator bool())
			this->ref->DelReference(this);
		this->ref = newref;
		if (this->operator bool())
			this->ref->AddReference(this);
	}
};

#endif // BASE_H