package mysqldao

import (
	"context"

	"git.lzcwy.cn/zhengchiliu/zt-closet-manage-backend/global"
	"git.lzcwy.cn/zhengchiliu/zt-closet-manage-backend/internal/mysqlmodel"
	"gorm.io/gorm"
)

// OrderDao 订单数据访问
type OrderDao struct {
	db *gorm.DB
}

// NewOrderDao 创建订单 DAO
func NewOrderDao() *OrderDao {
	return &OrderDao{db: global.MysqlEngine}
}

// OrderListParams 订单分页列表筛选参数。
type OrderListParams struct {
	OrderNo       string // 订单号（精确）
	TransactionID string // 微信交易号（精确）
	UserID        int64  // 用户 ID，0 表示全部
	Status        int8   // 订单状态，0 表示全部
	Page          int
	Size          int
}

// OrderList 分页查询订单
func (d *OrderDao) OrderList(ctx context.Context, p OrderListParams) ([]mysqlmodel.Order, int64, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderList")
	defer span.End()

	var list []mysqlmodel.Order
	var total int64
	q := d.db.WithContext(ctx).Model(&mysqlmodel.Order{})
	if p.OrderNo != "" {
		q = q.Where("order_no = ?", p.OrderNo)
	}
	if p.TransactionID != "" {
		q = q.Where("transaction_id = ?", p.TransactionID)
	}
	if p.UserID != 0 {
		q = q.Where("user_id = ?", p.UserID)
	}
	if p.Status != 0 {
		q = q.Where("order_status = ?", p.Status)
	}
	if err := q.Count(&total).Error; err != nil {
		span.RecordError(err)
		return nil, 0, err
	}
	if err := q.Offset((p.Page - 1) * p.Size).Limit(p.Size).Order("id DESC").Find(&list).Error; err != nil {
		span.RecordError(err)
		return nil, 0, err
	}
	return list, total, nil
}

// OrderGetByOrderNo 按订单号查询订单
func (d *OrderDao) OrderGetByOrderNo(ctx context.Context, orderNo string) (*mysqlmodel.Order, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderGetByOrderNo")
	defer span.End()

	var order mysqlmodel.Order
	if err := d.db.WithContext(ctx).Where("order_no = ?", orderNo).First(&order).Error; err != nil {
		span.RecordError(err)
		return nil, err
	}
	return &order, nil
}

// OrderGetByID 按主键 ID 查询订单
func (d *OrderDao) OrderGetByID(ctx context.Context, id int64) (*mysqlmodel.Order, error) {
	ctx, span := global.Tracer.Start(ctx, "mysqldao.OrderGetByID")
	defer span.End()

	var order mysqlmodel.Order
	if err := d.db.WithContext(ctx).First(&order, id).Error; err != nil {
		span.RecordError(err)
		return nil, err
	}
	return &order, nil
}