44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
// Package storage defines interfaces to interact with the files persisted such as the list of servers.
|
|
package storage
|
|
|
|
import (
|
|
"github.com/qdm12/gluetun/internal/models"
|
|
)
|
|
|
|
//go:generate mockgen -destination=infoerrorer_mock_test.go -package $GOPACKAGE . InfoErrorer
|
|
|
|
type Storage struct {
|
|
mergedServers models.AllServers
|
|
// this is stored in memory to avoid re-parsing
|
|
// the embedded JSON file on every call to the
|
|
// SyncServers method.
|
|
hardcodedServers models.AllServers
|
|
logger Infoer
|
|
filepath string
|
|
}
|
|
|
|
type Infoer interface {
|
|
Info(s string)
|
|
}
|
|
|
|
// New creates a new storage and reads the servers from the
|
|
// embedded servers file and the file on disk.
|
|
// Passing an empty filepath disables writing servers to a file.
|
|
func New(logger Infoer, filepath string) (storage *Storage, err error) {
|
|
// A unit test prevents any error from being returned
|
|
// and ensures all providers are part of the servers returned.
|
|
hardcodedServers, _ := parseHardcodedServers()
|
|
|
|
storage = &Storage{
|
|
hardcodedServers: hardcodedServers,
|
|
logger: logger,
|
|
filepath: filepath,
|
|
}
|
|
|
|
if err := storage.SyncServers(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return storage, nil
|
|
}
|