forked from tangled.org/core
Monorepo for Tangled — https://tangled.org
at master 2.2 kB view raw
1package models 2 3import ( 4 "strings" 5 6 "github.com/go-git/go-git/v5/plumbing" 7 "tangled.org/core/api/tangled" 8 "tangled.org/core/workflow" 9) 10 11// PipelineEnvVars extracts environment variables from pipeline trigger metadata. 12// These are framework-provided variables that are injected into workflow steps. 13func PipelineEnvVars(tr *tangled.Pipeline_TriggerMetadata, pipelineId PipelineId, devMode bool) map[string]string { 14 if tr == nil { 15 return nil 16 } 17 18 env := make(map[string]string) 19 20 // Standard CI environment variable 21 env["CI"] = "true" 22 23 env["TANGLED_PIPELINE_ID"] = pipelineId.Rkey 24 25 // Repo info 26 if tr.Repo != nil { 27 env["TANGLED_REPO_KNOT"] = tr.Repo.Knot 28 env["TANGLED_REPO_DID"] = tr.Repo.Did 29 env["TANGLED_REPO_NAME"] = tr.Repo.Repo 30 env["TANGLED_REPO_DEFAULT_BRANCH"] = tr.Repo.DefaultBranch 31 env["TANGLED_REPO_URL"] = BuildRepoURL(tr.Repo, devMode) 32 } 33 34 switch workflow.TriggerKind(tr.Kind) { 35 case workflow.TriggerKindPush: 36 if tr.Push != nil { 37 refName := plumbing.ReferenceName(tr.Push.Ref) 38 refType := "branch" 39 if refName.IsTag() { 40 refType = "tag" 41 } 42 43 env["TANGLED_REF"] = tr.Push.Ref 44 env["TANGLED_REF_NAME"] = refName.Short() 45 env["TANGLED_REF_TYPE"] = refType 46 env["TANGLED_SHA"] = tr.Push.NewSha 47 env["TANGLED_COMMIT_SHA"] = tr.Push.NewSha 48 } 49 50 case workflow.TriggerKindPullRequest: 51 if tr.PullRequest != nil { 52 // For PRs, the "ref" is the source branch 53 env["TANGLED_REF"] = "refs/heads/" + tr.PullRequest.SourceBranch 54 env["TANGLED_REF_NAME"] = tr.PullRequest.SourceBranch 55 env["TANGLED_REF_TYPE"] = "branch" 56 env["TANGLED_SHA"] = tr.PullRequest.SourceSha 57 env["TANGLED_COMMIT_SHA"] = tr.PullRequest.SourceSha 58 59 // PR-specific variables 60 env["TANGLED_PR_SOURCE_BRANCH"] = tr.PullRequest.SourceBranch 61 env["TANGLED_PR_TARGET_BRANCH"] = tr.PullRequest.TargetBranch 62 env["TANGLED_PR_SOURCE_SHA"] = tr.PullRequest.SourceSha 63 env["TANGLED_PR_ACTION"] = tr.PullRequest.Action 64 } 65 66 case workflow.TriggerKindManual: 67 // Manual triggers may not have ref/sha info 68 // Include any manual inputs if present 69 if tr.Manual != nil { 70 for _, pair := range tr.Manual.Inputs { 71 env["TANGLED_INPUT_"+strings.ToUpper(pair.Key)] = pair.Value 72 } 73 } 74 } 75 76 return env 77}