From 02def0d8ad79b7b7931b41c22f5fd8a39eea5337 Mon Sep 17 00:00:00 2001 From: Virgil Date: Thu, 2 Apr 2026 01:40:46 +0000 Subject: [PATCH] feat: add iterator for repository flags Co-Authored-By: Virgil --- repos.go | 6 ++++++ repos_test.go | 39 +++++++++++++++++++++++++++++++++++++++ 2 files changed, 45 insertions(+) diff --git a/repos.go b/repos.go index 6d4ae81..ffb3583 100644 --- a/repos.go +++ b/repos.go @@ -325,6 +325,12 @@ func (s *RepoService) ListFlags(ctx context.Context, owner, repo string) ([]stri return out, nil } +// IterFlags returns an iterator over all flags for a repository. +func (s *RepoService) IterFlags(ctx context.Context, owner, repo string) iter.Seq2[string, error] { + path := ResolvePath("/api/v1/repos/{owner}/{repo}/flags", pathParams("owner", owner, "repo", repo)) + return ListIter[string](ctx, s.client, path, nil) +} + // ReplaceFlags replaces all flags for a repository. func (s *RepoService) ReplaceFlags(ctx context.Context, owner, repo string, opts *types.ReplaceFlagsOption) error { path := ResolvePath("/api/v1/repos/{owner}/{repo}/flags", pathParams("owner", owner, "repo", repo)) diff --git a/repos_test.go b/repos_test.go index 50e1f0f..4c5ad06 100644 --- a/repos_test.go +++ b/repos_test.go @@ -1176,6 +1176,45 @@ func TestRepoService_ListFlags_Good(t *testing.T) { } } +func TestRepoService_IterFlags_Good(t *testing.T) { + requests := 0 + srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + requests++ + if r.Method != http.MethodGet { + t.Errorf("expected GET, got %s", r.Method) + } + if r.URL.Path != "/api/v1/repos/core/go-forge/flags" { + t.Errorf("wrong path: %s", r.URL.Path) + http.NotFound(w, r) + return + } + if got := r.URL.Query().Get("page"); got != "1" { + t.Errorf("got page=%q, want %q", got, "1") + } + if got := r.URL.Query().Get("limit"); got != "50" { + t.Errorf("got limit=%q, want %q", got, "50") + } + w.Header().Set("X-Total-Count", "2") + json.NewEncoder(w).Encode([]string{"alpha", "beta"}) + })) + defer srv.Close() + + f := NewForge(srv.URL, "tok") + var got []string + for flag, err := range f.Repos.IterFlags(context.Background(), "core", "go-forge") { + if err != nil { + t.Fatal(err) + } + got = append(got, flag) + } + if requests != 1 { + t.Fatalf("expected 1 request, got %d", requests) + } + if !reflect.DeepEqual(got, []string{"alpha", "beta"}) { + t.Fatalf("got %#v", got) + } +} + func TestRepoService_ReplaceFlags_Good(t *testing.T) { srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.Method != http.MethodPut {