package stress

import (
	"strings"

	"auto-check/pkg/sshclient"

	"golang.org/x/crypto/ssh"
)

// ============================
// 远程工具检测
// ============================

// ToolInfo 远程工具信息
type ToolInfo struct {
	Name    string
	Path    string
	Version string
}

// ToolSet 远程可用工具集合
type ToolSet struct {
	Tools map[string]ToolInfo // name → info
}

// Has 工具是否可用
func (ts ToolSet) Has(name string) bool {
	_, ok := ts.Tools[name]
	return ok
}

// Get 获取工具信息
func (ts ToolSet) Get(name string) (ToolInfo, bool) {
	t, ok := ts.Tools[name]
	return t, ok
}

// DetectTools 检测远程所有相关工具是否可用
func DetectTools(client *ssh.Client) ToolSet {
	ts := ToolSet{Tools: make(map[string]ToolInfo)}
	for _, name := range []string{"stress-ng", "stressapptest", "iperf3", "fio", "lm-sensors"} {
		if info, ok := detectOne(client, name); ok {
			ts.Tools[name] = info
		}
	}
	return ts
}

// detectOne 检测单个工具
func detectOne(client *ssh.Client, name string) (ToolInfo, bool) {
	out, err := sshclient.RunCommand(client, "which "+name+" 2>/dev/null")
	path := strings.TrimSpace(out)
	if err != nil || path == "" {
		return ToolInfo{}, false
	}

	var version string
	switch name {
	case "stress-ng":
		out, _ = sshclient.RunCommand(client, path+" --version 2>&1 | head -1")
		version = strings.TrimSpace(out)
	case "stressapptest":
		out, _ = sshclient.RunCommand(client, path+" --help 2>&1 | head -1")
		version = strings.TrimSpace(out)
	case "iperf3":
		out, _ = sshclient.RunCommand(client, path+" --version 2>&1 | head -1")
		version = strings.TrimSpace(out)
	case "fio":
		out, _ = sshclient.RunCommand(client, path+" --version 2>&1 | head -1")
		version = strings.TrimSpace(out)
	case "lm-sensors":
		out, _ = sshclient.RunCommand(client, path+" -v 2>&1 | head -1")
		version = strings.TrimSpace(out)
	}

	return ToolInfo{Name: name, Path: path, Version: version}, true
}