TenantProvisioningWizard.hpp

/* -*- mode: c++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
 *
 * Copyright (C) 2026 Marco Craveiro <marco.craveiro@gmail.com>
 *
 * 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 3 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 ORES_QT_TENANT_PROVISIONING_WIZARD_HPP
#define ORES_QT_TENANT_PROVISIONING_WIZARD_HPP
 
#include <QWizard>
#include <QWizardPage>
#include <QComboBox>
#include <QLabel>
#include <QLineEdit>
#include <QProgressBar>
#include <QRadioButton>
#include <QSpinBox>
#include <QCheckBox>
#include <QTextEdit>
#include <QPushButton>
#include <QListWidget>
#include <optional>
#include <cstdint>
#include <vector>
#include <string>
#include "ores.logging/make_logger.hpp"
#include "ores.qt/ClientManager.hpp"
 
namespace ores::qt {
 
class ClientDatasetBundleModel;
class LeiEntityPicker;
 
class TenantProvisioningWizard final : public QWizard {
    Q_OBJECT
 
private:
    inline static std::string_view logger_name =
        "ores.qt.tenant_provisioning_wizard";
 
    [[nodiscard]] static auto& lg() {
        using namespace ores::logging;
        static auto instance = make_logger(logger_name);
        return instance;
    }
 
public:
    enum PageId {
        Page_Welcome,
        Page_BundleSelection,
        Page_BundleInstall,
        Page_DataSourceSelection,
        Page_PartySetup,
        Page_CounterpartySetup,
        Page_OrganisationSetup,
        Page_ReportSetup,
        Page_ReportInstall,
        Page_Summary
    };
 
    struct ReportSpec {
        std::string name;
        std::string description;
        std::string schedule_expression;
        std::string report_type;
        std::string concurrency_policy;
    };
 
    enum class DataSourceMode {
        gleif,
        synthetic
    };
 
    explicit TenantProvisioningWizard(
        ClientManager* clientManager,
        QWidget* parent = nullptr);
 
    ~TenantProvisioningWizard() override = default;
 
    ClientManager* clientManager() const { return clientManager_; }
 
    QString selectedBundleCode() const { return selectedBundleCode_; }
    void setSelectedBundleCode(const QString& code) { selectedBundleCode_ = code; }
 
    QString selectedBundleName() const { return selectedBundleName_; }
    void setSelectedBundleName(const QString& name) { selectedBundleName_ = name; }
 
    QString rootLei() const { return rootLei_; }
    void setRootLei(const QString& lei) { rootLei_ = lei; }
 
    QString rootLeiName() const { return rootLeiName_; }
    void setRootLeiName(const QString& name) { rootLeiName_ = name; }
 
    QString leiDatasetSize() const { return leiDatasetSize_; }
    void setLeiDatasetSize(const QString& size) { leiDatasetSize_ = size; }
 
    bool organisationPublished() const { return organisationPublished_; }
    void setOrganisationPublished(bool v) { organisationPublished_ = v; }
 
    DataSourceMode dataSourceMode() const { return dataSourceMode_; }
    void setDataSourceMode(DataSourceMode m) { dataSourceMode_ = m; }
 
    // --- Synthetic generation options (set by DataSourceSelectionPage) ---
 
    QString syntheticCountry() const { return syntheticCountry_; }
    void setSyntheticCountry(const QString& c) { syntheticCountry_ = c; }
 
    int syntheticPartyCount() const { return syntheticPartyCount_; }
    void setSyntheticPartyCount(int c) { syntheticPartyCount_ = c; }
 
    int syntheticPartyMaxDepth() const { return syntheticPartyMaxDepth_; }
    void setSyntheticPartyMaxDepth(int d) { syntheticPartyMaxDepth_ = d; }
 
    int syntheticCounterpartyCount() const { return syntheticCounterpartyCount_; }
    void setSyntheticCounterpartyCount(int c) { syntheticCounterpartyCount_ = c; }
 
    int syntheticCounterpartyMaxDepth() const { return syntheticCounterpartyMaxDepth_; }
    void setSyntheticCounterpartyMaxDepth(int d) { syntheticCounterpartyMaxDepth_ = d; }
 
    int syntheticPortfolioLeafCount() const { return syntheticPortfolioLeafCount_; }
    void setSyntheticPortfolioLeafCount(int c) { syntheticPortfolioLeafCount_ = c; }
 
    int syntheticPortfolioMaxDepth() const { return syntheticPortfolioMaxDepth_; }
    void setSyntheticPortfolioMaxDepth(int d) { syntheticPortfolioMaxDepth_ = d; }
 
    int syntheticBooksPerPortfolio() const { return syntheticBooksPerPortfolio_; }
    void setSyntheticBooksPerPortfolio(int c) { syntheticBooksPerPortfolio_ = c; }
 
    int syntheticBusinessUnitCount() const { return syntheticBusinessUnitCount_; }
    void setSyntheticBusinessUnitCount(int c) { syntheticBusinessUnitCount_ = c; }
 
    int syntheticBusinessUnitMaxDepth() const { return syntheticBusinessUnitMaxDepth_; }
    void setSyntheticBusinessUnitMaxDepth(int d) { syntheticBusinessUnitMaxDepth_ = d; }
 
    bool syntheticGenerateAddresses() const { return syntheticGenerateAddresses_; }
    void setSyntheticGenerateAddresses(bool v) { syntheticGenerateAddresses_ = v; }
 
    int syntheticContactsPerParty() const { return syntheticContactsPerParty_; }
    void setSyntheticContactsPerParty(int c) { syntheticContactsPerParty_ = c; }
 
    int syntheticContactsPerCounterparty() const { return syntheticContactsPerCounterparty_; }
    void setSyntheticContactsPerCounterparty(int c) { syntheticContactsPerCounterparty_ = c; }
 
    bool syntheticGenerateIdentifiers() const { return syntheticGenerateIdentifiers_; }
    void setSyntheticGenerateIdentifiers(bool v) { syntheticGenerateIdentifiers_ = v; }
 
    std::optional<std::uint64_t> syntheticSeed() const { return syntheticSeed_; }
    void setSyntheticSeed(std::optional<std::uint64_t> s) { syntheticSeed_ = s; }
 
    std::vector<ReportSpec> selectedReports() const { return selectedReports_; }
    void setSelectedReports(std::vector<ReportSpec> r) { selectedReports_ = std::move(r); }
 
    void clearBootstrapFlag();
 
signals:
    void provisioningCompleted();
 
private:
    void setupPages();
 
    ClientManager* clientManager_;
    QString selectedBundleCode_;
    QString selectedBundleName_;
    QString rootLei_;
    QString rootLeiName_;
    QString leiDatasetSize_ = "large";
    bool organisationPublished_ = false;
    DataSourceMode dataSourceMode_ = DataSourceMode::gleif;
 
    // Synthetic generation options
    QString syntheticCountry_ = "GB";
    int syntheticPartyCount_ = 5;
    int syntheticPartyMaxDepth_ = 3;
    int syntheticCounterpartyCount_ = 10;
    int syntheticCounterpartyMaxDepth_ = 3;
    int syntheticPortfolioLeafCount_ = 8;
    int syntheticPortfolioMaxDepth_ = 4;
    int syntheticBooksPerPortfolio_ = 2;
    int syntheticBusinessUnitCount_ = 10;
    int syntheticBusinessUnitMaxDepth_ = 2;
    bool syntheticGenerateAddresses_ = true;
    int syntheticContactsPerParty_ = 2;
    int syntheticContactsPerCounterparty_ = 1;
    bool syntheticGenerateIdentifiers_ = true;
    std::optional<std::uint64_t> syntheticSeed_;
 
    std::vector<ReportSpec> selectedReports_;
};
 
// Forward declarations
class ProvisioningWelcomePage;
class BundleSelectionPage;
class BundleInstallPage;
class DataSourceSelectionPage;
class PartySetupPage;
class CounterpartySetupPage;
class OrganisationSetupPage;
class ReportSetupPage;
class ReportInstallPage;
class ApplyAndSummaryPage;
 
class ProvisioningWelcomePage final : public QWizardPage {
    Q_OBJECT
 
public:
    explicit ProvisioningWelcomePage(TenantProvisioningWizard* wizard);
 
private:
    void setupUI();
    TenantProvisioningWizard* wizard_;
};
 
class BundleSelectionPage final : public QWizardPage {
    Q_OBJECT
 
public:
    explicit BundleSelectionPage(TenantProvisioningWizard* wizard);
    void initializePage() override;
    bool validatePage() override;
    bool isComplete() const override;
 
private:
    void setupUI();
    void onBundleChanged(int index);
 
    TenantProvisioningWizard* wizard_;
    ClientDatasetBundleModel* bundleModel_;
    QComboBox* bundleCombo_;
    QLabel* descriptionLabel_;
    QLabel* statusLabel_;
};
 
class BundleInstallPage final : public QWizardPage {
    Q_OBJECT
 
private:
    inline static std::string_view logger_name =
        "ores.qt.bundle_install_page";
 
    [[nodiscard]] static auto& lg() {
        using namespace ores::logging;
        static auto instance = make_logger(logger_name);
        return instance;
    }
 
public:
    explicit BundleInstallPage(TenantProvisioningWizard* wizard);
    void initializePage() override;
    bool isComplete() const override;
 
private:
    void startPublish();
    void appendLog(const QString& message);
 
    TenantProvisioningWizard* wizard_;
    QLabel* statusLabel_;
    QProgressBar* progressBar_;
    QTextEdit* logOutput_;
    bool publishComplete_ = false;
    bool publishSuccess_ = false;
};
 
class DataSourceSelectionPage final : public QWizardPage {
    Q_OBJECT
 
public:
    explicit DataSourceSelectionPage(TenantProvisioningWizard* wizard);
    bool validatePage() override;
    int nextId() const override;
 
private:
    void setupUI();
    void onModeChanged();
 
    TenantProvisioningWizard* wizard_;
    QRadioButton* gleifRadio_;
    QRadioButton* syntheticRadio_;
    QWidget* syntheticOptions_;
 
    // Basic counts
    QComboBox* countryCombo_;
    QSpinBox* partyCountSpin_;
    QSpinBox* counterpartyCountSpin_;
    QSpinBox* portfolioLeafCountSpin_;
    QSpinBox* booksPerPortfolioSpin_;
    QSpinBox* businessUnitCountSpin_;
 
    // Hierarchy depths
    QSpinBox* partyMaxDepthSpin_;
    QSpinBox* counterpartyMaxDepthSpin_;
    QSpinBox* portfolioMaxDepthSpin_;
    QSpinBox* businessUnitMaxDepthSpin_;
 
    // Contact and identifier options
    QSpinBox* contactsPerPartySpin_;
    QSpinBox* contactsPerCounterpartySpin_;
    QCheckBox* generateAddressesCheck_;
    QCheckBox* generateIdentifiersCheck_;
 
    // Reproducibility
    QLineEdit* seedEdit_;
};
 
class PartySetupPage final : public QWizardPage {
    Q_OBJECT
 
public:
    explicit PartySetupPage(TenantProvisioningWizard* wizard);
    void initializePage() override;
    bool validatePage() override;
 
private:
    void setupUI();
 
    TenantProvisioningWizard* wizard_;
    LeiEntityPicker* leiPicker_;
    QComboBox* datasetSizeCombo_;
    QLabel* instructionLabel_;
    bool leiLoaded_ = false;
};
 
class CounterpartySetupPage final : public QWizardPage {
    Q_OBJECT
 
public:
    explicit CounterpartySetupPage(TenantProvisioningWizard* wizard);
 
private:
    void setupUI();
    TenantProvisioningWizard* wizard_;
};
 
class OrganisationSetupPage final : public QWizardPage {
    Q_OBJECT
 
private:
    inline static std::string_view logger_name =
        "ores.qt.organisation_setup_page";
 
    [[nodiscard]] static auto& lg() {
        using namespace ores::logging;
        static auto instance = make_logger(logger_name);
        return instance;
    }
 
public:
    explicit OrganisationSetupPage(TenantProvisioningWizard* wizard);
    void initializePage() override;
    bool isComplete() const override;
 
private:
    void startPublish();
    void startBundlePublish();
    void startSyntheticGeneration();
    void appendLog(const QString& message);
 
    TenantProvisioningWizard* wizard_;
    QLabel* statusLabel_;
    QProgressBar* progressBar_;
    QTextEdit* logOutput_;
    bool publishComplete_ = false;
    bool publishSuccess_ = false;
};
 
class ReportSetupPage final : public QWizardPage {
    Q_OBJECT
 
public:
    explicit ReportSetupPage(TenantProvisioningWizard* wizard);
    bool validatePage() override;
    int nextId() const override;
 
private:
    void setupUI();
 
    TenantProvisioningWizard* wizard_;
    QListWidget* reportList_;
};
 
class ReportInstallPage final : public QWizardPage {
    Q_OBJECT
 
private:
    inline static std::string_view logger_name =
        "ores.qt.report_install_page";
 
    [[nodiscard]] static auto& lg() {
        using namespace ores::logging;
        static auto instance = make_logger(logger_name);
        return instance;
    }
 
public:
    explicit ReportInstallPage(TenantProvisioningWizard* wizard);
    void initializePage() override;
    bool isComplete() const override;
 
private:
    void startInstall();
    void appendLog(const QString& message);
 
    TenantProvisioningWizard* wizard_;
    QLabel* statusLabel_;
    QProgressBar* progressBar_;
    QTextEdit* logOutput_;
    bool installComplete_ = false;
    bool installSuccess_ = false;
};
 
class ApplyAndSummaryPage final : public QWizardPage {
    Q_OBJECT
 
private:
    inline static std::string_view logger_name =
        "ores.qt.apply_and_summary_page";
 
    [[nodiscard]] static auto& lg() {
        using namespace ores::logging;
        static auto instance = make_logger(logger_name);
        return instance;
    }
 
public:
    explicit ApplyAndSummaryPage(TenantProvisioningWizard* wizard);
    void initializePage() override;
 
private:
    void setupUI();
 
    TenantProvisioningWizard* wizard_;
    QLabel* summaryLabel_;
};
 
}
 
#endif