tecamino-proccessSupervisor/htop/htopHandler.go

package htop

import (
	"embed"
	"encoding/json"
	"html/template"
	"net/http"
)

//go:embed templates
var templatesFS embed.FS

type HTopHandler struct {
	Table *HtopTable
}

func NewHTopHandler() (*HTopHandler, error) {
	table, err := NewTable()
	if err != nil {
		return nil, err
	}
	return &HTopHandler{
		Table: table,
	}, nil
}

func (h *HTopHandler) UpdateHTop(w http.ResponseWriter, r *http.Request) {
	err := h.Table.UpdateTable()
	if err != nil {
		http.Error(w, "Failed to get processes", 500)
		return
	}
	h.Table.Sort(r)

	funcMap := template.FuncMap{
		"div": divide,
	}

	w.Header().Set("Content-Type", "text/html; charset=utf-8")

	// Detect HTMX request via the HX-Request header
	if r.Header.Get("HX-Request") == "true" {
		tmpl := template.Must(
			template.New("table.html").Funcs(funcMap).ParseFS(
				templatesFS,
				"templates/table.html",
			),
		)
		tmpl.Execute(w, h.Table)
		return
	}

	tmpl := template.Must(
		template.New("htop.html").Funcs(funcMap).ParseFS(
			templatesFS,
			"templates/htop.html",
			"templates/table.html",
		),
	)
	tmpl.Execute(w, h.Table)
}

func divide(a any, b any) float64 {
	af := toFloat64(a)
	bf := toFloat64(b)
	if bf == 0 {
		return 0
	}
	return af / bf
}

func toFloat64(v any) float64 {
	switch val := v.(type) {
	case int:
		return float64(val)
	case int32:
		return float64(val)
	case int64:
		return float64(val)
	case uint64:
		return float64(val)
	case float32:
		return float64(val)
	case float64:
		return val
	default:
		return 0
	}
}

// GET /taskmanager/usage
func UsageHandler(w http.ResponseWriter, r *http.Request) {
	cpu, mem, err := GetSystemUsage()
	if err != nil {
		http.Error(w, "Failed to get usage", 500)
		return
	}

	usage := struct {
		CPU float64 `json:"cpu"`
		Mem float64 `json:"mem"`
	}{
		CPU: cpu,
		Mem: mem,
	}
	json.NewEncoder(w).Encode(usage)
}