Nikola JokicandGitHub
7442885c99
* Use strong typing and allow atomic change of the max runners * prepare type change in dockerscaleset
104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/actions/scaleset"
|
|
)
|
|
|
|
type Config struct {
|
|
RegistrationURL string
|
|
MaxRunners uint32
|
|
MinRunners uint32
|
|
ScaleSetName string
|
|
RunnerGroup string
|
|
GitHubApp scaleset.GitHubAppAuth
|
|
Token string
|
|
RunnerImage string
|
|
LogLevel string
|
|
LogFormat string
|
|
}
|
|
|
|
func (c *Config) defaults() {
|
|
if c.RunnerGroup == "" {
|
|
c.RunnerGroup = scaleset.DefaultRunnerGroup
|
|
}
|
|
if c.RunnerImage == "" {
|
|
c.RunnerImage = "ghcr.io/actions/actions-runner:latest"
|
|
}
|
|
}
|
|
|
|
func (c *Config) Validate() error {
|
|
c.defaults()
|
|
|
|
if _, err := url.ParseRequestURI(c.RegistrationURL); err != nil {
|
|
return fmt.Errorf("invalid registration URL: %w, it should be the full URL of where you want to register your scale set, e.g. 'https://github.com/org/repo'", err)
|
|
}
|
|
|
|
appError := c.GitHubApp.Validate()
|
|
if c.Token == "" && appError != nil {
|
|
return fmt.Errorf("no credentials provided: either GitHub App (client id, installation id and private key) (recommended) or a Personal Access Token are required")
|
|
}
|
|
|
|
if c.ScaleSetName == "" {
|
|
return fmt.Errorf("scale set name is required")
|
|
}
|
|
if c.MaxRunners < c.MinRunners {
|
|
return fmt.Errorf("max runners cannot be less than min-runners")
|
|
}
|
|
if c.RunnerGroup == "" {
|
|
return fmt.Errorf("runner group is required")
|
|
}
|
|
if c.RunnerImage == "" {
|
|
return fmt.Errorf("runner image is required")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Config) ActionsAuth() *scaleset.ActionsAuth {
|
|
if err := c.GitHubApp.Validate(); err == nil {
|
|
return &scaleset.ActionsAuth{
|
|
App: &c.GitHubApp,
|
|
}
|
|
}
|
|
|
|
return &scaleset.ActionsAuth{
|
|
Token: c.Token,
|
|
}
|
|
}
|
|
|
|
func (c *Config) Logger() *slog.Logger {
|
|
var lvl slog.Level
|
|
switch strings.ToLower(c.LogLevel) {
|
|
case "debug":
|
|
lvl = slog.LevelDebug
|
|
case "info":
|
|
lvl = slog.LevelInfo
|
|
case "warn":
|
|
lvl = slog.LevelWarn
|
|
case "error":
|
|
lvl = slog.LevelError
|
|
default:
|
|
lvl = slog.LevelInfo
|
|
}
|
|
|
|
switch c.LogFormat {
|
|
case "json":
|
|
return slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
|
|
AddSource: true,
|
|
Level: lvl,
|
|
}))
|
|
case "text":
|
|
return slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
|
|
AddSource: true,
|
|
Level: lvl,
|
|
}))
|
|
default:
|
|
return slog.New(slog.DiscardHandler)
|
|
}
|
|
}
|