using aeqw89.DataKeys;
using Beam.Abstractions;
using Beam.Data;
using Beam.Downloaders;
using Beam.Dynamic;
using Beam.Models;

namespace Beam.Fluent;

public static class FluentDownload {
    public static ITransformStage<RawType, OutType> Links<RawType, OutType>(params IEnumerable<string> links) where RawType : IDocument {
        return new TransformStage<RawType, OutType>(new DownloadContextBuilder<RawType>()
            .WithLinks(links));
    }

    public static ITransformStage<RawType, OutType>
        ResourceDefinition<RawType, OutType>(ResourceDefinition definition) where RawType : IDocument {
        if (definition.Location.States.Count == 0)
            throw new ArgumentException(Exceptions.Exceptions.resource_definition_invalid_states_count, nameof(definition));
        var linkGenerator = new OrderedLinkGenerator(definition.Location.Segments, (NumberedStateChanger)definition.Location.StateChanger.Behavior,
            definition.Location.States.First().Copy());
        return new TransformStage<RawType, OutType>(new DownloadContextBuilder<RawType>()
            .WithLinks(StringEnumerable.FromGenerator(linkGenerator!)));
    }
    
    public static ITransformStage<RawType, OutType> FromContext<RawType, OutType>(DownloadContext<RawType> existing) where RawType : IDocument {
        return new TransformStage<RawType, OutType>(DownloadContextBuilder<RawType>.FromContext(existing));
    }
}