Kavita/API/Helpers/Builders/ChapterBuilder.cs

using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using API.Entities;
using API.Entities.Enums;
using API.Entities.Person;
using API.Services.Tasks.Scanner.Parser;

namespace API.Helpers.Builders;
#nullable enable

public class ChapterBuilder : IEntityBuilder<Chapter>
{
    private readonly Chapter _chapter;
    public Chapter Build() => _chapter;

    public ChapterBuilder(string number, string? range=null)
    {
        _chapter = new Chapter()
        {
            Range = string.IsNullOrEmpty(range) ? number : Parser.RemoveExtensionIfSupported(range),
            Title = string.IsNullOrEmpty(range) ? number : range,
            Number = Parser.MinNumberFromRange(number).ToString(CultureInfo.InvariantCulture),
            MinNumber = Parser.MinNumberFromRange(number),
            MaxNumber = Parser.MaxNumberFromRange(number),
            SortOrder = Parser.MinNumberFromRange(number),
            Files = [],
            Pages = 1,
            CreatedUtc = DateTime.UtcNow
        };
    }

    public static ChapterBuilder FromParserInfo(ParserInfo info)
    {
        var specialTreatment = info.IsSpecialInfo();
        var specialTitle = specialTreatment ? Parser.RemoveExtensionIfSupported(info.Filename) : info.Chapters;
        var builder = new ChapterBuilder(Parser.DefaultChapter);

        return builder.WithNumber(Parser.RemoveExtensionIfSupported(info.Chapters)!)
            .WithRange(specialTreatment ? info.Filename : info.Chapters)
            .WithTitle(specialTreatment && info.Format is MangaFormat.Epub or MangaFormat.Pdf
            ? info.Title
            : specialTitle ?? string.Empty)
            .WithIsSpecial(specialTreatment);
    }

    public ChapterBuilder WithId(int id)
    {
        _chapter.Id = Math.Max(id, 0);
        return this;
    }


    private ChapterBuilder WithNumber(string number)
    {
        _chapter.Number = number;
        _chapter.MinNumber = Parser.MinNumberFromRange(number);
        _chapter.MaxNumber = Parser.MaxNumberFromRange(number);
        return this;
    }

    public ChapterBuilder WithSortOrder(float order)
    {
        _chapter.SortOrder = order;
        return this;
    }

    public ChapterBuilder WithStoryArc(string arc)
    {
        _chapter.StoryArc = arc;
        return this;
    }

    public ChapterBuilder WithStoryArcNumber(string number)
    {
        _chapter.StoryArcNumber = number;
        return this;
    }

    public ChapterBuilder WithRange(string range)
    {
        _chapter.Range = Parser.RemoveExtensionIfSupported(range);
        return this;
    }

    public ChapterBuilder WithReleaseDate(DateTime time)
    {
        _chapter.ReleaseDate = time;
        return this;
    }

    public ChapterBuilder WithAgeRating(AgeRating rating)
    {
        _chapter.AgeRating = rating;
        return this;
    }

    public ChapterBuilder WithPages(int pages)
    {
        _chapter.Pages = pages;
        return this;
    }
    public ChapterBuilder WithCoverImage(string cover)
    {
        _chapter.CoverImage = cover;
        return this;
    }
    public ChapterBuilder WithIsSpecial(bool isSpecial)
    {
        _chapter.IsSpecial = isSpecial;
        return this;
    }
    public ChapterBuilder WithTitle(string title)
    {
        _chapter.Title = title;
        return this;
    }

    public ChapterBuilder WithFile(MangaFile file)
    {
        _chapter.Files ??= new List<MangaFile>();
        _chapter.Files.Add(file);
        return this;
    }

    public ChapterBuilder WithFiles(IList<MangaFile> files)
    {
        _chapter.Files = files ?? new List<MangaFile>();
        return this;
    }

    public ChapterBuilder WithLastModified(DateTime lastModified)
    {
        _chapter.LastModified = lastModified;
        _chapter.LastModifiedUtc = lastModified.ToUniversalTime();
        return this;
    }

    public ChapterBuilder WithCreated(DateTime created)
    {
        _chapter.Created = created;
        _chapter.CreatedUtc = created.ToUniversalTime();
        return this;
    }

    public ChapterBuilder WithPerson(Person person, PersonRole role)
    {
        _chapter.People ??= new List<ChapterPeople>();
        _chapter.People.Add(new ChapterPeople()
        {
            Person = person,
            Role = role,
            Chapter = _chapter,
        });

        return this;
    }
}