mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-19 06:17:32 +00:00
84 lines
2.6 KiB
Go
84 lines
2.6 KiB
Go
package admin
|
|
|
|
import (
|
|
"context"
|
|
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
|
|
"github.com/caos/zitadel/internal/api/grpc/object"
|
|
org_grpc "github.com/caos/zitadel/internal/api/grpc/org"
|
|
"github.com/caos/zitadel/internal/domain"
|
|
usr_model "github.com/caos/zitadel/internal/user/model"
|
|
admin_pb "github.com/caos/zitadel/pkg/grpc/admin"
|
|
obj_pb "github.com/caos/zitadel/pkg/grpc/object"
|
|
)
|
|
|
|
func (s *Server) IsOrgUnique(ctx context.Context, req *admin_pb.IsOrgUniqueRequest) (*admin_pb.IsOrgUniqueResponse, error) {
|
|
isUnique, err := s.query.IsOrgUnique(ctx, req.Name, req.Domain)
|
|
return &admin_pb.IsOrgUniqueResponse{IsUnique: isUnique}, err
|
|
}
|
|
|
|
func (s *Server) GetOrgByID(ctx context.Context, req *admin_pb.GetOrgByIDRequest) (*admin_pb.GetOrgByIDResponse, error) {
|
|
org, err := s.query.OrgByID(ctx, req.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &admin_pb.GetOrgByIDResponse{Org: org_grpc.OrgViewToPb(org)}, nil
|
|
}
|
|
|
|
func (s *Server) ListOrgs(ctx context.Context, req *admin_pb.ListOrgsRequest) (*admin_pb.ListOrgsResponse, error) {
|
|
queries, err := listOrgRequestToModel(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
orgs, err := s.query.SearchOrgs(ctx, queries)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &admin_pb.ListOrgsResponse{
|
|
Result: org_grpc.OrgViewsToPb(orgs.Orgs),
|
|
Details: &obj_pb.ListDetails{
|
|
TotalResult: orgs.Count,
|
|
ProcessedSequence: orgs.Sequence,
|
|
ViewTimestamp: timestamppb.New(orgs.Timestamp),
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func (s *Server) SetUpOrg(ctx context.Context, req *admin_pb.SetUpOrgRequest) (*admin_pb.SetUpOrgResponse, error) {
|
|
userIDs, err := s.getClaimedUserIDsOfOrgDomain(ctx, domain.NewIAMDomainName(req.Org.Name, s.iamDomain))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
human := setUpOrgHumanToDomain(req.User.(*admin_pb.SetUpOrgRequest_Human_).Human) //TODO: handle machine
|
|
org := setUpOrgOrgToDomain(req.Org)
|
|
|
|
objectDetails, err := s.command.SetUpOrg(ctx, org, human, userIDs, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &admin_pb.SetUpOrgResponse{
|
|
Details: object.DomainToAddDetailsPb(objectDetails),
|
|
}, nil
|
|
}
|
|
|
|
func (s *Server) getClaimedUserIDsOfOrgDomain(ctx context.Context, orgDomain string) ([]string, error) {
|
|
users, err := s.users.SearchUsers(ctx, &usr_model.UserSearchRequest{
|
|
Queries: []*usr_model.UserSearchQuery{
|
|
{
|
|
Key: usr_model.UserSearchKeyPreferredLoginName,
|
|
Method: domain.SearchMethodEndsWithIgnoreCase,
|
|
Value: "@" + orgDomain,
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
userIDs := make([]string, len(users.Result))
|
|
for i, user := range users.Result {
|
|
userIDs[i] = user.ID
|
|
}
|
|
return userIDs, nil
|
|
}
|