srcmlhandler.cpp

/********************************************************************************************************************************************************
* @file srcmlhandler.cpp
*
* @Copyright (C) 2022 i-trace.org
*
* This file is part of iTrace Infrastructure http://www.i-trace.org/.
* iTrace Infrastructure is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
* iTrace Infrastructure 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 General Public License for more details.
* You should have received a copy of the GNU General Public License along with iTrace Infrastructure. If not, see <https://www.gnu.org/licenses/>.
********************************************************************************************************************************************************/

#include "srcmlhandler.h"

#include <iostream>

SRCMLHandler::SRCMLHandler(QString path) {
    file_path = path;
}

bool SRCMLHandler::isPositional() {
    QFile file(file_path);
    file.open(QIODevice::ReadOnly);
    QXmlStreamReader parser(&file);
    parser.readNextStartElement();

    if(parser.name() == "unit") {
        for(auto decl : parser.namespaceDeclarations()) {
            if(decl.prefix() == "pos") {
                file.close();
                return true;
            }
        }
    }
    file.close();
    return false;
}

QVector<QString> SRCMLHandler::getAllFilenames() {
    QFile file(file_path);
    file.open(QIODevice::ReadOnly);
    QXmlStreamReader parser(&file);
    QVector<QString> files;

    while(!parser.atEnd()) {
        if(parser.readNextStartElement()) {
            if(parser.name() == "unit") {
                for(auto attr : parser.attributes()) {
                    if(attr.name() == "filename") { files.push_back(attr.value().toString()); }
                }
            }
        }
    }
    file.close();
    return files;
}

QString SRCMLHandler::getUnitText(QString unit_filename) {
    QString srcml_namespace = "declare default element namespace \"http://www.srcML.org/srcML/src\"; declare namespace re=\"http://exslt.org/regular-expressions\";";
    QFile file(file_path);
    file.open(QIODevice::ReadOnly);

    QString query_string = QString("doc($file)//unit[@filename = \""+unit_filename+"\"]");
    QString full_xquery = srcml_namespace + query_string;

    QXmlQuery query_results;
    query_results.bindVariable("file",&file);
    query_results.setQuery(full_xquery);
    if(!query_results.isValid()) { return "INVALID QUERY"; }
    QString results;
    query_results.evaluateTo(&results);
    file.close();
    return results;
}

QString SRCMLHandler::getUnitBody(QString unit_filename) {
    QString srcml_namespace = "declare default element namespace \"http://www.srcML.org/srcML/src\"; declare namespace re=\"http://exslt.org/regular-expressions\";";
    QFile file(file_path);
    file.open(QIODevice::ReadOnly);

    QString query_string = QString("doc($file)//unit[@filename = \""+unit_filename+"\"]/string()");
    QString full_xquery = srcml_namespace + query_string;

    QXmlQuery query_results;
    query_results.bindVariable("file",&file);
    query_results.setQuery(full_xquery);
    if(!query_results.isValid()) { return "INVALID QUERY"; }
    QString results;
    query_results.evaluateTo(&results);
    file.close();
    return results;
}