rts-sim-testing-service/api/drafting.go

package api

import (
	"net/http"
	"strconv"

	jwt "github.com/appleboy/gin-jwt/v2"
	"github.com/gin-gonic/gin"
	"go.uber.org/zap"
	"google.golang.org/protobuf/proto"
	"joylink.club/bj-rtsts-server/ats/verify/protos/graphicData"
	"joylink.club/bj-rtsts-server/ats/verify/simulation/wayside/memory"
	"joylink.club/bj-rtsts-server/db/model"
	"joylink.club/bj-rtsts-server/dto"
	"joylink.club/bj-rtsts-server/middleware"
	"joylink.club/bj-rtsts-server/service"
)

func InitDraftingRouter(api *gin.RouterGroup, authMiddleware *jwt.GinJWTMiddleware) {
	authed := api.Group("/v1/drafting").Use(authMiddleware.MiddlewareFunc(), middleware.PermissMiddleware)
	authed.GET("/paging", pageQueryDrafting)
	authed.POST("", createDrafting)
	authed.POST("/:id/saveAs", saveAsDrafting)
	authed.GET("/:id", queryDraftingInfo)
	authed.PUT("/:id", updateDraftingInfo)
	authed.DELETE("/:id", deleteDrafting)
	authed.POST("/calculatelink", generateCalculateLinkData)
}

// 分页查询草稿
//
//	@Summary		分页查询草稿
//
//	@Security		JwtAuth
//
//	@Description	可以通过草稿名称过滤，分页查询草稿
//	@Tags			草稿Api
//	@Accept			json
//	@Produce		json
//	@Param			PageDraftingReqDto	query		dto.PageDraftingReqDto	true	"草稿查询条件带分页信息"
//	@Success		200	{object}	dto.PageDto
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/drafting/paging [get]
func pageQueryDrafting(c *gin.Context) {
	user, _ := c.Get(middleware.IdentityKey)
	zap.S().Debug("分页查询草稿", user)
	req := dto.PageDraftingReqDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
	}
	zap.S().Debug("分页查草稿参数", req)
	page, err := service.PageDraftingQuery(&req)
	if err != nil {
		panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
	}
	c.JSON(http.StatusOK, page)
}

// 创建草稿
//
//	@Summary		创建草稿
//
//	@Security		JwtAuth
//
//	@Description	创建草稿数据
//	@Tags			草稿Api
//	@Accept			json
//	@Produce		json
//	@Param			draftingDto	query	dto.DraftingDto	true	"创建的草稿信息"
//	@Success		200	{object}	nil
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/drafting [post]
func createDrafting(c *gin.Context) {
	user, _ := c.Get(middleware.IdentityKey)
	createId := user.(*model.User).ID
	req := dto.DraftingDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
	}
	zap.S().Debug("保存数据", req)
	data, err := service.CreateDrafting(createId, &req)
	if err != nil {
		panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
	}
	c.JSON(http.StatusOK, data)
}

// 草稿另存为
//
//	@Summary		草稿另存为
//
//	@Security		JwtAuth
//
//	@Description	草稿另存为
//	@Tags			草稿Api
//	@Accept			json
//	@Produce		json
//	@Param			id	path		int	true	"源草稿id"
//	@Param			draftingDto	query	dto.DraftingDto	true	"另存为草稿信息"
//	@Success		200	{object}	nil
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/drafting/{id}/saveAs [post]
func saveAsDrafting(c *gin.Context) {
	id, exist := c.Params.Get("id")
	if !exist {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: "必要参数id不存在"})
	}
	zap.S().Debug("传入参数id为" + id)
	user, _ := c.Get(middleware.IdentityKey)
	createrId := user.(*model.User).ID
	req := dto.DraftingDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
	}
	int64Id, _ := strconv.ParseInt(id, 10, 64)
	newData, err := service.SaveAsDrafting(createrId, int32(int64Id), &req)
	if err != nil {
		panic(dto.ErrorDto{Code: dto.DataOperationError, Message: err.Error()})
	}
	c.JSON(http.StatusOK, newData)
}

// 查询草稿详情
//
//	@Summary		查询草稿详情
//
//	@Security		JwtAuth
//
//	@Description	查询草稿详情
//	@Tags			草稿Api
//	@Accept			json
//	@Produce		json
//	@Param			id	path		int	true	"草稿ID"
//	@Success		200	{object}    model.Drafting
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/drafting/{id} [get]
func queryDraftingInfo(c *gin.Context) {
	id, exist := c.Params.Get("id")
	if !exist {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: "必要参数id不存在"})
	}
	zap.S().Debug("传入参数id为" + id)
	int64Id, _ := strconv.ParseInt(id, 10, 64)
	c.JSON(http.StatusOK, service.QueryDrafting(int32(int64Id)))
}

// 修改草稿信息
//
//	@Summary		修改草稿信息
//
//	@Security		JwtAuth
//
//	@Description	修改草稿信息
//	@Tags			草稿Api
//	@Accept			json
//	@Produce		json
//	@Param			id	path		int	true	"草稿ID"
//	@Param			draftingDto	query	dto.DraftingDto	true	"修改的草稿信息"
//	@Success		200	{object}	nil
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/drafting/{id} [put]
func updateDraftingInfo(c *gin.Context) {
	id, exist := c.Params.Get("id")
	if !exist {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: "必要参数id不存在"})
	}
	zap.S().Debug("传入参数id为" + id)
	req := dto.DraftingDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
	}
	int64Id, _ := strconv.ParseInt(id, 10, 64)
	result := service.UpdateDrafting(int32(int64Id), &req)
	if !result {
		panic(dto.ErrorDto{Code: dto.DataOperationError, Message: "保存出错"})
	}
	c.JSON(http.StatusOK, result)
}

// 删除草稿数据
//
//	@Summary		删除草稿数据
//
//	@Security		JwtAuth
//
//	@Description	删除草稿数据
//	@Tags			草稿Api
//	@Accept			json
//	@Produce		json
//	@Param			id	path		int	true	"草稿ID"
//	@Success		200	{object}	nil
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/drafting/{id} [delete]
func deleteDrafting(c *gin.Context) {
	user, _ := c.Get(middleware.IdentityKey)
	zap.S().Debug("id删除草稿的图形数据", user)
	idStr := c.Param("id")
	id, err := strconv.Atoi(idStr)
	if err != nil {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
	}
	zap.S().Debug("id查询草稿的图形数据", id)
	service.DeleteDraftingById(id)
	c.JSON(http.StatusOK, true)
}

// 根据地图数据新生成计算的link信息
//
//	@Summary		根据地图数据新生成计算的link信息
//
//	@Security		JwtAuth
//
//	@Description	根据地图数据新生成计算的link信息
//	@Tags			草稿Api
//	@Accept			json
//	@Produce		json
//	@Param			DraftingMapDataDto	query	dto.DraftingMapDataDto	true	"地图信息"
//	@Success		200	{object}	nil
//	@Failure		401	{object}	dto.ErrorDto
//	@Failure		404	{object}	dto.ErrorDto
//	@Failure		500	{object}	dto.ErrorDto
//	@Router			/api/v1/drafting/calculatelink [post]
func generateCalculateLinkData(c *gin.Context) {
	req := dto.DraftingMapDataDto{}
	if err := c.ShouldBind(&req); err != nil {
		panic(dto.ErrorDto{Code: dto.ArgumentParseError, Message: err.Error()})
	}
	gd := &graphicData.RtssGraphicStorage{}
	proto.Unmarshal(req.Proto, gd)
	c.JSON(http.StatusOK, memory.BuildCalculateLinkData(gd))
}