| 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 | }
|
|---|