-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathextend.go
66 lines (61 loc) · 2.11 KB
/
extend.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
package fw_openapi
import (
"github.com/sv-tools/openapi/spec"
"strings"
)
func (oa *OpenAPI) NewStringProp(prop *spec.RefOrSpec[spec.Schema], typeString string) (isArray bool, sa spec.SingleOrArray[string]) {
if strings.HasPrefix(typeString, "[]") {
isArray = true
prop.Spec.Items = spec.NewBoolOrSchema(true, nil)
prop.Spec.Items.Schema = spec.NewSchemaSpec()
tt := spec.NewSingleOrArray[string]("string")
prop.Spec.Items.Schema.Spec.Type = &tt
sa = spec.NewSingleOrArray[string]("array")
} else {
sa = spec.NewSingleOrArray[string]("string")
}
return
}
func (oa *OpenAPI) NewIntProp(prop *spec.RefOrSpec[spec.Schema], typeString string) (isArray bool, sa spec.SingleOrArray[string]) {
switch typeString {
case "int":
sa = spec.NewSingleOrArray[string]("integer")
prop.Spec.Format = "int32"
case "int64":
sa = spec.NewSingleOrArray[string]("integer")
prop.Spec.Format = "int64"
case "[]int":
sa = spec.NewSingleOrArray[string]("array")
prop.Spec.Items = spec.NewBoolOrSchema(true, nil)
prop.Spec.Items.Schema = spec.NewSchemaSpec()
tt := spec.NewSingleOrArray[string]("integer")
prop.Spec.Items.Schema.Spec.Type = &tt
prop.Spec.Items.Schema.Spec.Format = "int32"
isArray = true
case "[]int64":
sa = spec.NewSingleOrArray[string]("array")
prop.Spec.Items = spec.NewBoolOrSchema(true, nil)
prop.Spec.Items.Schema = spec.NewSchemaSpec()
tt := spec.NewSingleOrArray[string]("integer")
prop.Spec.Items.Schema.Spec.Type = &tt
prop.Spec.Items.Schema.Spec.Format = "int64"
isArray = true
}
return
}
func (oa *OpenAPI) NewFloatProp(prop *spec.RefOrSpec[spec.Schema], typeString string) (isArray bool, sa spec.SingleOrArray[string]) {
sa = spec.NewSingleOrArray[string]("number")
switch typeString {
case "float32", "float64":
prop.Spec.Format = "float"
case "[]float32", "[]float64":
sa = spec.NewSingleOrArray[string]("array")
prop.Spec.Items = spec.NewBoolOrSchema(true, nil)
prop.Spec.Items.Schema = spec.NewSchemaSpec()
tt := spec.NewSingleOrArray[string]("number")
prop.Spec.Items.Schema.Spec.Type = &tt
prop.Spec.Items.Schema.Spec.Format = "float"
isArray = true
}
return
}