package srv_seopush

import (
	"context"
	"fmt"
	"strings"

	"hkdj-web-server/pub/common/log"
	"hkdj-web-server/pub/common/srvdef"
	"hkdj-web-server/pub/models/playletthememod"
	"hkdj-web-server/pub/seo/h5path"
	seoindexnow "hkdj-web-server/pub/seo/indexnow"
	"hkdj-web-server/web/webg"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo/options"
)

// BulkIndexNowResult 批量提交结果，包含提交的 URL 列表和 IndexNow 原始回执。
type BulkIndexNowResult struct {
	URLs       []string               `json:"urls"`
	SubmitInfo *seoindexnow.BatchResult `json:"submitInfo"`
}

// SubmitHomeAndThemes 组装主页 + 前 N 个专题（按 Rank 降序）URL，批量提交 IndexNow。
// limit 为 URL 总数（包含主页），默认 10；小于等于 1 时视为仅主页。
func SubmitHomeAndThemes(ctx context.Context, limit int) (*BulkIndexNowResult, *srvdef.SrvErr) {
	if limit <= 0 {
		limit = 10
	}
	webAddr := SEOBaseURL(ctx)
	if webAddr == "" {
		return nil, srvdef.New(srvdef.ErrNoData, "WebAddr 未配置", "")
	}
	key := strings.TrimSpace(webg.Conf.IndexNow.Key)
	if key == "" {
		return nil, srvdef.New(srvdef.ErrNoData, "indexNow.key 未配置", "")
	}

	urls := make([]string, 0, limit)
	urls = append(urls, h5path.Abs(webAddr, "/"))
	if remain := limit - 1; remain > 0 {
		themes, err := playletthememod.FindManyPlayletTheme(ctx,
			bson.M{playletthememod.FN.PlayletTheme.Status: true},
			options.Find().
				SetSort(bson.M{playletthememod.FN.PlayletTheme.Rank: -1}).
				SetLimit(int64(remain)))
		if err == nil {
			for _, t := range themes {
				if t == nil || t.Id <= 0 {
					continue
				}
				path := h5path.TopicDetail(t.Slug, t.Id)
				if path == "" {
					continue
				}
				urls = append(urls, h5path.Abs(webAddr, path))
			}
		} else {
			log.Warn("seopush bulk: list themes failed", log.E(err))
		}
	}

	config, cerr := seoindexnow.BuildConfig(webAddr, key)
	if cerr != nil {
		return nil, srvdef.New(srvdef.Failure, fmt.Sprintf("构造 IndexNow 配置失败: %v", cerr), "")
	}
	result := seoindexnow.NewBatchIndexNowSubmitter(config).SubmitBatch(urls)
	log.Info("seopush bulk: indexnow submitted",
		log.Any("urls", urls),
		log.Any("statusCodes", result.StatusCodes),
		log.Any("success", result.SuccessCount),
		log.Any("fail", result.FailCount))
	return &BulkIndexNowResult{URLs: urls, SubmitInfo: result}, nil
}