Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
@@ -0,0 +1,109 @@
+package ipblockmod
+
+import (
+	"fmt"
+	"time"
+
+	"91porn-server/common"
+	"91porn-server/common/db"
+	"91porn-server/common/log"
+	"91porn-server/models"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+var mdb *db.MongoDB
+
+const table = models.IpBlock
+
+func coll(t *db.MongoTool) *db.MongoTool {
+	if t == nil {
+		return mdb.Coll(table)
+	}
+	return t.Coll(table)
+}
+
+func initIndex() {
+	many := []mongo.IndexModel{ //batch set indexes   //value is the type 1 or -1
+		{
+			Keys:    bson.D{{Key: "type", Value: 1}},
+			Options: options.Index(),
+		},
+		{
+			Keys: bson.D{{Key: "ip", Value: 1}},
+		},
+		{
+			Keys: bson.D{{Key: "createdAt", Value: 1}},
+		},
+	}
+	if _, err := coll(nil).CreateIndex(many); err != nil {
+		panic(fmt.Sprintf("%s model set index err ==>[%+v]", table, err))
+	}
+}
+
+// 添加
+func Insert(l *IPBlock) (err error) {
+	l.UpdatedAt = time.Now()
+	l.CreatedAt = time.Now()
+	if _, err = coll(nil).InsertOne(l); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Insert", table, "InsertOne", err))
+		return
+	}
+	return
+}
+
+// 删除
+func DeleteOne(objID primitive.ObjectID) (err error) {
+	if _, err = coll(nil).DeleteOne(bson.M{"_id": objID}); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "DeleteOne", table, "DeleteOne", err))
+		return
+	}
+	return
+}
+
+func DeleteMany(objID []primitive.ObjectID) (err error) {
+	if _, err = coll(nil).DeleteMany(bson.M{"_id": bson.M{"$in": objID}}); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "DeleteMany", table, "DeleteMany", err), log.Any("objID", objID))
+		return
+	}
+	return
+}
+
+// Update 修改
+func Update(id primitive.ObjectID, set IPBlockEdit) (err error) {
+	set.UpdatedAt = time.Now()
+	update, _ := common.ToBsonM(&set)
+	if _, err = coll(nil).UpdateOne(bson.M{"_id": id}, bson.M{"$set": update}); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Update", table, "UpdateOne", err),
+			log.Any("id", id),
+			log.Any("set", set),
+		)
+	}
+	return
+}
+
+// FindOneByType 根据Type查询ip信息
+func FindOneByType(blockType string) (data IPBlock, err error) {
+	if err = coll(nil).FindOne(&data, bson.M{"type": blockType}); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "FindOneByType", table, "FindOne", err), log.Any("blockType", blockType))
+		return
+	}
+	return
+}
+
+func AggregateForIPArray(blockType string) (webAggre WebAggregateResp, err error) {
+	if err = coll(nil).AggregateDecode(&webAggre, []bson.M{
+		{
+			"$match": bson.M{"type": blockType},
+		},
+		{
+			"$group": bson.M{"_id": bson.M{"blockType": "$type"}, "ips": bson.M{"$addToSet": "$ip"}},
+		}}); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "AggregateForIPArray", table, "FindOne", err), log.Any("blockType", blockType))
+		return
+	}
+	return
+}
@@ -0,0 +1,29 @@
+package ipblockmod
+
+import (
+	"time"
+
+	"91porn-server/common/constant"
+	"91porn-server/common/db"
+
+	"go.mongodb.org/mongo-driver/bson/primitive"
+)
+
+// BlockAct 封禁行为
+type BlockAct string
+
+const BannedComment BlockAct = constant.BlockComment //评论 限制评论
+
+type IPBlock struct {
+	ID        primitive.ObjectID `json:"id" bson:"_id,omitempty"`
+	IP        string             `json:"ip" bson:"ip" binding:"required"`     //ip
+	Type      BlockAct           `json:"type" bson:"type" binding:"required"` //block 行为
+	Remark    string             `json:"remark" bson:"remark"`                //备注
+	UpdatedAt time.Time          `json:"updatedAt" bson:"updatedAt"`
+	CreatedAt time.Time          `json:"createdAt" bson:"createdAt"`
+}
+
+func Init() {
+	mdb = db.Init(table)
+	initIndex()
+}
@@ -0,0 +1,69 @@
+package ipblockmod
+
+import (
+	"time"
+
+	"91porn-server/common/log"
+	"91porn-server/models/commod"
+
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+type IPBlockEdit struct {
+	IP        *string   `form:"ip" json:"ip,omitempty" bson:"ip,omitempty"`                           //ip
+	Type      *BlockAct `form:"type" json:"type,omitempty"  bson:"type,omitempty" binding:"required"` //block 行为
+	Remark    string    `form:"remark" json:"remark"`                                                 //备注
+	UpdatedAt time.Time `form:"updatedAt" json:"updatedAt" bson:"updatedAt,omitempty"`
+}
+
+type ListParam struct {
+	Type string `form:"type" json:"type"`
+	commod.Page
+}
+
+type WebResp struct {
+	Total int64       `json:"total"`
+	List  interface{} `json:"list"`
+}
+
+type WebAggregateResp struct {
+	ID struct {
+		BlockType string `json:"blockType"`
+	} `json:"_id"`
+	IPS []string `json:"ips"`
+}
+
+func List(param ListParam) (webResp WebResp, err error) {
+	stdqer := commod.StdQuery{Page: &commod.PageBy{Size: param.PageSize, Num: param.PageNumber},
+		Order: &[]commod.OrderBy{{Key: "updatedAt", Desc: true}}}
+	filter := bson.M{}
+	if param.Type != "" {
+		filter["type"] = param.Type
+	}
+	ipblocks, err := find(filter, stdqer)
+	if err != nil {
+		log.Error("web ipblock list  error", log.E(err), log.Any("param", param))
+		return
+	}
+	total := count(filter)
+	webResp.List = ipblocks
+	webResp.Total = total
+	return
+}
+
+func find(filter bson.M, stdQuery commod.StdQuery) (data []IPBlock, err error) {
+	if err = coll(nil).Find(&data, filter, commod.ConvertToListQuery(stdQuery)); err != nil {
+		log.Error("models ipblock Find error", log.E(err))
+		return
+	}
+	return
+}
+
+func count(query bson.M) (total int64) {
+	total, err := coll(nil).Count(query)
+	if err != nil {
+		log.Error("models ipblock  Count error", log.E(err))
+		return
+	}
+	return
+}