// Copyright 2019 threeSD Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#pragma once

#include <atomic>
#include <functional>
#include <QThread>
#include "common/progress_callback.h"
#include "core/importer.h"

class MultiJob : public QThread {
    Q_OBJECT

public:
    using ExecuteFunc = std::function<bool(Core::SDMCImporter&, const Core::ContentSpecifier&,
                                           const Common::ProgressCallback&)>;
    using AbortFunc = std::function<void(Core::SDMCImporter&)>;
    // (content, error log)
    using FailedContentList = std::vector<std::pair<Core::ContentSpecifier, std::string>>;

    explicit MultiJob(QObject* parent, Core::SDMCImporter& importer,
                      std::vector<Core::ContentSpecifier> contents, ExecuteFunc execute_func,
                      AbortFunc abort_func);
    ~MultiJob() override;

    void run() override;
    void Cancel();

    FailedContentList GetFailedContents() const;

signals:
    /**
     * Called when progress is updated on the current content.
     * @param current_imported_size Imported size of the current content.
     * @param total_imported_size Total imported size taking all previous contents into
     * consideration.
     * @param eta ETA in seconds, 0 when not determined.
     */
    void ProgressUpdated(u64 current_imported_size, u64 total_imported_size, int eta);

    /// Dumping of a content has been finished, go on to the next. Called at start as well.
    void NextContent(std::size_t count, u64 total_imported_size,
                     const Core::ContentSpecifier& next_content, int eta);

    void Completed();

private:
    std::atomic_bool cancelled{false};
    Core::SDMCImporter& importer;
    std::vector<Core::ContentSpecifier> contents;
    FailedContentList failed_contents;
    ExecuteFunc execute_func;
    AbortFunc abort_func;
};

Q_DECLARE_METATYPE(Core::ContentSpecifier)