mirror of
https://github.com/phishingclub/phishingclub.git
synced 2026-02-12 16:12:44 +00:00
39 lines
956 B
Go
39 lines
956 B
Go
package database
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
const (
|
|
RECIPIENT_GROUP_TABLE = "recipient_groups"
|
|
)
|
|
|
|
// RecipientGroup is a grouping of recipient
|
|
type RecipientGroup struct {
|
|
ID *uuid.UUID `gorm:"primary_key;not null;unique;type:uuid"`
|
|
CreatedAt *time.Time `gorm:"not null;index;"`
|
|
UpdatedAt *time.Time `gorm:"not null;index;"`
|
|
|
|
Name string `gorm:"not null;index;uniqueIndex:idx_recipient_groups_unique_name_and_company_id;"`
|
|
|
|
// can belong-to
|
|
CompanyID *uuid.UUID `gorm:"type:uuid;index;uniqueIndex:idx_recipient_groups_unique_name_and_company_id"`
|
|
Company *Company
|
|
|
|
// many-to-many
|
|
Recipients []Recipient `gorm:"many2many:recipient_group_recipients;"`
|
|
}
|
|
|
|
func (e *RecipientGroup) Migrate(db *gorm.DB) error {
|
|
// SQLITE
|
|
// ensure name + company id is unique
|
|
return UniqueIndexNameAndNullCompanyID(db, "recipient_groups")
|
|
}
|
|
|
|
func (RecipientGroup) TableName() string {
|
|
return RECIPIENT_GROUP_TABLE
|
|
}
|