package packagemanager

import (
	"log"
	"strings"
	"system_setup_tool/internal/shell"
)

type ChocoManager struct{}

func (c *ChocoManager) Install(packages []string) error {
	if len(packages) == 0 {
		return nil
	}
	err := InstallWithProgress(c, packages)
	if err != nil {
		return err
	}
	return nil
}

func (c *ChocoManager) Name() string {
	return "Chocolatey"
}

func (c *ChocoManager) InstallManager() error {
	if _, err := shell.ExecLookPath("choco"); err == nil {
		return nil
	}

	return installHomebrew()
}

func installChoco() error {
	return nil
}

func (c *ChocoManager) InstallPackage(pkg string) error {
	cmd := shell.ExecCommand("choco", "install", pkg)
	return cmd.Run()
}

func (c *ChocoManager) RemovePackage(pkg string) error {
	cmd := shell.ExecCommand("choco", "uninstall", pkg)
	return cmd.Run()
}

func (c *ChocoManager) SearchPackage(pkg string) []string {
	cmd := shell.ExecCommand("choco", "search", pkg)
	packages, err := cmd.Output()
	if err != nil {
		log.Printf("error fetching %s packages: %v", c.Name(), err)
	}
	packageList := strings.Split(strings.TrimSpace(string(packages)), "\n")
	return packageList
}

func (c *ChocoManager) UpdatePackage(pkg string) error {
	cmd := shell.ExecCommand("choco", "update", pkg)
	return cmd.Run()
}

func (c *ChocoManager) UpdateAllPackages() error {
	cmd := shell.ExecCommand("choco", "update")
	return cmd.Run()
}

func (c *ChocoManager) Check(pkg string) error {
	cmd := shell.ExecCommand("choco", "search", "--local-only", pkg)
	return cmd.Run()
}