125 lines
2.8 KiB
Go
125 lines
2.8 KiB
Go
package controllers
|
||
|
||
import (
|
||
"net/http"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/shirou/gopsutil/cpu"
|
||
"github.com/shirou/gopsutil/disk"
|
||
"github.com/shirou/gopsutil/mem"
|
||
)
|
||
|
||
type FolderStat struct {
|
||
Name string `json:"Name"`
|
||
Path string `json:"Path"`
|
||
Size int64 `json:"Size"`
|
||
IsDir bool `json:"IsDir"`
|
||
Children []FolderStat `json:"Children,omitempty"`
|
||
}
|
||
|
||
type SystemStats struct {
|
||
CPUUsage float64 `json:"CPUUsage"`
|
||
TotalRAM uint64 `json:"TotalRAM"`
|
||
UsedRAM uint64 `json:"UsedRAM"`
|
||
FreeRAM uint64 `json:"FreeRAM"`
|
||
UsedRAMPct float64 `json:"UsedRAMPct"`
|
||
DiskTotal uint64 `json:"DiskTotal"`
|
||
DiskFree uint64 `json:"DiskFree"`
|
||
DiskUsed uint64 `json:"DiskUsed"`
|
||
DiskUsedPct float64 `json:"DiskUsedPct"`
|
||
}
|
||
|
||
// calcule récursivement la taille d’un dossier
|
||
func getFolderStats(absPath, relPath string) (FolderStat, error) {
|
||
info, err := os.Stat(absPath)
|
||
if err != nil {
|
||
return FolderStat{}, err
|
||
}
|
||
|
||
stat := FolderStat{
|
||
Name: info.Name(),
|
||
Path: relPath,
|
||
IsDir: info.IsDir(),
|
||
Size: 0,
|
||
}
|
||
|
||
if !info.IsDir() {
|
||
stat.Size = info.Size()
|
||
return stat, nil
|
||
}
|
||
|
||
entries, err := os.ReadDir(absPath)
|
||
if err != nil {
|
||
return stat, err
|
||
}
|
||
|
||
var total int64
|
||
for _, entry := range entries {
|
||
childAbs := filepath.Join(absPath, entry.Name())
|
||
childRel := filepath.Join(relPath, entry.Name())
|
||
childStat, err := getFolderStats(childAbs, childRel)
|
||
if err != nil {
|
||
continue
|
||
}
|
||
stat.Children = append(stat.Children, childStat)
|
||
total += childStat.Size
|
||
}
|
||
stat.Size = total
|
||
|
||
return stat, nil
|
||
}
|
||
|
||
// Handler principal
|
||
func FolderStatsHandler() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
base := "upload"
|
||
rel := c.Query("path")
|
||
if rel == "" {
|
||
rel = ""
|
||
}
|
||
rel = filepath.Clean("/" + rel)
|
||
rel = strings.TrimPrefix(rel, "/")
|
||
absPath := filepath.Join(base, rel)
|
||
|
||
baseAbs, _ := filepath.Abs(base)
|
||
absPath, _ = filepath.Abs(absPath)
|
||
if !strings.HasPrefix(absPath, baseAbs) {
|
||
c.JSON(http.StatusForbidden, gin.H{"error": "Chemin invalide"})
|
||
return
|
||
}
|
||
|
||
// stats dossier
|
||
folderStats, err := getFolderStats(absPath, rel)
|
||
if err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Impossible de calculer les stats"})
|
||
return
|
||
}
|
||
|
||
// stats système
|
||
cpuPercent, _ := cpu.Percent(0, false)
|
||
memStats, _ := mem.VirtualMemory()
|
||
diskStats, _ := disk.Usage(baseAbs)
|
||
|
||
sysStats := SystemStats{
|
||
CPUUsage: cpuPercent[0],
|
||
TotalRAM: memStats.Total,
|
||
UsedRAM: memStats.Used,
|
||
FreeRAM: memStats.Available,
|
||
UsedRAMPct: memStats.UsedPercent,
|
||
DiskTotal: diskStats.Total,
|
||
DiskFree: diskStats.Free,
|
||
DiskUsed: diskStats.Used,
|
||
DiskUsedPct: diskStats.UsedPercent,
|
||
}
|
||
|
||
// réponse
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"folderStats": folderStats,
|
||
"systemStats": sysStats,
|
||
})
|
||
}
|
||
}
|