// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include <zencore/guid.h>
#include <zencore/zencore.h>

#include <filesystem>
#include <memory>
#include <string>
#include <vector>

namespace zen {

struct TraceSessionInfo
{
	Guid				  SessionGuid{};
	Guid				  TraceGuid{};
	uint16_t			  ControlPort	   = 0;
	uint8_t				  TransportVersion = 0;
	uint8_t				  ProtocolVersion  = 0;
	std::string			  RemoteAddress;
	uint64_t			  BytesRecorded = 0;
	std::filesystem::path TraceFilePath;
};

class TraceRecorder
{
public:
	TraceRecorder();
	~TraceRecorder();

	void Initialize(uint16_t InPort, const std::filesystem::path& OutputDir);
	void Shutdown();

	bool	 IsRunning() const;
	uint16_t GetPort() const;

	std::vector<TraceSessionInfo> GetActiveSessions() const;

private:
	struct Impl;
	std::unique_ptr<Impl> m_Impl;
};

}  // namespace zen