Add a type member to ServerInfo and code for it

Some of the code is waiting for ifdef removal.

2 files changed, 71 insertions, 8 deletions

diff --git a/src/net/net.cpp b/src/net/net.cpp
index 9740844d..50baac73 100644
--- a/src/net/net.cpp
+++ b/src/net/net.cpp
@@ -121,15 +121,13 @@ Net::TradeHandler *Net::getTradeHandler()
 
 namespace Net
 {
-    bool networkLoaded = false;
+    ServerInfo::Type networkType = ServerInfo::UNKNOWN;
 } // namespace Net
 
 void Net::connectToServer(const ServerInfo &server)
 {
-    // TODO: Actually query the server about itself and choose the netcode
-    // based on that
-
-    if (networkLoaded)
+    // Remove with ifdefs
+    if (networkType != ServerInfo::UNKNOWN)
     {
         getGeneralHandler()->reload();
     }
@@ -142,9 +140,50 @@ void Net::connectToServer(const ServerInfo &server)
 #endif
 
         getGeneralHandler()->load();
+
+        networkType = server.type;
     }
+    // End remove section
+
+    // Uncomment after ifdefs removed
+    /*ServerInfo server = ServerInfo(inServer);
+    if (server.type == ServerInfo::UNKNOWN)
+    {
+        // TODO: Query the server about itself and choose the netcode based on
+        // that
+    }
+
+    //if (networkType == server.type)
+    if (networkType != ServerInfo::UNKNOWN)
+    {
+        getGeneralHandler()->reload();
+    }
+    else
+    {
+        if (networkType != ServerInfo::UNKNOWN)
+        {
+            getGeneralHandler()->unload();
+        }
+
+        switch (server.type)
+        {
+            case ServerInfo::MANASERV:
+                new ManaServ::GeneralHandler;
+                break;
+
+            case ServerInfo::EATHENA:
+                new EAthena::GeneralHandler;
+                break;
+
+            default:
+                // Shouldn't happen...
+                break;
+        }
+
+        getGeneralHandler()->load();
 
-    networkLoaded = true;
+        networkType = server.type;
+    }*/
 
     getLoginHandler()->setServer(server);
 
diff --git a/src/net/serverinfo.h b/src/net/serverinfo.h
index 4f68c6d6..82d11fec 100644
--- a/src/net/serverinfo.h
+++ b/src/net/serverinfo.h
@@ -28,12 +28,34 @@
 class ServerInfo
 {
 public:
+    enum Type {
+        UNKNOWN,
+        MANASERV,
+        EATHENA
+    };
+
+    Type type;
     std::string name;
     std::string hostname;
     unsigned short port;
 
+    ServerInfo()
+    {
+        type = UNKNOWN;
+        port = 0;
+    }
+
+    ServerInfo(const ServerInfo &info)
+    {
+        type = info.type;
+        name = info.name;
+        hostname = info.hostname;
+        port = info.port;
+    }
+
     void clear()
     {
+        type = UNKNOWN;
         name.clear();
         hostname.clear();
         port = 0;
@@ -41,12 +63,14 @@ public:
 
     bool operator==(const ServerInfo &other)
     {
-        return (hostname == other.hostname && port == other.port);
+        return (type == other.type && hostname == other.hostname &&
+                port == other.port);
     }
 
     bool operator!=(const ServerInfo &other)
     {
-        return (hostname != other.hostname || port != other.port);
+        return (type != other.type || hostname != other.hostname ||
+                port != other.port);
     }
 };