data_source_github_ref.go

package github

import (
	"context"
	"errors"
	"fmt"
	"net/http"

	"github.com/google/go-github/v84/github"
	"github.com/hashicorp/terraform-plugin-log/tflog"
	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
)

func dataSourceGithubRef() *schema.Resource {
	return &schema.Resource{
		ReadContext: dataSourceGithubRefRead,

		Schema: map[string]*schema.Schema{
			"ref": {
				Type:     schema.TypeString,
				Required: true,
				ForceNew: true,
			},
			"repository": {
				Type:     schema.TypeString,
				Required: true,
				ForceNew: true,
			},
			"owner": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"etag": {
				Type:     schema.TypeString,
				Computed: true,
			},
			"sha": {
				Type:     schema.TypeString,
				Computed: true,
			},
		},
	}
}

func dataSourceGithubRefRead(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics {
	client := meta.(*Owner).v3client
	owner, ok := d.Get("owner").(string)
	if !ok {
		owner = meta.(*Owner).name
	}
	repoName := d.Get("repository").(string)
	ref := d.Get("ref").(string)

	refData, resp, err := client.Git.GetRef(ctx, owner, repoName, ref)
	if err != nil {
		var ghErr *github.ErrorResponse
		if errors.As(err, &ghErr) {
			if ghErr.Response.StatusCode == http.StatusNotFound {
				tflog.Debug(ctx, fmt.Sprintf("Missing GitHub ref %s/%s (%s)", owner, repoName, ref), map[string]any{
					"owner": owner,
					"repo":  repoName,
					"ref":   ref,
				})
				d.SetId("")
				return nil
			}
		}
		return diag.FromErr(err)
	}

	d.SetId(buildTwoPartID(repoName, ref))
	err = d.Set("etag", resp.Header.Get("ETag"))
	if err != nil {
		return diag.FromErr(err)
	}
	err = d.Set("sha", *refData.Object.SHA)
	if err != nil {
		return diag.FromErr(err)
	}

	return nil
}