You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
// Copyright 2017 Drone.IO Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package github
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"git.awesome-for.me/liuzhiguo/go-scm/scm"
|
|
)
|
|
|
|
type organizationService struct {
|
|
client *wrapper
|
|
}
|
|
|
|
func (s *organizationService) Find(ctx context.Context, name string) (*scm.Organization, *scm.Response, error) {
|
|
path := fmt.Sprintf("orgs/%s", name)
|
|
out := new(organization)
|
|
res, err := s.client.do(ctx, "GET", path, nil, out)
|
|
return convertOrganization(out), res, err
|
|
}
|
|
|
|
func (s *organizationService) FindMembership(ctx context.Context, name, username string) (*scm.Membership, *scm.Response, error) {
|
|
path := fmt.Sprintf("orgs/%s/memberships/%s", name, username)
|
|
out := new(membership)
|
|
res, err := s.client.do(ctx, "GET", path, nil, out)
|
|
return convertMembership(out), res, err
|
|
}
|
|
|
|
func (s *organizationService) List(ctx context.Context, opts scm.ListOptions) ([]*scm.Organization, *scm.Response, error) {
|
|
path := fmt.Sprintf("user/orgs?%s", encodeListOptions(opts))
|
|
out := []*organization{}
|
|
res, err := s.client.do(ctx, "GET", path, nil, &out)
|
|
return convertOrganizationList(out), res, err
|
|
}
|
|
|
|
func convertOrganizationList(from []*organization) []*scm.Organization {
|
|
to := []*scm.Organization{}
|
|
for _, v := range from {
|
|
to = append(to, convertOrganization(v))
|
|
}
|
|
return to
|
|
}
|
|
|
|
type organization struct {
|
|
Login string `json:"login"`
|
|
Avatar string `json:"avatar_url"`
|
|
}
|
|
|
|
type membership struct {
|
|
State string `json:"state"`
|
|
Role string `json:"role"`
|
|
}
|
|
|
|
func convertOrganization(from *organization) *scm.Organization {
|
|
return &scm.Organization{
|
|
Name: from.Login,
|
|
Avatar: from.Avatar,
|
|
}
|
|
}
|
|
|
|
func convertMembership(from *membership) *scm.Membership {
|
|
to := new(scm.Membership)
|
|
switch from.State {
|
|
case "active":
|
|
to.Active = true
|
|
}
|
|
switch from.Role {
|
|
case "admin":
|
|
to.Role = scm.RoleAdmin
|
|
case "member":
|
|
to.Role = scm.RoleMember
|
|
default:
|
|
to.Role = scm.RoleUndefined
|
|
}
|
|
return to
|
|
}
|