StabilityMatrix/StabilityMatrix.Avalonia/Helpers/TagCsvParser.cs

using System.Collections.Generic;
using System.IO;
using System.Threading.Tasks;
using StabilityMatrix.Avalonia.Models.TagCompletion;
using Sylvan.Data.Csv;
using Sylvan;

namespace StabilityMatrix.Avalonia.Helpers;

public class TagCsvParser
{
    private readonly Stream stream;

    public TagCsvParser(Stream stream)
    {
        this.stream = stream;
    }

    public async IAsyncEnumerable<TagCsvEntry> ParseAsync()
    {
        var pool = new StringPool();
        var options = new CsvDataReaderOptions
        {
            StringFactory = pool.GetString,
            HasHeaders = false,
        };

        using var textReader = new StreamReader(stream);
        await using var dataReader = await CsvDataReader.CreateAsync(textReader, options);

        while (await dataReader.ReadAsync())
        {
            var entry = new TagCsvEntry
            {
                Name = dataReader.GetString(0),
                Type = dataReader.GetInt32(1),
                Count = dataReader.GetInt32(2),
                Aliases = dataReader.GetString(3),
            };
            yield return entry;
        }

        /*var dataBinderOptions = new DataBinderOptions
        {
            BindingMode = DataBindingMode.Any
        };*/
        /*var results = dataReader.GetRecordsAsync<TagCsvEntry>(dataBinderOptions);
        return results;*/
    }

    public async Task<Dictionary<string, TagCsvEntry>> GetDictionaryAsync()
    {
        var dict = new Dictionary<string, TagCsvEntry>();

        await foreach (var entry in ParseAsync())
        {
            if (entry.Name is null || entry.Type is null)
            {
                continue;
            }

            dict.Add(entry.Name, entry);
        }

        return dict;
    }
}