6716bf018c
Cleans up a whole bunch of things I wanted to get out of the door right away: * depot internal references to //third_party/nixery have been replaced with //tools/nixery * cleaned up files from Github * fixed SPDX & Copyright headers * code formatting and inclusion in //tools/depotfmt checks Change-Id: Iea79f0fdf3aa04f71741d4f4032f88605ae415bb Reviewed-on: https://cl.tvl.fyi/c/depot/+/5486 Tested-by: BuildkiteCI Reviewed-by: tazjin <tazjin@tvl.su> Autosubmit: tazjin <tazjin@tvl.su>
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
// Copyright 2022 The TVL Contributors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package config implements structures to store Nixery's configuration at
|
|
// runtime as well as the logic for instantiating this configuration from the
|
|
// environment.
|
|
package config
|
|
|
|
import (
|
|
"os"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func getConfig(key, desc, def string) string {
|
|
value := os.Getenv(key)
|
|
if value == "" && def == "" {
|
|
log.WithFields(log.Fields{
|
|
"option": key,
|
|
"description": desc,
|
|
}).Fatal("missing required configuration envvar")
|
|
} else if value == "" {
|
|
return def
|
|
}
|
|
|
|
return value
|
|
}
|
|
|
|
// Backend represents the possible storage backend types
|
|
type Backend int
|
|
|
|
const (
|
|
GCS = iota
|
|
FileSystem
|
|
)
|
|
|
|
// Config holds the Nixery configuration options.
|
|
type Config struct {
|
|
Port string // Port on which to launch HTTP server
|
|
Pkgs PkgSource // Source for Nix package set
|
|
Timeout string // Timeout for a single Nix builder (seconds)
|
|
WebDir string // Directory with static web assets
|
|
PopUrl string // URL to the Nix package popularity count
|
|
Backend Backend // Storage backend to use for Nixery
|
|
}
|
|
|
|
func FromEnv() (Config, error) {
|
|
pkgs, err := pkgSourceFromEnv()
|
|
if err != nil {
|
|
return Config{}, err
|
|
}
|
|
|
|
var b Backend
|
|
switch os.Getenv("NIXERY_STORAGE_BACKEND") {
|
|
case "gcs":
|
|
b = GCS
|
|
case "filesystem":
|
|
b = FileSystem
|
|
default:
|
|
log.WithField("values", []string{
|
|
"gcs",
|
|
}).Fatal("NIXERY_STORAGE_BACKEND must be set to a supported value (gcs or filesystem)")
|
|
}
|
|
|
|
return Config{
|
|
Port: getConfig("PORT", "HTTP port", ""),
|
|
Pkgs: pkgs,
|
|
Timeout: getConfig("NIX_TIMEOUT", "Nix builder timeout", "60"),
|
|
WebDir: getConfig("WEB_DIR", "Static web file dir", ""),
|
|
PopUrl: os.Getenv("NIX_POPULARITY_URL"),
|
|
Backend: b,
|
|
}, nil
|
|
}
|