go-lemmy/cmd/gen/generator/routes.go

70 lines
1.7 KiB
Go
Raw Normal View History

package generator
import (
"io"
"strings"
"github.com/dave/jennifer/jen"
"go.arsenm.dev/go-lemmy/cmd/gen/parser"
)
type RoutesGenerator struct {
w io.Writer
PkgName string
}
func NewRoutes(w io.Writer, pkgName string) *RoutesGenerator {
return &RoutesGenerator{w, pkgName}
}
func (r *RoutesGenerator) Generate(routes []parser.Route, impls map[string]string) error {
f := jen.NewFile(r.PkgName)
2023-01-07 22:40:18 +00:00
f.HeaderComment("Code generated by go.arsenm.dev/go-lemmy/cmd/gen (routes generator). DO NOT EDIT.")
for _, r := range routes {
resStruct := impls[r.Struct]
f.Func().Params(
jen.Id("c").Id("*Client"),
2023-01-31 03:38:57 +00:00
).Id(transformName(r.Struct)).Params(
jen.Id("ctx").Qual("context", "Context"),
jen.Id("data").Qual("go.arsenm.dev/go-lemmy/types", r.Struct),
).Params(
jen.Op("*").Qual("go.arsenm.dev/go-lemmy/types", resStruct),
jen.Error(),
).BlockFunc(func(g *jen.Group) {
g.Id("resData").Op(":=").Op("&").Qual("go.arsenm.dev/go-lemmy/types", resStruct).Block()
var funcName string
switch r.Method {
case "GET":
funcName = "getReq"
default:
funcName = "req"
}
g.List(jen.Id("res"), jen.Err()).Op(":=").Id("c").Dot(funcName).Params(
jen.Id("ctx"), jen.Lit(r.Method), jen.Lit(r.Path), jen.Id("data"), jen.Op("&").Id("resData"),
)
g.If(jen.Err().Op("!=").Nil()).Block(
jen.Return(jen.Nil(), jen.Err()),
)
g.Err().Op("=").Id("resError").Params(jen.Id("res"), jen.Id("resData").Dot("LemmyResponse"))
g.If(jen.Err().Op("!=").Nil()).Block(
jen.Return(jen.Nil(), jen.Err()),
)
g.Return(jen.Id("resData"), jen.Nil())
})
}
return f.Render(r.w)
}
2023-01-31 03:38:57 +00:00
func transformName(s string) string {
s = strings.TrimPrefix(s, "Get")
s = strings.TrimPrefix(s, "List")
return s
}