121 lines
3.0 KiB
Go
121 lines
3.0 KiB
Go
package teams
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
)
|
|
|
|
// InfrastructureAccount defines an account of the team on some infrastructure (i.e AWS, Google) platform.
|
|
type infrastructureAccount struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Provider string `json:"provider"`
|
|
Type string `json:"type"`
|
|
Description string `json:"description"`
|
|
Owner string `json:"owner"`
|
|
OwnerDn string `json:"owner_dn"`
|
|
Disabled bool `json:"disabled"`
|
|
}
|
|
|
|
// Team defines informaiton for a single team, including the list of members and infrastructure accounts.
|
|
type team struct {
|
|
Dn string `json:"dn"`
|
|
ID string `json:"id"`
|
|
TeamName string `json:"id_name"`
|
|
TeamID string `json:"team_id"`
|
|
Type string `json:"type"`
|
|
FullName string `json:"name"`
|
|
Aliases []string `json:"alias"`
|
|
Mails []string `json:"mail"`
|
|
Members []string `json:"member"`
|
|
CostCenter string `json:"cost_center"`
|
|
DeliveryLead string `json:"delivery_lead"`
|
|
ParentTeamID string `json:"parent_team_id"`
|
|
|
|
InfrastructureAccounts []infrastructureAccount `json:"infrastructure-accounts"`
|
|
}
|
|
|
|
type Client interface {
|
|
TeamInfo(string, string) (*team, error)
|
|
}
|
|
|
|
type httpClient interface {
|
|
Do(req *http.Request) (*http.Response, error)
|
|
}
|
|
|
|
type API struct {
|
|
httpClient
|
|
url string
|
|
logger *logrus.Entry
|
|
}
|
|
|
|
// NewTeamsAPI creates an object to query the team API.
|
|
func NewTeamsAPI(url string, log *logrus.Logger) *API {
|
|
t := API{
|
|
url: strings.TrimRight(url, "/"),
|
|
httpClient: &http.Client{},
|
|
logger: log.WithField("pkg", "teamsapi"),
|
|
}
|
|
|
|
return &t
|
|
}
|
|
|
|
// TeamInfo returns information about a given team using its ID and a token to authenticate to the API service.
|
|
func (t *API) TeamInfo(teamID, token string) (tm *team, err error) {
|
|
var (
|
|
req *http.Request
|
|
resp *http.Response
|
|
)
|
|
|
|
url := fmt.Sprintf("%s/teams/%s", t.url, teamID)
|
|
t.logger.Debugf("Request url: %s", url)
|
|
req, err = http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
req.Header.Add("Authorization", "Bearer "+token)
|
|
resp, err = t.httpClient.Do(req)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func() {
|
|
closeErr := resp.Body.Close()
|
|
if closeErr != nil {
|
|
err = fmt.Errorf("error when closing response: %v", closeErr)
|
|
}
|
|
}()
|
|
if resp.StatusCode != 200 {
|
|
var raw map[string]json.RawMessage
|
|
d := json.NewDecoder(resp.Body)
|
|
err = d.Decode(&raw)
|
|
if err != nil {
|
|
err = fmt.Errorf("team API query failed with status code %d and malformed response: %v", resp.StatusCode, err)
|
|
return
|
|
}
|
|
|
|
if errMessage, ok := raw["error"]; ok {
|
|
err = fmt.Errorf("team API query failed with status code %d and message: '%v'", resp.StatusCode, string(errMessage))
|
|
return
|
|
}
|
|
err = fmt.Errorf("team API query failed with status code %d", resp.StatusCode)
|
|
|
|
return
|
|
}
|
|
|
|
tm = &team{}
|
|
d := json.NewDecoder(resp.Body)
|
|
err = d.Decode(tm)
|
|
if err != nil {
|
|
err = fmt.Errorf("could not parse team API response: %v", err)
|
|
tm = nil
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|