[1073] | 1 | #include "utilxmltools.h"
|
---|
| 2 |
|
---|
| 3 | namespace UtilXmlTools{
|
---|
| 4 |
|
---|
| 5 | // As this will not likely be modified we use QVector instead of QList since it is more cache friendly
|
---|
| 6 | QVector<QString> getAllXmlFilesByWildcard(const QString &wildcard){
|
---|
| 7 | QStringList validFilesMatching;
|
---|
| 8 | QStringList filesMatching;
|
---|
| 9 |
|
---|
| 10 | // Get all files matching the wildcard
|
---|
| 11 |
|
---|
| 12 | filesMatching=UtilXmlTools::getAllFilesByWildcard(wildcard);
|
---|
| 13 |
|
---|
| 14 | // Check if all are XmlFiles, only return valid XmlFiles
|
---|
| 15 |
|
---|
| 16 | for(int i=0; i<filesMatching.size(); i++){
|
---|
| 17 | if(filesMatching[i].endsWith(".xml",Qt::CaseInsensitive)){
|
---|
| 18 | validFilesMatching << filesMatching[i];
|
---|
| 19 | }
|
---|
| 20 | }
|
---|
| 21 |
|
---|
| 22 | return validFilesMatching.toVector();
|
---|
| 23 |
|
---|
| 24 | }
|
---|
| 25 |
|
---|
| 26 | QVector<QString> getAllPatchFilesByWildcard(const QString &wildcard){
|
---|
| 27 | QStringList validFilesMatching;
|
---|
| 28 | QStringList filesMatching;
|
---|
| 29 |
|
---|
| 30 | // Get all files matching the wildcard
|
---|
| 31 |
|
---|
| 32 | filesMatching=UtilXmlTools::getAllFilesByWildcard(wildcard);
|
---|
| 33 |
|
---|
| 34 | // Check if all are PatchFiles, only return valid PatchFiles
|
---|
| 35 |
|
---|
| 36 | for(int i=0; i<filesMatching.size(); i++){
|
---|
| 37 | if(filesMatching[i].endsWith(".patch",Qt::CaseInsensitive) || filesMatching[i].endsWith(".oni-patch",Qt::CaseInsensitive)){
|
---|
| 38 | validFilesMatching << filesMatching[i];
|
---|
| 39 | }
|
---|
| 40 | }
|
---|
| 41 |
|
---|
| 42 | return validFilesMatching.toVector();
|
---|
| 43 |
|
---|
| 44 | }
|
---|
| 45 |
|
---|
| 46 | void backupFile(const QString &file, bool verboseEnabled){
|
---|
| 47 | if(!QFile::exists(file+".bak")){
|
---|
| 48 | if(!Util::FileSystem::backupFile(file)){
|
---|
| 49 | std::cerr << "Couldn't back up file '" << file.toUtf8().constData() << "'. Aborting." << std::endl;
|
---|
| 50 | exit(1);
|
---|
| 51 | }
|
---|
| 52 | }
|
---|
| 53 | else{
|
---|
| 54 | if(verboseEnabled){
|
---|
| 55 | std::cout << "Backup file '" << file.toUtf8().constData() << "'' already exists. Skipping..." << std::endl;
|
---|
| 56 | }
|
---|
| 57 | }
|
---|
| 58 | }
|
---|
| 59 |
|
---|
| 60 | void getAllXpathElements(const QString &xPathExpression, pugi::xml_document &doc, QList<pugi::xml_node> &result){
|
---|
| 61 |
|
---|
| 62 | pugi::xpath_node_set selectedNodes;
|
---|
| 63 | pugi::xpath_node node;
|
---|
| 64 |
|
---|
| 65 | try
|
---|
| 66 | {
|
---|
| 67 | selectedNodes = doc.select_nodes(xPathExpression.toUtf8().constData());
|
---|
| 68 | }
|
---|
| 69 |
|
---|
| 70 | catch (const pugi::xpath_exception& e)
|
---|
| 71 | {
|
---|
| 72 | displayErrorMessage("XPath element selection","Selection of elements using the XPathExpression: '" + xPathExpression + "' failed:\n" + e.what());
|
---|
| 73 | }
|
---|
| 74 |
|
---|
| 75 | for (pugi::xpath_node_set::const_iterator currNode = selectedNodes.begin(); currNode != selectedNodes.end(); ++currNode)
|
---|
| 76 | {
|
---|
| 77 | node = *currNode;
|
---|
| 78 | if(node){ // if node != null
|
---|
| 79 | result << node.node();
|
---|
| 80 | }
|
---|
| 81 | }
|
---|
| 82 |
|
---|
| 83 | if(result.isEmpty()){
|
---|
| 84 | result << pugi::xml_node(); // add an empty node if none found
|
---|
| 85 | }
|
---|
| 86 |
|
---|
| 87 | }
|
---|
| 88 |
|
---|
| 89 | pugi::xml_node getFirstXpathElement(const QString &xPathExpression, pugi::xml_document &doc){
|
---|
| 90 | pugi::xpath_node selectedNode;
|
---|
| 91 |
|
---|
| 92 | try
|
---|
| 93 | {
|
---|
| 94 | selectedNode = doc.select_single_node(xPathExpression.toUtf8().constData());
|
---|
| 95 | }
|
---|
| 96 |
|
---|
| 97 | catch (const pugi::xpath_exception& e)
|
---|
| 98 | {
|
---|
| 99 | displayErrorMessage("XPath element selection","Selection of element using the XPathExpression: '" + xPathExpression + "' failed:\n" + e.what());
|
---|
| 100 | }
|
---|
| 101 |
|
---|
| 102 | return selectedNode.node();
|
---|
| 103 | }
|
---|
| 104 |
|
---|
| 105 | void getAllNamedElements(pugi::xml_node &node, QList<pugi::xml_node> &result, XmlFilter &filters){
|
---|
| 106 | for (pugi::xml_node_iterator currNode = node.begin(); currNode != node.end(); ++currNode)
|
---|
| 107 | {
|
---|
| 108 |
|
---|
| 109 | if ((*currNode).name() == filters.getElementName() && (filters.getParentElementName() == "" || filters.getParentElementName() == (*currNode).parent().name())
|
---|
| 110 | && (filters.getAttributeName() == "" ||
|
---|
| 111 | QString((*currNode).attribute(filters.getAttributeName().toUtf8().constData()).value()) == filters.getAttributeValue()) ){ // Seems node attribute must be converted to qtsring to remove \r\n needs to check in future!
|
---|
| 112 |
|
---|
| 113 | result << *currNode;
|
---|
| 114 | continue;
|
---|
| 115 | }
|
---|
| 116 | getAllNamedElements(*currNode,result,filters);
|
---|
| 117 | }
|
---|
| 118 | }
|
---|
| 119 |
|
---|
| 120 | pugi::xml_node getFirstNamedElement(pugi::xml_node &node, XmlFilter &filters){
|
---|
| 121 |
|
---|
| 122 | pugi::xml_node foundNode;
|
---|
| 123 |
|
---|
| 124 | for (pugi::xml_node_iterator currNode = node.begin(); currNode != node.end(); ++currNode)
|
---|
| 125 | {
|
---|
| 126 | if ((*currNode).name() == filters.getElementName() && (filters.getParentElementName() == "" || filters.getParentElementName() == (*currNode).parent().name())
|
---|
| 127 | && (filters.getAttributeName() == "" ||
|
---|
| 128 | QString((*currNode).attribute(filters.getAttributeName().toUtf8().constData()).value()) == filters.getAttributeValue()) ){
|
---|
| 129 | return *currNode;
|
---|
| 130 | }
|
---|
| 131 |
|
---|
| 132 | foundNode=getFirstNamedElement(*currNode,filters);
|
---|
| 133 |
|
---|
| 134 | if(foundNode.type()!=pugi::node_null){
|
---|
| 135 | return foundNode;
|
---|
| 136 | }
|
---|
| 137 |
|
---|
| 138 | }
|
---|
| 139 |
|
---|
| 140 | return foundNode;
|
---|
| 141 | }
|
---|
| 142 |
|
---|
| 143 | void displaySuccessMessage(const int numberOperations, const QString &operation){
|
---|
| 144 | std::cout << "------------------------------------------------------------------------" << std::endl;
|
---|
| 145 | std::cout << numberOperations << " " << operation.toUtf8().constData() << " operation(s) completed with success!" << std::endl;
|
---|
| 146 | std::cout << "------------------------------------------------------------------------" << std::endl;
|
---|
| 147 | }
|
---|
| 148 |
|
---|
| 149 | void displayErrorMessage(const QString& operation, const QString &message, bool exitProgram){
|
---|
| 150 | std::cerr << "************************************************************************" << std::endl;
|
---|
| 151 | std::cerr << operation.toUtf8().constData() << " operation failed!" << std::endl << std::endl;
|
---|
| 152 | std::cerr << message.toUtf8().constData() << std::endl << std::endl;
|
---|
| 153 | if(exitProgram) std::cerr << "Aborting..." << std::endl;
|
---|
| 154 | std::cerr << "************************************************************************" << std::endl;
|
---|
| 155 | if(exitProgram) exit(1);
|
---|
| 156 | }
|
---|
| 157 |
|
---|
| 158 | QStringList qStringListFromSpacedString(const QString &mySpacedString){
|
---|
| 159 | return mySpacedString.split(" ");
|
---|
| 160 | }
|
---|
| 161 |
|
---|
| 162 | QList<int> qListIntFromSpacedString(const QString &mySpacedString){
|
---|
| 163 | QStringList stringList;
|
---|
| 164 | QList<int> intList;
|
---|
| 165 |
|
---|
| 166 | stringList = mySpacedString.split(" ");
|
---|
| 167 |
|
---|
| 168 | foreach(QString value, stringList){
|
---|
| 169 | bool ok;
|
---|
| 170 |
|
---|
| 171 | intList << value.toInt(&ok);
|
---|
| 172 |
|
---|
| 173 | if(!ok){
|
---|
| 174 | throw std::runtime_error(QString("Impossible to convert string '" + value + "' to int!").toUtf8().constData());
|
---|
| 175 | }
|
---|
| 176 | }
|
---|
| 177 |
|
---|
| 178 | return intList;
|
---|
| 179 | }
|
---|
| 180 |
|
---|
| 181 | QList<double> qListDoubleFromSpacedString(const QString &mySpacedString){
|
---|
| 182 | QStringList stringList;
|
---|
| 183 | QList<double> doubleList;
|
---|
| 184 |
|
---|
| 185 | stringList = mySpacedString.split(" ");
|
---|
| 186 |
|
---|
| 187 | foreach(QString value, stringList){
|
---|
| 188 | bool ok;
|
---|
| 189 |
|
---|
| 190 | doubleList << value.toDouble(&ok);
|
---|
| 191 |
|
---|
| 192 | if(!ok){
|
---|
| 193 | throw std::runtime_error(QString("Impossible to convert string '" + value + "' to double!").toUtf8().constData());
|
---|
| 194 | }
|
---|
| 195 | }
|
---|
| 196 |
|
---|
| 197 | return doubleList;
|
---|
| 198 | }
|
---|
| 199 |
|
---|
| 200 | void loadXmlFile(const QString &file, pugi::xml_document &document, pugi::xml_node &rootNode, bool backupsEnabled, bool verboseEnabled, const QString &operationForErrorMessage){
|
---|
| 201 |
|
---|
| 202 | pugi::xml_parse_result result = document.load_file(file.toUtf8().constData());
|
---|
| 203 | rootNode=document.root();
|
---|
| 204 |
|
---|
| 205 | if(result.status==pugi::status_ok){
|
---|
| 206 | if(verboseEnabled){
|
---|
| 207 | std::cout << "File '" << file.toUtf8().constData() << "' loaded with sucess." << std::endl;
|
---|
| 208 | }
|
---|
| 209 | }
|
---|
| 210 | else{
|
---|
| 211 | UtilXmlTools::displayErrorMessage(operationForErrorMessage,"An error ocurred while loading '" +file + "' XML file\n" + result.description());
|
---|
| 212 | }
|
---|
| 213 |
|
---|
| 214 | if(backupsEnabled){
|
---|
| 215 | UtilXmlTools::backupFile(file,verboseEnabled); // bake a backup of the file.
|
---|
| 216 | }
|
---|
| 217 |
|
---|
| 218 | }
|
---|
| 219 |
|
---|
| 220 | void saveXmlFile(const QString &file, pugi::xml_document &document, const QString &operationForErrorMessage){
|
---|
| 221 | if(!document.save_file(file.toUtf8().constData(), "\t", pugi::format_indent | pugi::format_save_file_text | pugi::format_no_escapes)){ // output as the system new lines ending
|
---|
| 222 | UtilXmlTools::displayErrorMessage(operationForErrorMessage,"An error ocurred while saving '" + file + "' XML file");
|
---|
| 223 | }
|
---|
| 224 | }
|
---|
| 225 |
|
---|
| 226 | //TODO Needs optimization
|
---|
| 227 | QStringList QStringToArgsArray(const QString &args){
|
---|
| 228 | QStringList result;
|
---|
| 229 | int startIdx=0, endIdx=0;
|
---|
| 230 |
|
---|
| 231 | if(!args.isEmpty()){ // if non empty arguments
|
---|
| 232 |
|
---|
| 233 | while(endIdx<args.length()){
|
---|
| 234 |
|
---|
| 235 | startIdx=endIdx;
|
---|
| 236 |
|
---|
| 237 | if(args.at(startIdx)==' '){ // Ignore spaces until a different char is found
|
---|
| 238 | endIdx++;
|
---|
| 239 | continue;
|
---|
| 240 | }
|
---|
| 241 | else if(args.at(startIdx)!='"'){ // if first character is different from quote it ends with space
|
---|
| 242 | endIdx=args.indexOf(' ',startIdx+1);
|
---|
| 243 | }
|
---|
| 244 | else{ // If is a quote try to get the all the string until next quote
|
---|
| 245 | endIdx=args.indexOf('"',startIdx+1);
|
---|
| 246 | }
|
---|
| 247 |
|
---|
| 248 | if(endIdx==-1) break;
|
---|
| 249 |
|
---|
| 250 | endIdx++;
|
---|
| 251 |
|
---|
| 252 | result << args.mid(startIdx,endIdx-startIdx).replace("\"","").trimmed(); // remove quotes
|
---|
| 253 | }
|
---|
| 254 |
|
---|
| 255 | }
|
---|
| 256 |
|
---|
| 257 | return result;
|
---|
| 258 | }
|
---|
| 259 |
|
---|
| 260 | // TODO: Replace later with a more generic function and put in util library
|
---|
| 261 | // Supports wildcards, and directory with wildcard e.g.:
|
---|
| 262 | // *.xml
|
---|
| 263 | // C:/myXmls/*.xml
|
---|
| 264 | QStringList getAllFilesByWildcard(const QString &wildcard){
|
---|
| 265 |
|
---|
| 266 | QString pathNormalized;
|
---|
| 267 | QStringList nameWildCard; // entryList requires a QStringList
|
---|
| 268 | QStringList resultFiles; // result files with absolute path
|
---|
| 269 | int endOfPathIdx;
|
---|
| 270 | QDir directory;
|
---|
| 271 |
|
---|
| 272 | if(wildcard==""){
|
---|
| 273 | std::cout << "You need to specify a wildcard! Aborting..." << std::endl;
|
---|
| 274 | exit(1);
|
---|
| 275 | }
|
---|
| 276 |
|
---|
| 277 | pathNormalized=Util::FileSystem::normalizePath(wildcard); // Convert slashes to work in both mac and windows
|
---|
| 278 |
|
---|
| 279 | if(pathNormalized.contains("/")){ // If contains full path
|
---|
| 280 | endOfPathIdx=pathNormalized.lastIndexOf("/"); // get last slash
|
---|
| 281 |
|
---|
| 282 | nameWildCard.append(pathNormalized.right(pathNormalized.size()-1-endOfPathIdx)); // get the names wildcard // -1 because starts with zeo
|
---|
| 283 |
|
---|
| 284 | pathNormalized=pathNormalized.left(endOfPathIdx); // get the complete path
|
---|
| 285 |
|
---|
| 286 | directory=QDir(pathNormalized);
|
---|
| 287 | }
|
---|
| 288 | else{ // if relative
|
---|
| 289 | nameWildCard << wildcard;
|
---|
| 290 | }
|
---|
| 291 |
|
---|
| 292 | foreach (QFileInfo fileInfo, directory.entryInfoList(nameWildCard)){
|
---|
| 293 | resultFiles << fileInfo.absoluteFilePath();
|
---|
| 294 | }
|
---|
| 295 |
|
---|
| 296 | return resultFiles;
|
---|
| 297 | }
|
---|
| 298 |
|
---|
| 299 | }
|
---|