package config import ( "encoding/json" "fmt" "io" "os" "runtime" "gitea.com/gitea/act_runner/core" "github.com/appleboy/com/file" "github.com/joho/godotenv" "github.com/kelseyhightower/envconfig" ) type ( // Config provides the system configuration. Config struct { Debug bool `envconfig:"GITEA_DEBUG"` Trace bool `envconfig:"GITEA_TRACE"` Client Client Runner Runner Platform Platform } Client struct { Address string `ignored:"true"` Proto string `envconfig:"GITEA_RPC_PROTO" default:"http"` Host string `envconfig:"GITEA_RPC_HOST"` Secret string `envconfig:"GITEA_RPC_SECRET"` SkipVerify bool `envconfig:"GITEA_RPC_SKIP_VERIFY"` GRPC bool `envconfig:"GITEA_RPC_GRPC" default:"true"` GRPCWeb bool `envconfig:"GITEA_RPC_GRPC_WEB"` } Runner struct { Name string `envconfig:"GITEA_RUNNER_NAME"` Token string `envconfig:"GITEA_RUNNER_TOKEN" required:"true"` Capacity int `envconfig:"GITEA_RUNNER_CAPACITY" default:"1"` File string `envconfig:"GITEA_RUNNER_FILE" default:".runner"` Environ map[string]string `envconfig:"GITEA_RUNNER_ENVIRON"` EnvFile string `envconfig:"GITEA_RUNNER_ENV_FILE"` Labels []string `envconfig:"GITEA_RUNNER_LABELS"` } Platform struct { OS string `envconfig:"GITEA_PLATFORM_OS"` Arch string `envconfig:"GITEA_PLATFORM_ARCH"` } ) // FromEnviron returns the settings from the environment. func FromEnviron() (Config, error) { cfg := Config{} if err := envconfig.Process("", &cfg); err != nil { return cfg, err } // check runner config exist if file.IsFile(cfg.Runner.File) { jsonFile, _ := os.Open(cfg.Runner.File) defer jsonFile.Close() byteValue, _ := io.ReadAll(jsonFile) var runner core.Runner if err := json.Unmarshal(byteValue, &runner); err != nil { return cfg, err } if runner.UUID != "" { cfg.Client.Secret = runner.UUID } } // runner config if cfg.Runner.Environ == nil { cfg.Runner.Environ = map[string]string{} } if cfg.Runner.Name == "" { cfg.Runner.Name, _ = os.Hostname() } // platform config if cfg.Platform.OS == "" { cfg.Platform.OS = runtime.GOOS } if cfg.Platform.Arch == "" { cfg.Platform.Arch = runtime.GOARCH } cfg.Client.Address = fmt.Sprintf( "%s://%s", cfg.Client.Proto, cfg.Client.Host, ) if file := cfg.Runner.EnvFile; file != "" { envs, err := godotenv.Read(file) if err != nil { return cfg, err } for k, v := range envs { cfg.Runner.Environ[k] = v } } return cfg, nil }