package schema

import (
	"time"

	"bj_power_mes/constants"

	"entgo.io/ent"
	"entgo.io/ent/dialect/entsql"
	"entgo.io/ent/schema"
	"entgo.io/ent/schema/edge"
	"entgo.io/ent/schema/field"
	"entgo.io/ent/schema/index"
)

// Pallet 托盘——接驳台一次上料批次
type Pallet struct {
	ent.Schema
}

func (Pallet) Annotations() []schema.Annotation {
	return []schema.Annotation{
		entsql.Annotation{Table: "pallet"},
	}
}

func (Pallet) Fields() []ent.Field {
	return []ent.Field{
		field.Int("id").Positive(),
		field.String("palletCode").Default("").Comment("MOM 托盘码，预留"),
		field.Int("workOrderId").Comment("关联工单ID"),
		field.Int("dockNo").Comment("接驳台编号"),
		field.Enum("status").
			GoType(constants.PalletStatus("")).
			Default(string(constants.PalletStatus_PendingProcess)).
			Comment("托盘状态"),
		field.Int("passQty").Default(0).Comment("良品数"),
		field.Int("ngQty").Default(0).Comment("不良品数"),
		field.Int("totalQty").Default(0).Comment("工件总数"),
		field.Time("arrivedAt").Default(time.Now).Optional().Nillable().Comment("到达接驳台时间"),
		field.Time("departedAt").Optional().Nillable().Comment("离开接驳台时间"),
		field.Time("completedAt").Optional().Nillable().Comment("全部完工时间"),
		field.Time("reportedAt").Optional().Nillable().Comment("报工时间，预留"),
		field.Time("createdAt").Default(time.Now).Immutable().Comment("创建时间"),
		field.Time("updatedAt").Default(time.Now).UpdateDefault(time.Now).Optional().Nillable().Comment("更新时间"),
	}
}

func (Pallet) Edges() []ent.Edge {
	return []ent.Edge{
		edge.From("workOrder", WorkOrder.Type).Ref("pallets").Field("workOrderId").Unique().Required(),
		edge.To("jobs", Job.Type),
	}
}

func (Pallet) Indexes() []ent.Index {
	return []ent.Index{
		index.Fields("workOrderId"),
		index.Fields("status"),
	}
}