-
Notifications
You must be signed in to change notification settings - Fork 0
/
generator.go
52 lines (45 loc) · 1.25 KB
/
generator.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
package service
import (
"fmt"
"github.com/specgen-io/specgen/v2/goven/generator"
"github.com/specgen-io/specgen/v2/goven/kotlin/models"
"github.com/specgen-io/specgen/v2/goven/kotlin/types"
"github.com/specgen-io/specgen/v2/goven/spec"
)
type ServerGenerator interface {
ServiceImports() []string
ServicesControllers(version *spec.Version) []generator.CodeFile
ServiceImplAnnotation(api *spec.Api) (annotationImport, annotation string)
ExceptionController(responses *spec.ErrorResponses) *generator.CodeFile
ErrorsHelpers() *generator.CodeFile
ContentType() []generator.CodeFile
}
type Generator struct {
ServerGenerator
models.Generator
Jsonlib string
Types *types.Types
Packages *Packages
}
func NewGenerator(jsonlib, server string, packages *Packages) *Generator {
types := models.NewTypes(jsonlib)
models := models.NewGenerator(jsonlib, &(packages.Packages))
var serverGenerator ServerGenerator = nil
switch server {
case Spring:
serverGenerator = NewSpringGenerator(types, models, packages)
break
case Micronaut:
serverGenerator = NewMicronautGenerator(types, models, packages)
break
default:
panic(fmt.Sprintf(`Unsupported server: %s`, server))
}
return &Generator{
serverGenerator,
models,
jsonlib,
types,
packages,
}
}