Add org-slug and org-id flags to orb validate & process commands (#922)

api/api.go

@@ -513,7 +513,7 @@ func WhoamiQuery(cl *graphql.Client) (*WhoamiResponse, error) {
 }
 
 // OrbQuery validated and processes an orb.
-func OrbQuery(cl *graphql.Client, configPath string) (*ConfigResponse, error) {
+func OrbQuery(cl *graphql.Client, configPath string, ownerId string) (*ConfigResponse, error) {
 	var response OrbConfigResponse
 
 	config, err := loadYaml(configPath)
@@ -522,8 +522,8 @@ func OrbQuery(cl *graphql.Client, configPath string) (*ConfigResponse, error) {
 	}
 
 	query := `
-		query ValidateOrb ($config: String!) {
-			orbConfig(orbYaml: $config) {
+		query ValidateOrb ($config: String!, $owner: UUID) {
+			orbConfig(orbYaml: $config, ownerId: $owner) {
 				valid,
 				errors { message },
 				sourceYaml,
@@ -533,6 +533,11 @@ func OrbQuery(cl *graphql.Client, configPath string) (*ConfigResponse, error) {
 
 	request := graphql.NewRequest(query)
 	request.Var("config", config)
+
+	if ownerId != "" {
+		request.Var("owner", ownerId)
+	}
+
 	request.SetToken(cl.Token)
 
 	err = cl.Run(request, &response)

api/collaborators/collaborators.go

@@ -0,0 +1,14 @@
+package collaborators
+
+type CollaborationResult struct {
+	VcsType   string `json:"vcs_type"`
+	OrgSlug   string `json:"slug"`
+	OrgName   string `json:"name"`
+	OrgId     string `json:"id"`
+	AvatarUrl string `json:"avatar_url"`
+}
+
+type CollaboratorsClient interface {
+	GetCollaborationBySlug(slug string) (*CollaborationResult, error)
+	GetOrgCollaborations() ([]CollaborationResult, error)
+}

api/collaborators/collaborators_rest.go

@@ -0,0 +1,64 @@
+package collaborators
+
+import (
+	"net/url"
+	"strings"
+
+	"github.com/CircleCI-Public/circleci-cli/api/rest"
+	"github.com/CircleCI-Public/circleci-cli/settings"
+)
+
+var (
+	CollaborationsPath = "me/collaborations"
+)
+
+type collaboratorsRestClient struct {
+	client *rest.Client
+}
+
+// NewCollaboratorsRestClient returns a new collaboratorsRestClient satisfying the api.CollaboratorsClient
+// interface via the REST API.
+func NewCollaboratorsRestClient(config settings.Config) (*collaboratorsRestClient, error) {
+	client := &collaboratorsRestClient{
+		client: rest.NewFromConfig(config.Host, &config),
+	}
+	return client, nil
+}
+
+func (c *collaboratorsRestClient) GetOrgCollaborations() ([]CollaborationResult, error) {
+	req, err := c.client.NewRequest("GET", &url.URL{Path: CollaborationsPath}, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	var resp []CollaborationResult
+	_, err = c.client.DoRequest(req, &resp)
+	return resp, err
+}
+
+func (c *collaboratorsRestClient) GetCollaborationBySlug(slug string) (*CollaborationResult, error) {
+	// Support for <vcs-name>/<org-name> as well as <vcs-short>/<org-name> for the slug
+	// requires splitting
+	collaborations, err := c.GetOrgCollaborations()
+
+	if err != nil {
+		return nil, err
+	}
+
+	slugParts := strings.Split(slug, "/")
+
+	for _, v := range collaborations {
+		// The rest-api allways returns <vsc-short>/<org-name> as a slug
+		if v.OrgSlug == slug {
+			return &v, nil
+		}
+
+		// Compare first part of argument slug with the VCSType
+		splitted := strings.Split(v.OrgSlug, "/")
+		if len(slugParts) >= 2 && len(splitted) >= 2 && slugParts[0] == v.VcsType && slugParts[1] == splitted[1] {
+			return &v, nil
+		}
+	}
+
+	return nil, nil
+}