summaryrefslogtreecommitdiff
path: root/GetMedia.cs
blob: b01f38eeb61a77e77f6a97c63ecf58ef96d1a279 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
using HyperBooru.ApiClient;
using HyperBooru.ApiModels;
using System.Management.Automation;

namespace HyperBooru.PowerShell;

[Alias("ghbm")]
[Cmdlet(VerbsCommon.Get, "HyperBooruMedia")]
public class GetMediaCmdlet : PSCmdlet {
    [Parameter(Mandatory = true)]
    public HBSession Session { get; set; }

    [Parameter]
    public SwitchParameter     SelectIngest      { get; set; }
    [Parameter]
    public SwitchParameter     IncludeNsfw       { get; set; }
    [Parameter]
    public Guid?               ContinuationToken { get; set; }
    [Parameter]
    public int                 Count             { get; set; } = 50;
    [Parameter]
    public ApiModels.SortOrder SortOrder         { get; set; }
    [Parameter(ParameterSetName = "All", Mandatory = true)]
    public SwitchParameter     All               { get; set; }
    [Parameter(ParameterSetName = "TagId", Mandatory = true)]
    public Guid                TagId             { get; set; }
    [Parameter(ParameterSetName = "Query", Mandatory = true)]
    public string              Query             { get; set; }

    protected override void ProcessRecord() {
        FeedRequest feedRequest;

        switch(ParameterSetName) {
            case "TagId":
                feedRequest = new FeedTagRequest {
                    SelectIngest      = SelectIngest,
                    IncludeNsfw       = IncludeNsfw,
                    ContinuationToken = ContinuationToken,
                    Count             = Count,
                    SortOrder         = SortOrder,
                    TagId             = TagId
                };
                break;
            case "Query":
                feedRequest = new FeedSearchRequest {
                    SelectIngest      = SelectIngest,
                    IncludeNsfw       = IncludeNsfw,
                    ContinuationToken = ContinuationToken,
                    Count             = Count,
                    SortOrder         = SortOrder,
                    Query             = Query
                };
                break;
            default:
                feedRequest = new FeedRequest {
                    SelectIngest      = SelectIngest,
                    IncludeNsfw       = IncludeNsfw,
                    ContinuationToken = ContinuationToken,
                    Count             = Count,
                    SortOrder         = SortOrder
                };
                break;
        }

        var mediaIds = Session.Feed.LoadChunkAsync(feedRequest)
            .GetAwaiter()
            .GetResult();

        foreach(var m in mediaIds)
            WriteObject(m);
    }
}