Beam/Beam.Fluent/DownloadBuilder.TransformStage.cs

using Beam.Data;
using Beam.Downloaders;
using Beam.Dynamic;
using Beam.Models;

namespace Beam.Fluent {
public static partial class DownloadBuilder<RawType, OutType> {
        private sealed record TransformStage(
            WebResource Source,
            BeamDataContext Data,
            DownloadContextBuilder<RawType> CtxBuilder) : ITransformStage, IAlternativeTransformStage {
            public IContextStage WithTransformer(Func<DataBindings, AsyncTransformer<RawType, OutType>> factory) {
                var transformer = factory(Data.Bindings[Source.Bindings]);
                return new ContextStage(CtxBuilder, transformer);
            }

            public IContextStage WithTransformer(AsyncTransformer<RawType, OutType> transformer) {
                return new ContextStage(CtxBuilder, transformer);
            }
        }
    }
}