From 41c3ee72e5714c3b651500aa4740d6447a257488 Mon Sep 17 00:00:00 2001
From: Kyle Carberry <kyle@coder.com>
Date: Thu, 24 Feb 2022 10:47:51 -0600
Subject: [PATCH] ci: Update DataDog GitHub branch to fallback to GITHUB_REF
 (#356)

This was detecting branches, but not our "main" branch before.
Hopefully this fixes it!
---
 scripts/datadog-cireport/main.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/scripts/datadog-cireport/main.go b/scripts/datadog-cireport/main.go
index 2c0b3a3e67..51fb902b16 100644
--- a/scripts/datadog-cireport/main.go
+++ b/scripts/datadog-cireport/main.go
@@ -59,9 +59,16 @@ func main() {
 	}
 	commitParts := strings.Split(string(commitData), ",")
 
+	// On pull requests, this will be set!
 	branch := os.Getenv("GITHUB_HEAD_REF")
 	if branch == "" {
-		branch = os.Getenv("GITHUB_BASE_REF")
+		githubRef := os.Getenv("GITHUB_REF")
+		for _, prefix := range []string{"refs/heads/", "refs/tags/"} {
+			if !strings.HasPrefix(githubRef, prefix) {
+				continue
+			}
+			branch = strings.TrimPrefix(githubRef, prefix)
+		}
 	}
 
 	tags := map[string]string{