//---------------------------------------------------------------------------
//
// SCSI Target Emulator RaSCSI (*^..^*)
// for Raspberry Pi
//
// Copyright (C) 2021 Uwe Seimet
//
// The DeviceFactory singleton creates devices based on their type and the image file extension
//
//---------------------------------------------------------------------------

#pragma once

#include <set>
#include <list>
#include <map>
#include <string>
#include "rascsi_interface.pb.h"

using namespace std;
using namespace rascsi_interface;

typedef pair<uint32_t, uint32_t> Geometry;

class Device;

class DeviceFactory
{
public:

	DeviceFactory();
	~DeviceFactory() {};

	static DeviceFactory& instance();

	Device *CreateDevice(PbDeviceType, const string&);
	PbDeviceType GetTypeForFile(const string&);
	const set<uint32_t>& GetSectorSizes(PbDeviceType type) { return sector_sizes[type]; }
	const set<uint32_t>& GetSectorSizes(const string&);
	const set<uint64_t> GetCapacities(PbDeviceType);
	const map<string, string>& GetDefaultParams(PbDeviceType type) { return default_params[type]; }
	const list<string> GetNetworkInterfaces() const;
	const map<string, PbDeviceType> GetExtensionMapping() const { return extension_mapping; }

private:

	map<PbDeviceType, set<uint32_t>> sector_sizes;

	// Optional mapping of drive capacities to drive geometries
	map<PbDeviceType, map<uint64_t, Geometry>> geometries;

	map<PbDeviceType, map<string, string>> default_params;

	map<string, PbDeviceType> extension_mapping;

	string GetExtension(const string&) const;
};
