99 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			99 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
package teams
 | 
						|
 | 
						|
import (
 | 
						|
	"encoding/json"
 | 
						|
	"fmt"
 | 
						|
	"net/http"
 | 
						|
	"strings"
 | 
						|
 | 
						|
	"github.com/Sirupsen/logrus"
 | 
						|
)
 | 
						|
 | 
						|
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"`
 | 
						|
}
 | 
						|
 | 
						|
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 TeamsAPI struct {
 | 
						|
	url                string
 | 
						|
	httpClient         *http.Client
 | 
						|
	logger             *logrus.Entry
 | 
						|
	RefreshTokenAction func() (string, error)
 | 
						|
}
 | 
						|
 | 
						|
func NewTeamsAPI(url string, log *logrus.Logger) *TeamsAPI {
 | 
						|
	t := TeamsAPI{
 | 
						|
		url:        strings.TrimRight(url, "/"),
 | 
						|
		httpClient: &http.Client{},
 | 
						|
		logger:     log.WithField("pkg", "teamsapi"),
 | 
						|
	}
 | 
						|
 | 
						|
	return &t
 | 
						|
}
 | 
						|
 | 
						|
func (t *TeamsAPI) TeamInfo(teamId string) (*Team, error) {
 | 
						|
	// TODO: avoid getting a new token on every call to the Teams API.
 | 
						|
	token, err := t.RefreshTokenAction()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	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 nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	req.Header.Add("Authorization", "Bearer "+token)
 | 
						|
	resp, err := t.httpClient.Do(req)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	defer resp.Body.Close()
 | 
						|
	if resp.StatusCode != 200 {
 | 
						|
		var raw map[string]json.RawMessage
 | 
						|
		d := json.NewDecoder(resp.Body)
 | 
						|
		err = d.Decode(&raw)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
 | 
						|
		if errMessage, ok := raw["error"]; ok {
 | 
						|
			return nil, fmt.Errorf("Team API query failed with status code %d and message: '%s'", resp.StatusCode, string(errMessage))
 | 
						|
		} else {
 | 
						|
			return nil, fmt.Errorf("Team API query failed with status code %d", resp.StatusCode)
 | 
						|
		}
 | 
						|
	}
 | 
						|
	teamInfo := &Team{}
 | 
						|
	d := json.NewDecoder(resp.Body)
 | 
						|
	err = d.Decode(teamInfo)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return teamInfo, nil
 | 
						|
}
 |