mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
// Package aifalse 暂时只有服务器监控
|
|
package aifalse
|
|
|
|
import (
|
|
"math"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/shirou/gopsutil/cpu"
|
|
"github.com/shirou/gopsutil/disk"
|
|
"github.com/shirou/gopsutil/mem"
|
|
|
|
"github.com/FloatTech/ZeroBot-Plugin/control"
|
|
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
)
|
|
|
|
func init() { // 插件主体
|
|
engine := control.Register("aifalse", &control.Options{
|
|
DisableOnDefault: false,
|
|
Help: "AIfalse\n" +
|
|
"- 查询计算机当前活跃度[检查身体|自检|启动自检|系统状态",
|
|
})
|
|
engine.OnFullMatchGroup([]string{"检查身体", "自检", "启动自检", "系统状态"}, zero.AdminPermission).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
ctx.SendChain(message.Text(
|
|
"* CPU占用率: ", cpuPercent(), "%\n",
|
|
"* RAM占用率: ", memPercent(), "%\n",
|
|
"* 硬盘活动率: ", diskPercent(), "%",
|
|
),
|
|
)
|
|
})
|
|
engine.OnFullMatch("清理缓存", zero.SuperUserPermission).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
err := os.RemoveAll("data/cache/*")
|
|
if err != nil {
|
|
ctx.Send("错误: " + err.Error())
|
|
} else {
|
|
ctx.Send("成功!")
|
|
}
|
|
})
|
|
}
|
|
|
|
func cpuPercent() float64 {
|
|
percent, _ := cpu.Percent(time.Second, false)
|
|
return math.Round(percent[0])
|
|
}
|
|
|
|
func memPercent() float64 {
|
|
memInfo, _ := mem.VirtualMemory()
|
|
return math.Round(memInfo.UsedPercent)
|
|
}
|
|
|
|
func diskPercent() float64 {
|
|
parts, _ := disk.Partitions(true)
|
|
diskInfo, _ := disk.Usage(parts[0].Mountpoint)
|
|
return math.Round(diskInfo.UsedPercent)
|
|
}
|