/**
 * UGENE - Integrated Bioinformatics Tools.
 * Copyright (C) 2008-2021 UniPro <ugene@unipro.ru>
 * http://ugene.net
 *
 * This program 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 2
 * of the License, or (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301, USA.
 */

#ifndef _U2_QD_DOC_FORMAT_H_
#define _U2_QD_DOC_FORMAT_H_

#include <U2Core/DocumentModel.h>
#include <U2Core/GObject.h>

#include <U2Formats/TextDocumentFormat.h>

#include <U2Gui/ObjectViewModel.h>
#include <U2Gui/ObjectViewTasks.h>

#include "QueryDesignerPlugin.h"

namespace U2 {

class QDDocFormat : public TextDocumentFormat {
    Q_OBJECT
public:
    QDDocFormat(QObject *p);

    virtual Document *createNewLoadedDocument(IOAdapterFactory *io, const GUrl &url, U2OpStatus &os, const QVariantMap &fs = QVariantMap());

    virtual void storeDocument(Document *d, IOAdapter *io, U2OpStatus &os);

protected:
    virtual FormatCheckResult checkRawTextData(const QByteArray &rawData, const GUrl & = GUrl()) const;

    virtual Document *loadTextDocument(IOAdapter *io, const U2DbiRef &targetDb, const QVariantMap &hints, U2OpStatus &os);
};

class QueryScene;

class QDGObject : public GObject {
    Q_OBJECT
public:
    static const GObjectType TYPE;
    QDGObject(const QString &objectName, const QString &data, const QVariantMap &map = QVariantMap())
        : GObject(TYPE, objectName), serializedScene(data), scene(NULL) {
        Q_UNUSED(map);
    }

    virtual GObject *clone(const U2DbiRef &dbiRef, U2OpStatus &os, const QVariantMap &hints = QVariantMap()) const;
    QueryScene *getScene() const {
        return scene;
    }
    void setSceneRawData(const QString &d) {
        serializedScene = d;
    }
    QString getSceneRawData() const {
        return serializedScene;
    }

private:
    QString serializedScene;
    QueryScene *scene;
};

class QDViewFactory : public GObjectViewFactory {
    Q_OBJECT
public:
    static const GObjectViewFactoryId ID;
    QDViewFactory(QObject *p = NULL)
        : GObjectViewFactory(ID, U2::QueryDesignerPlugin::tr("Query Designer"), p) {
    }

    virtual bool canCreateView(const MultiGSelection &multiSelection);
    virtual Task *createViewTask(const MultiGSelection &multiSelection, bool single = false);
};

class OpenQDViewTask : public ObjectViewTask {
    Q_OBJECT
public:
    OpenQDViewTask(Document *doc);
    virtual void open();

private:
    Document *document;
};

}    // namespace U2

#endif
