package seo

import (
	"log/slog"
	"strings"

	"github.com/mlogclub/simple/common/strs"
	"github.com/mlogclub/simple/common/urls"

	"github.com/go-resty/resty/v2"

	"bbs-go/internal/pkg/config"
)

func Push(url string) {
	go func() {
		PushUrls([]string{url})
		PushSmUrls([]string{url})
	}()
}

// 百度链接推送
func PushUrls(urls []string) {
	if len(urls) == 0 {
		return
	}
	if len(config.Instance.BaiduSEO.Site) == 0 || len(config.Instance.BaiduSEO.Token) == 0 {
		return
	}
	api := "http://data.zz.baidu.com/urls?site=" + config.Instance.BaiduSEO.Site + "&token=" +
		config.Instance.BaiduSEO.Token
	body := strings.Join(urls, "\n")
	if response, err := resty.New().R().SetBody(body).Post(api); err != nil {
		slog.Error(err.Error(), slog.Any("err", err))
	} else {
		slog.Info("百度链接提交完成", slog.Any("resp", response.Body()))
	}
}

// 神马链接推送
func PushSmUrls(urlList []string) {
	if len(urlList) == 0 {
		return
	}
	conf := config.Instance.SmSEO
	if strs.IsBlank(conf.Site) || strs.IsBlank(conf.UserName) || strs.IsBlank(conf.Token) {
		return
	}

	u := urls.ParseUrl("https://data.zhanzhang.sm.cn/push")
	u.AddQuery("site", conf.Site)
	u.AddQuery("user_name", conf.UserName)
	u.AddQuery("resource_name", "mip_add")
	u.AddQuery("token", conf.Token)

	body := strings.Join(urlList, "\n")
	if response, err := resty.New().R().SetBody(body).Post(u.BuildStr()); err != nil {
		slog.Error(err.Error(), slog.Any("err", err))
	} else {
		slog.Info("神马搜索链接推送完成：", slog.Any("resp", response.Body()))
	}
}