102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package connector
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func (tc *TsConnector) TcServiceCall(ctx *gin.Context) {
|
|
var jsonData struct {
|
|
ServiceName string `json:"service"`
|
|
Command string `json:"command"`
|
|
Args string `json:"args"`
|
|
}
|
|
|
|
value, exists := ctx.Get("username")
|
|
if !exists {
|
|
ctx.JSON(http.StatusOK, gin.H{"message": "Server error: username not found in context", "ok": false})
|
|
return
|
|
}
|
|
|
|
username, ok := value.(string)
|
|
if !ok {
|
|
ctx.JSON(http.StatusOK, gin.H{"message": "Server error: invalid username type in context", "ok": false})
|
|
return
|
|
}
|
|
|
|
err := ctx.ShouldBindJSON(&jsonData)
|
|
if err != nil {
|
|
ctx.JSON(http.StatusOK, gin.H{"message": "error", "error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if jsonData.ServiceName == "" {
|
|
ctx.JSON(http.StatusOK, gin.H{"message": "error", "error": "service_name is required"})
|
|
return
|
|
}
|
|
|
|
go tc.teamserver.TsServiceCall(jsonData.ServiceName, username, jsonData.Command, jsonData.Args)
|
|
|
|
ctx.JSON(http.StatusOK, gin.H{"message": "success", "result": "ok"})
|
|
}
|
|
|
|
func (tc *TsConnector) TcServiceList(c *gin.Context) {
|
|
services, err := tc.teamserver.TsServiceList()
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"message": "error", "error": err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, gin.H{"message": "success", "services": services})
|
|
}
|
|
|
|
func (tc *TsConnector) TcServiceLoad(c *gin.Context) {
|
|
var jsonData struct {
|
|
ConfigPath string `json:"config_path"`
|
|
}
|
|
|
|
err := c.ShouldBindJSON(&jsonData)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"message": "error", "error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if jsonData.ConfigPath == "" {
|
|
c.JSON(http.StatusOK, gin.H{"message": "error", "error": "config_path is required"})
|
|
return
|
|
}
|
|
|
|
err = tc.teamserver.TsServiceLoad(jsonData.ConfigPath)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"message": "error", "error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "success"})
|
|
}
|
|
|
|
func (tc *TsConnector) TcServiceUnload(c *gin.Context) {
|
|
var jsonData struct {
|
|
ServiceName string `json:"service"`
|
|
}
|
|
|
|
err := c.ShouldBindJSON(&jsonData)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"message": "error", "error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if jsonData.ServiceName == "" {
|
|
c.JSON(http.StatusOK, gin.H{"message": "error", "error": "service is required"})
|
|
return
|
|
}
|
|
|
|
err = tc.teamserver.TsServiceUnload(jsonData.ServiceName)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, gin.H{"message": "error", "error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "success"})
|
|
}
|