/
spec.go
68 lines (57 loc) · 1.14 KB
/
spec.go
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package openapi
import (
"os"
"gopkg.in/yaml.v2"
)
type Spec struct {
Paths map[string]Path `yaml:"paths"`
}
type Path struct {
Get *Operation `yaml:"get"`
Post *Operation `yaml:"post"`
Put *Operation `yaml:"put"`
Patch *Operation `yaml:"patch"`
Delete *Operation `yaml:"delete"`
Head *Operation `yaml:"head"`
Options *Operation `yaml:"options"`
}
func (p Path) Operations() map[string]Operation {
rtn := make(map[string]Operation)
if p.Get != nil {
rtn["GET"] = *p.Get
}
if p.Post != nil {
rtn["POST"] = *p.Post
}
if p.Put != nil {
rtn["PUT"] = *p.Put
}
if p.Patch != nil {
rtn["PATCH"] = *p.Patch
}
if p.Delete != nil {
rtn["DELETE"] = *p.Delete
}
if p.Head != nil {
rtn["HEAD"] = *p.Head
}
if p.Options != nil {
rtn["OPTIONS"] = *p.Options
}
return rtn
}
type Operation struct {
OperationId string `yaml:"operationId"`
Tags []string `yaml:"tags"`
}
func LoadSpec(filename string) (*Spec, error) {
bytes, err := os.ReadFile(filename)
if err != nil {
return nil, err
}
var spec Spec
if err := yaml.Unmarshal(bytes, &spec); err != nil {
return nil, err
}
return &spec, nil
}