X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcommon%2Fpeerfactory.cpp;h=498495c719a310430ce3f33f5c61e23bcefec102;hp=9d55fccdee950e161a37bb3c3867a65ab1f48260;hb=c1cf157116de7fc3da96203aa6f03c38c7ebb650;hpb=31414ddaa421b1f8091eab8149da74545f7c47ab

diff --git a/src/common/peerfactory.cpp b/src/common/peerfactory.cpp
index 9d55fccd..498495c7 100644
--- a/src/common/peerfactory.cpp
+++ b/src/common/peerfactory.cpp
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-2014 by the Quassel Project                        *
+ *   Copyright (C) 2005-2018 by the Quassel Project                        *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
@@ -20,26 +20,40 @@
 
 #include "peerfactory.h"
 
+#include "protocols/datastream/datastreampeer.h"
 #include "protocols/legacy/legacypeer.h"
 
-RemotePeer *PeerFactory::createPeer(const ProtoDescriptor &protocol, AuthHandler *authHandler, QTcpSocket *socket, QObject *parent)
+PeerFactory::ProtoList PeerFactory::supportedProtocols()
 {
-    return createPeer(ProtoList() << protocol, authHandler, socket, parent);
+    ProtoList result;
+    result.append(ProtoDescriptor(Protocol::DataStreamProtocol, DataStreamPeer::supportedFeatures()));
+    result.append(ProtoDescriptor(Protocol::LegacyProtocol, 0));
+    return result;
 }
 
+RemotePeer* PeerFactory::createPeer(
+    const ProtoDescriptor& protocol, AuthHandler* authHandler, QTcpSocket* socket, Compressor::CompressionLevel level, QObject* parent)
+{
+    return createPeer(ProtoList() << protocol, authHandler, socket, level, parent);
+}
 
-RemotePeer *PeerFactory::createPeer(const ProtoList &protocols, AuthHandler *authHandler, QTcpSocket *socket, QObject *parent)
+RemotePeer* PeerFactory::createPeer(
+    const ProtoList& protocols, AuthHandler* authHandler, QTcpSocket* socket, Compressor::CompressionLevel level, QObject* parent)
 {
-    foreach(const ProtoDescriptor &protodesc, protocols) {
+    foreach (const ProtoDescriptor& protodesc, protocols) {
         Protocol::Type proto = protodesc.first;
         quint16 features = protodesc.second;
-        switch(proto) {
-            case Protocol::LegacyProtocol:
-                return new LegacyPeer(authHandler, socket, parent);
-            default:
-                break;
+        switch (proto) {
+        case Protocol::LegacyProtocol:
+            return new LegacyPeer(authHandler, socket, level, parent);
+        case Protocol::DataStreamProtocol:
+            if (DataStreamPeer::acceptsFeatures(features))
+                return new DataStreamPeer(authHandler, socket, features, level, parent);
+            break;
+        default:
+            break;
         }
     }
 
-    return 0;
+    return nullptr;
 }