package worker

import (
	"context"

	"github.com/mackerelio/go-osstat/cpu"
	"github.com/mackerelio/go-osstat/memory"

	"git.lastassault.de/speatzle/morffix/rpc"
	"git.lastassault.de/speatzle/morffix/types"
)

func init() {
	rpcServer.RegisterMethod("status", status)
}

func status(ctx context.Context, req rpc.Request) (any, error) {
	s := types.WorkerStatus{}
	cStats, _ := cpu.Get()
	if cStats != nil {
		s.CPUUsage = cStats.Total
		s.CPUCount = uint64(cStats.CPUCount)
	}
	mStats, _ := memory.Get()
	if mStats != nil {
		s.MemoryUsage = mStats.Used
		s.MemoryTotal = mStats.Total
	}

	return s, nil
}