File indexing completed on 2024-04-14 14:53:33

 /* This file is part of the KDE project
    Copyright (C) 2003-2011 Jarosław Staniek <staniek@kde.org>
 
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 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
    Library General Public License for more details.
 
    You should have received a copy of the GNU Library General Public License
    along with this library; see the file COPYING.LIB.  If not, write to
    the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  * Boston, MA 02110-1301, USA.
 */
 
 #include "KDbRecordEditBuffer.h"
 #include "KDb.h"
 #include "KDbQueryColumnInfo.h"
 #include "kdb_debug.h"
 
 KDbRecordEditBuffer::KDbRecordEditBuffer(bool dbAwareBuffer)
         : m_simpleBuffer(dbAwareBuffer ? nullptr : new SimpleMap())
         , m_simpleBufferIt(dbAwareBuffer ? nullptr : new SimpleMap::ConstIterator())
         , m_dbBuffer(dbAwareBuffer ? new DbHash() : nullptr)
         , m_dbBufferIt(dbAwareBuffer ? new DbHash::Iterator() : nullptr)
         , m_defaultValuesDbBuffer(dbAwareBuffer ? new QMap<KDbQueryColumnInfo*, bool>() : nullptr)
         , m_defaultValuesDbBufferIt(dbAwareBuffer ? new QMap<KDbQueryColumnInfo*, bool>::ConstIterator() : nullptr)
 {
 }
 
 KDbRecordEditBuffer::~KDbRecordEditBuffer()
 {
     delete m_simpleBuffer;
     delete m_simpleBufferIt;
     delete m_dbBuffer;
     delete m_dbBufferIt;
     delete m_defaultValuesDbBuffer;
     delete m_defaultValuesDbBufferIt;
 }
 
 bool KDbRecordEditBuffer::isDBAware() const
 {
     return m_dbBuffer != nullptr;
 }
 
 const QVariant* KDbRecordEditBuffer::at(KDbQueryColumnInfo* ci, bool useDefaultValueIfPossible) const
 {
     if (!ci) {
         kdbWarning() << "no KDbQueryColumnInfo provided";
         return nullptr;
     }
     if (!m_dbBuffer) {
         kdbWarning() << "not db-aware buffer!";
         return nullptr;
     }
     *m_dbBufferIt = m_dbBuffer->find(ci);
     QVariant* result = nullptr;
     if (*m_dbBufferIt != m_dbBuffer->end())
         result = &(*m_dbBufferIt).value();
     if (useDefaultValueIfPossible
             && (!result || result->isNull())
             && ci->field() && !ci->field()->defaultValue().isNull() && KDb::isDefaultValueAllowed(*ci->field())
             && !hasDefaultValueAt(*ci)) {
         //no buffered or stored value: try to get a default value declared in a field, so user can modify it
         if (!result)
             m_dbBuffer->insert(ci, ci->field()->defaultValue());
         result = &(*m_dbBuffer)[ ci ];
         m_defaultValuesDbBuffer->insert(ci, true);
     }
     return (const QVariant*)result;
 }
 
 const QVariant* KDbRecordEditBuffer::at(const KDbField &field) const
 {
     if (!m_simpleBuffer) {
         kdbWarning() << "this is db-aware buffer!";
         return nullptr;
     }
     *m_simpleBufferIt = m_simpleBuffer->constFind(field.name());
     if (*m_simpleBufferIt == m_simpleBuffer->constEnd()) {
         //kdbWarning() << "no such field:" << field;
         return nullptr;
     }
     return &(*m_simpleBufferIt).value();
 }
 
 const QVariant* KDbRecordEditBuffer::at(const QString& fname) const
 {
     if (!m_simpleBuffer) {
         kdbWarning() << "this is db-aware buffer!";
         return nullptr;
     }
     *m_simpleBufferIt = m_simpleBuffer->constFind(fname);
     if (*m_simpleBufferIt == m_simpleBuffer->constEnd()) {
         //kdbWarning() << "no such field:" << fname;
         return nullptr;
     }
     return &(*m_simpleBufferIt).value();
 }
 
 void KDbRecordEditBuffer::removeAt(const KDbQueryColumnInfo& ci)
 {
     if (!m_dbBuffer) {
         kdbWarning() << "not db-aware buffer!";
         return;
     }
     m_dbBuffer->remove(const_cast<KDbQueryColumnInfo*>(&ci)); // const_cast ok here, we won't modify ci
 }
 
 void KDbRecordEditBuffer::removeAt(const KDbField& field)
 {
     if (!m_simpleBuffer) {
         kdbWarning() << "this is db-aware buffer!";
         return;
     }
     m_simpleBuffer->remove(field.name());
 }
 
 void KDbRecordEditBuffer::removeAt(const QString& fname)
 {
     if (!m_simpleBuffer) {
         kdbWarning() << "this is db-aware buffer!";
         return;
     }
     m_simpleBuffer->remove(fname);
 }
 
 void KDbRecordEditBuffer::clear()
 {
     if (m_dbBuffer) {
         m_dbBuffer->clear();
         m_defaultValuesDbBuffer->clear();
     }
     if (m_simpleBuffer)
         m_simpleBuffer->clear();
 }
 
 bool KDbRecordEditBuffer::isEmpty() const
 {
     if (m_dbBuffer)
         return m_dbBuffer->isEmpty();
     if (m_simpleBuffer)
         return m_simpleBuffer->isEmpty();
     return true;
 }
 
 void KDbRecordEditBuffer::insert(KDbQueryColumnInfo *ci, const QVariant &val)
 {
     if (ci && m_dbBuffer) {
         m_dbBuffer->insert(ci, val);
         m_defaultValuesDbBuffer->remove(ci);
     }
 }
 
 void KDbRecordEditBuffer::insert(const QString &fname, const QVariant &val)
 {
     if (m_simpleBuffer) {
         m_simpleBuffer->insert(fname, val);
     }
 }
 
 bool KDbRecordEditBuffer::hasDefaultValueAt(const KDbQueryColumnInfo &ci) const
 {
     return m_defaultValuesDbBuffer->value(const_cast<KDbQueryColumnInfo*>(&ci), false);
 }
 
 KDbRecordEditBuffer::SimpleMap KDbRecordEditBuffer::simpleBuffer() const
 {
     return *m_simpleBuffer;
 }
 
 KDbRecordEditBuffer::DbHash KDbRecordEditBuffer::dbBuffer() const
 {
     return *m_dbBuffer;
 }
 
 KDB_EXPORT QDebug operator<<(QDebug dbg, const KDbRecordEditBuffer& buffer)
 {
     if (buffer.isDBAware()) {
         const KDbRecordEditBuffer::DbHash buf(buffer.dbBuffer());
         dbg.space() << "RecordEditBuffer type=DB-AWARE," << buf.count() << "field(s):\n";
         for (KDbRecordEditBuffer::DbHash::ConstIterator it = buf.constBegin();
              it != buf.constEnd(); ++it)
         {
             dbg.nospace() << "* field name=" << qPrintable(it.key()->field()->name()) << " val="
             << (it.value().isNull() ? QLatin1String("<NULL>") : it.value().toString())
             << (buffer.hasDefaultValueAt(*it.key()) ? " DEFAULT\n" : "\n");
         }
         return dbg.space();
     }
     const KDbRecordEditBuffer::SimpleMap map(buffer.simpleBuffer());
     dbg.space() << "RecordEditBuffer type=SIMPLE," << map.count() << "field(s):\n";
     for (KDbRecordEditBuffer::SimpleMap::ConstIterator it = map.constBegin();
          it != map.constEnd(); ++it)
     {
         dbg.nospace() << "* field name=" << qPrintable(it.key()) << " val="
         << (it.value().isNull() ? QLatin1String("<NULL>") : it.value().toString()) << "\n";
     }
     return dbg.space();
 }