gluetun/internal/publicip/loop.go

package publicip

import (
	"net/http"
	"time"

	"github.com/qdm12/gluetun/internal/configuration"
	"github.com/qdm12/gluetun/internal/constants"
	"github.com/qdm12/gluetun/internal/loopstate"
	"github.com/qdm12/gluetun/internal/models"
	"github.com/qdm12/gluetun/internal/publicip/state"
)

var _ Looper = (*Loop)(nil)

type Looper interface {
	Runner
	RestartTickerRunner
	loopstate.Getter
	loopstate.Applier
	SettingsGetSetter
	GetSetter
}

type Loop struct {
	statusManager loopstate.Manager
	state         state.Manager
	// Objects
	fetcher Fetcher
	client  *http.Client
	logger  Logger
	// Fixed settings
	puid int
	pgid int
	// Internal channels and locks
	start        chan struct{}
	running      chan models.LoopStatus
	stop         chan struct{}
	stopped      chan struct{}
	updateTicker chan struct{}
	backoffTime  time.Duration
	userTrigger  bool
	// Mock functions
	timeNow func() time.Time
}

const defaultBackoffTime = 5 * time.Second

func NewLoop(client *http.Client, logger Logger,
	settings configuration.PublicIP, puid, pgid int) *Loop {
	start := make(chan struct{})
	running := make(chan models.LoopStatus)
	stop := make(chan struct{})
	stopped := make(chan struct{})
	updateTicker := make(chan struct{})

	statusManager := loopstate.New(constants.Stopped, start, running, stop, stopped)
	state := state.New(statusManager, settings, updateTicker)

	return &Loop{
		statusManager: statusManager,
		state:         state,
		// Objects
		client:       client,
		fetcher:      NewFetch(client),
		logger:       logger,
		puid:         puid,
		pgid:         pgid,
		start:        start,
		running:      running,
		stop:         stop,
		stopped:      stopped,
		updateTicker: updateTicker,
		userTrigger:  true,
		backoffTime:  defaultBackoffTime,
		timeNow:      time.Now,
	}
}