package schema

import (
	"entgo.io/ent"
	"entgo.io/ent/schema/field"
	"entgo.io/ent/schema/index"

	"github.com/pingcap/test-infra/common/model"
	"github.com/pingcap/test-infra/services/tcms/service/store/v2/ent/schema/mixin"
)

// Triage holds the schema definition for the Triage entity.
type Triage struct {
	ent.Schema
}

// Mixin Common fields.
func (Triage) Mixin() []ent.Mixin {
	return []ent.Mixin{
		mixin.ID{},
		mixin.Time{},
	}
}

// Fields of the Triage.
func (Triage) Fields() []ent.Field {
	return []ent.Field{
		field.Uint64("planExecutionID").GoType(model.ID(0)).Immutable().Min(1),
		field.Uint64("planID").GoType(model.ID(0)).Immutable().Min(1),
		field.JSON("triagingUsers", []model.ID{}).Default([]model.ID{}),
		field.JSON("toTriageCaseExecIDs", []model.ID{}).Default([]model.ID{}),
		field.JSON("triagedCaseExecIDs", []model.ID{}).Default([]model.ID{}),
		field.Enum("triagePhase").GoType(model.TriagePhaseToTriage).Default(string(model.TriagePhaseToTriage)),
		field.Time("startedAt").SchemaType(SchemaTypeDatetime).Optional().Nillable(),
		field.Time("triagedAt").SchemaType(SchemaTypeDatetime).Optional().Nillable(),
		field.JSON("reasons", []model.CaseExecFailReason{}).Default([]model.CaseExecFailReason{}),
		field.Enum("solution").GoType(model.ExecFailSolutionKEEP).Default(string(model.ExecFailSolutionKEEP)),
		field.String("note").MaxLen(2047).Default(""),
	}
}

// Indexes of the Triage.
func (Triage) Indexes() []ent.Index {
	return []ent.Index{
		index.Fields("planExecutionID").Unique().StorageKey("uniq_plan_execution_id"),
	}
}