pjs/xpcom/glue/nsWeakReference.h

#ifndef nsWeakReference_h__
#define nsWeakReference_h__

// nsWeakReference.h

#include "nsIWeakReference.h"

class NS_COM nsSupportsWeakReference : public nsISupportsWeakReference
	{
		public:
			nsSupportsWeakReference()
					: mProxy(0)
				{
					// nothing else to do here
				}

			inline virtual ~nsSupportsWeakReference();

			NS_IMETHOD GetWeakReference( nsIWeakReference** );

		private:
			friend class nsWeakReference;

			void
			NoticeProxyDestruction()
					// ...called (only) by an |nsWeakReference| from _its_ dtor.
				{
					mProxy = 0;
				}

			nsWeakReference* mProxy;
	};

class NS_COM nsWeakReference : public nsIWeakReference
	{
		public:
		// nsISupports...
			NS_IMETHOD_(nsrefcnt) AddRef();
			NS_IMETHOD_(nsrefcnt) Release();
			NS_IMETHOD QueryInterface( const nsIID&, void** );

		// nsIWeakReference...
			NS_IMETHOD QueryReference( const nsIID&, void** );


		private:
			friend class nsSupportsWeakReference;

			nsWeakReference( nsSupportsWeakReference* referent )
					: mRefCount(0),
						mReferent(referent)
					// ...I can only be constructed by an |nsSupportsWeakReference|
				{
					// nothing else to do here
				}

		virtual ~nsWeakReference()
		 			// ...I will only be destroyed by calling |delete| myself.
				{
					if ( mReferent )
						mReferent->NoticeProxyDestruction();
				}

			void
			NoticeReferentDestruction()
					// ...called (only) by an |nsSupportsWeakReference| from _its_ dtor.
				{
					mReferent = 0;
				}

			nsrefcnt									mRefCount;
			nsSupportsWeakReference*	mReferent;
	};

inline nsSupportsWeakReference::~nsSupportsWeakReference()
	{
		if ( mProxy )
			mProxy->NoticeReferentDestruction();
	}

#endif