package ports

import (
	"strings"

	"github.com/gogrlx/snack"
)

// parseList parses the output of `pkg_info`.
// Format: "name-version description text"
func parseList(output string) []snack.Package {
	var pkgs []snack.Package
	for _, line := range strings.Split(output, "\n") {
		line = strings.TrimSpace(line)
		if line == "" {
			continue
		}
		// First field is name-version, rest is description
		parts := strings.SplitN(line, " ", 2)
		nameVer := parts[0]
		name, ver := splitNameVersion(nameVer)
		p := snack.Package{
			Name:      name,
			Version:   ver,
			Installed: true,
		}
		if len(parts) == 2 {
			p.Description = strings.TrimSpace(parts[1])
		}
		pkgs = append(pkgs, p)
	}
	return pkgs
}

// parseSearchResults parses the output of `pkg_info -Q <query>`.
// Each line is a package name-version.
func parseSearchResults(output string) []snack.Package {
	var pkgs []snack.Package
	for _, line := range strings.Split(output, "\n") {
		line = strings.TrimSpace(line)
		if line == "" {
			continue
		}
		name, ver := splitNameVersion(line)
		pkgs = append(pkgs, snack.Package{
			Name:    name,
			Version: ver,
		})
	}
	return pkgs
}

// parseInfoOutput parses `pkg_info <pkg>` output.
// The first line is typically "Information for name-version" or
// the package description block. We extract name/version from
// the stem or the provided pkg name.
func parseInfoOutput(output string, pkg string) *snack.Package {
	lines := strings.Split(output, "\n")
	p := &snack.Package{Installed: true}

	for _, line := range lines {
		line = strings.TrimSpace(line)
		if strings.HasPrefix(line, "Information for ") {
			nameVer := strings.TrimPrefix(line, "Information for ")
			nameVer = strings.TrimSuffix(nameVer, ":")
			p.Name, p.Version = splitNameVersion(nameVer)
			continue
		}
	}

	// If we got description lines (after the header), join them
	if p.Name != "" {
		var desc []string
		inDesc := false
		for _, line := range lines {
			trimmed := strings.TrimSpace(line)
			if strings.HasPrefix(trimmed, "Information for ") {
				inDesc = false
				continue
			}
			if strings.HasPrefix(trimmed, "Comment:") {
				p.Description = strings.TrimSpace(strings.TrimPrefix(trimmed, "Comment:"))
				continue
			}
			if strings.HasPrefix(trimmed, "Description:") {
				inDesc = true
				continue
			}
			if inDesc && trimmed != "" {
				desc = append(desc, trimmed)
			}
		}
		if p.Description == "" && len(desc) > 0 {
			p.Description = strings.Join(desc, " ")
		}
	}

	if p.Name == "" {
		// Fallback: try to parse from pkg argument
		p.Name, p.Version = splitNameVersion(pkg)
		if p.Name == "" {
			return nil
		}
	}
	return p
}

// parseUpgradeOutput parses the output of `pkg_add -u -n`.
// Lines like "name-oldver -> name-newver" indicate available upgrades.
func parseUpgradeOutput(output string) []snack.Package {
	var pkgs []snack.Package
	for _, line := range strings.Split(output, "\n") {
		line = strings.TrimSpace(line)
		if !strings.Contains(line, "->") {
			continue
		}
		parts := strings.Fields(line)
		// Expect: "name-oldver -> name-newver"
		if len(parts) < 3 || parts[1] != "->" {
			continue
		}
		name, _ := splitNameVersion(parts[0])
		_, newVer := splitNameVersion(parts[2])
		if name != "" {
			pkgs = append(pkgs, snack.Package{
				Name:      name,
				Version:   newVer,
				Installed: true,
			})
		}
	}
	return pkgs
}

// parseFileListOutput parses `pkg_info -L <pkg>` output.
// Lines starting with "/" after the header are file paths.
func parseFileListOutput(output string) []string {
	var files []string
	for _, line := range strings.Split(output, "\n") {
		line = strings.TrimSpace(line)
		if strings.HasPrefix(line, "/") {
			files = append(files, line)
		}
	}
	return files
}

// parseOwnerOutput parses `pkg_info -E <path>` output.
// Returns the package name that owns the file.
func parseOwnerOutput(output string) string {
	output = strings.TrimSpace(output)
	// pkg_info -E returns the package name-version
	name, _ := splitNameVersion(output)
	return name
}

// splitNameVersion splits "name-version" at the last hyphen.
// OpenBSD packages use the last hyphen before a version number as separator.
func splitNameVersion(s string) (string, string) {
	idx := strings.LastIndex(s, "-")
	if idx <= 0 {
		return s, ""
	}
	return s[:idx], s[idx+1:]
}