package model

import (
	"context"
	"fmt"
	"strings"

	"qnc-server/common/globalkey"

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

var _ ProductFeatureModel = (*customProductFeatureModel)(nil)

type (
	// ProductFeatureModel is an interface to be customized, add more methods here,
	// and implement the added methods in customProductFeatureModel.
	ProductFeatureModel interface {
		productFeatureModel
		// SumFeatureCostPriceByProductId 汇总该产品下已关联且未删除的模块成本（feature.cost_price 之和）
		SumFeatureCostPriceByProductId(ctx context.Context, productId string) (float64, error)
		// SumFeatureCostPriceByProductIds 批量汇总，未出现在结果中的产品视为 0
		SumFeatureCostPriceByProductIds(ctx context.Context, productIds []string) (map[string]float64, error)
	}

	customProductFeatureModel struct {
		*defaultProductFeatureModel
	}
)

// NewProductFeatureModel returns a model for the database table.
func NewProductFeatureModel(conn sqlx.SqlConn, c cache.CacheConf) ProductFeatureModel {
	return &customProductFeatureModel{
		defaultProductFeatureModel: newProductFeatureModel(conn, c),
	}
}

func (m *customProductFeatureModel) SumFeatureCostPriceByProductId(ctx context.Context, productId string) (float64, error) {
	query := `
SELECT COALESCE(SUM(f.cost_price), 0)
FROM product_feature pf
INNER JOIN feature f ON f.id = pf.feature_id AND f.del_state = ?
WHERE pf.product_id = ? AND pf.del_state = ?`
	var sum float64
	err := m.QueryRowNoCacheCtx(ctx, &sum, query, globalkey.DelStateNo, productId, globalkey.DelStateNo)
	if err != nil {
		return 0, err
	}
	return sum, nil
}

func (m *customProductFeatureModel) SumFeatureCostPriceByProductIds(ctx context.Context, productIds []string) (map[string]float64, error) {
	out := make(map[string]float64)
	if len(productIds) == 0 {
		return out, nil
	}
	placeholders := strings.TrimSuffix(strings.Repeat("?,", len(productIds)), ",")
	query := fmt.Sprintf(`
SELECT pf.product_id, COALESCE(SUM(f.cost_price), 0) AS total
FROM product_feature pf
INNER JOIN feature f ON f.id = pf.feature_id AND f.del_state = ?
WHERE pf.del_state = ? AND pf.product_id IN (%s)
GROUP BY pf.product_id`, placeholders)
	args := make([]interface{}, 0, 2+len(productIds))
	args = append(args, globalkey.DelStateNo, globalkey.DelStateNo)
	for _, id := range productIds {
		args = append(args, id)
	}
	var rows []struct {
		ProductId string  `db:"product_id"`
		Total     float64 `db:"total"`
	}
	err := m.QueryRowsNoCacheCtx(ctx, &rows, query, args...)
	if err != nil {
		return nil, err
	}
	for i := range rows {
		out[rows[i].ProductId] = rows[i].Total
	}
	return out, nil
}