package agent

import (
	"fmt"
	"strings"
	"time"

	discover "github.com/hashicorp/go-discover"

	// support retry-join only for the following providers
	// to add more providers import additional packages or 'all'
	// to support all providers of go-discover
	_ "github.com/hashicorp/go-discover/provider/aws"
	_ "github.com/hashicorp/go-discover/provider/azure"
	_ "github.com/hashicorp/go-discover/provider/gce"
)

// RetryJoin is used to handle retrying a join until it succeeds or all
// retries are exhausted.
func (a *Agent) retryJoin() {
	cfg := a.config
	if len(cfg.RetryJoin) == 0 {
		return
	}

	a.logger.Printf("[INFO] agent: Supporting retry join for %v", discover.ProviderNames())
	a.logger.Printf("[INFO] agent: Joining cluster...")
	attempt := 0
	for {
		var addrs []string
		var err error

		for _, addr := range cfg.RetryJoin {
			switch {
			case strings.Contains(addr, "provider="):
				servers, err := discover.Addrs(addr, a.logger)
				if err != nil {
					a.logger.Printf("[ERR] agent: %s", err)
				} else {
					addrs = append(addrs, servers...)
					a.logger.Printf("[INFO] agent: Discovered servers: %s", strings.Join(servers, " "))
				}

			default:
				addrs = append(addrs, addr)
			}
		}

		if len(addrs) > 0 {
			n, err := a.JoinLAN(addrs)
			if err == nil {
				a.logger.Printf("[INFO] agent: Join completed. Synced with %d initial agents", n)
				return
			}
		}

		if len(addrs) == 0 {
			err = fmt.Errorf("No servers to join")
		}

		attempt++
		if cfg.RetryMaxAttempts > 0 && attempt > cfg.RetryMaxAttempts {
			a.retryJoinCh <- fmt.Errorf("agent: max join retry exhausted, exiting")
			return
		}

		a.logger.Printf("[WARN] agent: Join failed: %v, retrying in %v", err, cfg.RetryInterval)
		time.Sleep(cfg.RetryInterval)
	}
}

// RetryJoinWan is used to handle retrying a join -wan until it succeeds or all
// retries are exhausted.
func (a *Agent) retryJoinWan() {
	cfg := a.config

	if len(cfg.RetryJoinWan) == 0 {
		return
	}

	a.logger.Printf("[INFO] agent: Joining WAN cluster...")

	attempt := 0
	for {
		n, err := a.JoinWAN(cfg.RetryJoinWan)
		if err == nil {
			a.logger.Printf("[INFO] agent: Join -wan completed. Synced with %d initial agents", n)
			return
		}

		attempt++
		if cfg.RetryMaxAttemptsWan > 0 && attempt > cfg.RetryMaxAttemptsWan {
			a.retryJoinCh <- fmt.Errorf("agent: max join -wan retry exhausted, exiting")
			return
		}

		a.logger.Printf("[WARN] agent: Join -wan failed: %v, retrying in %v", err, cfg.RetryIntervalWan)
		time.Sleep(cfg.RetryIntervalWan)
	}
}