403 lines
11 KiB
C++
403 lines
11 KiB
C++
/**
|
|
* @file lggautocorrect.cpp
|
|
* @brief Auto Correct Manager
|
|
* @copyright Copyright (c) 2011 LordGregGreg Back
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include "llviewerprecompiledheaders.h"
|
|
#include "lggautocorrect.h"
|
|
#include "llsdserialize.h"
|
|
#include "llboost.h"
|
|
#include "llcontrol.h"
|
|
#include "llviewercontrol.h"
|
|
#include "llnotificationsutil.h"
|
|
|
|
LGGAutoCorrect* LGGAutoCorrect::sInstance;
|
|
|
|
LGGAutoCorrect::LGGAutoCorrect()
|
|
{
|
|
sInstance = this;
|
|
sInstance->loadFromDisk();
|
|
}
|
|
|
|
LGGAutoCorrect::~LGGAutoCorrect()
|
|
{
|
|
sInstance = NULL;
|
|
}
|
|
|
|
LGGAutoCorrect* LGGAutoCorrect::getInstance()
|
|
{
|
|
if(sInstance)return sInstance;
|
|
else
|
|
{
|
|
sInstance = new LGGAutoCorrect();
|
|
return sInstance;
|
|
}
|
|
}
|
|
void LGGAutoCorrect::save()
|
|
{
|
|
saveToDisk(mAutoCorrects);
|
|
}
|
|
std::string LGGAutoCorrect::getFileName()
|
|
{
|
|
std::string path=gDirUtilp->getExpandedFilename(LL_PATH_USER_SETTINGS, "");
|
|
|
|
if (!path.empty())
|
|
{
|
|
path = gDirUtilp->getExpandedFilename(LL_PATH_USER_SETTINGS, "settings_autocorrect.xml");
|
|
}
|
|
return path;
|
|
}
|
|
std::string LGGAutoCorrect::getDefaultFileName()
|
|
{
|
|
std::string path=gDirUtilp->getExpandedFilename(LL_PATH_APP_SETTINGS, "");
|
|
|
|
if (!path.empty())
|
|
{
|
|
path = gDirUtilp->getExpandedFilename(LL_PATH_APP_SETTINGS, "settings_autocorrect.xml");
|
|
}
|
|
return path;
|
|
}
|
|
LLSD LGGAutoCorrect::exportList(std::string listName)
|
|
{
|
|
LLSD toReturn;
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
toReturn["listName"]=listName;
|
|
toReturn["data"]=mAutoCorrects[listName]["data"];
|
|
toReturn["author"]=mAutoCorrects[listName]["author"];
|
|
toReturn["wordStyle"]=mAutoCorrects[listName]["wordStyle"];
|
|
toReturn["priority"]=mAutoCorrects[listName]["priority"];
|
|
}
|
|
return toReturn;
|
|
}
|
|
BOOL LGGAutoCorrect::addCorrectionList(LLSD newList)
|
|
{
|
|
if(newList.has("listName"))
|
|
{
|
|
std::string name = newList["listName"];
|
|
//if(!mAutoCorrects.has(name)){
|
|
LLSD newPart;
|
|
newPart["data"]=newList["data"];
|
|
newPart["enabled"]=TRUE;
|
|
newPart["announce"]=FALSE;
|
|
newPart["author"]=newList["author"];
|
|
newPart["wordStyle"]=newList["wordStyle"];
|
|
newPart["priority"]=newList["priority"].asInteger();
|
|
llinfos << "adding new list with settings priority "<<newPart["priority"].asInteger() <<llendl;
|
|
mAutoCorrects[name]=newPart;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
return FALSE;
|
|
}
|
|
BOOL LGGAutoCorrect::removeCorrectionList(std::string listName)
|
|
{
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
mAutoCorrects.erase(listName);
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
BOOL LGGAutoCorrect::setListEnabled(std::string listName, BOOL enabled)
|
|
{
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
mAutoCorrects[listName]["enabled"]=enabled;
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
BOOL LGGAutoCorrect::setListAnnounceeState(std::string listName, BOOL announce)
|
|
{
|
|
|
|
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
mAutoCorrects[listName]["announce"]=announce;
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
BOOL LGGAutoCorrect::setListStyle(std::string listName, BOOL announce)
|
|
{
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
mAutoCorrects[listName]["wordStyle"]=announce;
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
BOOL LGGAutoCorrect::setListPriority(std::string listName, int priority)
|
|
{
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
mAutoCorrects[listName]["priority"]=priority;
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
LLSD LGGAutoCorrect::getAutoCorrects()
|
|
{
|
|
//loadFromDisk();
|
|
return mAutoCorrects;
|
|
}
|
|
void LGGAutoCorrect::loadFromDisk()
|
|
{
|
|
std::string filename=getFileName();
|
|
if (filename.empty())
|
|
{
|
|
llinfos << "no valid user directory." << llendl;
|
|
}
|
|
if(!gDirUtilp->fileExists(filename))
|
|
{
|
|
std::string defaultName = getDefaultFileName();
|
|
llinfos << " user settings file doesnt exist, going to try and read default one from "<<defaultName.c_str()<< llendl;
|
|
|
|
if(gDirUtilp->fileExists(defaultName))
|
|
{
|
|
LLSD blankllsd;
|
|
llifstream file;
|
|
file.open(defaultName.c_str());
|
|
if (file.is_open())
|
|
{
|
|
LLSDSerialize::fromXMLDocument(blankllsd, file);
|
|
}
|
|
file.close();
|
|
saveToDisk(blankllsd);
|
|
}else
|
|
saveToDisk(getExampleLLSD());
|
|
}
|
|
else
|
|
{
|
|
llifstream file;
|
|
file.open(filename.c_str());
|
|
if (file.is_open())
|
|
{
|
|
LLSDSerialize::fromXML(mAutoCorrects, file);
|
|
}
|
|
file.close();
|
|
}
|
|
}
|
|
void LGGAutoCorrect::saveToDisk(LLSD newSettings)
|
|
{
|
|
mAutoCorrects=newSettings;
|
|
std::string filename=getFileName();
|
|
llofstream file;
|
|
file.open(filename.c_str());
|
|
LLSDSerialize::toPrettyXML(mAutoCorrects, file);
|
|
file.close();
|
|
}
|
|
void LGGAutoCorrect::runTest()
|
|
{
|
|
std::string startS("He just abandonned all his abilties");
|
|
std::string endS = replaceWords(startS);
|
|
llinfos << "!!! Test of autoreplace; start with "<<startS.c_str() << " end with " << endS.c_str()<<llendl;
|
|
|
|
|
|
}
|
|
BOOL LGGAutoCorrect::saveListToDisk(std::string listName, std::string fileName)
|
|
{
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
llofstream file;
|
|
file.open(fileName.c_str());
|
|
LLSDSerialize::toPrettyXML(exportList(listName), file);
|
|
file.close();
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
LLSD LGGAutoCorrect::getAutoCorrectEntries(std::string listName)
|
|
{
|
|
LLSD toReturn;
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
toReturn=mAutoCorrects[listName];
|
|
}
|
|
return toReturn;
|
|
}
|
|
std::string LGGAutoCorrect::replaceWord(std::string currentWord)
|
|
{
|
|
static LLCachedControl<bool> doAnything(gSavedSettings, "FSEnableAutoCorrect");
|
|
if(!(doAnything))return currentWord;
|
|
//loop through priorities
|
|
for(int currentPriority = 10;currentPriority>=0;currentPriority--)
|
|
{
|
|
LLSD::map_const_iterator loc_it = mAutoCorrects.beginMap();
|
|
LLSD::map_const_iterator loc_end = mAutoCorrects.endMap();
|
|
for ( ; loc_it != loc_end; ++loc_it)
|
|
{
|
|
const std::string& location = (*loc_it).first;
|
|
//llinfos << "location is "<<location.c_str() << " word is "<<currentWord.c_str()<<llendl;
|
|
const LLSD& loc_map = (*loc_it).second;
|
|
if(loc_map["priority"].asInteger()==currentPriority)
|
|
{
|
|
if(!loc_map["wordStyle"].asBoolean())
|
|
{
|
|
//this means look for partial matches instead of a full word
|
|
if(loc_map["enabled"].asBoolean())
|
|
{
|
|
LLSD::map_const_iterator inner_it = loc_map["data"].beginMap();
|
|
LLSD::map_const_iterator inner_end = loc_map["data"].endMap();
|
|
for ( ; inner_it != inner_end; ++inner_it)
|
|
{
|
|
const std::string& wrong = (*inner_it).first;
|
|
const std::string& right = (*inner_it).second;
|
|
int location = currentWord.find(wrong);
|
|
if(location!=std::string::npos)
|
|
{
|
|
currentWord=currentWord.replace(location,wrong.length(),right);
|
|
}
|
|
}
|
|
}
|
|
|
|
}else
|
|
if((loc_map["data"].has(currentWord))&&(loc_map["enabled"].asBoolean()))
|
|
{
|
|
std::string replacement = loc_map["data"][currentWord];
|
|
if(loc_map["announce"].asBoolean())
|
|
{
|
|
LLSD args;
|
|
//"[Before]" has been auto replaced by "[Replacement]"
|
|
// based on your [ListName] list.
|
|
args["BEFORE"] = currentWord;
|
|
args["LISTNAME"]=location;
|
|
args["REPLACEMENT"]=replacement;
|
|
LLNotificationsUtil::add("PhoenixAutoReplace",args);
|
|
}
|
|
gSavedSettings.setS32("FSAutoCorrectCount",gSavedSettings.getS32("FSAutoCorrectCount")+1);
|
|
llinfos << "found a word in list " << location.c_str() << " and it will replace " << currentWord.c_str() << " => " << replacement.c_str() << llendl;
|
|
return replacement;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return currentWord;
|
|
}
|
|
std::string LGGAutoCorrect::replaceWords(std::string words)
|
|
{
|
|
static LLCachedControl<bool> doAnything(gSavedSettings, "FSEnableAutoCorrect");
|
|
if(!(doAnything))return words;
|
|
//TODO update this function to use the "wordStyle" thing,
|
|
//but so far this function is never used, so later
|
|
|
|
boost_tokenizer tokens(words, boost::char_separator<char>(" "));
|
|
for (boost_tokenizer::iterator token_iter = tokens.begin(); token_iter != tokens.end(); ++token_iter)
|
|
{
|
|
std::string currentWord(*token_iter);
|
|
LLSD::map_const_iterator loc_it = mAutoCorrects.beginMap();
|
|
LLSD::map_const_iterator loc_end = mAutoCorrects.endMap();
|
|
for ( ; loc_it != loc_end; ++loc_it)
|
|
{
|
|
const std::string& location = (*loc_it).first;
|
|
//llinfos << "location is "<<location.c_str() << " word is "<<currentWord.c_str()<<llendl;
|
|
const LLSD& loc_map = (*loc_it).second;
|
|
if((loc_map["data"].has(currentWord))&&(loc_map["enabled"].asBoolean()))
|
|
{
|
|
std::string replacement = loc_map["data"][currentWord];
|
|
if(loc_map["announce"].asBoolean())
|
|
{
|
|
LLSD args;
|
|
//"[Before]" has been auto replaced by "[Replacement]"
|
|
// based on your [ListName] list.
|
|
args["BEFORE"] = currentWord;
|
|
args["LISTNAME"]=location;
|
|
args["REPLACEMENT"]=replacement;
|
|
LLNotificationsUtil::add("PhoenixAutoReplace",args);
|
|
}
|
|
llinfos << "found a word in list " << location.c_str() << " and it will replace " << currentWord.c_str() << " => " << replacement.c_str() << llendl;
|
|
int wordStart = words.find(currentWord);
|
|
words.replace(wordStart,currentWord.length(),replacement);
|
|
return replaceWords(words);//lol recursion!
|
|
}
|
|
}
|
|
}
|
|
return words;
|
|
}
|
|
BOOL LGGAutoCorrect::addEntryToList(std::string wrong, std::string right, std::string listName)
|
|
{
|
|
// *HACK: Make sure the "Custom" list exists, because the design of this
|
|
// system prevents us from updating it by changing the original file...
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
mAutoCorrects[listName]["data"][wrong]=right;
|
|
return TRUE;
|
|
}
|
|
else if(listName == "Custom")
|
|
{
|
|
mAutoCorrects[listName]["announce"] = 0;
|
|
mAutoCorrects[listName]["author"] = "You";
|
|
mAutoCorrects[listName]["data"][wrong] = right;
|
|
mAutoCorrects[listName]["enabled"] = 1;
|
|
mAutoCorrects[listName]["priority"] = 10;
|
|
mAutoCorrects[listName]["wordStyle"] = 1;
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
BOOL LGGAutoCorrect::removeEntryFromList(std::string wrong, std::string listName)
|
|
{
|
|
if(mAutoCorrects.has(listName))
|
|
{
|
|
if(mAutoCorrects[listName]["data"].has(wrong))
|
|
{
|
|
mAutoCorrects[listName]["data"].erase(wrong);
|
|
return TRUE;
|
|
}
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
LLSD LGGAutoCorrect::getExampleLLSD()
|
|
{
|
|
LLSD toReturn;
|
|
|
|
LLSD listone;
|
|
LLSD listtwo;
|
|
|
|
LLSD itemOne;
|
|
itemOne["wrong"]="wrong1";
|
|
itemOne["right"]="right1";
|
|
listone[0]=itemOne;
|
|
|
|
LLSD itemTwo;
|
|
itemTwo["wrong"]="wrong2";
|
|
itemTwo["right"]="right2";
|
|
listone[1]=itemTwo;
|
|
|
|
toReturn["listOne"]=listone;
|
|
|
|
|
|
itemOne["wrong"]="secondwrong1";
|
|
itemOne["right"]="secondright1";
|
|
listone[0]=itemOne;
|
|
|
|
itemTwo["wrong"]="secondwrong2";
|
|
itemTwo["right"]="secondright2";
|
|
listone[1]=itemTwo;
|
|
|
|
toReturn["listTwo"]=listone;
|
|
|
|
return toReturn;
|
|
}
|
|
|