StarterNG/StarterNG/Infrastructure/Sceneries/SceneryRepository.cs

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using StarterNG.Application.Abstractions;
using StarterNG.Domain.Sceneries;
using StarterNG.Infrastructure.Adapters;

namespace StarterNG.Infrastructure.Sceneries;

public sealed class SceneryRepository : ISceneryRepository
{
    private const char ExcludedPrefix = '$';

    private readonly IFileSystem _files;
    private readonly IGamePaths _paths;
    private readonly IDiagnosticsLog _log;
    private readonly SceneryParser _parser;

    public SceneryRepository(IFileSystem files, IGamePaths paths, IDiagnosticsLog log, SceneryParser parser)
    {
        _files = files;
        _paths = paths;
        _log = log;
        _parser = parser;
    }

    public IReadOnlyList<Scenery> LoadAll(IProgress<SceneryLoadProgress>? progress = null)
    {
        var files = SceneryFiles().ToList();
        var loaded = new ConcurrentBag<Scenery>();
        int total = Math.Max(1, files.Count);
        int done = 0;

        Parallel.ForEach(files,
            new ParallelOptions { MaxDegreeOfParallelism = Environment.ProcessorCount },
            file =>
            {
                var scenery = Load(file);
                if (scenery is not null)
                    loaded.Add(scenery);

                progress?.Report(new SceneryLoadProgress(Interlocked.Increment(ref done), total,
                                                         Path.GetFileName(file)));
            });

        return loaded.ToList();
    }

    public Scenery? Load(string path)
    {
        try
        {
            if (!_files.FileExists(path))
                return null;

            var scenery = _parser.Parse(path, LegacyText.Decode(_files.ReadAllBytes(path)));
            scenery.HasCompanionTimetable = _files.FileExists(Path.ChangeExtension(path, ".sbt"));
            return scenery;
        }
        catch (Exception ex)
        {
            _log.Log($"scenery/{Path.GetFileName(path)}", ex);
            return null;
        }
    }

    private IEnumerable<string> SceneryFiles() =>
        _files.GetFiles(_paths.Scenery, "*.scn")
              .Where(path => Path.GetFileName(path).FirstOrDefault() != ExcludedPrefix);
}