package logic

import (
	"context"
	"errors"

	"bj_power_mes/ent"
	"bj_power_mes/ent/semiflow"
)

type SemiReq struct {
	Sn              string `json:"sn"`
	OrderNo         string `json:"orderNo"`
	DoneProcessCodes []int `json:"doneProcessCodes"`
	Action          string `json:"action"` // INBOUND / OUTBOUND
	TargetDock      string `json:"targetDock"`
}

// SemiFlow 半成品流转：记录已完成工序，并调用 WMS /api/internal/semi/*
func (s *Service) SemiFlow(ctx context.Context, req SemiReq, operator string) error {
	if req.Sn == "" {
		return errors.New("sn 不能为空")
	}
	action := req.Action
	if action == "" {
		action = "INBOUND"
	}
	_, err := s.ctx.EntClient.SemiFlow.Create().
		SetSn(req.Sn).SetOrderNo(req.OrderNo).SetDoneProcessCodes(req.DoneProcessCodes).
		SetAction(action).SetTargetDock(req.TargetDock).SetOperator(operator).Save(ctx)
	if err != nil {
		return err
	}
	// 调用 WMS 内部接口
	var wmsErr error
	if action == "INBOUND" {
		wmsErr = s.ctx.Wms.SemiInbound(ctx, req.Sn, req.OrderNo, req.DoneProcessCodes, operator)
	} else {
		wmsErr = s.ctx.Wms.SemiOutbound(ctx, req.Sn, req.OrderNo, req.DoneProcessCodes, operator)
	}
	if wmsErr != nil {
		_ = wmsErr
	}
	s.ctx.EventLog.Write(ctx, "semi.flow", req.OrderNo, operator, "semi_flow", req.Sn, "半成品流转",
		map[string]any{"action": action, "doneProcessCodes": req.DoneProcessCodes})
	return nil
}

// ListSemiFlows 查询半成品流转记录
func (s *Service) ListSemiFlows(ctx context.Context, sn, orderNo string) ([]*ent.SemiFlow, error) {
	q := s.ctx.EntClient.SemiFlow.Query()
	if sn != "" {
		q = q.Where(semiflow.Sn(sn))
	}
	if orderNo != "" {
		q = q.Where(semiflow.OrderNo(orderNo))
	}
	return q.Order(ent.Desc(semiflow.FieldID)).All(ctx)
}

// ListEventLogs 操作日志查询（按工单号/操作人）
func (s *Service) ListEventLogs(ctx context.Context, orderNo, operator, eventType string, page, pageSize int) ([]*ent.EventLog, int, error) {
	return s.ctx.EventLog.Query(ctx, eventType, orderNo, operator, page, pageSize)
}