camzalewski/personal-site
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Support optional fields being empty.

Browse source
This commit is contained in:
mikestefanello 2025-04-12 14:56:38 -04:00
parent 356cf21b3e
commit 56f8a619a5
5 changed files with 46 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

17
ent/admin/extension.go
View file

@ -8,6 +8,7 @@ import (
"entgo.io/ent/entc"
"entgo.io/ent/entc/gen"
"entgo.io/ent/schema/field"
)
var (
@ -24,8 +25,9 @@ func (*Extension) Templates() []*gen.Template {
gen.MustParse(
gen.NewTemplate("admin").
Funcs(template.FuncMap{
"fieldName": fieldName,
"fieldLabel": fieldLabel,
"fieldName": fieldName,
"fieldLabel": fieldLabel,
"fieldIsPointer": fieldIsPointer,
}).
ParseFS(templateDir, "templates/*tmpl"),
),
@ -58,6 +60,17 @@ func fieldLabel(name string) string {
return upperFirst(out)
}
func fieldIsPointer(f *gen.Field) bool {
switch {
case f.Type.Type == field.TypeBool:
return false
case f.Optional,
f.Default:
return true
}
return false
}
func upperFirst(s string) string {
if len(s) == 0 {
return s

14
ent/admin/handler.go
View file

@ -90,8 +90,9 @@ func (h *Handler) PasswordTokenCreate(ctx echo.Context) error {
op := h.client.PasswordToken.Create()
op.SetHash(payload.Hash)
op.SetUserID(payload.UserID)
op.SetCreatedAt(payload.CreatedAt)
// op.SetUserID(payload.User)
if payload.CreatedAt != nil {
op.SetCreatedAt(*payload.CreatedAt)
}
_, err := op.Save(ctx.Request().Context())
return err
}
@ -110,8 +111,9 @@ func (h *Handler) PasswordTokenUpdate(ctx echo.Context, id int) error {
op := entity.Update()
op.SetHash(payload.Hash)
op.SetUserID(payload.UserID)
op.SetCreatedAt(payload.CreatedAt)
// op.SetUserID(payload.User)
if payload.CreatedAt != nil {
op.SetCreatedAt(*payload.CreatedAt)
}
_, err = op.Save(ctx.Request().Context())
return err
}
@ -176,7 +178,9 @@ func (h *Handler) UserCreate(ctx echo.Context) error {
op.SetEmail(payload.Email)
op.SetPassword(payload.Password)
op.SetVerified(payload.Verified)
op.SetCreatedAt(payload.CreatedAt)
if payload.CreatedAt != nil {
op.SetCreatedAt(*payload.CreatedAt)
}
_, err := op.Save(ctx.Request().Context())
return err
}

24
ent/admin/templates/handler.tmpl
View file

@ -96,11 +96,12 @@
op := h.client.{{ $n.Name }}.Create()
{{- range $f := $n.Fields }}
op.Set{{ fieldName $f.Name }}(payload.{{ fieldName $f.Name }})
{{- end }}
{{- range $e := $n.Edges }}
{{- if not $e.Inverse}}
// op.Set{{ fieldName $e.Name }}ID(payload.{{ fieldName $e.Name }})
{{- if (fieldIsPointer $f) }}
if payload.{{ fieldName $f.Name }} != nil {
op.Set{{ fieldName $f.Name }}(*payload.{{ fieldName $f.Name }})
}
{{- else }}
op.Set{{ fieldName $f.Name }}(payload.{{ fieldName $f.Name }})
{{- end }}
{{- end }}
_, err := op.Save(ctx.Request().Context())
@ -121,12 +122,13 @@
op := entity.Update()
{{- range $f := $n.Fields }}
{{- if not $f.Immutable }}
op.Set{{ fieldName $f.Name }}(payload.{{ fieldName $f.Name }})
{{- end }}
{{- end }}
{{- range $e := $n.Edges }}
{{- if not $e.Inverse}}
// op.Set{{ fieldName $e.Name }}ID(payload.{{ fieldName $e.Name }})
{{- if (fieldIsPointer $f) }}
if payload.{{ fieldName $f.Name }} != nil {
op.Set{{ fieldName $f.Name }}(*payload.{{ fieldName $f.Name }})
}
{{- else }}
op.Set{{ fieldName $f.Name }}(payload.{{ fieldName $f.Name }})
{{- end }}
{{- end }}
{{- end }}
_, err = op.Save(ctx.Request().Context())

2
ent/admin/templates/types.tmpl
View file

@ -8,7 +8,7 @@
{{ range $n := $.Nodes }}
type {{ $n.Name }} struct {
{{- range $f := $n.Fields }}
{{ fieldName $f.Name }} {{ $f.Type }} `form:"{{ $f.Name }}"`
{{ fieldName $f.Name }} {{ if (fieldIsPointer $f) }}*{{ end }}{{ $f.Type }} `form:"{{ $f.Name }}"`
{{- end }}
}
{{ end }}

16
ent/admin/types.go
View file

@ -4,17 +4,17 @@ package admin
import "time"
type PasswordToken struct {
Hash string `form:"hash"`
UserID int `form:"user_id"`
CreatedAt time.Time `form:"created_at"`
Hash string `form:"hash"`
UserID int `form:"user_id"`
CreatedAt *time.Time `form:"created_at"`
}
type User struct {
Name string `form:"name"`
Email string `form:"email"`
Password string `form:"password"`
Verified bool `form:"verified"`
CreatedAt time.Time `form:"created_at"`
Name string `form:"name"`
Email string `form:"email"`
Password string `form:"password"`
Verified bool `form:"verified"`
CreatedAt *time.Time `form:"created_at"`
}
type EntityList struct {