Add custom autocomplate list from file customwords.txt

2 files changed, 38 insertions, 6 deletions

diff --git a/src/gui/chatwindow.cpp b/src/gui/chatwindow.cpp
index 1e530ecba..0b9945370 100644
--- a/src/gui/chatwindow.cpp
+++ b/src/gui/chatwindow.cpp
@@ -190,6 +190,7 @@ ChatWindow::ChatWindow():
 
     fillCommands();
     initTradeFilter();
+    loadCustomList();
 }
 
 ChatWindow::~ChatWindow()
@@ -1093,12 +1094,14 @@ void ChatWindow::autoComplete()
     if (newName == "" && spellManager)
         newName = spellManager->autoComplete(name);
     if (newName == "")
-        newName = autoCompleteCommands(name);
+        newName = autoComplete(name, &mCommands);
     if (newName == "" && actorSpriteManager)
     {
         actorSpriteManager->getMobNames(nameList);
         newName = autoComplete(nameList, name);
     }
+    if (newName == "")
+        newName = autoComplete(name, &mCustomWords);
 
     if (newName != "")
     {
@@ -1149,12 +1152,15 @@ std::string ChatWindow::autoComplete(std::vector<std::string> &names,
     return newName;
 }
 
-std::string ChatWindow::autoCompleteCommands(std::string partName)
+std::string ChatWindow::autoComplete(std::string partName, History *words)
 {
-    Commands::iterator i = mCommands.begin();
+    if (!words)
+        return "";
+
+    Commands::iterator i = words->begin();
     std::vector<std::string> nameList;
 
-    while (i != mCommands.end())
+    while (i != words->end())
     {
         std::string line = *i;
         std::string::size_type pos = line.find(partName, 0);
@@ -1414,3 +1420,25 @@ std::string ChatWindow::doReplace(const std::string &msg)
     replaceSpecialChars(str);
     return str;
 }
+
+void ChatWindow::loadCustomList()
+{
+    std::ifstream listFile;
+    struct stat statbuf;
+
+    std::string listName = Client::getServerConfigDirectory()
+        + "/customwords.txt";
+
+    if (!stat(listName.c_str(), &statbuf) && S_ISREG(statbuf.st_mode))
+    {
+        listFile.open(listName.c_str(), std::ios::in);
+        char line[101];
+        while (listFile.getline(line, 100))
+        {
+            std::string str = line;
+            if (!str.empty())
+                mCustomWords.push_back(str);
+        }
+        listFile.close();
+    }
+}
diff --git a/src/gui/chatwindow.h b/src/gui/chatwindow.h
index 97f586442..1b10405d4 100644
--- a/src/gui/chatwindow.h
+++ b/src/gui/chatwindow.h
@@ -242,6 +242,8 @@ class ChatWindow : public Window,
 
         void saveState();
 
+        void loadCustomList();
+
         std::string doReplace(const std::string &msg);
 
     protected:
@@ -249,6 +251,8 @@ class ChatWindow : public Window,
         friend class WhisperTab;
         friend class PopupMenu;
 
+        typedef std::list<std::string> History;
+
         /** Remove the given tab from the window */
         void removeTab(ChatTab *tab);
 
@@ -265,7 +269,7 @@ class ChatWindow : public Window,
 
         std::string autoCompleteHistory(std::string partName);
 
-        std::string autoCompleteCommands(std::string partName);
+        std::string autoComplete(std::string partName, History *words);
 
         std::string autoComplete(std::vector<std::string> &names,
                                  std::string partName) const;
@@ -294,7 +298,6 @@ class ChatWindow : public Window,
         /** Manage whisper tabs */
         TabMap mWhispers;
 
-        typedef std::list<std::string> History;
         typedef History::iterator HistoryIterator;
         History mHistory;           /**< Command history. */
         HistoryIterator mCurHist;   /**< History iterator. */
@@ -302,6 +305,7 @@ class ChatWindow : public Window,
         typedef std::list<std::string> Commands;
         typedef Commands::iterator CommandsIterator;
         History mCommands;           /**< Command list. */
+        History mCustomWords;
 
         bool mReturnToggles; /**< Marks whether <Return> toggles the chat log
                                 or not */