package trace

import (
	"be.ems/src/framework/logger"
	"be.ems/src/framework/middleware"
	"be.ems/src/framework/middleware/collectlogs"
	"be.ems/src/modules/trace/controller"

	"github.com/gin-gonic/gin"
)

// 模块路由注册
func Setup(router *gin.Engine) {
	logger.Infof("开始加载 ====> trace 模块路由")

	traceGroup := router.Group("/trace")

	// 信令抓包
	tcpdumpGroup := traceGroup.Group("/tcpdump")
	{
		tcpdumpGroup.POST("/start",
			middleware.PreAuthorize(nil),
			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.tcpdump", collectlogs.BUSINESS_TYPE_OTHER)),
			controller.NewTcpdump.DumpStart,
		)
		tcpdumpGroup.POST("/stop",
			middleware.PreAuthorize(nil),
			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.tcpdump", collectlogs.BUSINESS_TYPE_OTHER)),
			controller.NewTcpdump.DumpStop,
		)
		tcpdumpGroup.GET("/download",
			middleware.PreAuthorize(nil),
			controller.NewTcpdump.DumpDownload,
		)
		tcpdumpGroup.POST("/upf",
			middleware.PreAuthorize(nil),
			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.tcpdump", collectlogs.BUSINESS_TYPE_OTHER)),
			controller.NewTcpdump.UPFTrace,
		)
	}

	// 跟踪任务 网元HLR
	taskHLRGroup := traceGroup.Group("/task/hlr")
	{
		taskHLRGroup.GET("/list",
			middleware.PreAuthorize(nil),
			controller.NewTraceTaskHlr.List,
		)
		taskHLRGroup.DELETE("/:ids",
			middleware.PreAuthorize(nil),
			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_DELETE)),
			controller.NewTraceTaskHlr.Remove,
		)
		taskHLRGroup.POST("/start",
			middleware.PreAuthorize(nil),
			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_INSERT)),
			controller.NewTraceTaskHlr.Start,
		)
		taskHLRGroup.POST("/stop",
			middleware.PreAuthorize(nil),
			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_UPDATE)),
			controller.NewTraceTaskHlr.Stop,
		)
		taskHLRGroup.POST("/file",
			middleware.PreAuthorize(nil),
			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_OTHER)),
			controller.NewTraceTaskHlr.File,
		)
	}
}