0
0
mirror of https://github.com/fralx/LimeReport.git synced 2024-12-24 00:33:02 +03:00
LimeReport/limereport/lrdatasourcemanager.cpp
Андрей Лухнов 0fca7169d3 Define code style and format all source file using clang-format-14
except those placed in 3rdparty directories.
2024-09-19 21:09:38 +03:00

1923 lines
63 KiB
C++

/***************************************************************************
* This file is part of the Lime Report project *
* Copyright (C) 2021 by Alexander Arin *
* arin_a@bk.ru *
* *
** GNU General Public License Usage **
* *
* This library 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. *
* You should have received a copy of the GNU General Public License *
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
* *
** GNU Lesser General Public License **
* *
* 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 3 of the *
* License, or (at your option) any later version. *
* You should have received a copy of the GNU Lesser General Public *
* License along with this library. *
* If not, see <http://www.gnu.org/licenses/>. *
* *
* 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 General Public License for more details. *
****************************************************************************/
#include "lrdatasourcemanager.h"
#include "lrdatadesignintf.h"
#include <QSqlQuery>
#include <QStringList>
#if (QT_VERSION < QT_VERSION_CHECK(5, 15, 1))
#include <QRegExp>
#endif
#include <QFileInfo>
#include <QSqlError>
#include <QSqlQueryModel>
#include <stdexcept>
#ifdef BUILD_WITH_EASY_PROFILER
#include "easy/profiler.h"
#else
#define EASY_BLOCK(...)
#define EASY_END_BLOCK
#endif
namespace LimeReport {
DataNode::~DataNode()
{
for (int i = 0; i < m_childs.count(); ++i) {
delete m_childs[i];
}
}
DataNode* DataNode::addChild(const QString& name, DataNode::NodeType type, const QIcon& icon)
{
DataNode* res = new DataNode(name, type, this, icon);
m_childs.push_back(res);
return res;
}
int DataNode::row()
{
if (m_parent) {
return m_parent->m_childs.indexOf(const_cast<DataNode*>(this));
}
return 0;
}
void DataNode::clear()
{
for (int i = 0; i < m_childs.count(); ++i) {
delete m_childs[i];
}
m_childs.clear();
}
DataSourceModel::DataSourceModel(DataSourceManager* dataManager):
m_dataManager(0),
m_rootNode(new DataNode())
{
setDataSourceManager(dataManager);
}
DataSourceModel::~DataSourceModel() { delete m_rootNode; }
QModelIndex DataSourceModel::index(int row, int column, const QModelIndex& parent) const
{
if (!m_rootNode)
return QModelIndex();
if (!hasIndex(row, column, parent))
return QModelIndex();
DataNode* parentNode;
if (parent.isValid()) {
parentNode = nodeFromIndex(parent);
} else {
parentNode = m_rootNode;
}
DataNode* childNode = parentNode->child(row);
if (childNode) {
return createIndex(row, column, childNode);
} else
return QModelIndex();
}
QModelIndex DataSourceModel::parent(const QModelIndex& child) const
{
if (!child.isValid())
return QModelIndex();
DataNode* childNode = nodeFromIndex(child);
if (!childNode)
return QModelIndex();
DataNode* parentNode = childNode->parent();
if ((parentNode == m_rootNode) || (!parentNode))
return QModelIndex();
return createIndex(parentNode->row(), 0, parentNode);
}
int DataSourceModel::rowCount(const QModelIndex& parent) const
{
if (!m_rootNode)
return 0;
DataNode* parentNode;
if (parent.isValid())
parentNode = nodeFromIndex(parent);
else
parentNode = m_rootNode;
return parentNode->childCount();
}
int DataSourceModel::columnCount(const QModelIndex& parent) const
{
Q_UNUSED(parent)
return 1;
}
QVariant DataSourceModel::data(const QModelIndex& index, int role) const
{
DataNode* node = nodeFromIndex(index);
switch (role) {
case Qt::DisplayRole:
if (!node)
return QVariant();
return node->name();
break;
case Qt::DecorationRole:
if (!node)
return QIcon();
return node->icon();
break;
default:
return QVariant();
}
}
void DataSourceModel::setDataSourceManager(DataSourceManager* dataManager)
{
m_dataManager = dataManager;
connect(m_dataManager, SIGNAL(datasourcesChanged()), this, SLOT(slotDatasourcesChanged()));
updateModel();
}
void DataSourceModel::slotDatasourcesChanged()
{
beginResetModel();
updateModel();
endResetModel();
}
DataNode* DataSourceModel::nodeFromIndex(const QModelIndex& index) const
{
if (index.isValid()) {
return static_cast<DataNode*>(index.internalPointer());
} else
return m_rootNode;
}
void DataSourceModel::fillFields(DataNode* parent)
{
foreach (QString name, m_dataManager->fieldNames(parent->name())) {
parent->addChild(name, DataNode::Field, QIcon(":/report/images/field"));
}
}
void DataSourceModel::updateModel()
{
QMap<QString, DataNode*> connections;
m_rootNode->clear();
DataNode* ds = m_rootNode->addChild(tr("Datasources"), DataNode::DataSources,
QIcon(":/report/images/databases"));
foreach (QString name, m_dataManager->connectionNames()) {
DataNode* connection
= ds->addChild(name, DataNode::Connection, QIcon(":/report/images/database"));
connections.insert(name, connection);
}
foreach (QString name, m_dataManager->dataSourceNames()) {
DataNode* datasource;
if (m_dataManager->isQuery(name)) {
DataNode* connection
= connections.value(m_dataManager->queryByName(name)->connectionName());
if (connection)
datasource
= connection->addChild(name, DataNode::Query, QIcon(":/report/images/table"));
else
datasource = ds->addChild(name, DataNode::Query);
} else if (m_dataManager->isSubQuery(name)) {
DataNode* connection
= connections.value(m_dataManager->subQueryByName(name)->connectionName());
if (connection)
datasource = connection->addChild(name, DataNode::SubQuery,
QIcon(":/report/images/table"));
else
datasource = ds->addChild(name, DataNode::SubQuery);
} else {
datasource = ds->addChild(name, DataNode::Model, QIcon(":/report/images/table"));
}
fillFields(datasource);
}
DataNode* vars = m_rootNode->addChild(tr("Variables"), DataNode::Variables,
QIcon(":/report/images/folder"));
foreach (QString name, m_dataManager->variableNames()) {
vars->addChild(name, DataNode::Variable, QIcon(":/report/images/value"));
}
vars = m_rootNode->addChild(tr("External variables"), DataNode::Variables,
QIcon(":/report/images/folder"));
foreach (QString name, m_dataManager->userVariableNames()) {
vars->addChild(name, DataNode::Variable, QIcon(":/report/images/value"));
}
}
DataSourceManager::DataSourceManager(QObject* parent):
QObject(parent),
m_lastError(""),
m_designTime(false),
m_needUpdate(false),
m_dbCredentialsProvider(0),
m_hasChanges(false)
{
m_groupFunctionFactory.registerFunctionCreator(
QLatin1String("COUNT"), new ConstructorGroupFunctionCreator<CountGroupFunction>);
m_groupFunctionFactory.registerFunctionCreator(
QLatin1String("SUM"), new ConstructorGroupFunctionCreator<SumGroupFunction>);
m_groupFunctionFactory.registerFunctionCreator(
QLatin1String("AVG"), new ConstructorGroupFunctionCreator<AvgGroupFunction>);
m_groupFunctionFactory.registerFunctionCreator(
QLatin1String("MIN"), new ConstructorGroupFunctionCreator<MinGroupFunction>);
m_groupFunctionFactory.registerFunctionCreator(
QLatin1String("MAX"), new ConstructorGroupFunctionCreator<MaxGroupFunction>);
setSystemVariable(QLatin1String("#PAGE"), 1, SecondPass);
setSystemVariable(QLatin1String("#PAGE_COUNT"), 0, SecondPass);
setSystemVariable(QLatin1String("#IS_LAST_PAGEFOOTER"), false, FirstPass);
setSystemVariable(QLatin1String("#IS_FIRST_PAGEFOOTER"), false, FirstPass);
m_datasourcesModel.setDataSourceManager(this);
connect(&m_reportVariables, SIGNAL(variableHasBeenAdded(QString)), this,
SLOT(slotVariableHasBeenAdded(QString)));
connect(&m_reportVariables, SIGNAL(variableHasBeenChanged(QString)), this,
SLOT(slotVariableHasBeenChanged(QString)));
connect(&m_userVariables, SIGNAL(variableHasBeenAdded(QString)), this,
SLOT(slotVariableHasBeenAdded(QString)));
connect(&m_userVariables, SIGNAL(variableHasBeenChanged(QString)), this,
SLOT(slotVariableHasBeenChanged(QString)));
}
QString DataSourceManager::defaultDatabasePath() const { return m_defaultDatabasePath; }
void DataSourceManager::setDefaultDatabasePath(const QString& defaultDatabasePath)
{
m_defaultDatabasePath = defaultDatabasePath;
}
QString DataSourceManager::putGroupFunctionsExpressions(QString expression)
{
if (m_groupFunctionsExpressionsMap.contains(expression)) {
return QString::number(m_groupFunctionsExpressionsMap.value(expression));
} else {
m_groupFunctionsExpressions.append(expression);
m_groupFunctionsExpressionsMap.insert(expression, m_groupFunctionsExpressions.size() - 1);
return QString::number(m_groupFunctionsExpressions.size() - 1);
}
}
void DataSourceManager::clearGroupFuntionsExpressions()
{
m_groupFunctionsExpressionsMap.clear();
m_groupFunctionsExpressions.clear();
}
QString DataSourceManager::getExpression(QString index)
{
bool ok = false;
int i = index.toInt(&ok);
if (ok && m_groupFunctionsExpressions.size() > i)
return m_groupFunctionsExpressions.at(index.toInt());
else
return "";
}
bool DataSourceManager::designTime() const { return m_designTime; }
void DataSourceManager::setDesignTime(bool designTime) { m_designTime = designTime; }
DataSourceManager::~DataSourceManager()
{
clear(All);
clearGroupFunction();
}
void DataSourceManager::connectAllDatabases()
{
foreach (ConnectionDesc* conn, m_connections) {
try {
connectConnection(conn);
} catch (ReportError& e) {
putError(e.what());
setLastError(e.what());
qDebug() << e.what();
}
}
}
bool DataSourceManager::addModel(const QString& name, QAbstractItemModel* model, bool owned)
{
if (m_datasources.contains(name.toLower()))
removeDatasource(name.toLower());
ModelHolder* mh = new ModelHolder(model, owned);
try {
putHolder(name, mh);
connect(mh, SIGNAL(modelStateChanged()), this, SIGNAL(datasourcesChanged()));
} catch (ReportError& e) {
putError(e.what());
setLastError(e.what());
return false;
}
emit datasourcesChanged();
return true;
}
void DataSourceManager::removeModel(const QString& name)
{
if (m_datasources.contains(name.toLower()))
removeDatasource(name.toLower());
}
ICallbackDatasource* DataSourceManager::createCallbackDatasource(const QString& name)
{
ICallbackDatasource* ds = new CallbackDatasource();
IDataSourceHolder* holder = new CallbackDatasourceHolder(dynamic_cast<IDataSource*>(ds), true);
putHolder(name, holder);
emit datasourcesChanged();
m_needUpdate = true;
return ds;
}
void DataSourceManager::registerDbCredentialsProvider(IDbCredentialsProvider* provider)
{
m_dbCredentialsProvider = provider;
}
void DataSourceManager::addCallbackDatasource(ICallbackDatasource* datasource, const QString& name)
{
IDataSource* datasourceIntf = dynamic_cast<IDataSource*>(datasource);
if (datasourceIntf) {
IDataSourceHolder* holder = new CallbackDatasourceHolder(datasourceIntf, true);
putHolder(name, holder);
emit datasourcesChanged();
}
}
QSharedPointer<QAbstractItemModel> DataSourceManager::previewSQL(const QString& connectionName,
const QString& sqlText,
QString masterDatasource)
{
QSqlDatabase db = QSqlDatabase::database(connectionName);
if (db.isValid() && db.isOpen()) {
QSqlQueryModel* model = new QSqlQueryModel();
QMap<QString, QString> aliasesToParam;
QString queryText = replaceVariables(sqlText, aliasesToParam);
queryText = replaceFields(queryText, aliasesToParam, masterDatasource);
QSqlQuery query(db);
query.prepare(queryText);
foreach (QString param, aliasesToParam.keys()) {
QVariant value;
if (param.contains(".")) {
value = fieldData(aliasesToParam.value(param));
param = param.right(param.length() - param.indexOf('.') - 1);
} else {
value = variable(aliasesToParam.value(param));
}
if (value.isValid() || m_designTime)
query.bindValue(':' + param, value);
}
query.exec();
#if QT_VERSION >= QT_VERSION_CHECK(6, 2, 0)
model->setQuery(std::move(query));
#else
model->setQuery(query);
#endif
m_lastError = model->lastError().text();
putError(m_lastError);
if (model->query().isActive())
return QSharedPointer<QAbstractItemModel>(model);
else
return QSharedPointer<QAbstractItemModel>(0);
}
if (!db.isOpen()) {
m_lastError = tr("Connection \"%1\" is not open").arg(connectionName);
putError(m_lastError);
}
return QSharedPointer<QAbstractItemModel>(0);
}
void DataSourceManager::updateDatasourceModel()
{
m_datasourcesModel.updateModel();
emit datasourcesChanged();
m_needUpdate = false;
}
QString DataSourceManager::extractField(QString source)
{
if (source.contains('.')) {
return source.right(source.length() - (source.indexOf('.') + 1));
}
return source;
}
QString DataSourceManager::replaceVariables(QString value)
{
#if QT_VERSION >= QT_VERSION_CHECK(5, 15, 1)
QRegularExpression rx = getVariableRegEx();
QRegularExpressionMatchIterator iter = rx.globalMatch(value);
qsizetype pos = 0;
QString result;
while (iter.hasNext()) {
QRegularExpressionMatch match = iter.next();
QString var = match.captured(0);
var.remove("$V{");
var.remove("}");
result += value.mid(pos, match.capturedStart(0));
if (variable(var).isValid()) {
result += variable(var).toString();
} else {
result += QString(tr("Variable \"%1\" not found!").arg(var));
}
pos = match.capturedEnd(0);
}
result += value.mid(pos);
return result;
#else
QRegExp rx(Const::VARIABLE_RX);
if (value.contains(rx)) {
int pos = -1;
while ((pos = rx.indexIn(value)) != -1) {
QString var = rx.cap(0);
var.remove("$V{");
var.remove("}");
if (variable(var).isValid()) {
value.replace(pos, rx.cap(0).length(), variable(var).toString());
} else {
value.replace(pos, rx.cap(0).length(),
QString(tr("Variable \"%1\" not found!").arg(var)));
}
}
}
return value;
#endif
return QString();
}
QString DataSourceManager::replaceVariables(QString query, QMap<QString, QString>& aliasesToParam)
{
#if QT_VERSION >= QT_VERSION_CHECK(5, 15, 1)
QRegularExpression rx = getVariableRegEx();
int curentAliasIndex = 0;
if (query.contains(rx)) {
int pos = -1;
QRegularExpressionMatch match = rx.match(query);
while ((pos = match.capturedStart()) != -1) {
QString var = match.captured(0);
var.remove("$V{");
var.remove("}");
if (!match.captured(1).isEmpty()) {
if (aliasesToParam.contains(var)) {
curentAliasIndex++;
aliasesToParam.insert(var + "_v_alias" + QString::number(curentAliasIndex),
var);
var += "_v_alias" + QString::number(curentAliasIndex);
} else {
aliasesToParam.insert(var, var);
}
query.replace(pos, match.captured(0).length(), ":" + var);
} else {
QString varName = match.captured(2).trimmed();
QString varParam = match.captured(3).trimmed();
if (!varName.isEmpty()) {
if (!varParam.isEmpty() && varParam.compare("nobind") == 0) {
query.replace(pos, match.captured(0).length(),
variable(varName).toString());
} else {
query.replace(
pos, match.captured(0).length(),
QString(tr("Unknown parameter \"%1\" for variable \"%2\" found!")
.arg(varName)
.arg(varParam)));
}
} else {
query.replace(pos, match.captured(0).length(),
QString(tr("Variable \"%1\" not found!").arg(var)));
}
}
match = rx.match(query);
}
}
#else
QRegExp rx(Const::VARIABLE_RX);
int curentAliasIndex = 0;
if (query.contains(rx)) {
int pos = -1;
while ((pos = rx.indexIn(query)) != -1) {
QString var = rx.cap(0);
var.remove("$V{");
var.remove("}");
if (!rx.cap(1).isEmpty()) {
if (aliasesToParam.contains(var)) {
curentAliasIndex++;
aliasesToParam.insert(var + "_v_alias" + QString::number(curentAliasIndex),
var);
var += "_v_alias" + QString::number(curentAliasIndex);
} else {
aliasesToParam.insert(var, var);
}
query.replace(pos, rx.cap(0).length(), ":" + var);
} else {
QString varName = rx.cap(2).trimmed();
QString varParam = rx.cap(3).trimmed();
if (!varName.isEmpty()) {
if (!varParam.isEmpty() && varParam.compare("nobind") == 0) {
query.replace(pos, rx.cap(0).length(), variable(varName).toString());
} else {
query.replace(
pos, rx.cap(0).length(),
QString(tr("Unknown parameter \"%1\" for variable \"%2\" found!")
.arg(varName)
.arg(varParam)));
}
} else {
query.replace(pos, rx.cap(0).length(),
QString(tr("Variable \"%1\" not found!").arg(var)));
}
}
}
}
#endif
return query;
}
QString DataSourceManager::replaceFields(QString query, QMap<QString, QString>& aliasesToParam,
QString masterDatasource)
{
#if QT_VERSION >= QT_VERSION_CHECK(5, 15, 1)
QRegularExpression rx = getFieldRegEx();
int curentAliasIndex = 0;
if (query.contains(rx)) {
int pos = -1;
QRegularExpressionMatch match = rx.match(query);
while ((pos = match.capturedStart()) != -1) {
QString field = match.captured(0);
field.remove("$D{");
field.remove("}");
if (!aliasesToParam.contains(field)) {
if (field.contains("."))
aliasesToParam.insert(field, field);
else
aliasesToParam.insert(field, masterDatasource + "." + field);
} else {
curentAliasIndex++;
if (field.contains("."))
aliasesToParam.insert(field + "_f_alias" + QString::number(curentAliasIndex),
field);
else
aliasesToParam.insert(field + "_f_alias" + QString::number(curentAliasIndex),
masterDatasource + "." + field);
field += "_f_alias" + QString::number(curentAliasIndex);
}
query.replace(pos, match.capturedLength(), ":" + extractField(field));
match = rx.match(query);
}
}
#else
QRegExp rx(Const::FIELD_RX);
if (query.contains(rx)) {
int curentAliasIndex = 0;
int pos;
while ((pos = rx.indexIn(query)) != -1) {
QString field = rx.cap(0);
field.remove("$D{");
field.remove("}");
if (!aliasesToParam.contains(field)) {
if (field.contains("."))
aliasesToParam.insert(field, field);
else
aliasesToParam.insert(field, masterDatasource + "." + field);
} else {
curentAliasIndex++;
if (field.contains("."))
aliasesToParam.insert(field + "_f_alias" + QString::number(curentAliasIndex),
field);
else
aliasesToParam.insert(field + "_f_alias" + QString::number(curentAliasIndex),
masterDatasource + "." + field);
field += "_f_alias" + QString::number(curentAliasIndex);
}
query.replace(pos, rx.cap(0).length(), ":" + extractField(field));
}
}
return query;
#endif
return query;
}
void DataSourceManager::setReportVariable(const QString& name, const QVariant& value)
{
if (!containsVariable(name)) {
addVariable(name, value);
} else
changeVariable(name, value);
}
void DataSourceManager::addQuery(const QString& name, const QString& sqlText,
const QString& connectionName)
{
QueryDesc* queryDecs = new QueryDesc(name, sqlText, connectionName);
putQueryDesc(queryDecs);
putHolder(name, new QueryHolder(sqlText, connectionName, this));
m_hasChanges = true;
m_varToDataSource.clear();
emit datasourcesChanged();
}
void DataSourceManager::addSubQuery(const QString& name, const QString& sqlText,
const QString& connectionName, const QString& masterDatasource)
{
SubQueryDesc* subQueryDesc
= new SubQueryDesc(name.toLower(), sqlText, connectionName, masterDatasource);
putSubQueryDesc(subQueryDesc);
putHolder(name, new SubQueryHolder(sqlText, connectionName, masterDatasource, this));
m_hasChanges = true;
m_varToDataSource.clear();
emit datasourcesChanged();
}
void DataSourceManager::addProxy(const QString& name, const QString& master, const QString& detail,
QList<FieldsCorrelation> fields)
{
ProxyDesc* proxyDesc = new ProxyDesc();
proxyDesc->setName(name);
proxyDesc->setMaster(master);
proxyDesc->setDetail(detail);
foreach (FieldsCorrelation correlation, fields) {
proxyDesc->addFieldsCorrelation(correlation);
}
putProxyDesc(proxyDesc);
putHolder(name, new ProxyHolder(proxyDesc, this));
m_hasChanges = true;
emit datasourcesChanged();
}
void DataSourceManager::addCSV(const QString& name, const QString& csvText,
const QString& separator, bool firstRowIsHeader)
{
CSVDesc* csvDesc = new CSVDesc(name, csvText, separator, firstRowIsHeader);
putCSVDesc(csvDesc);
putHolder(name, new CSVHolder(*csvDesc, this));
m_hasChanges = true;
emit datasourcesChanged();
}
QString DataSourceManager::queryText(const QString& dataSourceName)
{
if (isQuery(dataSourceName))
return queryByName(dataSourceName)->queryText();
if (isSubQuery(dataSourceName))
return subQueryByName(dataSourceName)->queryText();
else
return QString();
}
QueryDesc* DataSourceManager::queryByName(const QString& datasourceName)
{
int queryIndex = queryIndexByName(datasourceName);
if (queryIndex != -1)
return m_queries.at(queryIndex);
return 0;
}
SubQueryDesc* DataSourceManager::subQueryByName(const QString& datasourceName)
{
int queryIndex = subQueryIndexByName(datasourceName);
if (queryIndex != -1)
return m_subqueries.at(queryIndex);
return 0;
}
ConnectionDesc* DataSourceManager::connectionByName(const QString& connectionName)
{
int queryIndex = connectionIndexByName(connectionName);
if (queryIndex != -1)
return m_connections.at(queryIndex);
return 0;
}
int DataSourceManager::queryIndexByName(const QString& dataSourceName)
{
for (int i = 0; i < m_queries.count(); i++) {
QueryDesc* desc = m_queries.at(i);
if (desc->queryName().compare(dataSourceName, Qt::CaseInsensitive) == 0)
return i;
}
return -1;
}
int DataSourceManager::subQueryIndexByName(const QString& dataSourceName)
{
for (int i = 0; i < m_subqueries.count(); ++i) {
QueryDesc* desc = m_subqueries.at(i);
if (desc->queryName().compare(dataSourceName, Qt::CaseInsensitive) == 0)
return i;
}
return -1;
}
int DataSourceManager::proxyIndexByName(const QString& dataSourceName)
{
for (int i = 0; i < m_proxies.count(); ++i) {
ProxyDesc* desc = m_proxies.at(i);
if (desc->name().compare(dataSourceName, Qt::CaseInsensitive) == 0)
return i;
}
return -1;
}
int DataSourceManager::csvIndexByName(const QString& dataSourceName)
{
for (int i = 0; i < m_csvs.count(); ++i) {
CSVDesc* desc = m_csvs.at(i);
if (desc->name().compare(dataSourceName, Qt::CaseInsensitive) == 0)
return i;
}
return -1;
}
int DataSourceManager::connectionIndexByName(const QString& connectionName)
{
for (int i = 0; i < m_connections.count(); ++i) {
ConnectionDesc* desc = m_connections.at(i);
if (desc->name().compare(connectionName, Qt::CaseInsensitive) == 0)
return i;
}
return -1;
}
QList<ConnectionDesc*>& DataSourceManager::conections() { return m_connections; }
bool DataSourceManager::dataSourceIsValid(const QString& name)
{
if (m_datasources.value(name.toLower()))
return !m_datasources.value(name.toLower())->isInvalid();
else
throw ReportError(tr("Datasource \"%1\" not found!").arg(name));
}
bool DataSourceManager::isQuery(const QString& dataSourceName)
{
return (queryByName(dataSourceName));
}
QString DataSourceManager::connectionName(const QString& dataSourceName)
{
if (isQuery(dataSourceName))
return queryByName(dataSourceName)->connectionName();
if (isSubQuery(dataSourceName))
return subQueryByName(dataSourceName)->connectionName();
return QString();
}
ProxyDesc* DataSourceManager::proxyByName(const QString& datasourceName)
{
int proxyIndex = proxyIndexByName(datasourceName);
if (proxyIndex > -1)
return m_proxies.at(proxyIndex);
else
return 0;
}
CSVDesc* DataSourceManager::csvByName(const QString& datasourceName)
{
int csvIndex = csvIndexByName(datasourceName);
if (csvIndex > -1)
return m_csvs.at(csvIndex);
else
return 0;
}
void DataSourceManager::removeDatasource(const QString& name)
{
if (m_datasources.contains(name)) {
IDataSourceHolder* holder;
holder = m_datasources.value(name);
m_datasources.remove(name);
delete holder;
}
if (isQuery(name)) {
int queryIndex = queryIndexByName(name);
delete m_queries.at(queryIndex);
m_queries.removeAt(queryIndex);
}
if (isSubQuery(name)) {
int queryIndex = subQueryIndexByName(name);
delete m_subqueries.at(queryIndex);
m_subqueries.removeAt(queryIndex);
}
if (isProxy(name)) {
int proxyIndex = proxyIndexByName(name);
delete m_proxies.at(proxyIndex);
m_proxies.removeAt(proxyIndex);
}
if (isCSV(name)) {
int csvIndex = csvIndexByName(name);
delete m_csvs.at(csvIndex);
m_csvs.removeAt(csvIndex);
}
invalidateLinkedDatasources(name);
m_hasChanges = true;
emit datasourcesChanged();
}
void DataSourceManager::removeConnection(const QString& connectionName)
{
QList<ConnectionDesc*>::iterator cit = m_connections.begin();
while (cit != m_connections.end()) {
if (((*cit)->name().compare(connectionName) == 0)) {
if ((*cit)->isInternal()) {
QSqlDatabase db = QSqlDatabase::database(connectionName);
db.close();
}
QSqlDatabase::removeDatabase(connectionName);
delete (*cit);
cit = m_connections.erase(cit);
} else {
++cit;
}
}
m_hasChanges = true;
emit datasourcesChanged();
}
void DataSourceManager::addConnectionDesc(ConnectionDesc* connection)
{
if (!isConnection(connection->name())) {
connect(connection, SIGNAL(nameChanged(QString, QString)), this,
SLOT(slotConnectionRenamed(QString, QString)));
m_connections.append(connection);
m_hasChanges = true;
if (connection->autoconnect()) {
try {
connectConnection(connection);
} catch (ReportError& exception) {
qDebug() << exception.what();
}
}
} else {
throw ReportError(
tr("Connection with name \"%1\" already exists!").arg(connection->name()));
}
}
bool DataSourceManager::checkConnectionDesc(ConnectionDesc* connection)
{
if (connectConnection(connection)) {
if (connection->isInternal()) {
QSqlDatabase::removeDatabase(connection->name());
if (designTime())
emit datasourcesChanged();
}
return true;
}
if (connection->isInternal())
QSqlDatabase::removeDatabase(connection->name());
return false;
}
void DataSourceManager::putHolder(const QString& name, IDataSourceHolder* dataSource)
{
if (!m_datasources.contains(name.toLower())) {
m_datasources.insert(name.toLower(), dataSource);
} else
throw ReportError(tr("Datasource with name \"%1\" already exists!").arg(name));
}
void DataSourceManager::putQueryDesc(QueryDesc* queryDesc)
{
if (!containsDatasource(queryDesc->queryName())) {
m_queries.append(queryDesc);
connect(queryDesc, SIGNAL(queryTextChanged(QString, QString)), this,
SLOT(slotQueryTextChanged(QString, QString)));
} else
throw ReportError(
tr("Datasource with name \"%1\" already exists!").arg(queryDesc->queryName()));
}
void DataSourceManager::putSubQueryDesc(SubQueryDesc* subQueryDesc)
{
if (!containsDatasource(subQueryDesc->queryName())) {
m_subqueries.append(subQueryDesc);
connect(subQueryDesc, SIGNAL(queryTextChanged(QString, QString)), this,
SLOT(slotQueryTextChanged(QString, QString)));
} else
throw ReportError(
tr("Datasource with name \"%1\" already exists!").arg(subQueryDesc->queryName()));
}
void DataSourceManager::putProxyDesc(ProxyDesc* proxyDesc)
{
if (!containsDatasource(proxyDesc->name())) {
m_proxies.append(proxyDesc);
} else
throw ReportError(tr("Datasource with name \"%1\" already exists!").arg(proxyDesc->name()));
}
void DataSourceManager::putCSVDesc(CSVDesc* csvDesc)
{
if (!containsDatasource(csvDesc->name())) {
m_csvs.append(csvDesc);
connect(csvDesc, SIGNAL(cvsTextChanged(QString, QString)), this,
SLOT(slotCSVTextChanged(QString, QString)));
} else
throw ReportError(tr("Datasource with name \"%1\" already exists!").arg(csvDesc->name()));
}
bool DataSourceManager::initAndOpenDB(QSqlDatabase& db, ConnectionDesc& connectionDesc)
{
bool connected = false;
db.setHostName(replaceVariables(connectionDesc.host()));
db.setUserName(replaceVariables(connectionDesc.userName()));
db.setPassword(replaceVariables(connectionDesc.password()));
db.setDatabaseName(replaceVariables(connectionDesc.databaseName()));
if (connectionDesc.port() != "")
db.setPort(replaceVariables(connectionDesc.port()).toInt());
if (!connectionDesc.keepDBCredentials() && m_dbCredentialsProvider) {
if (!m_dbCredentialsProvider->getUserName(connectionDesc.name()).isEmpty())
db.setUserName(m_dbCredentialsProvider->getUserName(connectionDesc.name()));
if (!m_dbCredentialsProvider->getPassword(connectionDesc.name()).isEmpty())
db.setPassword(m_dbCredentialsProvider->getPassword(connectionDesc.name()));
}
QString dbName = replaceVariables(connectionDesc.databaseName());
if (connectionDesc.driver().compare("QSQLITE") == 0) {
if (!defaultDatabasePath().isEmpty()) {
dbName = !QFileInfo(dbName).exists()
? defaultDatabasePath() + QFileInfo(dbName).fileName()
: dbName;
}
if (QFileInfo(dbName).exists()) {
db.setDatabaseName(dbName);
} else {
setLastError(tr("Database \"%1\" not found").arg(dbName));
return false;
}
} else {
db.setDatabaseName(dbName);
}
connected = db.open();
if (!connected)
setLastError(db.lastError().text());
return connected;
}
ReportSettings* DataSourceManager::reportSettings() const { return m_reportSettings; }
void DataSourceManager::setReportSettings(ReportSettings* reportSettings)
{
m_reportSettings = reportSettings;
}
bool DataSourceManager::checkConnection(QSqlDatabase db)
{
QSqlQuery query("Select 1", db);
return query.first();
}
bool DataSourceManager::connectConnection(ConnectionDesc* connectionDesc)
{
bool connected = false;
clearErrors();
QString lastError = "";
foreach (QString datasourceName, dataSourceNames()) {
dataSourceHolder(datasourceName)->clearErrors();
}
if (!QSqlDatabase::contains(connectionDesc->name())) {
QString dbError;
{
QSqlDatabase db
= QSqlDatabase::addDatabase(connectionDesc->driver(), connectionDesc->name());
connectionDesc->setInternal(true);
connected = initAndOpenDB(db, *connectionDesc);
dbError = db.lastError().text();
}
if (!connected) {
if (!dbError.trimmed().isEmpty())
setLastError(dbError);
QSqlDatabase::removeDatabase(connectionDesc->name());
return false;
}
} else {
QSqlDatabase db = QSqlDatabase::database(connectionDesc->name());
if (!connectionDesc->isEqual(db) && connectionDesc->isInternal()) {
db.close();
connected = initAndOpenDB(db, *connectionDesc);
} else {
connected = checkConnection(db);
if (!connected && connectionDesc->isInternal())
connected = initAndOpenDB(db, *connectionDesc);
}
}
if (!connected) {
if (connectionDesc->isInternal())
QSqlDatabase::removeDatabase(connectionDesc->name());
return false;
} else {
foreach (QString datasourceName, dataSourceNames()) {
if (isQuery(datasourceName)) {
QueryHolder* qh = dynamic_cast<QueryHolder*>(dataSourceHolder(datasourceName));
if (qh) {
qh->invalidate(designTime() ? IDataSource::DESIGN_MODE
: IDataSource::RENDER_MODE);
invalidateChildren(datasourceName);
}
}
}
foreach (QString datasourceName, dataSourceNames()) {
if (isProxy(datasourceName)) {
ProxyHolder* ph = dynamic_cast<ProxyHolder*>(dataSourceHolder(datasourceName));
if (ph) {
ph->invalidate(designTime() ? IDataSource::DESIGN_MODE
: IDataSource::RENDER_MODE);
}
}
}
if (designTime())
emit datasourcesChanged();
}
return true;
}
void DataSourceManager::clearReportVariables() { m_reportVariables.clearUserVariables(); }
void DataSourceManager::connectAutoConnections()
{
foreach (ConnectionDesc* conn, m_connections) {
if (conn->autoconnect()) {
try {
connectConnection(conn);
} catch (ReportError& e) {
setLastError(e.what());
putError(e.what());
qDebug() << e.what();
}
}
}
}
QList<QString> DataSourceManager::childDatasources(const QString& parentDatasourceName)
{
QList<QString> result;
foreach (QString datasourceName, dataSourceNames()) {
if (isSubQuery(datasourceName)) {
SubQueryHolder* sh = dynamic_cast<SubQueryHolder*>(dataSourceHolder(datasourceName));
if (sh
&& sh->masterDatasource().compare(parentDatasourceName, Qt::CaseInsensitive) == 0) {
result.append(datasourceName);
}
}
}
return result;
}
void DataSourceManager::invalidateChildren(const QString& parentDatasourceName)
{
foreach (QString datasourceName, childDatasources(parentDatasourceName)) {
SubQueryHolder* sh = dynamic_cast<SubQueryHolder*>(dataSourceHolder(datasourceName));
if (sh)
sh->invalidate(designTime() ? IDataSource::DESIGN_MODE : IDataSource::RENDER_MODE);
invalidateChildren(datasourceName);
}
}
bool DataSourceManager::containsDatasource(const QString& dataSourceName)
{
return m_datasources.contains(dataSourceName.toLower());
}
bool DataSourceManager::isSubQuery(const QString& dataSourceName)
{
return subQueryIndexByName(dataSourceName) != -1;
}
bool DataSourceManager::isProxy(const QString& dataSourceName)
{
return proxyIndexByName(dataSourceName) != -1;
}
bool DataSourceManager::isCSV(const QString& datasourceName)
{
return csvIndexByName(datasourceName) != -1;
}
bool DataSourceManager::isConnection(const QString& connectionName)
{
return connectionIndexByName(connectionName) != -1;
}
bool DataSourceManager::isConnectionConnected(const QString& connectionName)
{
if (isConnection(connectionName) && QSqlDatabase::contains(connectionName)) {
QSqlDatabase db = QSqlDatabase::database(connectionName);
return db.isValid() && QSqlDatabase::database(connectionName).isOpen();
}
return false;
}
bool DataSourceManager::connectConnection(const QString& connectionName)
{
return connectConnection(connectionByName(connectionName));
}
void DataSourceManager::disconnectConnection(const QString& connectionName)
{
foreach (QString datasourceName, dataSourceNames()) {
if (isQuery(datasourceName) || isSubQuery(datasourceName)) {
QueryHolder* qh = dynamic_cast<QueryHolder*>(dataSourceHolder(datasourceName));
if (qh && qh->connectionName().compare(connectionName, Qt::CaseInsensitive) == 0) {
qh->invalidate(designTime() ? IDataSource::DESIGN_MODE : IDataSource::RENDER_MODE,
true);
qh->setLastError(tr("invalid connection"));
}
}
}
ConnectionDesc* connectionDesc = connectionByName(connectionName);
if (connectionDesc->isInternal()) {
{
QSqlDatabase db = QSqlDatabase::database(connectionName);
if (db.isOpen())
db.close();
}
if (QSqlDatabase::contains(connectionName))
QSqlDatabase::removeDatabase(connectionName);
}
}
IDataSource* DataSourceManager::dataSource(const QString& name)
{
IDataSourceHolder* holder = m_datasources.value(name.toLower());
if (holder) {
if (holder->isInvalid()) {
setLastError(name + " : " + holder->lastError());
return 0;
} else {
return holder->dataSource(designTime() ? IDataSource::DESIGN_MODE
: IDataSource::RENDER_MODE);
}
} else {
setLastError(tr("Datasource \"%1\" not found!").arg(name));
return 0;
}
}
IDataSourceHolder* DataSourceManager::dataSourceHolder(const QString& name)
{
if (m_datasources.value(name.toLower()))
return m_datasources.value(name.toLower());
else
return 0;
}
QStringList DataSourceManager::dataSourceNames()
{
QStringList result;
for (int i = 0; i < m_datasources.keys().count(); i++) {
result.append(m_datasources.keys().at(i));
}
result.sort();
return result;
}
QStringList DataSourceManager::dataSourceNames(const QString& connectionName)
{
QStringList result;
foreach (QueryDesc* query, m_queries) {
if (query->connectionName().compare(connectionName, Qt::CaseInsensitive) == 0) {
result.append(query->queryName());
}
}
foreach (QueryDesc* query, m_subqueries) {
if (query->connectionName().compare(connectionName, Qt::CaseInsensitive) == 0) {
result.append(query->queryName());
}
}
return result;
}
QStringList DataSourceManager::connectionNames()
{
QStringList result;
foreach (ConnectionDesc* conDesc, m_connections) {
result.append(conDesc->name());
}
return result;
}
QStringList DataSourceManager::fieldNames(const QString& datasourceName)
{
QStringList result;
IDataSource* ds = dataSource(datasourceName);
if (ds && !ds->isInvalid()) {
for (int i = 0; i < ds->columnCount(); i++) {
result.append(ds->columnNameByIndex(i));
}
result.sort();
}
return result;
}
void DataSourceManager::addConnection(const QString& connectionName)
{
addConnectionDesc(new ConnectionDesc(QSqlDatabase::database(connectionName)));
m_hasChanges = true;
emit datasourcesChanged();
}
QObject* DataSourceManager::createElement(const QString& collectionName, const QString&)
{
if (collectionName == "connections") {
ConnectionDesc* connection = new ConnectionDesc;
m_connections.append(connection);
return connection;
}
if (collectionName == "queries") {
QueryDesc* queryDesc = new QueryDesc;
m_queries.append(queryDesc);
return queryDesc;
}
if (collectionName == "subqueries") {
SubQueryDesc* subQueryDesc = new SubQueryDesc;
m_subqueries.append(subQueryDesc);
return subQueryDesc;
}
if (collectionName == "subproxies") {
ProxyDesc* proxyDesc = new ProxyDesc;
m_proxies.append(proxyDesc);
return proxyDesc;
}
if (collectionName == "variables") {
VarDesc* var = new VarDesc;
m_tempVars.append(var);
return var;
}
if (collectionName == "csvs") {
CSVDesc* csvDesc = new CSVDesc;
m_csvs.append(csvDesc);
return csvDesc;
}
return 0;
}
int DataSourceManager::elementsCount(const QString& collectionName)
{
if (collectionName == "connections") {
return m_connections.count();
}
if (collectionName == "queries") {
return m_queries.count();
}
if (collectionName == "subqueries") {
return m_subqueries.count();
}
if (collectionName == "subproxies") {
return m_proxies.count();
}
if (collectionName == "variables") {
return m_reportVariables.variablesCount();
}
if (collectionName == "csvs") {
return m_csvs.count();
}
return 0;
}
QObject* DataSourceManager::elementAt(const QString& collectionName, int index)
{
if (collectionName == "connections") {
return m_connections.at(index);
}
if (collectionName == "queries") {
return m_queries.at(index);
}
if (collectionName == "subqueries") {
return m_subqueries.at(index);
}
if (collectionName == "subproxies") {
return m_proxies.at(index);
}
if (collectionName == "variables") {
return m_reportVariables.variableAt(index);
}
if (collectionName == "csvs") {
return m_csvs.at(index);
}
return 0;
}
void DataSourceManager::collectionLoadFinished(const QString& collectionName)
{
EASY_BLOCK("DataSourceManager::collectionLoadFinished");
if (collectionName.compare("connections", Qt::CaseInsensitive) == 0) { }
EASY_BLOCK("queryes");
if (collectionName.compare("queries", Qt::CaseInsensitive) == 0) {
QMutableListIterator<QueryDesc*> it(m_queries);
while (it.hasNext()) {
it.next();
if (!m_datasources.contains(it.value()->queryName().toLower())) {
connect(it.value(), SIGNAL(queryTextChanged(QString, QString)), this,
SLOT(slotQueryTextChanged(QString, QString)));
putHolder(
it.value()->queryName(),
new QueryHolder(it.value()->queryText(), it.value()->connectionName(), this));
} else {
delete it.value();
it.remove();
}
}
}
EASY_END_BLOCK;
EASY_BLOCK("subqueries")
if (collectionName.compare("subqueries", Qt::CaseInsensitive) == 0) {
QMutableListIterator<SubQueryDesc*> it(m_subqueries);
while (it.hasNext()) {
it.next();
if (!m_datasources.contains(it.value()->queryName().toLower())) {
connect(it.value(), SIGNAL(queryTextChanged(QString, QString)), this,
SLOT(slotQueryTextChanged(QString, QString)));
putHolder(it.value()->queryName(),
new SubQueryHolder(it.value()->queryText(), it.value()->connectionName(),
it.value()->master(), this));
} else {
delete it.value();
it.remove();
}
}
}
EASY_END_BLOCK;
EASY_BLOCK("subproxies");
if (collectionName.compare("subproxies", Qt::CaseInsensitive) == 0) {
QMutableListIterator<ProxyDesc*> it(m_proxies);
while (it.hasNext()) {
it.next();
if (!m_datasources.contains(it.value()->name().toLower())) {
putHolder(it.value()->name(), new ProxyHolder(it.value(), this));
} else {
delete it.value();
it.remove();
}
}
}
EASY_END_BLOCK;
EASY_BLOCK("variables");
if (collectionName.compare("variables", Qt::CaseInsensitive) == 0) {
foreach (VarDesc* item, m_tempVars) {
if (!m_reportVariables.containsVariable(item->name())) {
m_reportVariables.addVariable(item->name(), item->value(), VarDesc::Report,
FirstPass);
VarDesc* currentVar = m_reportVariables.variableByName(item->name());
currentVar->initFrom(item);
}
delete item;
}
m_tempVars.clear();
}
EASY_END_BLOCK;
if (collectionName.compare("csvs", Qt::CaseInsensitive) == 0) {
QMutableListIterator<CSVDesc*> it(m_csvs);
while (it.hasNext()) {
it.next();
if (!m_datasources.contains(it.value()->name().toLower())) {
connect(it.value(), SIGNAL(cvsTextChanged(QString, QString)), this,
SLOT(slotCSVTextChanged(QString, QString)));
putHolder(it.value()->name(), new CSVHolder(*it.value(), this));
} else {
delete it.value();
it.remove();
}
}
}
if (designTime()) {
EASY_BLOCK("emit datasourcesChanged()");
emit datasourcesChanged();
EASY_END_BLOCK;
}
EASY_BLOCK("emit loadCollectionFinished(collectionName)");
emit loadCollectionFinished(collectionName);
EASY_END_BLOCK;
EASY_END_BLOCK;
}
void DataSourceManager::addVariable(const QString& name, const QVariant& value,
VarDesc::VarType type, RenderPass pass)
{
if (type == VarDesc::User) {
m_userVariables.addVariable(name, value, type, pass);
} else {
m_reportVariables.addVariable(name, value, type, pass);
}
if (designTime()) {
EASY_BLOCK("DataSourceManager::addVariable emit ds changed");
emit datasourcesChanged();
EASY_END_BLOCK;
}
}
void DataSourceManager::deleteVariable(const QString& name)
{
m_userVariables.deleteVariable(name);
if (m_reportVariables.containsVariable(name)
&& m_reportVariables.variableType(name) == VarDesc::Report) {
m_reportVariables.deleteVariable(name);
if (designTime()) {
m_hasChanges = true;
emit datasourcesChanged();
}
}
}
void DataSourceManager::changeVariable(const QString& name, const QVariant& value)
{
if (m_userVariables.containsVariable(name)) {
m_userVariables.changeVariable(name, value);
}
if (m_reportVariables.containsVariable(name)) {
m_reportVariables.changeVariable(name, value);
}
}
void DataSourceManager::setSystemVariable(const QString& name, const QVariant& value,
RenderPass pass)
{
addVariable(name, value, VarDesc::System, pass);
}
void DataSourceManager::setLastError(const QString& value)
{
m_lastError = value;
if (!value.isEmpty() && !m_errorsList.contains(value)) {
m_errorsList.append(value);
}
}
void DataSourceManager::invalidateLinkedDatasources(QString datasourceName)
{
foreach (QString name, dataSourceNames()) {
if (isSubQuery(name)) {
if (subQueryByName(name)->master().compare(datasourceName) == 0)
dataSourceHolder(name)->invalidate(designTime() ? IDataSource::DESIGN_MODE
: IDataSource::RENDER_MODE);
}
if (isProxy(name)) {
ProxyDesc* proxy = proxyByName(name);
if ((proxy->master().compare(datasourceName) == 0)
|| (proxy->child().compare(datasourceName) == 0))
dataSourceHolder(name)->invalidate(designTime() ? IDataSource::DESIGN_MODE
: IDataSource::RENDER_MODE);
}
}
}
void DataSourceManager::slotConnectionRenamed(const QString& oldName, const QString& newName)
{
foreach (QueryDesc* query, m_queries) {
if (query->connectionName().compare(oldName, Qt::CaseInsensitive) == 0)
query->setConnectionName(newName);
}
foreach (SubQueryDesc* query, m_subqueries) {
if (query->connectionName().compare(oldName, Qt::CaseInsensitive) == 0)
query->setConnectionName(newName);
}
}
void DataSourceManager::slotQueryTextChanged(const QString& queryName, const QString& queryText)
{
QueryHolder* holder = dynamic_cast<QueryHolder*>(m_datasources.value(queryName));
if (holder) {
holder->setQueryText(queryText);
}
m_varToDataSource.clear();
}
void DataSourceManager::invalidateQueriesContainsVariable(const QString& variableName)
{
if (!variableIsSystem(variableName)) {
if (m_varToDataSource.contains(variableName)) {
foreach (QString datasourceName, m_varToDataSource.value(variableName)) {
QueryHolder* holder
= dynamic_cast<QueryHolder*>(m_datasources.value(datasourceName));
if (holder)
holder->invalidate(designTime() ? IDataSource::DESIGN_MODE
: IDataSource::RENDER_MODE);
}
} else {
QVector<QString> datasources;
foreach (const QString& datasourceName, dataSourceNames()) {
QueryHolder* holder
= dynamic_cast<QueryHolder*>(m_datasources.value(datasourceName));
if (holder) {
#if (QT_VERSION < QT_VERSION_CHECK(5, 15, 1))
QRegExp rx(QString(Const::NAMED_VARIABLE_RX).arg(variableName));
#else
QRegularExpression rx = getNamedVariableRegEx(variableName);
#endif
if (holder->queryText().contains(rx)) {
holder->invalidate(designTime() ? IDataSource::DESIGN_MODE
: IDataSource::RENDER_MODE);
datasources.append(datasourceName);
}
}
}
m_varToDataSource.insert(variableName, datasources);
}
}
}
void DataSourceManager::slotVariableHasBeenAdded(const QString& variableName)
{
invalidateQueriesContainsVariable(variableName);
if (variableType(variableName) == VarDesc::Report)
m_hasChanges = true;
}
void DataSourceManager::slotVariableHasBeenChanged(const QString& variableName)
{
invalidateQueriesContainsVariable(variableName);
if (variableType(variableName) == VarDesc::Report)
m_hasChanges = true;
}
void DataSourceManager::slotCSVTextChanged(const QString& csvName, const QString& csvText)
{
CSVHolder* holder = dynamic_cast<CSVHolder*>(m_datasources.value(csvName));
if (holder) {
holder->setCSVText(csvText);
}
}
void DataSourceManager::clear(ClearMethod method)
{
m_varToDataSource.clear();
DataSourcesMap::iterator dit;
for (dit = m_datasources.begin(); dit != m_datasources.end();) {
bool owned = (*dit)->isOwned() && (*dit)->isRemovable();
switch (method) {
case All:
invalidateLinkedDatasources(dit.key());
delete dit.value();
dit = m_datasources.erase(dit);
break;
default:
if (owned) {
invalidateLinkedDatasources(dit.key());
delete dit.value();
dit = m_datasources.erase(dit);
} else {
++dit;
}
}
}
QList<ConnectionDesc*>::iterator cit = m_connections.begin();
while (cit != m_connections.end()) {
if ((*cit)->isInternal())
QSqlDatabase::removeDatabase((*cit)->name());
delete (*cit);
cit = m_connections.erase(cit);
}
// TODO: add smart pointes to collections
foreach (QueryDesc* desc, m_queries)
delete desc;
foreach (SubQueryDesc* desc, m_subqueries)
delete desc;
foreach (ProxyDesc* desc, m_proxies)
delete desc;
m_queries.clear();
m_subqueries.clear();
m_proxies.clear();
// if (method == All)
// clearUserVariables();
clearReportVariables();
emit cleared();
}
void DataSourceManager::clearGroupFunction()
{
foreach (GroupFunction* gf, m_groupFunctions.values()) {
delete gf;
}
m_groupFunctions.clear();
}
void DataSourceManager::clearGroupFunctionValues(const QString& bandObjectName)
{
foreach (GroupFunction* gf, m_groupFunctions.values(bandObjectName)) {
gf->values().clear();
}
}
GroupFunction* DataSourceManager::addGroupFunction(const QString& name, const QString& expression,
const QString& band, const QString& dataBand)
{
GroupFunction* gf
= m_groupFunctionFactory.createGroupFunction(name, expression, dataBand, this);
if (gf) {
m_groupFunctions.insert(band, gf);
}
return gf;
}
GroupFunction* DataSourceManager::groupFunction(const QString& name, const QString& expression,
const QString& band)
{
foreach (GroupFunction* gf, m_groupFunctions.values(band)) {
if ((gf->name().compare(name, Qt::CaseInsensitive) == 0)
&& (gf->data().compare(expression, Qt::CaseInsensitive) == 0))
return gf;
}
return 0;
}
QList<GroupFunction*> DataSourceManager::groupFunctionsByBand(const QString& band)
{
return m_groupFunctions.values(band);
}
void DataSourceManager::updateChildrenData(const QString& datasourceName)
{
foreach (SubQueryDesc* subquery, m_subqueries) {
if (subquery->master().compare(datasourceName, Qt::CaseInsensitive) == 0) {
SubQueryHolder* holder
= dynamic_cast<SubQueryHolder*>(dataSourceHolder(subquery->queryName()));
if (holder)
holder->runQuery();
}
}
foreach (ProxyDesc* subproxy, m_proxies) {
if (subproxy->master().compare(datasourceName, Qt::CaseInsensitive) == 0) {
ProxyHolder* holder = dynamic_cast<ProxyHolder*>(dataSourceHolder(subproxy->name()));
holder->filterModel();
}
}
}
QString DataSourceManager::extractDataSource(const QString& fieldName)
{
return fieldName.left(fieldName.indexOf('.'));
}
QString DataSourceManager::extractFieldName(const QString& fieldName)
{
return fieldName.right((fieldName.length() - fieldName.indexOf('.')) - 1);
}
bool DataSourceManager::containsField(const QString& fieldName)
{
IDataSource* ds = dataSource(extractDataSource(fieldName));
if (ds) {
return ds->columnIndexByName(extractFieldName(fieldName)) != -1;
}
return false;
}
bool DataSourceManager::containsVariable(const QString& variableName)
{
if (m_userVariables.containsVariable(variableName))
return true;
return m_reportVariables.containsVariable(variableName);
}
void DataSourceManager::clearUserVariables()
{
m_userVariables.clearUserVariables();
m_reportVariables.clearUserVariables();
}
QVariant DataSourceManager::fieldData(const QString& fieldName)
{
if (containsField(fieldName)) {
IDataSource* ds = dataSource(extractDataSource(fieldName));
if (ds)
return ds->data(extractFieldName(fieldName));
}
return QVariant();
}
QVariant DataSourceManager::fieldDataByRowIndex(const QString& fieldName, int rowIndex)
{
if (containsField(fieldName)) {
IDataSource* ds = dataSource(extractDataSource(fieldName));
if (ds) {
return ds->dataByRowIndex(extractFieldName(fieldName), rowIndex);
}
}
return QVariant();
}
QVariant DataSourceManager::fieldDataByRowIndex(const QString& fieldName, int rowIndex, int role)
{
if (containsField(fieldName)) {
IDataSource* ds = dataSource(extractDataSource(fieldName));
if (ds) {
return ds->dataByRowIndex(extractFieldName(fieldName), rowIndex, role);
}
}
return QVariant();
}
QVariant DataSourceManager::fieldDataByRowIndex(const QString& fieldName, int rowIndex,
const QString& roleName)
{
if (containsField(fieldName)) {
IDataSource* ds = dataSource(extractDataSource(fieldName));
if (ds) {
return ds->dataByRowIndex(extractFieldName(fieldName), rowIndex, roleName);
}
}
return QVariant();
}
QVariant DataSourceManager::fieldDataByKey(const QString& datasourceName,
const QString& valueFieldName,
const QString& keyFieldName, QVariant keyValue)
{
IDataSource* ds = dataSource(datasourceName);
if (ds) {
return ds->dataByKeyField(valueFieldName, keyFieldName, keyValue);
}
return QVariant();
}
QVariant DataSourceManager::headerData(const QString& fieldName, const QString& roleName)
{
if (containsField(fieldName)) {
IDataSource* ds = dataSource(extractDataSource(fieldName));
if (ds) {
return ds->headerData(extractFieldName(fieldName), roleName);
}
}
return QVariant();
}
QString DataSourceManager::columnName(const QString& datasourceName, int index)
{
IDataSource* ds = dataSource(datasourceName);
if (ds && !ds->isInvalid() && ds->columnCount() > index) {
return ds->columnNameByIndex(index);
}
return QString("unknown");
}
int DataSourceManager::columnCount(const QString& datasourceName)
{
IDataSource* ds = dataSource(datasourceName);
if (ds && !ds->isInvalid()) {
return ds->columnCount();
}
return -1;
}
void DataSourceManager::reopenDatasource(const QString& datasourceName)
{
QueryHolder* qh = dynamic_cast<QueryHolder*>(dataSourceHolder(datasourceName));
if (qh) {
qh->invalidate(designTime() ? IDataSource::DESIGN_MODE : IDataSource::RENDER_MODE);
invalidateChildren(datasourceName);
}
}
QVariant DataSourceManager::variable(const QString& variableName)
{
if (m_userVariables.containsVariable(variableName))
return m_userVariables.variable(variableName);
return m_reportVariables.variable(variableName);
}
RenderPass DataSourceManager::variablePass(const QString& name)
{
if (m_userVariables.containsVariable(name))
return m_userVariables.variablePass(name);
return m_reportVariables.variablePass(name);
}
bool DataSourceManager::variableIsSystem(const QString& name)
{
if (m_reportVariables.containsVariable(name))
return (m_reportVariables.variableType(name) == VarDesc::System);
return false;
}
bool DataSourceManager::variableIsMandatory(const QString& name)
{
if (m_reportVariables.containsVariable(name))
return m_reportVariables.variableByName(name)->isMandatory();
return false;
}
void DataSourceManager::setVarableMandatory(const QString& name, bool value)
{
if (m_reportVariables.containsVariable(name))
m_reportVariables.variableByName(name)->setMandatory(value);
}
QStringList DataSourceManager::variableNames() { return m_reportVariables.variableNames(); }
QStringList DataSourceManager::variableNamesByRenderPass(RenderPass pass)
{
QStringList result;
foreach (QString variableName, m_reportVariables.variableNames()) {
if (m_reportVariables.variablePass(variableName) == pass) {
result.append(variableName);
}
}
return result;
}
QStringList DataSourceManager::userVariableNames() { return m_userVariables.variableNames(); }
VarDesc::VarType DataSourceManager::variableType(const QString& name)
{
if (m_reportVariables.containsVariable(name))
return m_reportVariables.variableType(name);
return VarDesc::User;
}
VariableDataType DataSourceManager::variableDataType(const QString& name)
{
if (m_reportVariables.containsVariable(name))
return m_reportVariables.variableByName(name)->dataType();
return Enums::Undefined;
}
void DataSourceManager::setVariableDataType(const QString& name, VariableDataType value)
{
if (m_reportVariables.containsVariable(name))
m_reportVariables.variableByName(name)->setDataType(value);
}
void DataSourceManager::setAllDatasourcesToFirst()
{
foreach (IDataSourceHolder* ds, m_datasources.values()) {
if (ds->dataSource())
ds->dataSource()->first();
}
}
} // namespace LimeReport