96 lines
2.3 KiB
Go
96 lines
2.3 KiB
Go
// Copyright 2023 wanderer <a_mirre at utb dot cz>
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
package schema
|
|
|
|
import (
|
|
"time"
|
|
|
|
"entgo.io/ent"
|
|
"entgo.io/ent/schema/edge"
|
|
"entgo.io/ent/schema/field"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// LocalBreach holds the schema definition for the LocalBreach entity.
|
|
type LocalBreach struct {
|
|
ent.Schema
|
|
}
|
|
|
|
// ImportSchema is the LocalBreach model.
|
|
type ImportSchema struct {
|
|
Name string `yaml:"name" validate:"required,name"`
|
|
Description string `yaml:"description"`
|
|
CreatedAt time.Time `yaml:"createdAt"`
|
|
Date time.Time `yaml:"time"`
|
|
IsVerified bool `yaml:"isVerified"`
|
|
ContainsUsernames bool `yaml:"containsUsernames"`
|
|
ContainsEmails bool `yaml:"containsEmails"`
|
|
ContainsPasswords bool `yaml:"containsPasswds"`
|
|
ContainsHashes bool `yaml:"containsHashes"`
|
|
HashType string `yaml:"hashType"`
|
|
HashSalted bool `yaml:"hashSalted"`
|
|
HashPeppered bool `yaml:"hashPeppered"`
|
|
Data Data `yaml:"data"`
|
|
}
|
|
|
|
type (
|
|
DataUsernames []string
|
|
DataEmails []string
|
|
DataHashes []string
|
|
DataPasswords []string
|
|
Data struct {
|
|
Usernames *DataUsernames
|
|
Emails *DataEmails
|
|
Hashes *DataHashes
|
|
Passwords *DataPasswords
|
|
}
|
|
)
|
|
|
|
// Fields of the LocalBreach.
|
|
func (LocalBreach) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
field.UUID("id", uuid.UUID{}).
|
|
Default(uuid.New).
|
|
Unique().
|
|
Immutable(),
|
|
// Unique but may change.
|
|
field.String("name").
|
|
Unique().
|
|
NotEmpty(),
|
|
field.Time("date").
|
|
Default(time.Now()),
|
|
field.String("description").
|
|
Optional().
|
|
Nillable(),
|
|
field.Bool("is_verified"),
|
|
field.Bool("contains_passwords"),
|
|
field.Bool("contains_hashes"),
|
|
field.Bool("contains_emails"),
|
|
field.Bool("contains_usernames"),
|
|
field.String("hash_type").
|
|
Comment("MD5, SHA1, Argon2id...").
|
|
Optional().
|
|
Nillable(),
|
|
field.Bool("hash_salted").
|
|
Optional().
|
|
Nillable(),
|
|
field.Bool("hash_peppered").
|
|
Optional().
|
|
Nillable(),
|
|
field.JSON("data", &Data{}).
|
|
Optional(),
|
|
field.Time("created_at").
|
|
Default(time.Now()).
|
|
Immutable(),
|
|
}
|
|
}
|
|
|
|
// Edges of the LocalBreach.
|
|
func (LocalBreach) Edges() []ent.Edge {
|
|
return []ent.Edge{
|
|
edge.From("tracked_breaches", TrackedBreaches.Type).
|
|
Ref("localbreach"),
|
|
}
|
|
}
|