package main

import (
        "time"
        "github.com/pires/configo"
)

type configuration struct {
	Pubsub    []string
	Keepalive bool
	Heartbeat *configo.Duration
	Timeout   *configo.Duration
}

var (
	done = make (chan struct{})
)

func main() {
	var config configuration
	if err := configo.LoadEnvironment("stage", &config); err != nil {
		panic(err)
	}

	for _, pubsub := range config.Pubsub {
		println("Connecting to", pubsub)
	}

	if config.Keepalive {
		println("Enabling keep-alive...")
		heartbeat := time.NewTicker(config.Heartbeat.Duration)
		defer heartbeat.Stop()

		// run timeout
		timeout := time.NewTimer(config.Timeout.Duration)
		go func(){
			<-timeout.C
			close(done)
		}()

		// run heartbeat
		for {
			select {
			case <-done:
				println("Done")
				return
			case <- heartbeat.C:
				println("Received heartbeat")
			}
		}
	} else {
		println("Keep-alive disabled.")
	}
}