mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-19 20:32:35 +02:00
71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package provider
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/yusing/go-proxy/internal/common"
|
|
E "github.com/yusing/go-proxy/internal/error"
|
|
"github.com/yusing/go-proxy/internal/proxy/entry"
|
|
R "github.com/yusing/go-proxy/internal/route"
|
|
U "github.com/yusing/go-proxy/internal/utils"
|
|
W "github.com/yusing/go-proxy/internal/watcher"
|
|
)
|
|
|
|
type FileProvider struct {
|
|
fileName string
|
|
path string
|
|
}
|
|
|
|
func FileProviderImpl(filename string) (ProviderImpl, E.Error) {
|
|
impl := &FileProvider{
|
|
fileName: filename,
|
|
path: path.Join(common.ConfigBasePath, filename),
|
|
}
|
|
_, err := os.Stat(impl.path)
|
|
switch {
|
|
case err == nil:
|
|
return impl, nil
|
|
case errors.Is(err, os.ErrNotExist):
|
|
return nil, E.NotExist("file", impl.path)
|
|
default:
|
|
return nil, E.UnexpectedError(err)
|
|
}
|
|
}
|
|
|
|
func Validate(data []byte) E.Error {
|
|
return U.ValidateYaml(U.GetSchema(common.FileProviderSchemaPath), data)
|
|
}
|
|
|
|
func (p FileProvider) String() string {
|
|
return p.fileName
|
|
}
|
|
|
|
func (p *FileProvider) LoadRoutesImpl() (routes R.Routes, res E.Error) {
|
|
routes = R.NewRoutes()
|
|
|
|
b := E.NewBuilder("validation failure")
|
|
defer b.To(&res)
|
|
|
|
entries := entry.NewProxyEntries()
|
|
|
|
data, err := E.Check(os.ReadFile(p.path))
|
|
if err != nil {
|
|
b.Add(E.FailWith("read file", err))
|
|
return
|
|
}
|
|
|
|
if err = entries.UnmarshalFromYAML(data); err != nil {
|
|
b.Add(err)
|
|
return
|
|
}
|
|
|
|
b.Add(Validate(data), true)
|
|
|
|
return R.FromEntries(entries)
|
|
}
|
|
|
|
func (p *FileProvider) NewWatcher() W.Watcher {
|
|
return W.NewConfigFileWatcher(p.fileName)
|
|
}
|