101 lines
2.1 KiB
Go
101 lines
2.1 KiB
Go
package schema
|
|
|
|
import (
|
|
"entgo.io/ent"
|
|
"entgo.io/ent/dialect/entsql"
|
|
"entgo.io/ent/schema"
|
|
"entgo.io/ent/schema/edge"
|
|
"entgo.io/ent/schema/field"
|
|
"entgo.io/ent/schema/index"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// Session holds the schema definition for the Session entity.
|
|
type Session struct {
|
|
ent.Schema
|
|
}
|
|
|
|
// Fields of the Session.
|
|
func (Session) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
field.UUID("id", uuid.Nil).
|
|
Default(uuid.New),
|
|
field.Bool("active").
|
|
Default(false),
|
|
field.Time("issued_at"),
|
|
field.Time("expires_at").
|
|
Optional(),
|
|
field.String("token").
|
|
Optional(),
|
|
field.JSON("method", map[string]any{}).
|
|
Optional(),
|
|
field.UUID("device_id", uuid.Nil).
|
|
Optional(),
|
|
}
|
|
}
|
|
|
|
// Edges of the Session.
|
|
func (Session) Edges() []ent.Edge {
|
|
return []ent.Edge{
|
|
edge.From("device", SessionDevice.Type).
|
|
Ref("sessions").
|
|
Unique().
|
|
Field("device_id"),
|
|
}
|
|
}
|
|
|
|
// Indexes of the Session.
|
|
func (Session) Indexes() []ent.Index {
|
|
return []ent.Index{
|
|
index.Fields("active", "expires_at"),
|
|
}
|
|
}
|
|
|
|
// Annotations of the Session.
|
|
func (Session) Annotations() []schema.Annotation {
|
|
return []schema.Annotation{
|
|
// Named check constraints are compared by their name.
|
|
// Thus, the definition does not need to be normalized.
|
|
entsql.Checks(map[string]string{
|
|
"token_length": "(LENGTH(`token`) = 64)",
|
|
}),
|
|
}
|
|
}
|
|
|
|
// SessionDevice holds the schema definition for the SessionDevice entity.
|
|
type SessionDevice struct {
|
|
ent.Schema
|
|
}
|
|
|
|
// Fields of the SessionDevice.
|
|
func (SessionDevice) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
field.UUID("id", uuid.Nil).
|
|
Default(uuid.New),
|
|
field.String("ip_address").
|
|
MaxLen(50),
|
|
field.String("user_agent").
|
|
MaxLen(512),
|
|
field.String("location").
|
|
MaxLen(512),
|
|
field.Time("created_at"),
|
|
field.Time("updated_at").
|
|
Optional(),
|
|
}
|
|
}
|
|
|
|
// Edges of the SessionDevice.
|
|
func (SessionDevice) Edges() []ent.Edge {
|
|
return []ent.Edge{
|
|
edge.To("sessions", Session.Type),
|
|
}
|
|
}
|
|
|
|
// Indexes of the SessionDevice.
|
|
func (SessionDevice) Indexes() []ent.Index {
|
|
return []ent.Index{
|
|
index.Fields("ip_address", "user_agent"),
|
|
index.Fields("location", "updated_at"),
|
|
}
|
|
}
|