73 lines
2.4 KiB
Go
73 lines
2.4 KiB
Go
// This file is part of the happyDeliver (R) project.
|
|
// Copyright (c) 2025 happyDomain
|
|
// Authors: Pierre-Olivier Mercier, et al.
|
|
//
|
|
// This program is offered under a commercial and under the AGPL license.
|
|
// For commercial licensing, contact us at <contact@happydomain.org>.
|
|
//
|
|
// For AGPL licensing:
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package storage
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// TestStatus represents the status of a deliverability test
|
|
type TestStatus string
|
|
|
|
const (
|
|
StatusPending TestStatus = "pending"
|
|
StatusReceived TestStatus = "received"
|
|
StatusAnalyzed TestStatus = "analyzed"
|
|
StatusFailed TestStatus = "failed"
|
|
)
|
|
|
|
// Test represents a deliverability test instance
|
|
type Test struct {
|
|
ID uuid.UUID `gorm:"type:uuid;primaryKey"`
|
|
Status TestStatus `gorm:"type:varchar(20);not null;index"`
|
|
CreatedAt time.Time `gorm:"not null"`
|
|
UpdatedAt time.Time `gorm:"not null"`
|
|
Report *Report `gorm:"foreignKey:TestID;constraint:OnDelete:CASCADE"`
|
|
}
|
|
|
|
// BeforeCreate is a GORM hook that generates a UUID before creating a test
|
|
func (t *Test) BeforeCreate(tx *gorm.DB) error {
|
|
if t.ID == uuid.Nil {
|
|
t.ID = uuid.New()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Report represents the analysis report for a test
|
|
type Report struct {
|
|
ID uuid.UUID `gorm:"type:uuid;primaryKey"`
|
|
TestID uuid.UUID `gorm:"type:uuid;uniqueIndex;not null"`
|
|
RawEmail []byte `gorm:"type:bytea;not null"` // Full raw email with headers
|
|
ReportJSON []byte `gorm:"type:bytea;not null"` // JSON-encoded report data
|
|
CreatedAt time.Time `gorm:"not null"`
|
|
}
|
|
|
|
// BeforeCreate is a GORM hook that generates a UUID before creating a report
|
|
func (r *Report) BeforeCreate(tx *gorm.DB) error {
|
|
if r.ID == uuid.Nil {
|
|
r.ID = uuid.New()
|
|
}
|
|
return nil
|
|
}
|