DankMaterialShell/core/internal/server/sysupdate/handlers.go

package sysupdate

import (
	"net"

	"github.com/AvengeMedia/DankMaterialShell/core/internal/server/models"
	"github.com/AvengeMedia/DankMaterialShell/core/internal/server/params"
)

func HandleRequest(conn net.Conn, req models.Request, m *Manager) {
	switch req.Method {
	case "sysupdate.getState":
		models.Respond(conn, req.ID, m.GetState())
	case "sysupdate.refresh":
		force := params.BoolOpt(req.Params, "force", false)
		m.Refresh(RefreshOptions{Force: force})
		models.Respond(conn, req.ID, m.GetState())
	case "sysupdate.upgrade":
		handleUpgrade(conn, req, m)
	case "sysupdate.cancel":
		m.Cancel()
		models.Respond(conn, req.ID, m.GetState())
	case "sysupdate.acquire":
		m.Acquire()
		models.Respond(conn, req.ID, models.SuccessResult{Success: true})
	case "sysupdate.release":
		m.Release()
		models.Respond(conn, req.ID, models.SuccessResult{Success: true})
	case "sysupdate.setInterval":
		seconds, err := params.Int(req.Params, "seconds")
		if err != nil {
			models.RespondError(conn, req.ID, err.Error())
			return
		}
		m.SetInterval(seconds)
		models.Respond(conn, req.ID, m.GetState())
	default:
		models.RespondError(conn, req.ID, "unknown method: "+req.Method)
	}
}

func handleUpgrade(conn net.Conn, req models.Request, m *Manager) {
	opts := UpgradeOptions{
		IncludeFlatpak: params.BoolOpt(req.Params, "includeFlatpak", true),
		IncludeAUR:     params.BoolOpt(req.Params, "includeAUR", true),
		DryRun:         params.BoolOpt(req.Params, "dry", false),
		CustomCommand:  params.StringOpt(req.Params, "customCommand", ""),
		Terminal:       params.StringOpt(req.Params, "terminal", ""),
		Ignored:        stringSliceOpt(req.Params, "ignored"),
	}
	if err := m.Upgrade(opts); err != nil {
		models.RespondError(conn, req.ID, err.Error())
		return
	}
	models.Respond(conn, req.ID, m.GetState())
}

func stringSliceOpt(p map[string]any, key string) []string {
	val, ok := params.Any(p, key)
	if !ok {
		return nil
	}
	arr, ok := val.([]any)
	if !ok {
		return nil
	}
	out := make([]string, 0, len(arr))
	for _, v := range arr {
		if s, ok := v.(string); ok && s != "" {
			out = append(out, s)
		}
	}
	return out
}