package schema

import (
	"time"

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

// WorkOrder 工单：一个合同 = 一个工单
type WorkOrder struct {
	ent.Schema
}

func (WorkOrder) Annotations() []schema.Annotation {
	return []schema.Annotation{entsql.Annotation{Table: "work_order"}}
}

func (WorkOrder) Fields() []ent.Field {
	return []ent.Field{
		field.Int("id").Positive(),
		field.String("workOrderNo").MaxLen(64).Comment("工单号"),
		field.Int("productTypeId").Comment("产品类型ID"),
		field.String("productCode").Default("").MaxLen(64).Comment("产品编码"),
		field.String("productName").Default("").MaxLen(128),
		field.Int("quantity").Default(1).Comment("工件数量"),
		field.Int("finishedNum").Default(0).Comment("完成数量"),
		field.Int("failNum").Default(0).Comment("NG数量"),
		field.String("processSeq").Default("").MaxLen(64).Comment("工位组合，如 1,3,5（本工单要经过的工位，按序执行）"),
		// 工程编号体系（客户口径 问题记录 L580）：合同 → 工程编号 → 产品序号
		field.String("contractNo").Default("").MaxLen(64).Comment("合同编号"),
		field.String("projectNo").Default("").MaxLen(64).Comment("工程编号（一个合同对应一个工程编号）"),
		field.String("productSerial").Default("").MaxLen(64).Comment("产品序号（一个工程编号下各产品的序号）"),
		// 创建人（取当前登录用户）+ 完成日期（必填）
		field.String("createdBy").Default("").MaxLen(64).Comment("创建人"),
		field.Time("dueDate").Optional().Nillable().Comment("完成日期(必填)"),
		// CREATED / RELEASED / IN_PROGRESS / COMPLETED / CLOSED
		field.String("status").Default("CREATED").MaxLen(20),
		field.Time("planStart").Optional().Nillable().Comment("计划开始"),
		field.Time("planEnd").Optional().Nillable().Comment("计划结束"),
		field.JSON("context", map[string]any{}).Optional().SchemaType(map[string]string{
			dialect.Postgres: "jsonb",
		}).Comment("扩展字段"),
		field.Time("completedAt").Optional().Nillable(),
		field.Time("createdAt").Default(time.Now).Immutable(),
		field.Time("updatedAt").Default(time.Now).UpdateDefault(time.Now).Optional().Nillable(),
	}
}

func (WorkOrder) Indexes() []ent.Index {
	return []ent.Index{
		index.Fields("workOrderNo").Unique(),
		index.Fields("status"),
		index.Fields("productTypeId"),
	}
}