yay/repo.go

package main

import (
	"fmt"
	"os"
	"os/exec"
	"strings"
)

// RepoResult describes a Repository package
type RepoResult struct {
	Description string
	Repository  string
	Version     string
	Name        string
}

// RepoSearch describes a Repository search
type RepoSearch struct {
	Resultcount int
	Results     []RepoResult
}

func getInstalledPackage(pkg string) (err error) {
	cmd := exec.Command(PacmanBin, "-Qi", pkg)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	err = cmd.Run()
	return
}

// InstallPackage handles repo installs
func InstallPackage(pkg string, flags ...string) (err error) {
	cmd := exec.Command(PacmanBin, "-S", pkg, strings.Join(flags, " "))
	cmd.Stdout = os.Stdout
	cmd.Stdin = os.Stdin
	cmd.Stderr = os.Stderr
	err = cmd.Run()
	return nil
}

// SearchPackages handles repo searches
func SearchPackages(pkg string) (search RepoSearch, err error) {
	cmdOutput, err := exec.Command(PacmanBin, "-Ss", pkg).Output()
	outputSlice := strings.Split(string(cmdOutput), "\n")
	if outputSlice[0] == "" {
		return search, nil
	}

	i := true
	var tempStr string
	var rRes *RepoResult
	for _, pkgStr := range outputSlice {
		if i {
			rRes = new(RepoResult)
			fmt.Sscanf(pkgStr, "%s %s\n", &tempStr, &rRes.Version)
			repoNameSlc := strings.Split(tempStr, "/")
			rRes.Repository = repoNameSlc[0]
			rRes.Name = repoNameSlc[1]
			i = false
		} else {
			rRes.Description = pkgStr
			search.Resultcount++
			search.Results = append(search.Results, *rRes)
			i = true
		}
	}
	return
}

func isInRepo(pkg string) bool {
	if _, err := exec.Command(PacmanBin, "-Sp", pkg).Output(); err != nil {
		return false
	}

	return true
}

func (s RepoSearch) printSearch(index int) (err error) {
	for i, result := range s.Results {
		if index != SearchMode {
			fmt.Printf("%d \033[1m%s/\x1B[33m%s \x1B[36m%s\033[0m\n%s\n",
				i, result.Repository, result.Name, result.Version, result.Description)
		} else {
			fmt.Printf("\033[1m%s/\x1B[33m%s \x1B[36m%s\033[0m\n%s\n",
				result.Repository, result.Name, result.Version, result.Description)
		}
	}

	return nil
}