using System.Collections.Generic;
using System.Linq;

namespace Wizard;

public class AIStyleFileNameList
{
	private const int STYLE_ID_INDEX = 0;

	private const int FILE_NAME_INDEX = 1;

	private readonly Dictionary<int, string> _dataTable = new Dictionary<int, string>();

	public AIStyleFileNameList(List<string[]> csvData)
	{
		foreach (string[] csvDatum in csvData)
		{
			int key = int.Parse(csvDatum[0]);
			string value = csvDatum[1];
			_dataTable.Add(key, value);
		}
	}

	public string GetFileName(int id)
	{
		if (_dataTable.ContainsKey(id))
		{
			return _dataTable[id];
		}
		return "";
	}

	public List<string> GetFileNameList()
	{
		return _dataTable.Values.ToList();
	}
}