// 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
}