config.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"os"
	"time"
)

type PoolConfig struct {
	Image    string `json:"image"`
	Capacity int    `json:"capacity"`
	Standby  int    `json:"standby"`
}

type Config struct {
	Listen              string        `json:"listen"`
	DockerHost          string        `json:"docker_host"`
	LanguagesPath       string        `json:"languages_path"`
	SharedPath          string        `json:"shared_path"`
	RunDuration         time.Duration `json:"run_duration"`
	ThrottleQuota       int           `json:"throttle_quota"`
	ThrottleConcurrency int           `json:"throttle_concurrency"`
	ThrottleWhitelist   []string      `json:"throttle_whitelist"`
	NetworkDisabled     bool          `json:"network_disabled"`
	MemoryLimit         int64         `json:"memory_limit"`
	Pools               []PoolConfig  `json:"pools"`
	ApiToken            string        `json:"api_token"`
	FetchImages         bool          `json:"fetch_images"`
	Namespaces          bool          `json:"namespaces"`
}

func NewConfig() *Config {
	cfg := Config{
		DockerHost: os.Getenv("DOCKER_HOST"),
		SharedPath: os.Getenv("SHARED_PATH"),
	}

	cfg.Listen = "127.0.0.1:5000"
	cfg.SharedPath = expandPath(cfg.SharedPath)
	cfg.RunDuration = time.Second * 10
	cfg.ThrottleQuota = 5
	cfg.ThrottleConcurrency = 1
	cfg.ThrottleWhitelist = []string{}
	cfg.NetworkDisabled = false
	cfg.MemoryLimit = 67108864
	cfg.Pools = []PoolConfig{}
	cfg.FetchImages = false
	cfg.Namespaces = false
	cfg.LanguagesPath = "./languages.json"

	return &cfg
}

func NewConfigFromFile(path string) (*Config, error) {
	data, err := ioutil.ReadFile(path)
	if err != nil {
		return nil, err
	}

	config := Config{}

	err = json.Unmarshal(data, &config)

	if err == nil {
		config.SharedPath = expandPath(config.SharedPath)
		config.RunDuration = config.RunDuration * time.Second

		if config.Listen == "" {
			config.Listen = "127.0.0.1:5000"
		}
	}

	return &config, err
}