jiang13-forum/service/community_test.go

package service

import (
	"encoding/json"
	"errors"
	"io"
	"net/http"
	"net/http/httptest"
	"sync/atomic"
	"testing"

	"github.com/glebarez/sqlite"
	"gorm.io/gorm"

	"git.iioio.com/freefire/jiang13-forum/model"
)

func setupCommunityTest(t *testing.T) (*ForumSettingsService, *CommunityService) {
	t.Helper()
	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	if err != nil {
		t.Fatal(err)
	}
	if err := db.AutoMigrate(
		&model.ForumSetting{},
		&model.CommunityInstance{},
		&model.User{},
		&model.Post{},
	); err != nil {
		t.Fatal(err)
	}
	prev := model.DB
	model.DB = db
	t.Cleanup(func() { model.DB = prev })

	settings := NewForumSettingsService()
	svc := NewCommunityService(settings)
	return settings, svc
}

func TestCommunityHeartbeatHubDisabled(t *testing.T) {
	_, svc := setupCommunityTest(t)
	err := svc.ReceiveHeartbeat(CommunityHeartbeatPayload{
		InstanceID: "aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee",
		SiteURL:    "https://example.com",
		SiteName:   "测试站",
		Version:    "1.0.0",
		Users:      1,
		Posts:      2,
	}, "127.0.0.1", "https://other.example")
	if !errors.Is(err, ErrCommunityHubDisabled) {
		t.Fatalf("want ErrCommunityHubDisabled, got %v", err)
	}
}

func TestCommunityHeartbeatAcceptAndList(t *testing.T) {
	settings, svc := setupCommunityTest(t)
	settings.SetCommunityHubEnabled(true)

	payload := CommunityHeartbeatPayload{
		InstanceID: "11111111-2222-3333-4444-555555555555",
		SiteURL:    "https://forum.example.org",
		SiteName:   "示例论坛",
		Version:    "1.2.3",
		Users:      10,
		Posts:      20,
	}
	if err := svc.ReceiveHeartbeat(payload, "203.0.113.9", ""); err != nil {
		t.Fatal(err)
	}
	payload.Users = 11
	payload.Posts = 21
	if err := svc.ReceiveHeartbeat(payload, "203.0.113.9", ""); err != nil {
		t.Fatal(err)
	}

	list, err := svc.ListInstances()
	if err != nil {
		t.Fatal(err)
	}
	if len(list) != 1 {
		t.Fatalf("want 1 instance, got %d", len(list))
	}
	got := list[0]
	if got.Users != 11 || got.Posts != 21 || !got.Online {
		t.Fatalf("unexpected row: %+v", got)
	}
	if got.SiteURL != payload.SiteURL || got.SiteName != payload.SiteName {
		t.Fatalf("site fields mismatch: %+v", got)
	}
}

func TestCommunityUpdateIgnoresHubFields(t *testing.T) {
	settings, _ := setupCommunityTest(t)
	if settings.CommunityConfig().HubEnabled {
		t.Fatal("hub should be off by default")
	}
	if _, err := settings.UpdateCommunityConfig(CommunityConfig{
		ReportEnabled: true,
		HubEnabled:    true,
		HubURL:        "https://evil.example",
		SiteURL:       "https://should-be-ignored.example",
	}); err != nil {
		t.Fatal(err)
	}
	cfg := settings.CommunityConfig()
	if cfg.HubEnabled {
		t.Fatal("UpdateCommunityConfig must not enable hub")
	}
	if cfg.HubURL != DefaultCommunityHubURL {
		t.Fatalf("hub_url must stay official, got %s", cfg.HubURL)
	}
	if cfg.SiteURL == "https://should-be-ignored.example" {
		t.Fatal("client site_url must be ignored")
	}
	if !cfg.ReportEnabled {
		t.Fatal("report should be enabled")
	}
	settings.SetCommunityHubEnabled(true)
	if !settings.CommunityConfig().HubEnabled {
		t.Fatal("SetCommunityHubEnabled should enable hub")
	}
}

func TestCommunityHeartbeatBadURL(t *testing.T) {
	settings, svc := setupCommunityTest(t)
	settings.SetCommunityHubEnabled(true)
	err := svc.ReceiveHeartbeat(CommunityHeartbeatPayload{
		InstanceID: "aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee",
		SiteURL:    "javascript:alert(1)",
		SiteName:   "坏",
	}, "127.0.0.1", "")
	if !errors.Is(err, ErrCommunityBadPayload) {
		t.Fatalf("want ErrCommunityBadPayload, got %v", err)
	}
}

func TestCommunityOutboundHeartbeat(t *testing.T) {
	settings, svc := setupCommunityTest(t)
	var hits atomic.Int32
	var lastBody CommunityHeartbeatPayload

	hub := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/api/community/heartbeat" {
			http.NotFound(w, r)
			return
		}
		defer r.Body.Close()
		raw, _ := io.ReadAll(r.Body)
		_ = json.Unmarshal(raw, &lastBody)
		hits.Add(1)
		w.WriteHeader(http.StatusOK)
		_, _ = w.Write([]byte(`{"ok":true}`))
	}))
	t.Cleanup(hub.Close)

	prevHub := communityHubBaseURL
	communityHubBaseURL = hub.URL
	t.Cleanup(func() { communityHubBaseURL = prevHub })

	svc.trySendHeartbeat()
	if hits.Load() != 0 {
		t.Fatal("report disabled should not send")
	}

	if err := settings.setString(SettingOIDCRootURL, "http://reporter.local"); err != nil {
		t.Fatal(err)
	}
	if err := settings.setString(SettingSiteName, "上报测试站"); err != nil {
		t.Fatal(err)
	}
	if _, err := settings.UpdateCommunityConfig(CommunityConfig{ReportEnabled: true}); err != nil {
		t.Fatal(err)
	}

	if err := svc.SendHeartbeatOnce(""); err != nil {
		t.Fatal(err)
	}
	if hits.Load() != 1 {
		t.Fatalf("want 1 outbound hit, got %d", hits.Load())
	}
	if lastBody.InstanceID == "" || lastBody.SiteURL == "" {
		t.Fatalf("empty payload: %+v", lastBody)
	}

	if _, err := settings.UpdateCommunityConfig(CommunityConfig{ReportEnabled: false}); err != nil {
		t.Fatal(err)
	}
	if err := svc.SendHeartbeatOnce(""); err != nil {
		t.Fatal(err)
	}
	if hits.Load() != 1 {
		t.Fatalf("after disable want still 1 hit, got %d", hits.Load())
	}
}

func TestCommunityFeatureAndShowcase(t *testing.T) {
	settings, svc := setupCommunityTest(t)
	settings.SetCommunityHubEnabled(true)

	payload := CommunityHeartbeatPayload{
		InstanceID: "aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee",
		SiteURL:    "https://forum.example.org",
		SiteName:   "示例论坛",
		Version:    "2.0.0",
	}
	if err := svc.ReceiveHeartbeat(payload, "127.0.0.1", ""); err != nil {
		t.Fatal(err)
	}

	empty, err := svc.ListShowcase("")
	if err != nil {
		t.Fatal(err)
	}
	if len(empty) != 0 {
		t.Fatal("showcase should be empty before feature")
	}

	view, err := svc.SetInstanceFeatured(payload.InstanceID, CommunityFeatureInput{
		Featured:     true,
		FeaturedNote: "精选自托管",
	})
	if err != nil {
		t.Fatal(err)
	}
	if !view.Featured || view.FeaturedNote != "精选自托管" {
		t.Fatalf("unexpected view: %+v", view)
	}

	items, err := svc.ListShowcase("")
	if err != nil {
		t.Fatal(err)
	}
	if len(items) != 1 || items[0].SiteURL != payload.SiteURL {
		t.Fatalf("showcase=%+v", items)
	}

	// 心跳更新不得清掉精选
	payload.Users = 9
	if err := svc.ReceiveHeartbeat(payload, "127.0.0.1", ""); err != nil {
		t.Fatal(err)
	}
	items, err = svc.ListShowcase("")
	if err != nil {
		t.Fatal(err)
	}
	if len(items) != 1 || items[0].FeaturedNote != "精选自托管" {
		t.Fatalf("featured lost after heartbeat: %+v", items)
	}

	settings.SetCommunityHubEnabled(false)
	items, err = svc.ListShowcase("")
	if err != nil {
		t.Fatal(err)
	}
	if len(items) != 0 {
		t.Fatal("hub off should hide showcase")
	}
}

func TestCommunitySiteURLFromOrigin(t *testing.T) {
	settings, svc := setupCommunityTest(t)
	if _, err := settings.UpdateCommunityConfig(CommunityConfig{ReportEnabled: true}); err != nil {
		t.Fatal(err)
	}
	u, err := settings.EnsureCommunitySiteURL("http://localhost:5173")
	if err != nil {
		t.Fatal(err)
	}
	if u != "http://localhost:5173" {
		t.Fatalf("got %s", u)
	}
	if settings.CommunitySiteURL("") != "http://localhost:5173" {
		t.Fatal("should persist for ticker")
	}
	payload, err := svc.buildPayload("")
	if err != nil {
		t.Fatal(err)
	}
	if payload.SiteURL != "http://localhost:5173" {
		t.Fatalf("payload site_url=%s", payload.SiteURL)
	}
}

func TestCommunityHubByOfficialHost(t *testing.T) {
	settings, svc := setupCommunityTest(t)

	// 其它域名、未开运维开关 → 拒绝
	err := svc.ReceiveHeartbeat(CommunityHeartbeatPayload{
		InstanceID: "bbbbbbbb-cccc-dddd-eeee-ffffffffffff",
		SiteURL:    "https://forum.example.org",
		SiteName:   "他站",
		Version:    "1.0.0",
	}, "127.0.0.1", "https://other.example")
	if !errors.Is(err, ErrCommunityHubDisabled) {
		t.Fatalf("want disabled for other host, got %v", err)
	}

	// 请求 Host 为官网 → 自动枢纽
	payload := CommunityHeartbeatPayload{
		InstanceID: "cccccccc-dddd-eeee-ffff-000000000001",
		SiteURL:    "https://forum.example.org",
		SiteName:   "上报站",
		Version:    "1.1.0",
		Users:      3,
		Posts:      5,
	}
	if err := svc.ReceiveHeartbeat(payload, "203.0.113.1", "https://bbs.iioio.com"); err != nil {
		t.Fatal(err)
	}
	if !settings.CommunityHubEnabled("https://www.bbs.iioio.com") {
		t.Fatal("www 前缀也应识别为官网")
	}
	if !settings.CommunityConfigForRequest("https://bbs.iioio.com").HubEnabled {
		t.Fatal("CommunityConfigForRequest should enable hub for official host")
	}

	list, err := svc.ListInstances()
	if err != nil {
		t.Fatal(err)
	}
	if len(list) != 1 {
		t.Fatalf("want 1 instance, got %d", len(list))
	}

	// 已存 ROOT_URL 为官网时，无请求上下文也应开启
	if err := settings.setString(SettingOIDCRootURL, DefaultCommunityHubURL); err != nil {
		t.Fatal(err)
	}
	if !settings.CommunityHubEnabled("") {
		t.Fatal("ROOT_URL as official should enable hub without request hint")
	}
}

func TestCommunityHubSkipsOutboundHeartbeat(t *testing.T) {
	settings, svc := setupCommunityTest(t)
	var hits atomic.Int32

	hub := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		hits.Add(1)
		w.WriteHeader(http.StatusOK)
		_, _ = w.Write([]byte(`{"ok":true}`))
	}))
	t.Cleanup(hub.Close)

	prevHub := communityHubBaseURL
	communityHubBaseURL = hub.URL
	t.Cleanup(func() { communityHubBaseURL = prevHub })

	if err := settings.setString(SettingOIDCRootURL, DefaultCommunityHubURL); err != nil {
		t.Fatal(err)
	}
	if _, err := settings.UpdateCommunityConfig(CommunityConfig{ReportEnabled: true}); err != nil {
		t.Fatal(err)
	}
	if err := svc.SendHeartbeatOnce(""); err != nil {
		t.Fatal(err)
	}
	if hits.Load() != 0 {
		t.Fatal("hub site must not send outbound heartbeat")
	}

	// 运维开关开启同样跳过
	settings2, svc2 := setupCommunityTest(t)
	settings2.SetCommunityHubEnabled(true)
	if err := settings2.setString(SettingOIDCRootURL, "http://mirror.local"); err != nil {
		t.Fatal(err)
	}
	if _, err := settings2.UpdateCommunityConfig(CommunityConfig{ReportEnabled: true}); err != nil {
		t.Fatal(err)
	}
	prevHub2 := communityHubBaseURL
	communityHubBaseURL = hub.URL
	t.Cleanup(func() { communityHubBaseURL = prevHub2 })
	if err := svc2.SendHeartbeatOnce(""); err != nil {
		t.Fatal(err)
	}
	if hits.Load() != 0 {
		t.Fatal("ops hub must not send outbound heartbeat")
	}
}

func TestHostFromURLOrHost(t *testing.T) {
	cases := []struct {
		in, want string
	}{
		{"https://bbs.iioio.com", "bbs.iioio.com"},
		{"https://www.bbs.iioio.com/", "bbs.iioio.com"},
		{"https://bbs.iioio.com:443/path", "bbs.iioio.com"},
		{"BBS.IIOIO.COM", "bbs.iioio.com"},
		{"https://other.example", "other.example"},
		{"", ""},
	}
	for _, tc := range cases {
		if got := hostFromURLOrHost(tc.in); got != tc.want {
			t.Fatalf("hostFromURLOrHost(%q)=%q want %q", tc.in, got, tc.want)
		}
	}
}

func TestAsideShowcaseEntrySync(t *testing.T) {
	settings, _ := setupCommunityTest(t)
	if settings.NavShowShowcase() || settings.FooterShowShowcase() || settings.AsideShowShowcase() {
		t.Fatal("showcase entry should be off by default")
	}
	if err := settings.SetNavShowShowcase(true); err != nil {
		t.Fatal(err)
	}
	if err := settings.SetFooterShowShowcase(true); err != nil {
		t.Fatal(err)
	}
	if err := settings.SetAsideShowcaseEnabled(true); err != nil {
		t.Fatal(err)
	}
	if !settings.NavShowShowcase() || !settings.FooterShowShowcase() || !settings.AsideShowShowcase() {
		t.Fatal("showcase entry should be enabled")
	}
	cfg := settings.Limits()
	if !cfg.NavShowShowcase || !cfg.FooterShowShowcase || !cfg.AsideShowShowcase {
		t.Fatalf("limits missing showcase flags: %+v", cfg)
	}
	found := false
	for _, w := range cfg.AsideWidgets {
		if w.ID == AsideWidgetShowcase {
			found = true
			if !w.Enabled {
				t.Fatal("aside showcase widget should be enabled")
			}
		}
	}
	if !found {
		t.Fatal("aside widgets should include showcase")
	}
}