// Copyright (C) 2016 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0

#ifndef DRIVER_H
#define DRIVER_H

#include "option.h"
#include <qhash.h>
#include <qstring.h>
#include <qstringlist.h>
#include <qtextstream.h>

QT_BEGIN_NAMESPACE

class QTextStream;
class DomUI;
class DomWidget;
class DomSpacer;
class DomLayout;
class DomLayoutItem;
class DomActionGroup;
class DomAction;
class DomButtonGroup;

class Driver
{
    Q_DISABLE_COPY_MOVE(Driver)
public:
    Driver();
    virtual ~Driver();

    // tools
    bool printDependencies(const QString &fileName);
    bool uic(const QString &fileName, QTextStream *output = nullptr);
    bool uic(const QString &fileName, DomUI *ui, QTextStream *output = nullptr);

    // configuration
    inline QTextStream &output() const { return *m_output; }
    inline Option &option() { return m_option; }

    // utils
    static QString headerFileName(const QString &fileName);
    QString headerFileName() const;

    static QString normalizedName(const QString &name);
    static QString qtify(const QString &name);
    QString unique(const QString &instanceName=QString(),
                   const QString &className=QString());

    // symbol table
    QString findOrInsertWidget(const DomWidget *ui_widget);
    QString findOrInsertSpacer(const DomSpacer *ui_spacer);
    QString findOrInsertLayout(const DomLayout *ui_layout);
    QString findOrInsertLayoutItem(const DomLayoutItem *ui_layoutItem);
    QString findOrInsertName(const QString &name);
    QString findOrInsertActionGroup(const DomActionGroup *ui_group);
    QString findOrInsertAction(const DomAction *ui_action);
    QString findOrInsertButtonGroup(const DomButtonGroup *ui_group);
    // Find a group by its non-uniqified name
    const DomButtonGroup *findButtonGroup(const QString &attributeName) const;

    const DomWidget *widgetByName(const QString &attributeName) const;
    QString widgetVariableName(const QString &attributeName) const;
    const DomActionGroup *actionGroupByName(const QString &attributeName) const;
    const DomAction *actionByName(const QString &attributeName) const;

    bool useIdBasedTranslations() const { return m_idBasedTranslations; }
    void setUseIdBasedTranslations(bool u) { m_idBasedTranslations = u; }

private:
    template <class DomClass> using DomObjectHash = QHash<const DomClass *, QString>;
    template <class DomClass> using DomObjectHashConstIt =
        typename DomObjectHash<DomClass>::ConstIterator;

    template <class DomClass>
    DomObjectHashConstIt<DomClass> findByAttributeNameIt(const DomObjectHash<DomClass> &domHash,
                                                         const QString &name) const;
    template <class DomClass>
    const DomClass *findByAttributeName(const DomObjectHash<DomClass> &domHash,
                                        const QString &name) const;
    template <class DomClass>
    QString findOrInsert(DomObjectHash<DomClass> *domHash, const DomClass *dom, const QString &className,
                         bool isMember = true);

    Option m_option;
    QTextStream m_stdout;
    QTextStream *m_output;

    // symbol tables
    DomObjectHash<DomWidget> m_widgets;
    DomObjectHash<DomSpacer> m_spacers;
    DomObjectHash<DomLayout> m_layouts;
    DomObjectHash<DomActionGroup> m_actionGroups;
    DomObjectHash<DomButtonGroup> m_buttonGroups;
    DomObjectHash<DomAction> m_actions;
    QHash<QString, bool> m_nameRepository;
    bool m_idBasedTranslations = false;
};

QT_END_NAMESPACE

#endif // DRIVER_H