blob: 1ea7750adbecf578eabfcb65eabbe7dd38a363de (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
package main
import (
_ "embed"
"net/url"
"os"
"path"
"sigs.k8s.io/yaml"
)
type Upstream struct {
URLs []string `json:"urls"`
Name string `json:"name"`
Desc string `json:"desc"`
}
func ReadUpstreams(filename string) ([]Upstream, error) {
bs, err := os.ReadFile(filename)
if err != nil {
return nil, err
}
var ret []Upstream
if err := yaml.UnmarshalStrict(bs, &ret); err != nil {
return []Upstream{}, err
}
for i := range ret {
upstream := ret[i]
if err := upstream.Fill(); err != nil {
return nil, err
}
ret[i] = upstream
}
return ret, nil
}
func (upstream *Upstream) Fill() error {
if upstream.Name == "" {
u, err := url.Parse(upstream.URLs[0])
if err != nil {
return err
}
_, upstream.Name = path.Split(path.Clean(u.Path))
}
return nil
}
|
