defaults.go

package main

import (
	"github.com/spf13/viper"
	"github.com/teamnsrg/mida/log"
	"os"
)

// initViperConfig
func initViperConfig() {
	// Initialize the hardcoded defaults
	setDefaults()

	// We will read environment variables with the "MIDA" prefix
	viper.SetEnvPrefix("MIDA")
	viper.AutomaticEnv()

	cwd, err := os.Getwd()
	if err != nil {
		log.Log.Fatal(err)
	}

	viper.Set("cwd", cwd)
}

// Hardcoded default configuration values
func setDefaults() {
	// MIDA-Wide Configuration Defaults
	viper.SetDefault("crawlers", 1)
	viper.SetDefault("postprocessers", 1)
	viper.SetDefault("storers", 1)
	viper.SetDefault("prom_port", 8001)
	viper.SetDefault("monitor", false)
	viper.SetDefault("log_level", 2)
	viper.SetDefault("task_file", "examples/example_task.json")
	viper.SetDefault("rate_limit", 200)
	viper.SetDefault("tempdir", ".midatmp")

	viper.SetDefault("amqp_user", "")
	viper.SetDefault("amqp_pass", "")
	viper.SetDefault("amqp_uri", "amqp://localhost:5672")
	viper.SetDefault("amqp_task_queue", "mida-tasks")
}