Swiften/StreamStack/StreamStack.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

#pragma once

#include <boost/shared_ptr.hpp>
#include <boost/signal.hpp>
#include <vector>

#include "Swiften/Elements/Stanza.h"
#include "Swiften/Base/foreach.h"

namespace Swift {
	class XMPPLayer;
	class LowLayer;
	class StreamLayer;

	class StreamStack {
		public:
			StreamStack(boost::shared_ptr<XMPPLayer> xmppLayer, boost::shared_ptr<LowLayer> physicalLayer);
			~StreamStack();

			void addLayer(boost::shared_ptr<StreamLayer>);

			boost::shared_ptr<XMPPLayer> getXMPPLayer() const {
				return xmppLayer_;
			}

			template<typename T> boost::shared_ptr<T> getLayer() {
				foreach(const boost::shared_ptr<StreamLayer>& streamLayer, layers_) {
					boost::shared_ptr<T> layer = boost::dynamic_pointer_cast<T>(streamLayer);
					if (layer) {
						return layer;
					}
				}
				return boost::shared_ptr<T>();
			}

		private:
			boost::shared_ptr<XMPPLayer> xmppLayer_;
			boost::shared_ptr<LowLayer> physicalLayer_;
			std::vector< boost::shared_ptr<StreamLayer> > layers_;
			boost::bsignals::connection xmppReadSlotConnection_;
			boost::bsignals::connection xmppWriteSignalConnection_;
			std::vector< boost::bsignals::connection > writeSignalConnections_;
	};
}