FranklinWH-Toolbox/service/controllers/sitesAnalysisControllers.go

/*
 * Sites Analysis API List
 * /sitesAnalysis/manage/analysisList - GET
 * /sitesAnalysis/manage/config - POST 创建新配置
 * /sitesAnalysis/manage/config/{id} - GET & POST & DELETE 获取、更新、删除配置
 * /sitesAnalysis/data/{id} - GET 获取指定ID的分析数据
 */
package controllers

import (
	"fwh-toolbox-server/domain"
	"fwh-toolbox-server/services"
	"strconv"

	"github.com/gofiber/fiber/v2"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

// 获取站点分析管理名称表
// /sitesAnalysis/manage/analysisList
func GetSitesAnalysisList(c *fiber.Ctx) error {
	pageInt, _ := strconv.Atoi(c.Query("pageNum", "1"))
	sizeInt, _ := strconv.Atoi(c.Query("pageSize", "20"))

	if pageInt <= 0 || sizeInt <= 0 {
		return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
			"code":    400,
			"message": "Invalid request para",
		})
	}

	sn := c.Query("sn", "")
	progress := c.Query("progress", "")

	var snP *string = nil
	if sn != "" {
		snP = &sn
	}

	var progressP *string = nil
	if progress != "" {
		progressP = &progress
	}

	listPage, err := services.GetSiteAnalysisManageList(uint(pageInt), uint(sizeInt), snP, progressP)
	if err != nil {
		return JSONResponseNoData(c, 500, "Database read failed. (MongoDb)")
	}
	return JSONResponse(c, 0, "success", listPage)
}

// 创建一个分析配置
// /sitesAnalysis/manage/config
func CreateSitesAnalysis(c *fiber.Ctx) error {
	var body domain.SiteAnalysisConfigWrite
	if err := c.BodyParser(&body); err != nil {
		return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
			"code":    400,
			"message": "Invalid request body",
		})
	}

	if err := services.InsertSiteAnalysisConfig(body); err != nil {
		return JSONResponseNoData(c, 500, "Database write failed. (MongoDb)")
	}
	return JSONResponse(c, 0, "success", body)
}

// 获取单个分析配置
// /sitesAnalysis/manage/config/:id
func GetOneSiteAnalysisConfig(c *fiber.Ctx) error {
	id := c.Params("id")

	objID, err := primitive.ObjectIDFromHex(id)
	if err != nil {
		return JSONResponseNoData(c, 400, "Invalid request para :id")
	}

	config, err := services.GetSiteAnalysisConfig(objID)
	if err != nil {
		return JSONResponseNoData(c, 500, "Database read failed. (MongoDb)")
	}
	return JSONResponse(c, 0, "success", config)
}

// 更新一个分析配置
// /sitesAnalysis/manage/config/:id
func UpdateOneSiteAnalysisConfig(c *fiber.Ctx) error {
	id := c.Params("id")
	var body domain.SiteAnalysisConfigWrite
	if err := c.BodyParser(&body); err != nil {
		return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
			"code":    400,
			"message": "Invalid request body",
		})
	}

	objID, err := primitive.ObjectIDFromHex(id)
	if err != nil {
		return JSONResponseNoData(c, 400, "Invalid request para :id")
	}

	if err := services.UpdateSiteAnalysisConfig(objID, body); err != nil {
		return JSONResponseNoData(c, 500, "Database write failed. (MongoDb)")
	}
	return JSONResponse(c, 0, "success", body)
}

// 更新一个分析配置
// /sitesAnalysis/manage/description/:id
func UpdateOneSiteAnalysisDescription(c *fiber.Ctx) error {
	id := c.Params("id")
	var body string
	if err := c.BodyParser(&body); err != nil {
		return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
			"code":    400,
			"message": "Invalid request body",
		})
	}

	objID, err := primitive.ObjectIDFromHex(id)
	if err != nil {
		return JSONResponseNoData(c, 400, "Invalid request para :id")
	}

	if err := services.UpdateSiteAnalysisDescription(objID, body); err != nil {
		return JSONResponseNoData(c, 500, "Database write failed. (MongoDb)")
	}
	return JSONResponse(c, 0, "success", body)
}

// 删除一个分析数据
// /sitesAnalysis/manage/config/:id
func DeleteOneSiteAnalysis(c *fiber.Ctx) error {
	id := c.Params("id")

	objID, err := primitive.ObjectIDFromHex(id)
	if err != nil {
		return JSONResponseNoData(c, 400, "Invalid request para :id")
	}

	err = services.DeleteSiteAnalysisConfig(objID)
	if err != nil {
		return JSONResponseNoData(c, 500, "Database delete failed. (MongoDb)")
	}
	return JSONResponseNoData(c, 0, "success")
}

// 获取指定ID的分析数据
// /sitesAnalysis/data/:id
func GetOneSiteAnalysisData(c *fiber.Ctx) error {
	id := c.Params("id")

	objID, err := primitive.ObjectIDFromHex(id)
	if err != nil {
		return JSONResponseNoData(c, 400, "Invalid request para :id")
	}

	data, err := services.GetSiteAnalysisData(objID)
	if err != nil {
		return JSONResponseNoData(c, 500, "Database read failed. (MongoDb)")
	}
	return JSONResponse(c, 0, "success", data)
}