mirror of
https://github.com/gravitational/teleport.git
synced 2026-11-02 06:08:23 +08:00
Trent Clarke
e29fabd3b1
As we're now using a GitHub app to dispatch workflow requests from Drone to GHA, this patch updates the workflow runner script to handle app authentication.
115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
// Copyright 2022 Gravitational, Inc
|
|
//
|
|
// 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.
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gravitational/trace"
|
|
)
|
|
|
|
// inputMap represents the input values to a workflow. The odd `interface{}`
|
|
// value type (rather than the more obvious `string`) is so that we match the
|
|
// type that the GitHub API expects, and we can plug an `inputMap` straight
|
|
// in without any further conversion.
|
|
type inputMap map[string]interface{}
|
|
|
|
func (m inputMap) String() string {
|
|
text := strings.Builder{}
|
|
for k, v := range m {
|
|
fmt.Fprintf(&text, "%s=%v, ", k, v)
|
|
}
|
|
return text.String()
|
|
}
|
|
|
|
func (m inputMap) Set(s string) error {
|
|
parts := strings.SplitN(s, "=", 2)
|
|
|
|
if len(parts) != 2 {
|
|
return trace.BadParameter("Invalid input. Must be name=value")
|
|
}
|
|
|
|
key := parts[0]
|
|
value := parts[1]
|
|
|
|
m[key] = value
|
|
return nil
|
|
}
|
|
|
|
// key holds the bytes of a GutHub app key, stored as a slice of bytes so that
|
|
// it can be passed into the ghinstance library without further conversion.
|
|
type key []byte
|
|
|
|
func (k *key) String() string {
|
|
return string(*k)
|
|
}
|
|
|
|
func (k *key) Set(s string) error {
|
|
*k = []byte(s)
|
|
return nil
|
|
}
|
|
|
|
// args holds the parsed command-line arguments for the command.
|
|
type args struct {
|
|
appID int64
|
|
appKey key
|
|
owner string
|
|
repo string
|
|
workflow string
|
|
workflowRef string
|
|
useWorkflowTag bool
|
|
timeout time.Duration
|
|
inputs inputMap
|
|
}
|
|
|
|
func parseCommandLine() (args, error) {
|
|
|
|
cliArgs := args{
|
|
workflowRef: "main",
|
|
inputs: make(inputMap),
|
|
}
|
|
|
|
// 274696 is the Github-assigned app id for the default Drone interface app.
|
|
flag.Int64Var(&cliArgs.appID, "app-id", 274696, "ID of the Drone interface GitHub App")
|
|
flag.Var(&cliArgs.appKey, "app-key", "App key in PEM format for the Drone interface GitHub App. Can also be supplied via $GHA_APP_KEY.")
|
|
flag.StringVar(&cliArgs.owner, "owner", "", "Owner of the repo to target")
|
|
flag.StringVar(&cliArgs.repo, "repo", "", "Repo to target")
|
|
flag.StringVar(&cliArgs.workflow, "workflow", "", "Path to workflow")
|
|
flag.StringVar(&cliArgs.workflowRef, "workflow-ref", cliArgs.workflowRef, "Revision reference")
|
|
flag.BoolVar(&cliArgs.useWorkflowTag, "tag-workflow", false, "Use a workflow input to tag and ID workflows spawned by the event")
|
|
flag.DurationVar(&cliArgs.timeout, "timeout", 30*time.Minute, "Timeout")
|
|
flag.Var(cliArgs.inputs, "input", "Input to target workflow")
|
|
|
|
flag.Parse()
|
|
|
|
if cliArgs.appKey == nil {
|
|
keyText := os.Getenv("GHA_APP_KEY")
|
|
if keyText == "" {
|
|
return args{}, trace.BadParameter("No app key supplied")
|
|
}
|
|
cliArgs.appKey = key(keyText)
|
|
}
|
|
|
|
if cliArgs.appID == 0 {
|
|
return args{}, trace.BadParameter("No app ID supplied")
|
|
}
|
|
|
|
return cliArgs, nil
|
|
}
|