package model

import (
	"context"
	"fmt"
	"time"

	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

var _ InvProductBatchModel = (*customInvProductBatchModel)(nil)

type (
	InvProductBatchModel interface {
		invProductBatchModel
		FindByProductId(ctx context.Context, tenantId, productId string) ([]*InvProductBatch, error)
		FindLatestByProductId(ctx context.Context, tenantId, productId string) (*InvProductBatch, error)
		FindOneByProductAndBatchId(ctx context.Context, tenantId, productId, batchId string) (*InvProductBatch, error)
		NextBatchNo(ctx context.Context, tenantId, productCode string) (string, error)
		SoftDeleteByBatchId(ctx context.Context, batchId string) error
	}

	customInvProductBatchModel struct {
		*defaultInvProductBatchModel
	}
)

func NewInvProductBatchModel(conn sqlx.SqlConn) InvProductBatchModel {
	return &customInvProductBatchModel{
		defaultInvProductBatchModel: newInvProductBatchModel(conn),
	}
}

func (m *customInvProductBatchModel) FindByProductId(ctx context.Context, tenantId, productId string) ([]*InvProductBatch, error) {
	var list []*InvProductBatch
	query := fmt.Sprintf("SELECT %s FROM %s WHERE tenant_id = ? AND product_id = ? AND deleted_at IS NULL ORDER BY batch_no", invProductBatchRows, m.table)
	if err := m.conn.QueryRowsCtx(ctx, &list, query, tenantId, productId); err != nil {
		return nil, err
	}
	return list, nil
}

func (m *customInvProductBatchModel) FindLatestByProductId(ctx context.Context, tenantId, productId string) (*InvProductBatch, error) {
	var resp InvProductBatch
	query := fmt.Sprintf("SELECT %s FROM %s WHERE tenant_id = ? AND product_id = ? AND deleted_at IS NULL AND status = 1 ORDER BY batch_no DESC LIMIT 1", invProductBatchRows, m.table)
	if err := m.conn.QueryRowCtx(ctx, &resp, query, tenantId, productId); err != nil {
		return nil, err
	}
	return &resp, nil
}

func (m *customInvProductBatchModel) FindOneByProductAndBatchId(ctx context.Context, tenantId, productId, batchId string) (*InvProductBatch, error) {
	var resp InvProductBatch
	query := fmt.Sprintf("SELECT %s FROM %s WHERE tenant_id = ? AND product_id = ? AND batch_id = ? AND deleted_at IS NULL LIMIT 1", invProductBatchRows, m.table)
	if err := m.conn.QueryRowCtx(ctx, &resp, query, tenantId, productId, batchId); err != nil {
		return nil, err
	}
	return &resp, nil
}

func (m *customInvProductBatchModel) NextBatchNo(ctx context.Context, tenantId, productCode string) (string, error) {
	var next int64
	query := fmt.Sprintf(`SELECT COALESCE(MAX(CAST(SUBSTRING(batch_no, CHAR_LENGTH(?) + 3) AS UNSIGNED)), 0) + 1
		FROM %s
		WHERE tenant_id = ? AND batch_no LIKE CONCAT(?, '-B%%')`, m.table)
	if err := m.conn.QueryRowCtx(ctx, &next, query, productCode, tenantId, productCode); err != nil {
		return "", err
	}
	return fmt.Sprintf("%s-B%03d", productCode, next), nil
}

func (m *customInvProductBatchModel) SoftDeleteByBatchId(ctx context.Context, batchId string) error {
	batch, err := m.FindOneByBatchId(ctx, batchId)
	if err != nil {
		return err
	}
	batch.Status = 0
	batch.DeletedAt.Time = time.Now()
	batch.DeletedAt.Valid = true
	return m.Update(ctx, batch)
}