k8s.io/test-infra@v0.0.0-20240520184403-27c6b4c223d8/experiment/manual-trigger/manual-trigger.go

/*
Copyright 2017 The Kubernetes Authors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

// manual-trigger triggers jenkins jobs based a specified github pull request
package main

import (
	"flag"
	"fmt"
	"log"
	"net/url"
	"strings"

	"github.com/sirupsen/logrus"

	prowapi "sigs.k8s.io/prow/pkg/apis/prowjobs/v1"
	"sigs.k8s.io/prow/pkg/config/secret"
	"sigs.k8s.io/prow/pkg/github"
	"sigs.k8s.io/prow/pkg/jenkins"
	"sigs.k8s.io/prow/pkg/pod-utils/downwardapi"
)

type options struct {
	githubEndpoint         string
	graphqlEndpoint        string
	githubTokenFile        string
	jenkinsBearerTokenFile string
	jenkinsURL             string
	jenkinsTokenFile       string
	jenkinsUserName        string
	jobName                string
	num                    int
	org                    string
	repo                   string
}

func flagOptions() options {
	o := options{}

	flag.StringVar(&o.jenkinsBearerTokenFile, "jenkins-bearer-token-file", "", "Path to the file containing the Jenkins API bearer token.")
	flag.StringVar(&o.jenkinsURL, "jenkins-url", "", "Jenkins URL.")
	flag.StringVar(&o.jenkinsTokenFile, "jenkins-token-file", "", "Path to the file containing the Jenkins API token.")
	flag.StringVar(&o.jenkinsUserName, "jenkins-user-name", "", "Jenkins username.")

	flag.StringVar(&o.githubEndpoint, "github-endpoint", github.DefaultAPIEndpoint, "GitHub's API endpoint.")
	flag.StringVar(&o.graphqlEndpoint, "graphql-endpoint", github.DefaultGraphQLEndpoint, "GitHub's GraphQL API endpoint.")
	flag.StringVar(&o.githubTokenFile, "github-token-file", "", "Path to file containing GitHub OAuth token.")

	flag.StringVar(&o.jobName, "job-name", "", "Name of Jenkins job")

	flag.IntVar(&o.num, "num", 0, "GitHub issue number")
	flag.StringVar(&o.org, "org", "", "GitHub organization")
	flag.StringVar(&o.repo, "repo", "", "GitHub repository")
	flag.Parse()
	return o
}

func sanityCheckFlags(o options) error {
	if o.num <= 0 {
		return fmt.Errorf("empty or invalid --num")
	}
	if o.org == "" {
		return fmt.Errorf("empty --org")
	}
	if o.repo == "" {
		return fmt.Errorf("empty --repo")
	}
	if o.githubTokenFile == "" {
		return fmt.Errorf("empty --github-token-file")
	}
	if o.jobName == "" {
		return fmt.Errorf("empty --job-name")
	}

	if o.jenkinsBearerTokenFile == "" && (o.jenkinsUserName == "" || o.jenkinsTokenFile == "") {
		return fmt.Errorf("neither --jenkins-bearer-token-file nor the combination of --jenkins-user-name and --jenkins-token-file were provided")
	}

	if o.githubEndpoint == "" {
		return fmt.Errorf("empty --github-endpoint")
	} else if _, err := url.Parse(o.githubEndpoint); err != nil {
		return fmt.Errorf("bad --github-endpoint provided: %w", err)
	}

	if o.graphqlEndpoint == "" {
		return fmt.Errorf("empty --graphql-endpoint")
	} else if _, err := url.Parse(o.graphqlEndpoint); err != nil {
		return fmt.Errorf("bad --graphql-endpoint provided: %w", err)
	}

	if o.jenkinsURL == "" {
		return fmt.Errorf("empty --jenkins-url")
	} else if _, err := url.Parse(o.jenkinsURL); err != nil {
		return fmt.Errorf("bad --jenkins-url provided: %w", err)
	}

	return nil
}

func main() {
	o := flagOptions()
	err := sanityCheckFlags(o)
	if err != nil {
		log.Fatal(err)
	}

	var tokens []string
	tokens = append(tokens, o.githubTokenFile)

	if o.jenkinsTokenFile != "" {
		tokens = append(tokens, o.jenkinsTokenFile)
	}

	if o.jenkinsBearerTokenFile != "" {
		tokens = append(tokens, o.jenkinsBearerTokenFile)
	}

	if err := secret.Add(tokens...); err != nil {
		logrus.WithError(err).Fatal("Error starting secrets agent.")
	}

	// TODO: dry this out
	ac := jenkins.AuthConfig{}
	if o.jenkinsTokenFile != "" {
		ac.Basic = &jenkins.BasicAuthConfig{
			User:     o.jenkinsUserName,
			GetToken: secret.GetTokenGenerator(o.jenkinsTokenFile),
		}
	} else if o.jenkinsBearerTokenFile != "" {
		ac.BearerToken = &jenkins.BearerTokenAuthConfig{
			GetToken: secret.GetTokenGenerator(o.jenkinsBearerTokenFile),
		}
	} else {
		log.Fatalf("no jenkins auth token provided")
	}

	jc, err := jenkins.NewClient(o.jenkinsURL, false, nil, &ac, nil, nil)
	if err != nil {
		log.Fatalf("cannot setup Jenkins client: %v", err)
	}

	gc, err := github.NewClient(secret.GetTokenGenerator(o.githubTokenFile), secret.Censor, o.graphqlEndpoint, o.githubEndpoint)
	if err != nil {
		log.Fatalf("failed to construct GitHub client: %v", err)
	}

	pr, err := gc.GetPullRequest(o.org, o.repo, o.num)
	if err != nil {
		log.Fatalf("Unable to get information on pull request %s/%s#%d: %v", o.org, o.repo, o.num, err)
	}

	spec := prowapi.ProwJobSpec{
		Type: prowapi.PresubmitJob,
		Job:  o.jobName,
		Refs: &prowapi.Refs{
			Org:     o.org,
			Repo:    o.repo,
			BaseRef: pr.Base.Ref,
			BaseSHA: pr.Base.SHA,
			Pulls: []prowapi.Pull{
				{
					Number:  pr.Number,
					Author:  pr.User.Login,
					SHA:     pr.Head.SHA,
					HeadRef: pr.Head.Ref,
				},
			},
		},

		Report:         false,
		Context:        "",
		RerunCommand:   "",
		MaxConcurrency: 1,
	}

	if err = jc.BuildFromSpec(&spec, "0", o.jobName); err != nil {
		log.Println("Submitting the following to Jenkins:")
		env, _ := downwardapi.EnvForSpec(downwardapi.NewJobSpec(spec, "0", o.jobName))
		for k, v := range env {
			log.Printf("  %s=%s\n", k, v)
		}
		log.Fatalf("for %s/%s#%d resulted in an error: %v", o.org, o.repo, o.num, err)
	} else {
		slash := "/"
		if strings.HasSuffix(o.jenkinsURL, "/") {
			slash = ""
		}
		log.Printf("Successfully submitted job to %s%sjob/%s", o.jenkinsURL, slash, o.jobName)
	}
}