mirror of
https://github.com/Jguer/yay
synced 2024-10-31 04:12:51 +00:00
Merge pull request #97 from samosaara/master
Support for ranges when selecting packages
This commit is contained in:
commit
479e2f0ce0
2 changed files with 200 additions and 43 deletions
167
cmd.go
167
cmd.go
|
@ -3,6 +3,7 @@ package main
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
|
@ -13,7 +14,7 @@ import (
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
var cmdArgs *arguments = makeArguments()
|
var cmdArgs = makeArguments()
|
||||||
|
|
||||||
func usage() {
|
func usage() {
|
||||||
fmt.Println(`Usage:
|
fmt.Println(`Usage:
|
||||||
|
@ -95,7 +96,8 @@ func initYay() (err error) {
|
||||||
if _, err = os.Stat(configFile); os.IsNotExist(err) {
|
if _, err = os.Stat(configFile); os.IsNotExist(err) {
|
||||||
err = os.MkdirAll(filepath.Dir(configFile), 0755)
|
err = os.MkdirAll(filepath.Dir(configFile), 0755)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = fmt.Errorf("Unable to create config directory:", filepath.Dir(configFile), err)
|
err = fmt.Errorf("Unable to create config directory:\n%s\n"+
|
||||||
|
"The error was:\n%s", filepath.Dir(configFile), err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
// Save the default config if nothing is found
|
// Save the default config if nothing is found
|
||||||
|
@ -103,13 +105,14 @@ func initYay() (err error) {
|
||||||
} else {
|
} else {
|
||||||
cfile, errf := os.OpenFile(configFile, os.O_RDWR|os.O_CREATE, 0644)
|
cfile, errf := os.OpenFile(configFile, os.O_RDWR|os.O_CREATE, 0644)
|
||||||
if errf != nil {
|
if errf != nil {
|
||||||
fmt.Println("Error reading config: %s", err)
|
fmt.Printf("Error reading config: %s\n", err)
|
||||||
} else {
|
} else {
|
||||||
defer cfile.Close()
|
defer cfile.Close()
|
||||||
decoder := json.NewDecoder(cfile)
|
decoder := json.NewDecoder(cfile)
|
||||||
err = decoder.Decode(&config)
|
err = decoder.Decode(&config)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println("Loading default Settings.\nError reading config:", err)
|
fmt.Println("Loading default Settings.\nError reading config:",
|
||||||
|
err)
|
||||||
defaultSettings(&config)
|
defaultSettings(&config)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -181,7 +184,7 @@ func initAlpm() (err error) {
|
||||||
|
|
||||||
alpmHandle, err = alpmConf.CreateHandle()
|
alpmHandle, err = alpmConf.CreateHandle()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = fmt.Errorf("Unable to CreateHandle", err)
|
err = fmt.Errorf("Unable to CreateHandle: %s", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -261,11 +264,11 @@ cleanup:
|
||||||
func handleCmd() (changedConfig bool, err error) {
|
func handleCmd() (changedConfig bool, err error) {
|
||||||
changedConfig = false
|
changedConfig = false
|
||||||
|
|
||||||
for option, _ := range cmdArgs.options {
|
for option := range cmdArgs.options {
|
||||||
changedConfig = changedConfig || handleConfig(option)
|
changedConfig = changedConfig || handleConfig(option)
|
||||||
}
|
}
|
||||||
|
|
||||||
for option, _ := range cmdArgs.globals {
|
for option := range cmdArgs.globals {
|
||||||
changedConfig = changedConfig || handleConfig(option)
|
changedConfig = changedConfig || handleConfig(option)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -461,18 +464,78 @@ func handleRemove() (err error) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// BuildIntRange build the range from start to end
|
||||||
|
func BuildIntRange(rangeStart, rangeEnd int) []int {
|
||||||
|
if rangeEnd-rangeStart == 0 {
|
||||||
|
// rangeEnd == rangeStart, which means no range
|
||||||
|
return []int{rangeStart}
|
||||||
|
}
|
||||||
|
if rangeEnd < rangeStart {
|
||||||
|
swap := rangeEnd
|
||||||
|
rangeEnd = rangeStart
|
||||||
|
rangeStart = swap
|
||||||
|
}
|
||||||
|
|
||||||
|
final := make([]int, 0)
|
||||||
|
for i := rangeStart; i <= rangeEnd; i++ {
|
||||||
|
final = append(final, i)
|
||||||
|
}
|
||||||
|
return final
|
||||||
|
}
|
||||||
|
|
||||||
|
// BuildRange construct a range of ints from the format 1-10
|
||||||
|
func BuildRange(input string) ([]int, error) {
|
||||||
|
multipleNums := strings.Split(input, "-")
|
||||||
|
if len(multipleNums) != 2 {
|
||||||
|
return nil, errors.New("Invalid range")
|
||||||
|
}
|
||||||
|
|
||||||
|
rangeStart, err := strconv.Atoi(multipleNums[0])
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
rangeEnd, err := strconv.Atoi(multipleNums[1])
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return BuildIntRange(rangeStart, rangeEnd), err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Contains returns wheter e is present in s
|
||||||
|
func contains(s []string, e string) bool {
|
||||||
|
for _, a := range s {
|
||||||
|
if a == e {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemoveIntListFromList removes all src's elements that are present in target
|
||||||
|
func removeListFromList(src, target []string) []string {
|
||||||
|
max := len(target)
|
||||||
|
for i := 0; i < max; i++ {
|
||||||
|
if contains(src, target[i]) {
|
||||||
|
target = append(target[:i], target[i+1:]...)
|
||||||
|
max--
|
||||||
|
i--
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return target
|
||||||
|
}
|
||||||
|
|
||||||
// NumberMenu presents a CLI for selecting packages to install.
|
// NumberMenu presents a CLI for selecting packages to install.
|
||||||
func numberMenu(pkgS []string, flags []string) (err error) {
|
func numberMenu(pkgS []string, flags []string) (err error) {
|
||||||
//func numberMenu(cmdArgs *arguments) (err error) {
|
//func numberMenu(cmdArgs *arguments) (err error) {
|
||||||
var num int
|
var num int
|
||||||
|
|
||||||
aq, err := narrowSearch(pkgS, true)
|
aurQ, err := narrowSearch(pkgS, true)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println("Error during AUR search:", err)
|
fmt.Println("Error during AUR search:", err)
|
||||||
}
|
}
|
||||||
numaq := len(aq)
|
numaq := len(aurQ)
|
||||||
|
repoQ, numpq, err := queryRepo(pkgS)
|
||||||
pq, numpq, err := queryRepo(pkgS)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
@ -482,14 +545,16 @@ func numberMenu(pkgS []string, flags []string) (err error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
if config.SortMode == BottomUp {
|
if config.SortMode == BottomUp {
|
||||||
aq.printSearch(numpq + 1)
|
aurQ.printSearch(numpq + 1)
|
||||||
pq.printSearch()
|
repoQ.printSearch()
|
||||||
} else {
|
} else {
|
||||||
pq.printSearch()
|
repoQ.printSearch()
|
||||||
aq.printSearch(numpq + 1)
|
aurQ.printSearch(numpq + 1)
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Printf("\x1b[32m%s\x1b[0m\nNumbers: ", "Type numbers to install. Separate each number with a space.")
|
fmt.Printf("\x1b[32m%s %s\x1b[0m\nNumbers: ",
|
||||||
|
"Type the numbers or ranges (e.g. 1-10) you want to install.",
|
||||||
|
"Separate each one of them with a space.")
|
||||||
reader := bufio.NewReader(os.Stdin)
|
reader := bufio.NewReader(os.Stdin)
|
||||||
numberBuf, overflow, err := reader.ReadLine()
|
numberBuf, overflow, err := reader.ReadLine()
|
||||||
if err != nil || overflow {
|
if err != nil || overflow {
|
||||||
|
@ -498,33 +563,69 @@ func numberMenu(pkgS []string, flags []string) (err error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
numberString := string(numberBuf)
|
numberString := string(numberBuf)
|
||||||
var aurI []string
|
var aurI, aurNI, repoNI, repoI []string
|
||||||
var repoI []string
|
|
||||||
result := strings.Fields(numberString)
|
result := strings.Fields(numberString)
|
||||||
for _, numS := range result {
|
for _, numS := range result {
|
||||||
|
negate := numS[0] == '^'
|
||||||
|
if negate {
|
||||||
|
numS = numS[1:]
|
||||||
|
}
|
||||||
|
var numbers []int
|
||||||
num, err = strconv.Atoi(numS)
|
num, err = strconv.Atoi(numS)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
continue
|
numbers, err = BuildRange(numS)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
numbers = []int{num}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Install package
|
// Install package
|
||||||
if num > numaq+numpq || num <= 0 {
|
for _, x := range numbers {
|
||||||
continue
|
var target string
|
||||||
} else if num > numpq {
|
if x > numaq+numpq || x <= 0 {
|
||||||
if config.SortMode == BottomUp {
|
continue
|
||||||
aurI = append(aurI, aq[numaq+numpq-num].Name)
|
} else if x > numpq {
|
||||||
|
if config.SortMode == BottomUp {
|
||||||
|
target = aurQ[numaq+numpq-x].Name
|
||||||
|
} else {
|
||||||
|
target = aurQ[x-numpq-1].Name
|
||||||
|
}
|
||||||
|
if negate {
|
||||||
|
aurNI = append(aurNI, target)
|
||||||
|
} else {
|
||||||
|
aurI = append(aurI, target)
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
aurI = append(aurI, aq[num-numpq-1].Name)
|
if config.SortMode == BottomUp {
|
||||||
}
|
target = repoQ[numpq-x].Name()
|
||||||
} else {
|
} else {
|
||||||
if config.SortMode == BottomUp {
|
target = repoQ[x-1].Name()
|
||||||
repoI = append(repoI, pq[numpq-num].Name())
|
}
|
||||||
} else {
|
if negate {
|
||||||
repoI = append(repoI, pq[num-1].Name())
|
repoNI = append(repoNI, target)
|
||||||
|
} else {
|
||||||
|
repoI = append(repoI, target)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if len(repoI) == 0 && len(aurI) == 0 &&
|
||||||
|
(len(aurNI) > 0 || len(repoNI) > 0) {
|
||||||
|
// If no package was specified, only exclusions, exclude from all the
|
||||||
|
// packages
|
||||||
|
for _, pack := range aurQ {
|
||||||
|
aurI = append(aurI, pack.Name)
|
||||||
|
}
|
||||||
|
for _, pack := range repoQ {
|
||||||
|
repoI = append(repoI, pack.Name())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
aurI = removeListFromList(aurNI, aurI)
|
||||||
|
repoI = removeListFromList(repoNI, repoI)
|
||||||
|
|
||||||
if len(repoI) != 0 {
|
if len(repoI) != 0 {
|
||||||
arguments := makeArguments()
|
arguments := makeArguments()
|
||||||
arguments.addArg("S")
|
arguments.addArg("S")
|
||||||
|
@ -542,8 +643,8 @@ func numberMenu(pkgS []string, flags []string) (err error) {
|
||||||
// Complete provides completion info for shells
|
// Complete provides completion info for shells
|
||||||
func complete() error {
|
func complete() error {
|
||||||
path := completionFile + config.Shell + ".cache"
|
path := completionFile + config.Shell + ".cache"
|
||||||
|
info, err := os.Stat(path)
|
||||||
if info, err := os.Stat(path); os.IsNotExist(err) || time.Since(info.ModTime()).Hours() > 48 {
|
if os.IsNotExist(err) || time.Since(info.ModTime()).Hours() > 48 {
|
||||||
os.MkdirAll(filepath.Dir(completionFile), 0755)
|
os.MkdirAll(filepath.Dir(completionFile), 0755)
|
||||||
out, errf := os.Create(path)
|
out, errf := os.Create(path)
|
||||||
if errf != nil {
|
if errf != nil {
|
||||||
|
|
76
upgrade.go
76
upgrade.go
|
@ -264,13 +264,36 @@ func upRepo(local []alpm.Package) (upSlice, error) {
|
||||||
return slice, nil
|
return slice, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//Contains returns wheter e is present in s
|
||||||
|
func containsInt(s []int, e int) bool {
|
||||||
|
for _, a := range s {
|
||||||
|
if a == e {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemoveIntListFromList removes all src's elements that are present in target
|
||||||
|
func removeIntListFromList(src, target []int) []int {
|
||||||
|
max := len(target)
|
||||||
|
for i := 0; i < max; i++ {
|
||||||
|
if containsInt(src, target[i]) {
|
||||||
|
target = append(target[:i], target[i+1:]...)
|
||||||
|
max--
|
||||||
|
i--
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return target
|
||||||
|
}
|
||||||
|
|
||||||
// upgradePkgs handles updating the cache and installing updates.
|
// upgradePkgs handles updating the cache and installing updates.
|
||||||
func upgradePkgs(flags []string) error {
|
func upgradePkgs(flags []string) error {
|
||||||
aurUp, repoUp, err := upList()
|
aurUp, repoUp, err := upList()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
} else if len(aurUp)+len(repoUp) == 0 {
|
} else if len(aurUp)+len(repoUp) == 0 {
|
||||||
fmt.Println("\nthere is nothing to do")
|
fmt.Println("\nThere is nothing to do")
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -292,21 +315,54 @@ func upgradePkgs(flags []string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
result := strings.Fields(string(numberBuf))
|
result := strings.Fields(string(numberBuf))
|
||||||
|
excludeAur := make([]int, 0)
|
||||||
|
excludeRepo := make([]int, 0)
|
||||||
for _, numS := range result {
|
for _, numS := range result {
|
||||||
|
negate := numS[0] == '^'
|
||||||
|
if negate {
|
||||||
|
numS = numS[1:]
|
||||||
|
}
|
||||||
|
var numbers []int
|
||||||
num, err := strconv.Atoi(numS)
|
num, err := strconv.Atoi(numS)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
continue
|
numbers, err = BuildRange(numS)
|
||||||
}
|
if err != nil {
|
||||||
if num > len(aurUp)+len(repoUp)-1 || num < 0 {
|
continue
|
||||||
continue
|
}
|
||||||
} else if num < len(aurUp) {
|
|
||||||
num = len(aurUp) - num - 1
|
|
||||||
aurNums = append(aurNums, num)
|
|
||||||
} else {
|
} else {
|
||||||
num = len(aurUp) + len(repoUp) - num - 1
|
numbers = []int{num}
|
||||||
repoNums = append(repoNums, num)
|
}
|
||||||
|
for _, target := range numbers {
|
||||||
|
if target > len(aurUp)+len(repoUp)-1 || target < 0 {
|
||||||
|
continue
|
||||||
|
} else if target < len(aurUp) {
|
||||||
|
target = len(aurUp) - target - 1
|
||||||
|
if negate {
|
||||||
|
excludeAur = append(excludeAur, target)
|
||||||
|
} else {
|
||||||
|
aurNums = append(aurNums, target)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
target = len(aurUp) + len(repoUp) - target - 1
|
||||||
|
if negate {
|
||||||
|
excludeRepo = append(excludeRepo, target)
|
||||||
|
} else {
|
||||||
|
repoNums = append(repoNums, target)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if len(repoNums) == 0 && len(aurNums) == 0 &&
|
||||||
|
(len(excludeRepo) > 0 || len(excludeAur) > 0) {
|
||||||
|
if len(repoUp) > 0 {
|
||||||
|
repoNums = BuildIntRange(0, len(repoUp)-1)
|
||||||
|
}
|
||||||
|
if len(aurUp) > 0 {
|
||||||
|
aurNums = BuildIntRange(0, len(aurUp)-1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
aurNums = removeIntListFromList(excludeAur, aurNums)
|
||||||
|
repoNums = removeIntListFromList(excludeRepo, repoNums)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(repoUp) != 0 {
|
if len(repoUp) != 0 {
|
||||||
|
|
Loading…
Reference in a new issue