terraform-provider-github/github/resource_github_repository_pull_request_creation_policy.go at efeb9d41169f5801986d323af733cdcd74a096d6 · integrations/terraform-provider-github · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package github

import (
	"context"
	"fmt"

	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
)

func resourceGithubRepositoryPullRequestCreationPolicy() *schema.Resource {
	return &schema.Resource{
		Description:   "Manages the pull request creation policy for a repository.",
		CreateContext: resourceGithubRepositoryPullRequestCreationPolicyCreate,
		ReadContext:   resourceGithubRepositoryPullRequestCreationPolicyRead,
		UpdateContext: resourceGithubRepositoryPullRequestCreationPolicyUpdate,
		DeleteContext: resourceGithubRepositoryPullRequestCreationPolicyDelete,
		Importer: &schema.ResourceImporter{
			StateContext: resourceGithubRepositoryPullRequestCreationPolicyImport,
		},

		Schema: map[string]*schema.Schema{
			"repository": {
				Type:        schema.TypeString,
				Required:    true,
				ForceNew:    true,
				Description: "The name of the GitHub repository.",
			},
			"policy": {
				Type:             schema.TypeString,
				Required:         true,
				Description:      "Controls who can create pull requests for the repository. Can be `all` or `collaborators_only`.",
				ValidateDiagFunc: validation.ToDiagFunc(validation.StringInSlice([]string{"all", "collaborators_only"}, false)),
			},
		},
	}
}

func resourceGithubRepositoryPullRequestCreationPolicyCreate(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics {
	repoName := d.Get("repository").(string)
	policy := d.Get("policy").(string)

	nodeID, err := getRepositoryID(repoName, meta)
	if err != nil {
		return diag.Errorf("error resolving repository node ID for %s: %s", repoName, err)
	}

	if err := updateRepositoryPullRequestCreationPolicy(ctx, nodeID, policy, meta); err != nil {
		return diag.Errorf("error setting pull request creation policy for %s: %s", repoName, err)
	}

	d.SetId(repoName)
	return resourceGithubRepositoryPullRequestCreationPolicyRead(ctx, d, meta)
}

func resourceGithubRepositoryPullRequestCreationPolicyRead(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics {
	owner := meta.(*Owner).name
	repoName := d.Id()

	policy, err := getRepositoryPullRequestCreationPolicy(ctx, owner, repoName, meta)
	if err != nil {
		return diag.Errorf("error reading pull request creation policy for %s/%s: %s", owner, repoName, err)
	}

	if err := d.Set("policy", policy); err != nil {
		return diag.FromErr(err)
	}
	if err := d.Set("repository", repoName); err != nil {
		return diag.FromErr(err)
	}

	return nil
}

func resourceGithubRepositoryPullRequestCreationPolicyUpdate(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics {
	repoName := d.Id()
	policy := d.Get("policy").(string)

	nodeID, err := getRepositoryID(repoName, meta)
	if err != nil {
		return diag.Errorf("error resolving repository node ID for %s: %s", repoName, err)
	}

	if err := updateRepositoryPullRequestCreationPolicy(ctx, nodeID, policy, meta); err != nil {
		return diag.Errorf("error updating pull request creation policy for %s: %s", repoName, err)
	}

	return resourceGithubRepositoryPullRequestCreationPolicyRead(ctx, d, meta)
}

func resourceGithubRepositoryPullRequestCreationPolicyDelete(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics {
	repoName := d.Id()

	nodeID, err := getRepositoryID(repoName, meta)
	if err != nil {
		return diag.Errorf("error resolving repository node ID for %s: %s", repoName, err)
	}

	if err := updateRepositoryPullRequestCreationPolicy(ctx, nodeID, "all", meta); err != nil {
		return diag.Errorf("error resetting pull request creation policy for %s: %s", repoName, err)
	}

	return nil
}

func resourceGithubRepositoryPullRequestCreationPolicyImport(ctx context.Context, d *schema.ResourceData, meta any) ([]*schema.ResourceData, error) {
	repoName := d.Id()

	if err := d.Set("repository", repoName); err != nil {
		return nil, err
	}

	diags := resourceGithubRepositoryPullRequestCreationPolicyRead(ctx, d, meta)
	if diags.HasError() {
		return nil, fmt.Errorf("%s", diags[0].Summary)
	}

	return []*schema.ResourceData{d}, nil
}