yay/callbacks.go

package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"

	alpm "github.com/jguer/go-alpm"
)

func questionCallback(question alpm.QuestionAny) {
	if qi, err := question.QuestionInstallIgnorepkg(); err == nil {
		qi.SetInstall(true)
	}

	qp, err := question.QuestionSelectProvider()
	if err != nil {
		return
	}

	size := 0

	qp.Providers(alpmHandle).ForEach(func(pkg alpm.Package) error {
		size++
		return nil
	})

	fmt.Print(bold(cyan(":: ")))
	str := bold(fmt.Sprintf(bold("There are %d providers available for %s:"), size, qp.Dep()))

	size = 1
	var db string

	qp.Providers(alpmHandle).ForEach(func(pkg alpm.Package) error {
		thisDb := pkg.DB().Name()

		if db != thisDb {
			db = thisDb
			str += bold(cyan("\n:: ")) + bold("Repository "+db+"\n    ")
		}
		str += fmt.Sprintf("%d) %s ", size, pkg.Name())
		size++
		return nil
	})

	fmt.Println(str)

	for {
		fmt.Print("\nEnter a number (default=1): ")

		if config.NoConfirm {
			fmt.Println()
			break
		}

		reader := bufio.NewReader(os.Stdin)
		numberBuf, overflow, err := reader.ReadLine()

		if err != nil {
			fmt.Println(err)
			break
		}

		if overflow {
			fmt.Println("Input too long")
			continue
		}

		if string(numberBuf) == "" {
			break
		}

		num, err := strconv.Atoi(string(numberBuf))
		if err != nil {
			fmt.Printf("%s invalid number: %s\n", red("error:"), string(numberBuf))
			continue
		}

		if num < 1 || num > size {
			fmt.Printf("%s invalid value: %d is not between %d and %d\n", red("error:"), num, 1, size)
			continue
		}

		qp.SetUseIndex(num - 1)
		break
	}
}

func logCallback(level alpm.LogLevel, str string) {
	switch level {
	case alpm.LogWarning:
		fmt.Print(bold(yellow(smallArrow)), " ", str)
	case alpm.LogError:
		fmt.Print(bold(red(smallArrow)), " ", str)
	}
}