package wincmd

import (
	"golang.org/x/sys/windows/registry"
)

// EnsureAutoRun idempotently enables or disables HKLM Run startup.
func EnsureAutoRun(key, path string, enable bool) (changed bool, err error) {
	current, exists, err := getAutoRunValue(key)
	if err != nil {
		return false, err
	}

	if enable {
		if exists && current == path {
			return false, nil
		}
		_, err := AutoRun(key, path)
		return err == nil, err
	}

	if !exists {
		return false, nil
	}
	_, err = DeleteAutoRun(key)
	return err == nil, err
}

func getAutoRunValue(key string) (value string, exists bool, err error) {
	reg, err := registry.OpenKey(registry.LOCAL_MACHINE, `Software\Microsoft\Windows\CurrentVersion\Run`, registry.ALL_ACCESS)
	if err != nil {
		return "", false, err
	}
	defer reg.Close()

	value, _, err = reg.GetStringValue(key)
	if err != nil {
		if err == registry.ErrNotExist {
			return "", false, nil
		}
		return "", false, err
	}
	return value, true, nil
}