videoplayer/handler/shell.go

package handler

import (
	"github.com/gin-gonic/gin"
	"videoplayer/proto"
	"videoplayer/service"
)

type ShellHandler struct {
}

type CreateShellReq struct {
	ShellName    string `json:"shell_name"`
	ShellContent string `json:"shell_content"`
	Server       string `json:"server"`
}
type UpdateShellReq struct {
	ID           uint   `json:"id"`
	ShellName    string `json:"shell_name"`
	ShellContent string `json:"shell_content"`
	Server       string `json:"server"`
	Status       int    `json:"status"`
	ShellResult  string `json:"shell_result"`
}

type UpdateShellReqV2 struct {
	Shells []UpdateShellReq `json:"shells"`
}

type UpdateShellResp struct {
	ID     uint `json:"id" form:"id"`
	Status int  `json:"status" form:"status"`
}

func SetUpShellGroup(router *gin.Engine) {
	shellGroup := router.Group("/shell") //持续集成、部署
	shellHandler := ShellHandler{}
	shellGroup.POST("/create", shellHandler.CreateShell)
	shellGroup.POST("/list", shellHandler.ListShell)
	shellGroup.POST("/update", shellHandler.UpdateShell)
	shellGroup.POST("/server_will_run_list", shellHandler.ServerWillRun)
}

func (s *ShellHandler) CreateShell(c *gin.Context) {
	user_id, _ := c.Get("id")
	uid := int(user_id.(float64))
	var req CreateShellReq
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(200, gin.H{"code": proto.ShellCreateFailed, "message": "参数错误", "data": err.Error()})
	} else {
		id := service.CreateShell(req.ShellName, req.ShellContent, req.Server, uid)
		if id == 0 {
			c.JSON(200, gin.H{"code": proto.ShellCreateFailed, "message": "创建失败,id is 0", "data": ""})
			return
		}
		c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "创建成功", "data": id})
	}
}

func (s *ShellHandler) ListShell(c *gin.Context) {
	userId, _ := c.Get("id")
	id := int(userId.(float64))
	shells := service.FindShellByAuthID(id)
	c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "获取成功", "data": shells})
}

func (s *ShellHandler) UpdateShell(c *gin.Context) {
	var req UpdateShellReqV2
	userId, _ := c.Get("id")
	id := int(userId.(float64))
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(200, gin.H{"code": proto.ShellUpdateFailed, "message": "参数错误", "data": err.Error()})
	} else {
		var resp []UpdateShellResp
		for _, v := range req.Shells {
			if service.UpdateShellByID(v.ID, uint(id), v.ShellName, v.ShellContent, v.Server, v.Status, v.ShellResult) {
				resp = append(resp, UpdateShellResp{ID: v.ID, Status: v.Status})
			} else {
				resp = append(resp, UpdateShellResp{ID: v.ID, Status: -1})
			}
		}
		c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "更新成功", "data": resp})
	}
}

type ServerWillRunReq struct {
	Server string `json:"server"`
}

func (s *ShellHandler) ServerWillRun(c *gin.Context) {
	userId, _ := c.Get("id")
	id := int(userId.(float64))
	var req ServerWillRunReq
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(200, gin.H{"code": proto.ShellUpdateFailed, "message": "参数错误", "data": err.Error()})
	} else {
		willRunShells, err2 := service.FindShellWillRunByServer(req.Server, id)
		if err2 != nil {
			c.JSON(200, gin.H{"code": proto.ShellUpdateFailed, "message": "获取失败", "data": err2.Error()})
			return
		}
		c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "获取成功", "data": willRunShells})
	}
}