Files
teleport/lib/msgraph/models.go
T
Justinas Stankevičius c5a5f8e7a2 Replace MS Graph SDK with an in-house client (#44069)
* Add MS Graph client skeleton

* Complete retry logic

* Add GetID() to GroupMember

* Parse MS Graph API error responses

* Pull out page size into config

* Add missing "patch" methods to client interface

* Move code around

* Factor out roundtrip()

* Fill out remaining methods

* Add godocs and solve lints

* Run gci

* Move godocs to the interface

* Add missing method

* Finish errors_test

* close body in test

* nit

* Add license headers

* Move azureoidc onboarding code to msgraph.Client

* Make AzureTokenProvider public

* Use cmp.Or to shorten config defaults handling

* Address review nits

* Parallelize tests

* nit

* Respect context cancellation in retry logic

* Utilize BlockUntil in tests

* Remove Client interface, make client struct public

* Prefer defaults.HTTPClient to http.DefaultClient

* Iterate entra apps, do not buffer all in memory

* nit

* Set additional timeouts on default HTTP client

* Escape path segments when constructing URLs

* Add a general timeout for default HTTP client
2024-07-17 15:51:39 +00:00

117 lines
3.2 KiB
Go

// Teleport
// Copyright (C) 2024 Gravitational, Inc.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package msgraph
import (
"encoding/json"
"github.com/gravitational/trace"
)
type GroupMember interface {
GetID() *string
isGroupMember()
}
type DirectoryObject struct {
ID *string `json:"id,omitempty"`
DisplayName *string `json:"displayName,omitempty"`
}
type Group struct {
DirectoryObject
}
func (g *Group) isGroupMember() {}
func (g *Group) GetID() *string { return g.ID }
type User struct {
DirectoryObject
Mail *string `json:"mail,omitempty"`
OnPremisesSAMAccountName *string `json:"onPremisesSamAccountName,omitempty"`
UserPrincipalName *string `json:"userPrincipalName,omitempty"`
}
func (g *User) isGroupMember() {}
func (u *User) GetID() *string { return u.ID }
type Application struct {
DirectoryObject
AppID *string `json:"appId,omitempty"`
IdentifierURIs *[]string `json:"identifierUris,omitempty"`
Web *WebApplication `json:"web,omitempty"`
}
type WebApplication struct {
RedirectURIs *[]string `json:"redirectUris,omitempty"`
}
type ServicePrincipal struct {
DirectoryObject
AppRoleAssignmentRequired *bool `json:"appRoleAssignmentRequired,omitempty"`
PreferredSingleSignOnMode *string `json:"preferredSingleSignOnMode,omitempty"`
PreferredTokenSigningKeyThumbprint *string `json:"preferredTokenSigningKeyThumbprint,omitempty"`
}
type ApplicationServicePrincipal struct {
Application *Application `json:"application,omitempty"`
ServicePrincipal *ServicePrincipal `json:"servicePrincipal,omitempty"`
}
type FederatedIdentityCredential struct {
Audiences *[]string `json:"audiences,omitempty"`
Issuer *string `json:"issuer,omitempty"`
Name *string `json:"name,omitempty"`
Subject *string `json:"subject,omitempty"`
}
type SelfSignedCertificate struct {
Thumbprint *string `json:"thumbprint,omitempty"`
}
type AppRoleAssignment struct {
ID *string `json:"id,omitempty"`
AppRoleID *string `json:"appRoleId,omitempty"`
PrincipalID *string `json:"principalId,omitempty"`
ResourceID *string `json:"resourceId,omitempty"`
}
func decodeGroupMember(msg json.RawMessage) (GroupMember, error) {
var temp struct {
Type string `json:"@odata.type"`
}
if err := json.Unmarshal(msg, &temp); err != nil {
return nil, trace.Wrap(err)
}
var err error
var member GroupMember
switch temp.Type {
case "#microsoft.graph.user":
var u *User
err = json.Unmarshal(msg, &u)
member = u
default:
err = trace.BadParameter("unsupported group member type: %s", temp.Type)
}
return member, trace.Wrap(err)
}