FranklinWH-Toolbox/service/repositories/mongo/sitesAnalysisRepo.go

package mongo

import (
	"context"
	"fwh-toolbox-server/database"
	"fwh-toolbox-server/domain"
	"fwh-toolbox-server/dto"
	"time"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.mongodb.org/mongo-driver/mongo/options"
)

// 从数据库获取站点分析列表
func GetSiteAnalysisManageList(page, size uint, sn, progress, trigger *string) (*dto.ResPage[domain.SiteAnalysisInfo], error) {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	collection := database.GetCollection("SiteAnalysisConfig")

	filter := bson.M{}
	// 条件
	if sn != nil {
		filter["sn"] = *sn
	}
	if progress != nil {
		filter["progress"] = *progress
	}
	if trigger != nil {
		filter["trigger"] = *trigger
	}

	findOptions := options.Find().
		SetProjection(bson.M{
			"_id":         1,
			"sn":          1,
			"description": 1,
			"item":        1,
			"progress":    1,
			"trigger":     1,
		}).
		SetSkip(int64((page - 1) * size)).
		SetLimit(int64(size))

	cursor, err := collection.Find(ctx, filter, findOptions)
	if err != nil {
		return nil, err
	}
	defer cursor.Close(ctx)

	var configList []domain.SiteAnalysisInfo
	for cursor.Next(ctx) {
		var cfg domain.SiteAnalysisInfo
		if err := cursor.Decode(&cfg); err != nil {
			return nil, err
		}
		configList = append(configList, cfg)
	}

	// 查询总数
	total, err := collection.CountDocuments(ctx, filter)
	if err != nil {
		return nil, err
	}

	res := dto.ResPage[domain.SiteAnalysisInfo]{
		Data:     configList,
		PageSize: size,
		PageNum:  page,
		Total:    uint(total),
	}

	return &res, nil
}

// 从数据库获取一条站点分析配置
func GetSiteAnalysisConfig(id primitive.ObjectID) (*domain.SiteAnalysisConfig, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	collection := database.GetCollection("SiteAnalysisConfig")
	var cfg domain.SiteAnalysisConfig
	err := collection.FindOne(ctx, bson.M{"_id": id}).Decode(&cfg)
	if err != nil {
		return nil, err
	}

	return &cfg, nil
}

// 插入一条站点分析配置到数据库
// auto - 是否是自动触发数据
func InsertSiteAnalysisConfig(writeConfig domain.SiteAnalysisConfigWrite, auto bool) (primitive.ObjectID, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	trigger := domain.TriggerManual
	if auto {
		trigger = domain.TriggerAuto
	}
	config := domain.SiteAnalysisConfig{
		Id:          writeConfig.Id,
		Sn:          writeConfig.Sn,
		Description: writeConfig.Description,
		TimePeriod:  writeConfig.TimePeriod,
		Options:     writeConfig.Options,
		Item:        []string{},
		Progress:    domain.ProgressPending,
		Trigger:     trigger,
	}

	// 生成Item标签
	if writeConfig.Options.EnableTimeline {
		config.Item = append(config.Item, "时间线")
	}
	if writeConfig.Options.EnableRuntime {
		config.Item = append(config.Item, "运行时")
	}

	collection := database.GetCollection("SiteAnalysisConfig")

	res, err := collection.InsertOne(ctx, config)
	if err != nil {
		return primitive.ObjectID{}, err
	}

	return res.InsertedID.(primitive.ObjectID), nil
}

// 更新一条站点分析配置到数据库
func UpdateSiteAnalysisConfig(id primitive.ObjectID, writeConfig domain.SiteAnalysisConfigWrite) error {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	config := domain.SiteAnalysisConfig{
		Id:          writeConfig.Id,
		Sn:          writeConfig.Sn,
		Description: writeConfig.Description,
		Options:     writeConfig.Options,
		Item:        []string{},
		Progress:    domain.ProgressPending,
	}

	// 生成Item标签
	if writeConfig.Options.EnableTimeline {
		config.Item = append(config.Item, "时间线")
	}
	if writeConfig.Options.EnableRuntime {
		config.Item = append(config.Item, "运行时")
	}

	collection := database.GetCollection("SiteAnalysisConfig")

	// _, err := collection.InsertOne(ctx, config)
	filter := bson.M{"_id": id} // 按ID匹配
	update := bson.M{
		// 如果存在就更新全部字段
		"$set": config,
	}

	opts := options.Update().SetUpsert(true)

	_, err := collection.UpdateOne(ctx, filter, update, opts)
	if err != nil {
		return err
	}

	return nil
}

// 更新一条站点分析描述到数据库
func UpdateSiteAnalysisDescription(id primitive.ObjectID, desc string) error {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	collection := database.GetCollection("SiteAnalysisConfig")

	// _, err := collection.InsertOne(ctx, config)
	filter := bson.M{"_id": id} // 按ID匹配
	update := bson.M{
		// 如果存在就更新全部字段
		"$set": bson.M{
			"description": desc,
		},
	}

	opts := options.Update().SetUpsert(true)

	_, err := collection.UpdateOne(ctx, filter, update, opts)
	if err != nil {
		return err
	}

	return nil
}

// 更新一条站点分析结果到数据库
func UpdateSiteAnalysisResult(id primitive.ObjectID, result domain.Progress) error {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	collection := database.GetCollection("SiteAnalysisConfig")

	// _, err := collection.InsertOne(ctx, config)
	filter := bson.M{"_id": id} // 按ID匹配
	update := bson.M{
		// 如果存在就更新非progress字段
		"$set": bson.M{
			"progress": result,
		},
	}

	opts := options.Update().SetUpsert(true)

	_, err := collection.UpdateOne(ctx, filter, update, opts)
	if err != nil {
		return err
	}

	return nil
}

// 删除一条站点分析配置
func DeleteSiteAnalysisConfig(id primitive.ObjectID) error {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	filter := bson.M{"_id": id} // 删除条件
	collection := database.GetCollection("SiteAnalysisConfig")
	_, err := collection.DeleteOne(ctx, filter)
	if err != nil {
		return err
	}

	return nil
}

// 插入完整分析结果数据
func InsertSiteAnalysisData(id primitive.ObjectID, data domain.SiteAnalysisData) error {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	collection := database.GetCollection("SiteAnalysisConfig")

	// _, err := collection.InsertOne(ctx, config)
	filter := bson.M{"_id": id} // 按ID匹配
	update := bson.M{
		// 如果存在就更新非progress字段
		"$set": bson.M{
			"data": data,
		},
	}

	opts := options.Update().SetUpsert(true)

	_, err := collection.UpdateOne(ctx, filter, update, opts)
	if err != nil {
		return err
	}

	return nil
}

// 插入分析结果timeline数据
func InsertSiteAnalysisTimeline(id primitive.ObjectID, timeline []domain.TimelineItem) error {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	collection := database.GetCollection("SiteAnalysisConfig")

	// _, err := collection.InsertOne(ctx, config)
	filter := bson.M{"_id": id} // 按ID匹配
	update := bson.M{
		// 如果存在就更新非progress字段
		"$set": bson.M{
			"data.timeline": timeline,
		},
	}

	opts := options.Update().SetUpsert(true)

	_, err := collection.UpdateOne(ctx, filter, update, opts)
	if err != nil {
		return err
	}

	return nil
}

// 插入分析结果runtime数据
func InsertSiteAnalysisRuntime(id primitive.ObjectID, runtime []domain.RuntimeItem) error {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	collection := database.GetCollection("SiteAnalysisConfig")

	// _, err := collection.InsertOne(ctx, config)
	filter := bson.M{"_id": id} // 按ID匹配
	update := bson.M{
		// 如果存在就更新非progress字段
		"$set": bson.M{
			"data.runtime": runtime,
		},
	}

	opts := options.Update().SetUpsert(true)

	_, err := collection.UpdateOne(ctx, filter, update, opts)
	if err != nil {
		return err
	}

	return nil
}