130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package github
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/google/go-github/github"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
)
|
|
|
|
const pullPermission string = "pull"
|
|
const pushPermission string = "push"
|
|
const adminPermission string = "admin"
|
|
|
|
func resourceGithubTeamRepository() *schema.Resource {
|
|
return &schema.Resource{
|
|
Create: resourceGithubTeamRepositoryCreate,
|
|
Read: resourceGithubTeamRepositoryRead,
|
|
Update: resourceGithubTeamRepositoryUpdate,
|
|
Delete: resourceGithubTeamRepositoryDelete,
|
|
|
|
Schema: map[string]*schema.Schema{
|
|
"team_id": &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
ForceNew: true,
|
|
},
|
|
"repository": &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
ForceNew: true,
|
|
},
|
|
"permission": &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Optional: true,
|
|
Default: "pull",
|
|
ValidateFunc: validateValueFunc([]string{"pull", "push", "admin"}),
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func resourceGithubTeamRepositoryCreate(d *schema.ResourceData, meta interface{}) error {
|
|
client := meta.(*Organization).client
|
|
t := d.Get("team_id").(string)
|
|
r := d.Get("repository").(string)
|
|
p := d.Get("permission").(string)
|
|
|
|
_, err := client.Organizations.AddTeamRepo(toGithubID(t), meta.(*Organization).name, r,
|
|
&github.OrganizationAddTeamRepoOptions{Permission: p})
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
d.SetId(buildTwoPartID(&t, &r))
|
|
|
|
return resourceGithubTeamRepositoryRead(d, meta)
|
|
}
|
|
|
|
func resourceGithubTeamRepositoryRead(d *schema.ResourceData, meta interface{}) error {
|
|
client := meta.(*Organization).client
|
|
t := d.Get("team_id").(string)
|
|
r := d.Get("repository").(string)
|
|
|
|
repo, _, repoErr := client.Organizations.IsTeamRepo(toGithubID(t), meta.(*Organization).name, r)
|
|
|
|
if repoErr != nil {
|
|
d.SetId("")
|
|
return nil
|
|
}
|
|
|
|
repositoryName := repo.Name
|
|
|
|
d.Set("team_id", t)
|
|
d.Set("repository", repositoryName)
|
|
|
|
permName, permErr := getRepoPermission(repo.Permissions)
|
|
|
|
if permErr != nil {
|
|
return permErr
|
|
}
|
|
|
|
d.Set("permission", permName)
|
|
|
|
return nil
|
|
}
|
|
|
|
func resourceGithubTeamRepositoryUpdate(d *schema.ResourceData, meta interface{}) error {
|
|
client := meta.(*Organization).client
|
|
t := d.Get("team_id").(string)
|
|
r := d.Get("repository").(string)
|
|
p := d.Get("permission").(string)
|
|
|
|
// the go-github library's AddTeamRepo method uses the add/update endpoint from Github API
|
|
_, err := client.Organizations.AddTeamRepo(toGithubID(t), meta.(*Organization).name, r,
|
|
&github.OrganizationAddTeamRepoOptions{Permission: p})
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return resourceGithubTeamRepositoryRead(d, meta)
|
|
}
|
|
|
|
func resourceGithubTeamRepositoryDelete(d *schema.ResourceData, meta interface{}) error {
|
|
client := meta.(*Organization).client
|
|
t := d.Get("team_id").(string)
|
|
r := d.Get("repository").(string)
|
|
|
|
_, err := client.Organizations.RemoveTeamRepo(toGithubID(t), meta.(*Organization).name, r)
|
|
|
|
return err
|
|
}
|
|
|
|
func getRepoPermission(p *map[string]bool) (string, error) {
|
|
|
|
// Permissions are returned in this map format such that if you have a certain level
|
|
// of permission, all levels below are also true. For example, if a team has push
|
|
// permission, the map will be: {"pull": true, "push": true, "admin": false}
|
|
if (*p)[adminPermission] {
|
|
return adminPermission, nil
|
|
} else if (*p)[pushPermission] {
|
|
return pushPermission, nil
|
|
} else {
|
|
if (*p)[pullPermission] {
|
|
return pullPermission, nil
|
|
}
|
|
return "", errors.New("At least one permission expected from permissions map.")
|
|
}
|
|
}
|