mirror of
https://github.com/gazer-x/komari.git
synced 2026-06-22 00:05:52 +08:00
Akizon77
96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
package admin
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gorilla/websocket"
|
|
"github.com/komari-monitor/komari/api"
|
|
"github.com/komari-monitor/komari/database/auditlog"
|
|
"github.com/komari-monitor/komari/database/models"
|
|
"github.com/komari-monitor/komari/database/tasks"
|
|
"github.com/komari-monitor/komari/utils"
|
|
"github.com/komari-monitor/komari/ws"
|
|
)
|
|
|
|
// 接受数据类型：
|
|
// - command: string
|
|
// - clients: []string (客户端 UUID 列表)
|
|
func Exec(c *gin.Context) {
|
|
var req struct {
|
|
Command string `json:"command" binding:"required"`
|
|
Clients []string `json:"clients" binding:"required"`
|
|
}
|
|
var onlineClients []string
|
|
var offlineClients []string
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
api.RespondError(c, 400, "Invalid or missing request body: "+err.Error())
|
|
return
|
|
}
|
|
// for uuid := range ws.GetConnectedClients() {
|
|
// if contain(req.Clients, uuid) {
|
|
// onlineClients = append(onlineClients, uuid)
|
|
// }
|
|
// // else {
|
|
// // api.RespondError(c, 400, "Client not connected: "+uuid)
|
|
// // return
|
|
// // }
|
|
// }
|
|
for _, uuid := range req.Clients {
|
|
if client := ws.GetConnectedClients()[uuid]; client != nil {
|
|
onlineClients = append(onlineClients, uuid)
|
|
} else {
|
|
offlineClients = append(offlineClients, uuid)
|
|
}
|
|
}
|
|
if len(onlineClients) == 0 {
|
|
api.RespondError(c, 400, "No clients connected")
|
|
return
|
|
}
|
|
taskId := utils.GenerateRandomString(16)
|
|
if err := tasks.CreateTask(taskId, append(onlineClients, offlineClients...), req.Command); err != nil {
|
|
api.RespondError(c, 500, "Failed to create task: "+err.Error())
|
|
return
|
|
}
|
|
for _, uuid := range onlineClients {
|
|
var send struct {
|
|
Message string `json:"message"`
|
|
Command string `json:"command"`
|
|
TaskId string `json:"task_id"`
|
|
}
|
|
send.Message = "exec"
|
|
send.Command = req.Command
|
|
send.TaskId = taskId
|
|
|
|
payload, _ := json.Marshal(send)
|
|
client := ws.GetConnectedClients()[uuid]
|
|
if client != nil {
|
|
client.WriteMessage(websocket.TextMessage, payload)
|
|
} else {
|
|
api.RespondError(c, 400, "Client connection is null: "+uuid)
|
|
return
|
|
}
|
|
}
|
|
uuid, _ := c.Get("uuid")
|
|
auditlog.Log(c.ClientIP(), uuid.(string), "REC, task id: "+taskId, "warn")
|
|
api.RespondSuccess(c, gin.H{
|
|
"task_id": taskId,
|
|
"clients": onlineClients,
|
|
})
|
|
if len(offlineClients) > 0 {
|
|
for _, uuid := range offlineClients {
|
|
tasks.SaveTaskResult(taskId, uuid, "Client offline!", -1, models.FromTime(time.Now()))
|
|
}
|
|
}
|
|
}
|
|
|
|
// func contain(clients []string, uuid string) bool {
|
|
// for _, client := range clients {
|
|
// if client == uuid {
|
|
// return true
|
|
// }
|
|
// }
|
|
// return false
|
|
// }
|