using System.Xml; using Bugsnag; using ErsatzTV.Core; using ErsatzTV.Core.Errors; using ErsatzTV.Scanner.Core.Interfaces.Metadata.Nfo; using Microsoft.Extensions.Logging; using Microsoft.IO; namespace ErsatzTV.Scanner.Core.Metadata.Nfo; public class MovieNfoReader : NfoReader, IMovieNfoReader { private readonly IClient _client; private readonly ILogger _logger; public MovieNfoReader( RecyclableMemoryStreamManager recyclableMemoryStreamManager, IClient client, ILogger logger) : base(recyclableMemoryStreamManager, logger) { _client = client; _logger = logger; } public async Task> ReadFromFile(string fileName) { // ReSharper disable once ConvertToUsingDeclaration await using (Stream s = await SanitizedStreamForFile(fileName)) { return await Read(s); } } internal async Task> Read(Stream input) { MovieNfo? nfo = null; try { var settings = new XmlReaderSettings { Async = true, ConformanceLevel = ConformanceLevel.Fragment }; using var reader = XmlReader.Create(input, settings); var done = false; while (!done && await reader.ReadAsync()) { switch (reader.NodeType) { case XmlNodeType.Element: switch (reader.Name.ToLowerInvariant()) { case "movie": nfo = new MovieNfo(); break; case "title": await ReadStringContent(reader, nfo, (movie, title) => movie.Title = title); break; case "sorttitle": await ReadStringContent(reader, nfo, (movie, sortTitle) => movie.SortTitle = sortTitle); break; case "outline": await ReadStringContent(reader, nfo, (movie, outline) => movie.Outline = outline); break; case "year": await ReadIntContent(reader, nfo, (movie, year) => movie.Year = year); break; case "mpaa": await ReadStringContent( reader, nfo, (movie, contentRating) => movie.ContentRating = contentRating); break; case "premiered": await ReadDateTimeContent( reader, nfo, (movie, premiered) => movie.Premiered = premiered); break; case "plot": await ReadStringContent(reader, nfo, (movie, plot) => movie.Plot = plot); break; case "genre": await ReadStringContent(reader, nfo, (movie, genre) => movie.Genres.Add(genre)); break; case "tag": await ReadStringContent(reader, nfo, (movie, tag) => movie.Tags.Add(tag)); break; case "studio": await ReadStringContent(reader, nfo, (movie, studio) => movie.Studios.Add(studio)); break; case "actor": ReadActor(reader, nfo, (movie, actor) => movie.Actors.Add(actor)); break; case "credits": await ReadStringContent(reader, nfo, (movie, writer) => movie.Writers.Add(writer)); break; case "director": await ReadStringContent( reader, nfo, (movie, director) => movie.Directors.Add(director)); break; case "uniqueid": await ReadUniqueId(reader, nfo, (movie, uniqueid) => movie.UniqueIds.Add(uniqueid)); break; } break; case XmlNodeType.EndElement: if (reader.Name == "movie") { done = true; } break; } } return Optional(nfo).ToEither((BaseError)new FailedToReadNfo()); } catch (XmlException) { _logger.LogWarning("Invalid XML detected; returning incomplete metadata"); return Optional(nfo).ToEither((BaseError)new FailedToReadNfo()); } catch (Exception ex) { _client.Notify(ex); return new FailedToReadNfo(ex.ToString()); } } }