2017-08-01 16:43:20 +00:00
|
|
|
package main
|
2017-07-14 17:03:54 +00:00
|
|
|
|
|
|
|
import (
|
2017-08-01 16:43:20 +00:00
|
|
|
"bufio"
|
2017-07-14 17:03:54 +00:00
|
|
|
"fmt"
|
2017-08-01 16:43:20 +00:00
|
|
|
"os"
|
|
|
|
"sort"
|
2017-07-17 22:44:46 +00:00
|
|
|
"unicode"
|
2017-07-14 17:03:54 +00:00
|
|
|
|
|
|
|
alpm "github.com/jguer/go-alpm"
|
|
|
|
pkgb "github.com/mikkeloscar/gopkgbuild"
|
|
|
|
)
|
|
|
|
|
2017-08-01 16:43:20 +00:00
|
|
|
// upgrade type describes a system upgrade.
|
|
|
|
type upgrade struct {
|
2017-07-14 17:03:54 +00:00
|
|
|
Name string
|
|
|
|
Repository string
|
|
|
|
LocalVersion string
|
|
|
|
RemoteVersion string
|
|
|
|
}
|
|
|
|
|
2017-10-14 16:11:47 +00:00
|
|
|
// upSlice is a slice of Upgrades
|
2017-08-01 16:43:20 +00:00
|
|
|
type upSlice []upgrade
|
2017-07-17 22:44:46 +00:00
|
|
|
|
2017-08-01 16:43:20 +00:00
|
|
|
func (u upSlice) Len() int { return len(u) }
|
|
|
|
func (u upSlice) Swap(i, j int) { u[i], u[j] = u[j], u[i] }
|
2017-07-17 22:44:46 +00:00
|
|
|
|
2017-08-01 16:43:20 +00:00
|
|
|
func (u upSlice) Less(i, j int) bool {
|
|
|
|
iRunes := []rune(u[i].Repository)
|
|
|
|
jRunes := []rune(u[j].Repository)
|
2017-07-17 22:44:46 +00:00
|
|
|
|
|
|
|
max := len(iRunes)
|
|
|
|
if max > len(jRunes) {
|
|
|
|
max = len(jRunes)
|
|
|
|
}
|
|
|
|
|
|
|
|
for idx := 0; idx < max; idx++ {
|
|
|
|
ir := iRunes[idx]
|
|
|
|
jr := jRunes[idx]
|
|
|
|
|
|
|
|
lir := unicode.ToLower(ir)
|
|
|
|
ljr := unicode.ToLower(jr)
|
|
|
|
|
|
|
|
if lir != ljr {
|
|
|
|
return lir > ljr
|
|
|
|
}
|
|
|
|
|
|
|
|
// the lowercase runes are the same, so compare the original
|
|
|
|
if ir != jr {
|
|
|
|
return ir > jr
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2018-02-19 17:01:19 +00:00
|
|
|
func getVersionDiff(oldVersion, newversion string) (left, right string) {
|
2018-02-21 08:41:25 +00:00
|
|
|
old, errOld := pkgb.NewCompleteVersion(oldVersion)
|
|
|
|
new, errNew := pkgb.NewCompleteVersion(newversion)
|
2018-02-19 17:01:19 +00:00
|
|
|
|
2018-02-21 08:41:25 +00:00
|
|
|
if errOld != nil {
|
2018-03-02 04:03:09 +00:00
|
|
|
left = red("Invalid Version")
|
2018-02-21 08:41:25 +00:00
|
|
|
}
|
|
|
|
if errNew != nil {
|
2018-03-02 04:03:09 +00:00
|
|
|
right = red("Invalid Version")
|
2018-02-21 08:41:25 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if errOld == nil && errNew == nil {
|
|
|
|
if old.Version == new.Version {
|
2018-03-02 04:03:09 +00:00
|
|
|
left = string(old.Version) + "-" + red(string(old.Pkgrel))
|
|
|
|
right = string(new.Version) + "-" + green(string(new.Pkgrel))
|
2018-02-21 08:41:25 +00:00
|
|
|
} else {
|
2018-03-02 04:03:09 +00:00
|
|
|
left = red(string(old.Version)) + "-" + string(old.Pkgrel)
|
|
|
|
right = bold(green(string(new.Version))) + "-" + string(new.Pkgrel)
|
2018-02-19 17:01:19 +00:00
|
|
|
}
|
2018-02-21 08:41:25 +00:00
|
|
|
}
|
2018-02-19 17:01:19 +00:00
|
|
|
|
2018-02-21 08:41:25 +00:00
|
|
|
return
|
2018-02-19 17:01:19 +00:00
|
|
|
}
|
|
|
|
|
2017-10-14 16:11:47 +00:00
|
|
|
// upList returns lists of packages to upgrade from each source.
|
2018-02-18 23:46:25 +00:00
|
|
|
func upList(dt *depTree) (aurUp upSlice, repoUp upSlice, err error) {
|
2017-08-07 13:43:25 +00:00
|
|
|
local, remote, _, remoteNames, err := filterPackages()
|
2017-07-14 17:03:54 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2017-08-01 16:43:20 +00:00
|
|
|
repoC := make(chan upSlice)
|
|
|
|
aurC := make(chan upSlice)
|
2017-07-14 17:03:54 +00:00
|
|
|
errC := make(chan error)
|
|
|
|
|
2018-03-02 04:03:09 +00:00
|
|
|
fmt.Println(bold(cyan("::") + " Searching databases for updates..."))
|
2017-07-14 17:03:54 +00:00
|
|
|
go func() {
|
2017-08-01 16:43:20 +00:00
|
|
|
repoUpList, err := upRepo(local)
|
2017-07-14 17:03:54 +00:00
|
|
|
errC <- err
|
|
|
|
repoC <- repoUpList
|
|
|
|
}()
|
|
|
|
|
2018-03-02 04:03:09 +00:00
|
|
|
fmt.Println(bold(cyan("::") + " Searching AUR for updates..."))
|
2017-07-14 17:03:54 +00:00
|
|
|
go func() {
|
2018-02-18 23:46:25 +00:00
|
|
|
aurUpList, err := upAUR(remote, remoteNames, dt)
|
2017-07-14 17:03:54 +00:00
|
|
|
errC <- err
|
|
|
|
aurC <- aurUpList
|
|
|
|
}()
|
|
|
|
|
|
|
|
var i = 0
|
|
|
|
loop:
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case repoUp = <-repoC:
|
|
|
|
i++
|
|
|
|
case aurUp = <-aurC:
|
|
|
|
i++
|
|
|
|
case err := <-errC:
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println(err)
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
if i == 2 {
|
|
|
|
close(repoC)
|
|
|
|
close(aurC)
|
|
|
|
close(errC)
|
|
|
|
break loop
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2018-01-10 00:38:32 +00:00
|
|
|
func upDevel(remote []alpm.Package, packageC chan upgrade, done chan bool) {
|
2018-03-05 22:15:34 +00:00
|
|
|
for vcsName, e := range savedInfo {
|
2017-10-18 02:12:16 +00:00
|
|
|
if e.needsUpdate() {
|
2017-10-18 05:46:21 +00:00
|
|
|
found := false
|
2018-01-10 00:38:32 +00:00
|
|
|
var pkg alpm.Package
|
|
|
|
for _, r := range remote {
|
2018-03-05 22:15:34 +00:00
|
|
|
if r.Name() == vcsName {
|
2017-10-18 05:46:21 +00:00
|
|
|
found = true
|
2018-01-10 00:38:32 +00:00
|
|
|
pkg = r
|
2017-10-18 05:46:21 +00:00
|
|
|
}
|
|
|
|
}
|
2018-01-10 00:38:32 +00:00
|
|
|
if found {
|
|
|
|
if pkg.ShouldIgnore() {
|
2018-03-02 04:54:38 +00:00
|
|
|
fmt.Print(magenta("Warning: "))
|
|
|
|
fmt.Printf("%s ignoring package upgrade (%s => %s)\n", cyan(pkg.Name()), pkg.Version(), "git")
|
2018-01-10 00:38:32 +00:00
|
|
|
} else {
|
2018-03-05 22:15:34 +00:00
|
|
|
packageC <- upgrade{pkg.Name(), "devel", pkg.Version(), "latest-commit"}
|
2018-01-10 00:38:32 +00:00
|
|
|
}
|
2017-10-18 02:12:16 +00:00
|
|
|
} else {
|
2018-03-05 22:15:34 +00:00
|
|
|
removeVCSPackage([]string{vcsName})
|
2017-10-18 02:12:16 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2017-10-18 05:46:21 +00:00
|
|
|
done <- true
|
2017-10-18 02:12:16 +00:00
|
|
|
}
|
|
|
|
|
2017-10-14 16:11:47 +00:00
|
|
|
// upAUR gathers foreign packages and checks if they have new versions.
|
2017-07-14 17:03:54 +00:00
|
|
|
// Output: Upgrade type package list.
|
2018-02-18 23:46:25 +00:00
|
|
|
func upAUR(remote []alpm.Package, remoteNames []string, dt *depTree) (toUpgrade upSlice, err error) {
|
2017-07-14 17:03:54 +00:00
|
|
|
var routines int
|
|
|
|
var routineDone int
|
|
|
|
|
2017-08-01 16:43:20 +00:00
|
|
|
packageC := make(chan upgrade)
|
2017-07-14 17:03:54 +00:00
|
|
|
done := make(chan bool)
|
|
|
|
|
2017-10-18 02:12:16 +00:00
|
|
|
if config.Devel {
|
2017-10-18 05:46:21 +00:00
|
|
|
routines++
|
2018-01-10 00:38:32 +00:00
|
|
|
go upDevel(remote, packageC, done)
|
2018-03-03 23:42:12 +00:00
|
|
|
fmt.Println(bold(cyan("::") + " Checking development packages..."))
|
2017-10-18 02:12:16 +00:00
|
|
|
}
|
|
|
|
|
2018-02-18 23:46:25 +00:00
|
|
|
routines++
|
|
|
|
go func(remote []alpm.Package, remoteNames []string, dt *depTree) {
|
|
|
|
for _, pkg := range remote {
|
|
|
|
aurPkg, ok := dt.Aur[pkg.Name()]
|
|
|
|
if !ok {
|
|
|
|
continue
|
2017-07-14 17:03:54 +00:00
|
|
|
}
|
2018-02-18 23:46:25 +00:00
|
|
|
|
|
|
|
if (config.TimeUpdate && (int64(aurPkg.LastModified) > pkg.BuildDate().Unix())) ||
|
|
|
|
(alpm.VerCmp(pkg.Version(), aurPkg.Version) < 0) {
|
|
|
|
if pkg.ShouldIgnore() {
|
2018-02-19 17:01:19 +00:00
|
|
|
left, right := getVersionDiff(pkg.Version(), aurPkg.Version)
|
2018-03-02 04:54:38 +00:00
|
|
|
fmt.Print(magenta("Warning: "))
|
|
|
|
fmt.Printf("%s ignoring package upgrade (%s => %s)\n", cyan(pkg.Name()), left, right)
|
2017-07-14 17:03:54 +00:00
|
|
|
} else {
|
2018-02-18 23:46:25 +00:00
|
|
|
packageC <- upgrade{aurPkg.Name, "aur", pkg.Version(), aurPkg.Version}
|
2017-07-14 17:03:54 +00:00
|
|
|
}
|
|
|
|
}
|
2018-02-18 23:46:25 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
done <- true
|
|
|
|
}(remote, remoteNames, dt)
|
2018-02-21 08:41:25 +00:00
|
|
|
|
2018-02-18 23:46:25 +00:00
|
|
|
if routineDone == routines {
|
|
|
|
err = nil
|
|
|
|
return
|
2017-07-14 17:03:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case pkg := <-packageC:
|
2017-10-30 12:36:50 +00:00
|
|
|
for _, w := range toUpgrade {
|
|
|
|
if w.Name == pkg.Name {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
}
|
2017-07-14 17:03:54 +00:00
|
|
|
toUpgrade = append(toUpgrade, pkg)
|
|
|
|
case <-done:
|
|
|
|
routineDone++
|
|
|
|
if routineDone == routines {
|
|
|
|
err = nil
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-10-14 16:11:47 +00:00
|
|
|
// upRepo gathers local packages and checks if they have new versions.
|
2017-07-14 17:03:54 +00:00
|
|
|
// Output: Upgrade type package list.
|
2017-08-01 16:43:20 +00:00
|
|
|
func upRepo(local []alpm.Package) (upSlice, error) {
|
2017-10-18 02:38:19 +00:00
|
|
|
dbList, err := alpmHandle.SyncDbs()
|
2017-07-14 17:03:54 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2017-08-01 16:43:20 +00:00
|
|
|
slice := upSlice{}
|
2017-10-18 02:38:19 +00:00
|
|
|
|
2017-07-14 17:03:54 +00:00
|
|
|
for _, pkg := range local {
|
2017-07-18 09:51:58 +00:00
|
|
|
newPkg := pkg.NewVersion(dbList)
|
2018-01-10 00:38:32 +00:00
|
|
|
if newPkg != nil {
|
|
|
|
if pkg.ShouldIgnore() {
|
2018-03-02 04:54:38 +00:00
|
|
|
fmt.Print(magenta("Warning: "))
|
2018-01-26 01:18:49 +00:00
|
|
|
fmt.Printf("%s ignoring package upgrade (%s => %s)\n", pkg.Name(), pkg.Version(), newPkg.Version())
|
2018-01-10 00:38:32 +00:00
|
|
|
} else {
|
|
|
|
slice = append(slice, upgrade{pkg.Name(), newPkg.DB().Name(), pkg.Version(), newPkg.Version()})
|
|
|
|
}
|
2017-07-14 17:03:54 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return slice, nil
|
|
|
|
}
|
2017-08-01 16:43:20 +00:00
|
|
|
|
2018-01-31 21:04:21 +00:00
|
|
|
//Contains returns whether e is present in s
|
2018-01-14 18:53:37 +00:00
|
|
|
func containsInt(s []int, e int) bool {
|
2018-01-14 17:48:16 +00:00
|
|
|
for _, a := range s {
|
|
|
|
if a == e {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2018-01-14 18:53:37 +00:00
|
|
|
// RemoveIntListFromList removes all src's elements that are present in target
|
|
|
|
func removeIntListFromList(src, target []int) []int {
|
2018-01-14 17:48:16 +00:00
|
|
|
max := len(target)
|
|
|
|
for i := 0; i < max; i++ {
|
2018-01-14 18:53:37 +00:00
|
|
|
if containsInt(src, target[i]) {
|
2018-01-14 17:48:16 +00:00
|
|
|
target = append(target[:i], target[i+1:]...)
|
|
|
|
max--
|
|
|
|
i--
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return target
|
|
|
|
}
|
|
|
|
|
2017-10-14 16:11:47 +00:00
|
|
|
// upgradePkgs handles updating the cache and installing updates.
|
2018-02-18 23:46:25 +00:00
|
|
|
func upgradePkgs(dt *depTree) (stringSet, stringSet, error) {
|
|
|
|
repoNames := make(stringSet)
|
|
|
|
aurNames := make(stringSet)
|
|
|
|
|
|
|
|
aurUp, repoUp, err := upList(dt)
|
2017-08-01 16:43:20 +00:00
|
|
|
if err != nil {
|
2018-02-18 23:46:25 +00:00
|
|
|
return repoNames, aurNames, err
|
2017-08-01 16:43:20 +00:00
|
|
|
} else if len(aurUp)+len(repoUp) == 0 {
|
2018-02-18 23:46:25 +00:00
|
|
|
return repoNames, aurNames, err
|
2017-08-01 16:43:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
sort.Sort(repoUp)
|
2018-03-02 04:54:38 +00:00
|
|
|
fmt.Println(bold(blue("::")), len(aurUp)+len(repoUp), bold("Packages to upgrade."))
|
2018-01-16 10:18:36 +00:00
|
|
|
repoUp.Print(len(aurUp) + 1)
|
|
|
|
aurUp.Print(1)
|
2017-08-01 16:43:20 +00:00
|
|
|
|
2018-03-09 03:41:45 +00:00
|
|
|
if config.NoConfirm {
|
|
|
|
for _, up := range repoUp {
|
|
|
|
repoNames.set(up.Name)
|
|
|
|
}
|
|
|
|
for _, up := range aurUp {
|
|
|
|
aurNames.set(up.Name)
|
|
|
|
}
|
|
|
|
return repoNames, aurNames, nil
|
|
|
|
}
|
|
|
|
|
2018-03-09 22:11:30 +00:00
|
|
|
fmt.Println(bold(green(arrow + " Packages to not upgrade (eg: 1 2 3, 1-3, ^4 or repo name)")))
|
2018-03-09 03:41:45 +00:00
|
|
|
fmt.Print(bold(green(arrow + " ")))
|
|
|
|
reader := bufio.NewReader(os.Stdin)
|
|
|
|
|
|
|
|
numberBuf, overflow, err := reader.ReadLine()
|
|
|
|
if err != nil {
|
|
|
|
return nil, nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
if overflow {
|
|
|
|
return nil, nil, fmt.Errorf("Input too long")
|
|
|
|
}
|
|
|
|
|
|
|
|
//upgrade menu asks you which packages to NOT upgrade so in this case
|
|
|
|
//include and exclude are kind of swaped
|
|
|
|
//include, exclude, other := parseNumberMenu(string(numberBuf))
|
|
|
|
include, exclude, otherInclude, otherExclude := parseNumberMenu(string(numberBuf))
|
2017-08-01 16:43:20 +00:00
|
|
|
|
2018-03-09 03:41:45 +00:00
|
|
|
isInclude := len(exclude) == 0 && len(otherExclude) == 0
|
|
|
|
|
|
|
|
for i, pkg := range repoUp {
|
|
|
|
if isInclude && otherInclude.get(pkg.Repository) {
|
|
|
|
continue
|
2017-08-01 16:43:20 +00:00
|
|
|
}
|
2017-08-02 21:56:45 +00:00
|
|
|
|
2018-03-09 03:41:45 +00:00
|
|
|
if isInclude && !include.get(len(repoUp)-i+len(aurUp)) {
|
|
|
|
repoNames.set(pkg.Name)
|
2017-08-01 16:43:20 +00:00
|
|
|
}
|
2018-03-09 03:41:45 +00:00
|
|
|
|
|
|
|
if !isInclude && (exclude.get(len(repoUp)-i+len(aurUp)) || otherExclude.get(pkg.Repository)) {
|
|
|
|
repoNames.set(pkg.Name)
|
2018-01-14 17:48:16 +00:00
|
|
|
}
|
2017-08-01 16:43:20 +00:00
|
|
|
}
|
|
|
|
|
2018-03-09 03:41:45 +00:00
|
|
|
for i, pkg := range aurUp {
|
|
|
|
if isInclude && otherInclude.get(pkg.Repository) {
|
|
|
|
continue
|
2017-08-01 16:43:20 +00:00
|
|
|
}
|
|
|
|
|
2018-03-09 03:41:45 +00:00
|
|
|
if isInclude && !include.get(len(aurUp)-i) {
|
|
|
|
aurNames.set(pkg.Name)
|
|
|
|
}
|
|
|
|
|
|
|
|
if !isInclude && (exclude.get(len(aurUp)-i) || otherExclude.get(pkg.Repository)) {
|
|
|
|
aurNames.set(pkg.Name)
|
2017-08-01 16:43:20 +00:00
|
|
|
}
|
|
|
|
}
|
2018-01-20 22:37:10 +00:00
|
|
|
|
2018-02-18 23:46:25 +00:00
|
|
|
return repoNames, aurNames, err
|
2017-08-01 16:43:20 +00:00
|
|
|
}
|