mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 12:42:34 +02:00
76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package provider
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/yusing/go-proxy/internal/common"
|
|
"github.com/yusing/go-proxy/internal/gperr"
|
|
"github.com/yusing/go-proxy/internal/logging"
|
|
"github.com/yusing/go-proxy/internal/route"
|
|
"github.com/yusing/go-proxy/internal/utils"
|
|
W "github.com/yusing/go-proxy/internal/watcher"
|
|
)
|
|
|
|
type FileProvider struct {
|
|
fileName string
|
|
path string
|
|
l zerolog.Logger
|
|
}
|
|
|
|
func FileProviderImpl(filename string) (ProviderImpl, error) {
|
|
impl := &FileProvider{
|
|
fileName: filename,
|
|
path: path.Join(common.ConfigBasePath, filename),
|
|
l: logging.With().Str("type", "file").Str("name", filename).Logger(),
|
|
}
|
|
_, err := os.Stat(impl.path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return impl, nil
|
|
}
|
|
|
|
func validate(data []byte) (routes route.Routes, err gperr.Error) {
|
|
err = utils.UnmarshalValidateYAML(data, &routes)
|
|
return
|
|
}
|
|
|
|
func Validate(data []byte) (err gperr.Error) {
|
|
_, err = validate(data)
|
|
return
|
|
}
|
|
|
|
func (p *FileProvider) String() string {
|
|
return p.fileName
|
|
}
|
|
|
|
func (p *FileProvider) ShortName() string {
|
|
return strings.Split(p.fileName, ".")[0]
|
|
}
|
|
|
|
func (p *FileProvider) IsExplicitOnly() bool {
|
|
return false
|
|
}
|
|
|
|
func (p *FileProvider) Logger() *zerolog.Logger {
|
|
return &p.l
|
|
}
|
|
|
|
func (p *FileProvider) loadRoutesImpl() (route.Routes, gperr.Error) {
|
|
data, err := os.ReadFile(p.path)
|
|
if err != nil {
|
|
return nil, gperr.Wrap(err)
|
|
}
|
|
routes, err := validate(data)
|
|
if err != nil && len(routes) == 0 {
|
|
return nil, gperr.Wrap(err)
|
|
}
|
|
return routes, gperr.Wrap(err)
|
|
}
|
|
|
|
func (p *FileProvider) NewWatcher() W.Watcher {
|
|
return W.NewConfigFileWatcher(p.fileName)
|
|
}
|