yay/upgrade.go

229 lines
5.8 KiB
Go
Raw Normal View History

2017-08-01 16:43:20 +00:00
package main
2017-07-14 17:03:54 +00:00
import (
"fmt"
2017-08-01 16:43:20 +00:00
"sort"
2020-07-10 00:36:45 +00:00
"strings"
"sync"
2017-07-14 17:03:54 +00:00
2021-05-13 05:27:24 +00:00
aur "github.com/Jguer/aur"
2020-10-01 11:38:03 +00:00
alpm "github.com/Jguer/go-alpm/v2"
"github.com/leonelquinteros/gotext"
2020-07-31 23:20:00 +00:00
"github.com/Jguer/yay/v10/pkg/db"
"github.com/Jguer/yay/v10/pkg/intrange"
2020-10-01 12:06:21 +00:00
"github.com/Jguer/yay/v10/pkg/multierror"
"github.com/Jguer/yay/v10/pkg/query"
2020-07-05 00:45:23 +00:00
"github.com/Jguer/yay/v10/pkg/settings"
2020-10-01 12:06:21 +00:00
"github.com/Jguer/yay/v10/pkg/stringset"
"github.com/Jguer/yay/v10/pkg/text"
2020-07-31 23:20:00 +00:00
"github.com/Jguer/yay/v10/pkg/upgrade"
2017-07-14 17:03:54 +00:00
)
func filterUpdateList(list []db.Upgrade, filter upgrade.Filter) []db.Upgrade {
tmp := list[:0]
for _, pkg := range list {
if filter(pkg) {
tmp = append(tmp, pkg)
}
}
return tmp
}
2017-10-14 16:11:47 +00:00
// upList returns lists of packages to upgrade from each source.
func upList(warnings *query.AURWarnings, dbExecutor db.Executor, enableDowngrade bool,
filter upgrade.Filter) (aurUp, repoUp upgrade.UpSlice, err error) {
2020-07-31 23:20:00 +00:00
remote, remoteNames := query.GetRemotePackages(dbExecutor)
2017-07-14 17:03:54 +00:00
var wg sync.WaitGroup
2020-07-31 23:20:00 +00:00
var develUp upgrade.UpSlice
var repoSlice []db.Upgrade
var errs multierror.MultiError
2017-07-14 17:03:54 +00:00
2021-05-13 05:27:24 +00:00
aurdata := make(map[string]*aur.Pkg)
2019-11-11 07:15:04 +00:00
for _, pkg := range remote {
if pkg.ShouldIgnore() {
warnings.Ignore.Set(pkg.Name())
}
}
2020-07-05 00:45:23 +00:00
if config.Runtime.Mode == settings.ModeAny || config.Runtime.Mode == settings.ModeRepo {
text.OperationInfoln(gotext.Get("Searching databases for updates..."))
wg.Add(1)
go func() {
repoSlice, err = dbExecutor.RepoUpgrades(enableDowngrade)
2018-08-02 16:14:57 +00:00
errs.Add(err)
wg.Done()
}()
}
2020-07-05 00:45:23 +00:00
if config.Runtime.Mode == settings.ModeAny || config.Runtime.Mode == settings.ModeAUR {
text.OperationInfoln(gotext.Get("Searching AUR for updates..."))
2021-05-13 05:27:24 +00:00
var _aurdata []*aur.Pkg
_aurdata, err = query.AURInfo(config.Runtime.AURClient, remoteNames, warnings, config.RequestSplitN)
2018-08-02 16:14:57 +00:00
errs.Add(err)
if err == nil {
for _, pkg := range _aurdata {
aurdata[pkg.Name] = pkg
}
wg.Add(1)
go func() {
2020-10-01 12:06:21 +00:00
aurUp = upgrade.UpAUR(remote, aurdata, config.TimeUpdate)
wg.Done()
}()
if config.Devel {
text.OperationInfoln(gotext.Get("Checking development packages..."))
wg.Add(1)
go func() {
2020-10-01 12:06:21 +00:00
develUp = upgrade.UpDevel(remote, aurdata, config.Runtime.VCSStore)
wg.Done()
}()
}
}
}
wg.Wait()
printLocalNewerThanAUR(remote, aurdata)
names := make(stringset.StringSet)
for _, up := range develUp.Up {
names.Set(up.Name)
}
for _, up := range aurUp.Up {
if !names.Get(up.Name) {
develUp.Up = append(develUp.Up, up)
}
}
aurUp = develUp
2021-04-19 11:43:13 +00:00
aurUp.Repos = []string{"aur", "devel"}
2021-04-19 11:43:13 +00:00
repoUp = upgrade.UpSlice{Up: repoSlice, Repos: dbExecutor.Repos()}
aurUp.Up = filterUpdateList(aurUp.Up, filter)
repoUp.Up = filterUpdateList(repoUp.Up, filter)
return aurUp, repoUp, errs.Return()
2017-07-14 17:03:54 +00:00
}
func printLocalNewerThanAUR(
2021-05-13 05:27:24 +00:00
remote []alpm.IPackage, aurdata map[string]*aur.Pkg) {
for _, pkg := range remote {
aurPkg, ok := aurdata[pkg.Name()]
if !ok {
continue
}
2020-07-31 23:20:00 +00:00
left, right := upgrade.GetVersionDiff(pkg.Version(), aurPkg.Version)
if !isDevelPackage(pkg) && alpm.VerCmp(pkg.Version(), aurPkg.Version) > 0 {
text.Warnln(gotext.Get("%s: local (%s) is newer than AUR (%s)",
2020-08-16 22:09:43 +00:00
text.Cyan(pkg.Name()),
2018-08-04 13:22:49 +00:00
left, right,
))
}
}
}
2020-07-10 00:36:45 +00:00
func isDevelName(name string) bool {
for _, suffix := range []string{"git", "svn", "hg", "bzr", "nightly", "insiders-bin"} {
2020-07-10 00:36:45 +00:00
if strings.HasSuffix(name, "-"+suffix) {
return true
}
}
return strings.Contains(name, "-always-")
}
2020-10-01 11:38:03 +00:00
func isDevelPackage(pkg alpm.IPackage) bool {
2020-07-10 00:36:45 +00:00
return isDevelName(pkg.Name()) || isDevelName(pkg.Base())
}
2021-03-14 21:41:32 +00:00
// upgradePkgsMenu handles updating the cache and installing updates.
func upgradePkgsMenu(aurUp, repoUp upgrade.UpSlice) (stringset.StringSet, []string, error) {
ignore := make(stringset.StringSet)
targets := []string{}
allUpLen := len(repoUp.Up) + len(aurUp.Up)
2018-05-02 15:46:21 +00:00
if allUpLen == 0 {
2021-03-14 21:41:32 +00:00
return ignore, nil, nil
2017-08-01 16:43:20 +00:00
}
if !config.UpgradeMenu {
for _, pkg := range aurUp.Up {
2021-03-14 21:41:32 +00:00
targets = append(targets, pkg.Name)
}
2021-03-14 21:41:32 +00:00
return ignore, targets, nil
}
2017-08-01 16:43:20 +00:00
sort.Sort(repoUp)
sort.Sort(aurUp)
2021-04-19 11:43:13 +00:00
allUp := upgrade.UpSlice{Up: append(repoUp.Up, aurUp.Up...), Repos: append(repoUp.Repos, aurUp.Repos...)}
2020-08-16 22:09:43 +00:00
fmt.Printf("%s"+text.Bold(" %d ")+"%s\n", text.Bold(text.Cyan("::")), allUpLen, text.Bold(gotext.Get("Packages to upgrade.")))
2020-07-31 23:20:00 +00:00
allUp.Print()
2017-08-01 16:43:20 +00:00
text.Infoln(gotext.Get("Packages to exclude: (eg: \"1 2 3\", \"1-3\", \"^4\" or repo name)"))
numbers, err := getInput(config.AnswerUpgrade)
if err != nil {
return nil, nil, err
}
// upgrade menu asks you which packages to NOT upgrade so in this case
// include and exclude are kind of swapped
include, exclude, otherInclude, otherExclude := intrange.ParseNumberMenu(numbers)
2017-08-01 16:43:20 +00:00
isInclude := len(exclude) == 0 && len(otherExclude) == 0
for i, pkg := range repoUp.Up {
if isInclude && otherInclude.Get(pkg.Repository) {
ignore.Set(pkg.Name)
2017-08-01 16:43:20 +00:00
}
2017-08-02 21:56:45 +00:00
if isInclude && !include.Get(len(repoUp.Up)-i+len(aurUp.Up)) {
2021-03-14 21:41:32 +00:00
targets = append(targets, pkg.Name)
continue
2017-08-01 16:43:20 +00:00
}
if !isInclude && (exclude.Get(len(repoUp.Up)-i+len(aurUp.Up)) || otherExclude.Get(pkg.Repository)) {
2021-03-14 21:41:32 +00:00
targets = append(targets, pkg.Name)
continue
2018-01-14 17:48:16 +00:00
}
ignore.Set(pkg.Name)
2017-08-01 16:43:20 +00:00
}
for i, pkg := range aurUp.Up {
if isInclude && otherInclude.Get(pkg.Repository) {
continue
2017-08-01 16:43:20 +00:00
}
if isInclude && !include.Get(len(aurUp.Up)-i) {
2021-03-14 21:41:32 +00:00
targets = append(targets, "aur/"+pkg.Name)
}
if !isInclude && (exclude.Get(len(aurUp.Up)-i) || otherExclude.Get(pkg.Repository)) {
2021-03-14 21:41:32 +00:00
targets = append(targets, "aur/"+pkg.Name)
2017-08-01 16:43:20 +00:00
}
}
2021-03-14 21:41:32 +00:00
return ignore, targets, err
}
// Targets for sys upgrade
func sysupgradeTargets(dbExecutor db.Executor, enableDowngrade bool) (stringset.StringSet, []string, error) {
warnings := query.NewWarnings()
aurUp, repoUp, err := upList(warnings, dbExecutor, enableDowngrade, func(upgrade.Upgrade) bool { return true })
if err != nil {
return nil, nil, err
}
warnings.Print()
ignore, targets, errUp := upgradePkgsMenu(aurUp, repoUp)
return ignore, targets, errUp
2017-08-01 16:43:20 +00:00
}