/* This file is part of the KDE project

   Copyright 2009 David Faure <faure@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 ) version 3 or, at the discretion of KDE e.V.
   ( which shall act as a proxy as in section 14 of the GPLv3 ), 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.
*/

#ifndef KSYCOCADEVICES_P_H
#define KSYCOCADEVICES_P_H

class KSycocaAbstractDevice
{
public:
    KSycocaAbstractDevice() : m_stream(0)
    {
    }

    virtual ~KSycocaAbstractDevice() { delete m_stream; }

    virtual QIODevice* device() = 0;

    QDataStream* & stream() {
        if (!m_stream) {
            m_stream = new QDataStream(device());
            m_stream->setVersion(QDataStream::Qt_4_0);
        }
        return m_stream;
    }

private:
    QDataStream* m_stream;
};

#ifdef HAVE_MMAP
// Reading from a mmap'ed file
class KSycocaMmapDevice : public KSycocaAbstractDevice
{
public:
    KSycocaMmapDevice(const char* sycoca_mmap, size_t sycoca_size) {
        m_buffer = new QBuffer;
        m_buffer->setData(QByteArray::fromRawData(sycoca_mmap, sycoca_size));
    }
    ~KSycocaMmapDevice() {
        delete m_buffer;
    }
    virtual QIODevice* device() {
        return m_buffer;
    }
private:
    QBuffer* m_buffer;
};
#endif

// Reading from a QFile
class KSycocaFileDevice : public KSycocaAbstractDevice
{
public:
    KSycocaFileDevice(const QString& path) {
        m_database = new QFile(path);
        (void)fcntl(m_database->handle(), F_SETFD, FD_CLOEXEC);
    }
    ~KSycocaFileDevice() {
        delete m_database;
    }
    virtual QIODevice* device() {
        return m_database;
    }
private:
    QFile* m_database;
};

#ifndef QT_NO_SHAREDMEMORY
// Reading from a KMemFile
class KSycocaMemFileDevice : public KSycocaAbstractDevice
{
public:
    KSycocaMemFileDevice(const QString& path) {
        m_database = new KMemFile(path);
    }
    ~KSycocaMemFileDevice() {
        delete m_database;
    }
    virtual QIODevice* device() {
        return m_database;
    }
private:
    KMemFile* m_database;
};
#endif

// Reading from a dummy memory buffer
class KSycocaBufferDevice : public KSycocaAbstractDevice
{
public:
    KSycocaBufferDevice() {
        m_buffer = new QBuffer;
    }
    ~KSycocaBufferDevice() {
        delete m_buffer;
    }
    virtual QIODevice* device() {
        return m_buffer;
    }
private:
    QBuffer* m_buffer;
};

#endif /* KSYCOCADEVICES_P_H */