feat: add Grayscale test of ai summary

Author: Crnaneo

apis/hole/apis.go

@@ -5,6 +5,7 @@ import (
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
+	"hash/fnv"
 	"io"
 	"math/rand"
 	"net/http"
@@ -955,10 +956,14 @@ func DeleteHole(c *fiber.Ctx) error {
 func GenerateSummary(c *fiber.Ctx) error {
 	uid, _ := common.GetUserID(c)
 	if config.Config.WhiteListUserIds != nil && !slices.Contains(config.Config.WhiteListUserIds, uid) {
-		return c.Status(404).JSON(fiber.Map{
-			"code":    404,
-			"message": "Cannot GET " + c.Path(),
-		})
+		h := fnv.New32a()
+		h.Write([]byte(strconv.Itoa(uid)))
+		if int(h.Sum32())%100 >= int(100*config.Config.WhiteListRate) {
+			return c.Status(404).JSON(fiber.Map{
+				"code":    404,
+				"message": "Cannot GET " + c.Path(),
+			})
+		}
 	}
 	id, _ := c.ParamsInt("id")
 	forceRefresh := c.QueryBool("force_refresh")

config/config.go

@@ -64,6 +64,7 @@ var Config struct {
 	SummaryReplyBoundary3        int      `env:"SUMMARY_REPLY_BOUNDARY_1" envDefault:"1000"`
 	SummaryLogLimit              int      `env:"SUMMARY_LOG_LIMIT" envDefault:"1000"`
 	WhiteListUserIds             []int    `env:"WHITE_LIST_USER_IDS"`
+	WhiteListRate                float32  `env:"WHITE_LIST_RATE" envDefault:"1"`
 	MaxSummaryFloors             int      `env:"MAX_FLOORS_PER_HOLE" envDefault:"50"`
 }
 

models/hole.go

@@ -2,6 +2,7 @@ package models
 
 import (
 	"fmt"
+	"strconv"
 	"time"
 
 	"golang.org/x/exp/maps"
@@ -13,6 +14,7 @@ import (
 	"gorm.io/gorm/clause"
 	"gorm.io/plugin/dbresolver"
 
+	"hash/fnv"
 	"treehole_next/config"
 	"treehole_next/utils"
 )
@@ -293,6 +295,13 @@ func (holes Holes) Preprocess(c *fiber.Ctx) error {
 
 		// for users in whitelist or whitelist is empty, AISummaryAvailable is true,
 		hole.AISummaryAvailable = config.Config.WhiteListUserIds == nil || slices.Contains(config.Config.WhiteListUserIds, uid)
+		if !hole.AISummaryAvailable {
+			h := fnv.New32a()
+			h.Write([]byte(strconv.Itoa(uid)))
+			if int(h.Sum32())%100 < int(100*config.Config.WhiteListRate) {
+				hole.AISummaryAvailable = true
+			}
+		}
 		hole.AISummaryAvailable = hole.AISummaryAvailable && !(hole.Locked || hole.Hidden || hole.Frozen)
 		for _, tag := range hole.Tags {
 			if len(tag.Name) > 0 && tag.Name[0] == '*' {