// Copyright (C) 2021 The Qt Company Ltd.
// Copyright (C) 2019 Luxoft Sweden AB
// Copyright (C) 2018 Pelagicore AG
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only

#ifndef PLUGINCONTAINER_H
#define PLUGINCONTAINER_H

#include <memory>
#include <QtAppManManager/abstractcontainer.h>
#include <QtAppManPluginInterfaces/containerinterface.h>

QT_BEGIN_NAMESPACE_AM

class PluginContainerHelperFunctions final : public ContainerHelperFunctions
{
public:
    void closeAndClearFileDescriptors(QVector<int> &fdList) override;
    QStringList substituteCommand(const QStringList &debugWrapperCommand,
                                  const QString &program, const QStringList &arguments) override;
    bool hasRootPrivileges() override;

    void bindMountFileSystem(const QString &from, const QString &to, bool readOnly,
                             quint64 namespacePid) override;

    int watchdogSignal() override;

    QString checkDBusSocketPath(const QString &dbusAddress, const QByteArray &typeHint) override;
    QString checkWaylandSocketPath(const QString &xdgRuntimeDir, const QString &waylandDisplay) override;
};

class PluginContainerManager : public AbstractContainerManager
{
    Q_OBJECT
public:
    PluginContainerManager(ContainerManagerInterface *managerInterface, QObject *parent = nullptr);
    ~PluginContainerManager() override;

    bool initialize();

    static QString defaultIdentifier();
    bool supportsQuickLaunch() const override;

    AbstractContainer *create(Application *app, QVector<int> &&stdioRedirections,
                              const QMap<QString, QString> &debugWrapperEnvironment,
                              const QStringList &debugWrapperCommand) override;

    void setConfiguration(const QVariantMap &configuration) override;

private:
    ContainerManagerInterface *m_interface;
    std::unique_ptr<PluginContainerHelperFunctions> m_helpers;
};

class PluginContainer;

class PluginContainerProcess : public AbstractContainerProcess
{
    Q_OBJECT

public:
    PluginContainerProcess(PluginContainer *container);

    qint64 processId() const override;
    Am::RunState state() const override;

public Q_SLOTS:
    void stop(Am::ExitStatus exitStatus) override;

private:
    PluginContainer *m_container;
};

class PluginContainer : public AbstractContainer
{
    Q_OBJECT

public:
    virtual ~PluginContainer() override;

    QString controlGroup() const override;
    bool setControlGroup(const QString &groupName) override;

    bool setProgram(const QString &program) override;
    void setBaseDirectory(const QString &baseDirectory) override;

    bool isReady() override;
    bool hasDebugWrapper() const override;

    QString mapContainerPathToHost(const QString &containerPath) const override;
    QString mapHostPathToContainer(const QString &hostPath) const override;

    AbstractContainerProcess *start(const QStringList &arguments, const QMap<QString, QString> &env,
                                    const QVariantMap &amConfig) override;

protected:
    explicit PluginContainer(AbstractContainerManager *manager, Application *app,
                             ContainerInterface *containerInterface, bool hasDebugWrapper);
    std::unique_ptr<ContainerInterface> m_interface;
    bool m_startCalled = false;
    bool m_hasDebugWrapper = false;

    friend class PluginContainerProcess;
    friend class PluginContainerManager;
};

QT_END_NAMESPACE_AM

#endif // PLUGINCONTAINER_H