Added support for customdata directory.

6 files changed, 53 insertions, 5 deletions

diff --git a/src/gui/char_server.cpp b/src/gui/char_server.cpp
index cccdcff1..a5fda25f 100644
--- a/src/gui/char_server.cpp
+++ b/src/gui/char_server.cpp
@@ -108,7 +108,7 @@ ServerSelectDialog::action(const gcn::ActionEvent &event)
         state = CHAR_CONNECT_STATE;
     }
     else if (event.getId() == "cancel") {
-        state = LOGIN_STATE;
+        state = LOADDATA_STATE;
     }
 }
 
diff --git a/src/gui/updatewindow.cpp b/src/gui/updatewindow.cpp
index 24084fe4..b66bcfbd 100644
--- a/src/gui/updatewindow.cpp
+++ b/src/gui/updatewindow.cpp
@@ -179,7 +179,7 @@ void UpdaterWindow::action(const gcn::ActionEvent &event)
     }
     else if (event.getId() == "play")
     {
-        state = LOGIN_STATE;
+        state = LOADDATA_STATE;
     }
 }
 
diff --git a/src/main.cpp b/src/main.cpp
index a9bcd077..383e7f39 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -535,7 +535,7 @@ int main(int argc, char *argv[])
     SDL_Event event;
 
     if (options.skipUpdate && state != ERROR_STATE) {
-        state = LOGIN_STATE;
+        state = LOADDATA_STATE;
     }
     else {
         state = UPDATE_STATE;
@@ -623,6 +623,7 @@ int main(int argc, char *argv[])
                     break;
 
                     // Those states don't cause a network disconnect
+                case LOADDATA_STATE:
                 case ACCOUNT_STATE:
                 case CHAR_CONNECT_STATE:
                 case CONNECTING_STATE:
@@ -643,13 +644,24 @@ int main(int argc, char *argv[])
             }
 
             switch (state) {
-                case LOGIN_STATE:
-                    logger->log("State: LOGIN");
+                case LOADDATA_STATE:
+                    logger->log("State: LOADDATA");
+
+                    //add customdata directory
+                    ResourceManager::getInstance()->searchAndAddArchives(
+                        "customdata/",
+                        "zip",
+                        false);
 
                     // Load XML databases
                     EquipmentDB::load();
                     ItemDB::load();
                     MonsterDB::load();
+                    state = LOGIN_STATE;
+                    break;
+
+                case LOGIN_STATE:
+                    logger->log("State: LOGIN");
 
                     if (!loginData.password.empty()) {
                         state = ACCOUNT_STATE;
diff --git a/src/main.h b/src/main.h
index 91fd8552..5787756a 100644
--- a/src/main.h
+++ b/src/main.h
@@ -39,6 +39,7 @@
 
 enum {
     EXIT_STATE,
+    LOADDATA_STATE,
     LOGIN_STATE,
     ACCOUNT_STATE,
     REGISTER_STATE,
diff --git a/src/resources/resourcemanager.cpp b/src/resources/resourcemanager.cpp
index 4ed316cf..37a82b0c 100644
--- a/src/resources/resourcemanager.cpp
+++ b/src/resources/resourcemanager.cpp
@@ -109,6 +109,33 @@ ResourceManager::addToSearchPath(const std::string &path, bool append)
     PHYSFS_addToSearchPath(path.c_str(), append ? 1 : 0);
 }
 
+void
+ResourceManager::searchAndAddArchives(const std::string &path,
+                                      const std::string &ext,
+                                      bool append)
+{
+    const char *dirSep = PHYSFS_getDirSeparator();
+    char **list = PHYSFS_enumerateFiles(path.c_str());
+
+    for (char **i = list; *i != NULL; i++)
+    {
+        size_t len = strlen(*i);
+
+        if (len > ext.length() && !ext.compare((*i)+(len - ext.length())))
+        {
+            std::string file, realPath, archive;
+
+            file = path + (*i);
+            realPath = std::string(PHYSFS_getRealDir(file.c_str()));
+            archive = realPath + dirSep + file;
+
+            addToSearchPath(archive, append);
+        }
+    }
+
+    PHYSFS_freeList(list);
+}
+
 bool
 ResourceManager::mkdir(const std::string &path)
 {
diff --git a/src/resources/resourcemanager.h b/src/resources/resourcemanager.h
index d458f96e..e176e337 100644
--- a/src/resources/resourcemanager.h
+++ b/src/resources/resourcemanager.h
@@ -83,6 +83,14 @@ class ResourceManager
         addToSearchPath(const std::string &path, bool append);
 
         /**
+        * Searches for zip files and adds them to the search path.
+        */
+        void
+        searchAndAddArchives(const std::string &path,
+                             const std::string &ext,
+                             bool append);
+
+        /**
          * Creates a directory in the write path
          */
         bool