package options

import (
	"strings"

	"github.com/cortezaproject/corteza/server/pkg/options"
)

type (
	EsOpt struct {
		Addresses            []string `env:"ES_ADDRESS"`
		Username             string   `env:"ES_USERNAME"`
		Password             string   `env:"ES_PASSWORD"`
		CertFile             string   `env:"ES_CERT_FILE"`
		Secure               bool     `env:"ES_SECURE"`
		EnableRetryOnTimeout bool     `env:"ES_ENABLE_RETRY_ON_TIMEOUT"`
		MaxRetries           int      `env:"ES_MAX_RETRIES"`
		IndexInterval        int      `env:"ES_INDEX_INTERVAL"`
		IndexBackFillMonths  int      `env:"ES_INDEX_BACKFILL_MONTHS"`
	}
)

func ES() (o *EsOpt, err error) {
	o = &EsOpt{}
	return o, func() error {
		o.Username = options.EnvString("ES_USERNAME", "")
		o.Password = options.EnvString("ES_PASSWORD", "")
		o.Secure = options.EnvBool("ES_SECURE", true)
		o.CertFile = options.EnvString("ES_CERT_FILE", "")

		o.EnableRetryOnTimeout = options.EnvBool("ES_ENABLE_RETRY_ON_TIMEOUT", true)
		o.MaxRetries = options.EnvInt("ES_MAX_RETRIES", 5)
		o.IndexInterval = options.EnvInt("ES_INDEX_INTERVAL", 30)
		o.IndexBackFillMonths = options.EnvInt("ES_INDEX_BACKFILL_MONTHS", 0)

		for _, a := range strings.Split(options.EnvString("ES_ADDRESS", "http://es:9200"), " ") {
			if a = strings.TrimSpace(a); a != "" {
				o.Addresses = append(o.Addresses, a)
			}
		}
		return nil
	}()
}