98 lines
1.9 KiB
Go
98 lines
1.9 KiB
Go
package list
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/cli/cli/v2/api"
|
|
)
|
|
|
|
type OrganizationList struct {
|
|
Organizations []Organization
|
|
TotalCount int
|
|
User string
|
|
}
|
|
|
|
type Organization struct {
|
|
Login string
|
|
}
|
|
|
|
func listOrgs(httpClient *http.Client, hostname string, limit int) (*OrganizationList, error) {
|
|
type response struct {
|
|
User struct {
|
|
Login string
|
|
Organizations struct {
|
|
TotalCount int
|
|
Nodes []Organization
|
|
PageInfo struct {
|
|
HasNextPage bool
|
|
EndCursor string
|
|
}
|
|
} `graphql:"organizations(first: $limit, after: $endCursor)"`
|
|
}
|
|
}
|
|
|
|
query := `query OrganizationList($user: String!, $limit: Int!, $endCursor: String) {
|
|
user(login: $user) {
|
|
login
|
|
organizations(first: $limit, after: $endCursor) {
|
|
totalCount
|
|
nodes {
|
|
login
|
|
}
|
|
pageInfo {
|
|
hasNextPage
|
|
endCursor
|
|
}
|
|
}
|
|
}
|
|
}`
|
|
|
|
client := api.NewClientFromHTTP(httpClient)
|
|
|
|
user, err := api.CurrentLoginName(client, hostname)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
listResult := OrganizationList{}
|
|
listResult.User = user
|
|
pageLimit := min(limit, 100)
|
|
variables := map[string]interface{}{
|
|
"user": user,
|
|
}
|
|
|
|
loop:
|
|
for {
|
|
variables["limit"] = pageLimit
|
|
var data response
|
|
err := client.GraphQL(hostname, query, variables, &data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
listResult.TotalCount = data.User.Organizations.TotalCount
|
|
|
|
for _, org := range data.User.Organizations.Nodes {
|
|
listResult.Organizations = append(listResult.Organizations, org)
|
|
if len(listResult.Organizations) == limit {
|
|
break loop
|
|
}
|
|
}
|
|
|
|
if data.User.Organizations.PageInfo.HasNextPage {
|
|
variables["endCursor"] = data.User.Organizations.PageInfo.EndCursor
|
|
pageLimit = min(pageLimit, limit-len(listResult.Organizations))
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
|
|
return &listResult, nil
|
|
}
|
|
|
|
func min(a, b int) int {
|
|
if a < b {
|
|
return a
|
|
}
|
|
return b
|
|
}
|