//go:build windows

package process

import (
	"fmt"
	"os"
	"os/exec"
	"path/filepath"
	"strconv"
	"strings"
	"syscall"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
	"golang.org/x/sys/windows"
)

func TestWindowsSetProcGroupDoesNotSuspendSharedHelpers(t *testing.T) {
	cmd := exec.Command("kandev")
	setProcGroup(cmd)

	require.NotNil(t, cmd.SysProcAttr)
	require.NotZero(t, cmd.SysProcAttr.CreationFlags&syscall.CREATE_NEW_PROCESS_GROUP)
	require.Zero(t, cmd.SysProcAttr.CreationFlags&windows.CREATE_SUSPENDED)
}

func TestWindowsSetAgentProcGroupStartsSuspended(t *testing.T) {
	cmd := exec.Command("kandev")
	setAgentProcGroup(cmd)

	require.NotNil(t, cmd.SysProcAttr)
	require.NotZero(t, cmd.SysProcAttr.CreationFlags&syscall.CREATE_NEW_PROCESS_GROUP)
	require.NotZero(t, cmd.SysProcAttr.CreationFlags&windows.CREATE_SUSPENDED)
}

func TestWindowsProcessLifecycleJobKillsDescendants(t *testing.T) {
	pidFile := filepath.Join(t.TempDir(), "child.pid")
	cmd := fixtureCmd(fmt.Sprintf("delay-then-child %s 200 30", pidFile))
	setAgentProcGroup(cmd)
	require.NoError(t, cmd.Start())
	parentPID := cmd.Process.Pid
	waited := false
	t.Cleanup(func() {
		if waited {
			return
		}
		_ = killProcessGroup(parentPID)
		_ = cmd.Process.Kill()
		_ = cmd.Wait()
	})

	lifecycle, err := installProcessLifecycle(cmd)
	require.NoError(t, err)
	childPID := waitForWindowsChildPID(t, pidFile, 5*time.Second)

	releaseProcessLifecycle(lifecycle)
	_ = cmd.Wait()
	waited = true

	require.Eventually(t, func() bool {
		return !windowsProcessAlive(childPID)
	}, 5*time.Second, 50*time.Millisecond,
		"child process %d should be killed when the job handle is released", childPID)
}

func TestWindowsProcessLifecycleInstallFailsForInvalidPID(t *testing.T) {
	_, err := installProcessLifecycle(&exec.Cmd{Process: &os.Process{Pid: -1}})
	require.Error(t, err)
}

func waitForWindowsChildPID(t *testing.T, pidFile string, timeout time.Duration) int {
	t.Helper()
	deadline := time.Now().Add(timeout)
	for {
		raw, err := os.ReadFile(pidFile)
		if err == nil {
			pid, perr := strconv.Atoi(strings.TrimSpace(string(raw)))
			if perr == nil && pid > 0 {
				return pid
			}
		}
		if time.Now().After(deadline) {
			t.Fatalf("timed out waiting for fixture to write child pid to %s", pidFile)
		}
		time.Sleep(20 * time.Millisecond)
	}
}

func windowsProcessAlive(pid int) bool {
	if pid <= 0 {
		return false
	}
	output, err := exec.Command("tasklist", "/FI", fmt.Sprintf("PID eq %d", pid), "/NH").Output()
	if err != nil {
		return false
	}
	text := strings.ToLower(string(output))
	if strings.Contains(text, "no tasks") {
		return false
	}
	return strings.Contains(text, strconv.Itoa(pid))
}