20251209

archive/bbs-go/server/internal/pkg/seo/push.go

@@ -0,0 +1,62 @@
+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()))
+	}
+}