#pragma once

#include <atomic>
#include <mutex>
#include <unordered_set>
#include <vector>
#include <functional>

#include "warp.h"
#include "binaryninjaapi.h"
#include "uitypes.h"

enum WarpFetchCompletionStatus
{
	KeepCallback,
	RemoveCallback,
};

// Responsible for fetching data from the containers, to later be queried from the container interface.
class WarpFetcher
{
	LoggerRef m_logger;

	std::mutex m_requestMutex;
	std::vector<FunctionRef> m_pendingRequests;
	std::unordered_set<Warp::FunctionGUID> m_processedGuids;

public:
	using CallbackId = uint64_t;
	using CompletionCallback = std::function<WarpFetchCompletionStatus()>;

	explicit WarpFetcher();

	// The global fetcher instance, this is used for the fetch dialog and the sidebar.
	static std::shared_ptr<WarpFetcher> Global();

	std::atomic<bool> m_requestInProgress = false;

	[[nodiscard]] CallbackId AddCompletionCallback(CompletionCallback cb)
	{
		std::lock_guard<std::mutex> lock(m_requestMutex);
		const CallbackId id = m_nextCallbackId++;
		m_completionCallbacks.emplace(id, std::move(cb));
		return id;
	}

	void RemoveCompletionCallback(CallbackId id)
	{
		std::lock_guard<std::mutex> lock(m_requestMutex);
		m_completionCallbacks.erase(id);
	}

	void AddPendingFunction(const FunctionRef& func);

	void FetchPendingFunctions(const std::vector<Warp::SourceTag>& allowedTags);

	void ClearProcessed();

private:
	// List of callbacks to call when done fetching data, assume that others are using this as well.
	std::atomic<CallbackId> m_nextCallbackId = 1;
	std::unordered_map<CallbackId, CompletionCallback> m_completionCallbacks;
	void ExecuteCompletionCallback();

	std::vector<FunctionRef> FlushPendingFunctions();
};