install.go 27 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177
  1. package main
  2. import (
  3. "errors"
  4. "fmt"
  5. "os"
  6. "os/exec"
  7. "path/filepath"
  8. "strconv"
  9. "strings"
  10. "sync"
  11. alpm "github.com/Jguer/go-alpm"
  12. gosrc "github.com/Morganamilo/go-srcinfo"
  13. "github.com/leonelquinteros/gotext"
  14. "github.com/Jguer/yay/v10/pkg/completion"
  15. "github.com/Jguer/yay/v10/pkg/intrange"
  16. "github.com/Jguer/yay/v10/pkg/multierror"
  17. "github.com/Jguer/yay/v10/pkg/query"
  18. "github.com/Jguer/yay/v10/pkg/settings"
  19. "github.com/Jguer/yay/v10/pkg/stringset"
  20. "github.com/Jguer/yay/v10/pkg/text"
  21. )
  22. func asdeps(parser *settings.Arguments, pkgs []string) error {
  23. if len(pkgs) == 0 {
  24. return nil
  25. }
  26. parser = parser.CopyGlobal()
  27. _ = parser.AddArg("D", "asdeps")
  28. parser.AddTarget(pkgs...)
  29. _, stderr, err := capture(passToPacman(parser))
  30. if err != nil {
  31. return fmt.Errorf("%s %s", stderr, err)
  32. }
  33. return nil
  34. }
  35. func asexp(parser *settings.Arguments, pkgs []string) error {
  36. if len(pkgs) == 0 {
  37. return nil
  38. }
  39. parser = parser.CopyGlobal()
  40. _ = parser.AddArg("D", "asexplicit")
  41. parser.AddTarget(pkgs...)
  42. _, stderr, err := capture(passToPacman(parser))
  43. if err != nil {
  44. return fmt.Errorf("%s %s", stderr, err)
  45. }
  46. return nil
  47. }
  48. // Install handles package installs
  49. func install(parser *settings.Arguments, alpmHandle *alpm.Handle) (err error) {
  50. var incompatible stringset.StringSet
  51. var do *depOrder
  52. var aurUp upSlice
  53. var repoUp upSlice
  54. var srcinfos map[string]*gosrc.Srcinfo
  55. warnings := makeWarnings()
  56. if config.Runtime.Mode == settings.ModeAny || config.Runtime.Mode == settings.ModeRepo {
  57. if config.CombinedUpgrade {
  58. if parser.ExistsArg("y", "refresh") {
  59. err = earlyRefresh(parser)
  60. if err != nil {
  61. return fmt.Errorf(gotext.Get("error refreshing databases"))
  62. }
  63. }
  64. } else if parser.ExistsArg("y", "refresh") || parser.ExistsArg("u", "sysupgrade") || len(parser.Targets) > 0 {
  65. err = earlyPacmanCall(parser, alpmHandle)
  66. if err != nil {
  67. return err
  68. }
  69. }
  70. }
  71. // we may have done -Sy, our handle now has an old
  72. // database.
  73. alpmHandle, err = initAlpmHandle(config.Runtime.PacmanConf, alpmHandle)
  74. if err != nil {
  75. return err
  76. }
  77. config.Runtime.AlpmHandle = alpmHandle
  78. _, _, localNames, remoteNames, err := query.FilterPackages(alpmHandle)
  79. if err != nil {
  80. return err
  81. }
  82. remoteNamesCache := stringset.FromSlice(remoteNames)
  83. localNamesCache := stringset.FromSlice(localNames)
  84. requestTargets := parser.Copy().Targets
  85. // create the arguments to pass for the repo install
  86. arguments := parser.Copy()
  87. arguments.DelArg("asdeps", "asdep")
  88. arguments.DelArg("asexplicit", "asexp")
  89. arguments.Op = "S"
  90. arguments.ClearTargets()
  91. if config.Runtime.Mode == settings.ModeAUR {
  92. arguments.DelArg("u", "sysupgrade")
  93. }
  94. // if we are doing -u also request all packages needing update
  95. if parser.ExistsArg("u", "sysupgrade") {
  96. aurUp, repoUp, err = upList(warnings, alpmHandle)
  97. if err != nil {
  98. return err
  99. }
  100. warnings.print()
  101. ignore, aurUp, errUp := upgradePkgs(aurUp, repoUp)
  102. if errUp != nil {
  103. return errUp
  104. }
  105. for _, up := range repoUp {
  106. if !ignore.Get(up.Name) {
  107. requestTargets = append(requestTargets, up.Name)
  108. parser.AddTarget(up.Name)
  109. }
  110. }
  111. for up := range aurUp {
  112. requestTargets = append(requestTargets, "aur/"+up)
  113. parser.AddTarget("aur/" + up)
  114. }
  115. value, _, exists := cmdArgs.GetArg("ignore")
  116. if len(ignore) > 0 {
  117. ignoreStr := strings.Join(ignore.ToSlice(), ",")
  118. if exists {
  119. ignoreStr += "," + value
  120. }
  121. if arguments.Options["ignore"] == nil {
  122. arguments.Options["ignore"] = &settings.Option{
  123. Args: []string{ignoreStr},
  124. }
  125. } else {
  126. arguments.Options["ignore"].Add(ignoreStr)
  127. }
  128. }
  129. }
  130. targets := stringset.FromSlice(parser.Targets)
  131. dp, err := getDepPool(requestTargets, warnings, alpmHandle)
  132. if err != nil {
  133. return err
  134. }
  135. err = dp.CheckMissing()
  136. if err != nil {
  137. return err
  138. }
  139. if len(dp.Aur) == 0 {
  140. if !config.CombinedUpgrade {
  141. if parser.ExistsArg("u", "sysupgrade") {
  142. fmt.Println(gotext.Get(" there is nothing to do"))
  143. }
  144. return nil
  145. }
  146. parser.Op = "S"
  147. parser.DelArg("y", "refresh")
  148. parser.Options["ignore"] = arguments.Options["ignore"]
  149. return show(passToPacman(parser))
  150. }
  151. if len(dp.Aur) > 0 && os.Geteuid() == 0 {
  152. return fmt.Errorf(gotext.Get("refusing to install AUR packages as root, aborting"))
  153. }
  154. conflicts, err := dp.CheckConflicts()
  155. if err != nil {
  156. return err
  157. }
  158. do = getDepOrder(dp)
  159. if err != nil {
  160. return err
  161. }
  162. for _, pkg := range do.Repo {
  163. arguments.AddTarget(pkg.DB().Name() + "/" + pkg.Name())
  164. }
  165. for _, pkg := range dp.Groups {
  166. arguments.AddTarget(pkg)
  167. }
  168. if len(do.Aur) == 0 && len(arguments.Targets) == 0 && (!parser.ExistsArg("u", "sysupgrade") || config.Runtime.Mode == settings.ModeAUR) {
  169. fmt.Println(gotext.Get(" there is nothing to do"))
  170. return nil
  171. }
  172. do.Print()
  173. fmt.Println()
  174. if config.CleanAfter {
  175. defer cleanAfter(do.Aur)
  176. }
  177. if do.HasMake() {
  178. switch config.RemoveMake {
  179. case "yes":
  180. defer func() {
  181. err = removeMake(do)
  182. }()
  183. case "no":
  184. break
  185. default:
  186. if continueTask(gotext.Get("Remove make dependencies after install?"), false) {
  187. defer func() {
  188. err = removeMake(do)
  189. }()
  190. }
  191. }
  192. }
  193. if config.CleanMenu {
  194. if anyExistInCache(do.Aur) {
  195. askClean := pkgbuildNumberMenu(do.Aur, remoteNamesCache)
  196. toClean, errClean := cleanNumberMenu(do.Aur, remoteNamesCache, askClean)
  197. if errClean != nil {
  198. return errClean
  199. }
  200. cleanBuilds(toClean)
  201. }
  202. }
  203. toSkip := pkgbuildsToSkip(do.Aur, targets)
  204. cloned, err := downloadPkgbuilds(do.Aur, toSkip, config.BuildDir)
  205. if err != nil {
  206. return err
  207. }
  208. var toDiff []Base
  209. var toEdit []Base
  210. if config.DiffMenu {
  211. pkgbuildNumberMenu(do.Aur, remoteNamesCache)
  212. toDiff, err = diffNumberMenu(do.Aur, remoteNamesCache)
  213. if err != nil {
  214. return err
  215. }
  216. if len(toDiff) > 0 {
  217. err = showPkgbuildDiffs(toDiff, cloned)
  218. if err != nil {
  219. return err
  220. }
  221. }
  222. }
  223. if len(toDiff) > 0 {
  224. oldValue := config.NoConfirm
  225. config.NoConfirm = false
  226. fmt.Println()
  227. if !continueTask(gotext.Get("Proceed with install?"), true) {
  228. return fmt.Errorf(gotext.Get("aborting due to user"))
  229. }
  230. err = updatePkgbuildSeenRef(toDiff)
  231. if err != nil {
  232. text.Errorln(err.Error())
  233. }
  234. config.NoConfirm = oldValue
  235. }
  236. err = mergePkgbuilds(do.Aur)
  237. if err != nil {
  238. return err
  239. }
  240. srcinfos, err = parseSrcinfoFiles(do.Aur, true)
  241. if err != nil {
  242. return err
  243. }
  244. if config.EditMenu {
  245. pkgbuildNumberMenu(do.Aur, remoteNamesCache)
  246. toEdit, err = editNumberMenu(do.Aur, remoteNamesCache)
  247. if err != nil {
  248. return err
  249. }
  250. if len(toEdit) > 0 {
  251. err = editPkgbuilds(toEdit, srcinfos)
  252. if err != nil {
  253. return err
  254. }
  255. }
  256. }
  257. if len(toEdit) > 0 {
  258. oldValue := config.NoConfirm
  259. config.NoConfirm = false
  260. fmt.Println()
  261. if !continueTask(gotext.Get("Proceed with install?"), true) {
  262. return errors.New(gotext.Get("aborting due to user"))
  263. }
  264. config.NoConfirm = oldValue
  265. }
  266. incompatible, err = getIncompatible(do.Aur, srcinfos, alpmHandle)
  267. if err != nil {
  268. return err
  269. }
  270. if config.PGPFetch {
  271. err = checkPgpKeys(do.Aur, srcinfos)
  272. if err != nil {
  273. return err
  274. }
  275. }
  276. if !config.CombinedUpgrade {
  277. arguments.DelArg("u", "sysupgrade")
  278. }
  279. if len(arguments.Targets) > 0 || arguments.ExistsArg("u") {
  280. if errShow := show(passToPacman(arguments)); errShow != nil {
  281. return errors.New(gotext.Get("error installing repo packages"))
  282. }
  283. deps := make([]string, 0)
  284. exp := make([]string, 0)
  285. for _, pkg := range do.Repo {
  286. if !dp.Explicit.Get(pkg.Name()) && !localNamesCache.Get(pkg.Name()) && !remoteNamesCache.Get(pkg.Name()) {
  287. deps = append(deps, pkg.Name())
  288. continue
  289. }
  290. if parser.ExistsArg("asdeps", "asdep") && dp.Explicit.Get(pkg.Name()) {
  291. deps = append(deps, pkg.Name())
  292. } else if parser.ExistsArg("asexp", "asexplicit") && dp.Explicit.Get(pkg.Name()) {
  293. exp = append(exp, pkg.Name())
  294. }
  295. }
  296. if errDeps := asdeps(parser, deps); errDeps != nil {
  297. return errDeps
  298. }
  299. if errExp := asexp(parser, exp); errExp != nil {
  300. return errExp
  301. }
  302. }
  303. go exitOnError(completion.Update(alpmHandle, config.AURURL, config.Runtime.CompletionPath, config.CompletionInterval, false))
  304. err = downloadPkgbuildsSources(do.Aur, incompatible)
  305. if err != nil {
  306. return err
  307. }
  308. err = buildInstallPkgbuilds(dp, do, srcinfos, parser, incompatible, conflicts, alpmHandle)
  309. if err != nil {
  310. return err
  311. }
  312. return nil
  313. }
  314. func removeMake(do *depOrder) error {
  315. removeArguments := settings.MakeArguments()
  316. err := removeArguments.AddArg("R", "u")
  317. if err != nil {
  318. return err
  319. }
  320. for _, pkg := range do.getMake() {
  321. removeArguments.AddTarget(pkg)
  322. }
  323. oldValue := config.NoConfirm
  324. config.NoConfirm = true
  325. err = show(passToPacman(removeArguments))
  326. config.NoConfirm = oldValue
  327. return err
  328. }
  329. func inRepos(syncDB alpm.DBList, pkg string) bool {
  330. target := toTarget(pkg)
  331. if target.DB == "aur" {
  332. return false
  333. } else if target.DB != "" {
  334. return true
  335. }
  336. previousHideMenus := hideMenus
  337. hideMenus = false
  338. _, err := syncDB.FindSatisfier(target.DepString())
  339. hideMenus = previousHideMenus
  340. if err == nil {
  341. return true
  342. }
  343. return !syncDB.FindGroupPkgs(target.Name).Empty()
  344. }
  345. func earlyPacmanCall(parser *settings.Arguments, alpmHandle *alpm.Handle) error {
  346. arguments := parser.Copy()
  347. arguments.Op = "S"
  348. targets := parser.Targets
  349. parser.ClearTargets()
  350. arguments.ClearTargets()
  351. syncDB, err := alpmHandle.SyncDBs()
  352. if err != nil {
  353. return err
  354. }
  355. if config.Runtime.Mode == settings.ModeRepo {
  356. arguments.Targets = targets
  357. } else {
  358. // separate aur and repo targets
  359. for _, target := range targets {
  360. if inRepos(syncDB, target) {
  361. arguments.AddTarget(target)
  362. } else {
  363. parser.AddTarget(target)
  364. }
  365. }
  366. }
  367. if parser.ExistsArg("y", "refresh") || parser.ExistsArg("u", "sysupgrade") || len(arguments.Targets) > 0 {
  368. err = show(passToPacman(arguments))
  369. if err != nil {
  370. return errors.New(gotext.Get("error installing repo packages"))
  371. }
  372. }
  373. return nil
  374. }
  375. func earlyRefresh(parser *settings.Arguments) error {
  376. arguments := parser.Copy()
  377. parser.DelArg("y", "refresh")
  378. arguments.DelArg("u", "sysupgrade")
  379. arguments.DelArg("s", "search")
  380. arguments.DelArg("i", "info")
  381. arguments.DelArg("l", "list")
  382. arguments.ClearTargets()
  383. return show(passToPacman(arguments))
  384. }
  385. func getIncompatible(bases []Base, srcinfos map[string]*gosrc.Srcinfo, alpmHandle *alpm.Handle) (stringset.StringSet, error) {
  386. incompatible := make(stringset.StringSet)
  387. basesMap := make(map[string]Base)
  388. alpmArch, err := alpmHandle.Arch()
  389. if err != nil {
  390. return nil, err
  391. }
  392. nextpkg:
  393. for _, base := range bases {
  394. for _, arch := range srcinfos[base.Pkgbase()].Arch {
  395. if arch == "any" || arch == alpmArch {
  396. continue nextpkg
  397. }
  398. }
  399. incompatible.Set(base.Pkgbase())
  400. basesMap[base.Pkgbase()] = base
  401. }
  402. if len(incompatible) > 0 {
  403. text.Warnln(gotext.Get("The following packages are not compatible with your architecture:"))
  404. for pkg := range incompatible {
  405. fmt.Print(" " + cyan(basesMap[pkg].String()))
  406. }
  407. fmt.Println()
  408. if !continueTask(gotext.Get("Try to build them anyway?"), true) {
  409. return nil, errors.New(gotext.Get("aborting due to user"))
  410. }
  411. }
  412. return incompatible, nil
  413. }
  414. func parsePackageList(dir string) (pkgdests map[string]string, pkgVersion string, err error) {
  415. stdout, stderr, err := capture(passToMakepkg(dir, "--packagelist"))
  416. if err != nil {
  417. return nil, "", fmt.Errorf("%s %s", stderr, err)
  418. }
  419. lines := strings.Split(stdout, "\n")
  420. pkgdests = make(map[string]string)
  421. for _, line := range lines {
  422. if line == "" {
  423. continue
  424. }
  425. fileName := filepath.Base(line)
  426. split := strings.Split(fileName, "-")
  427. if len(split) < 4 {
  428. return nil, "", errors.New(gotext.Get("cannot find package name: %v", split))
  429. }
  430. // pkgname-pkgver-pkgrel-arch.pkgext
  431. // This assumes 3 dashes after the pkgname, Will cause an error
  432. // if the PKGEXT contains a dash. Please no one do that.
  433. pkgName := strings.Join(split[:len(split)-3], "-")
  434. pkgVersion = strings.Join(split[len(split)-3:len(split)-1], "-")
  435. pkgdests[pkgName] = line
  436. }
  437. return pkgdests, pkgVersion, nil
  438. }
  439. func anyExistInCache(bases []Base) bool {
  440. for _, base := range bases {
  441. pkg := base.Pkgbase()
  442. dir := filepath.Join(config.BuildDir, pkg)
  443. if _, err := os.Stat(dir); !os.IsNotExist(err) {
  444. return true
  445. }
  446. }
  447. return false
  448. }
  449. func pkgbuildNumberMenu(bases []Base, installed stringset.StringSet) bool {
  450. toPrint := ""
  451. askClean := false
  452. for n, base := range bases {
  453. pkg := base.Pkgbase()
  454. dir := filepath.Join(config.BuildDir, pkg)
  455. toPrint += fmt.Sprintf(magenta("%3d")+" %-40s", len(bases)-n,
  456. bold(base.String()))
  457. anyInstalled := false
  458. for _, b := range base {
  459. anyInstalled = anyInstalled || installed.Get(b.Name)
  460. }
  461. if anyInstalled {
  462. toPrint += bold(green(gotext.Get(" (Installed)")))
  463. }
  464. if _, err := os.Stat(dir); !os.IsNotExist(err) {
  465. toPrint += bold(green(gotext.Get(" (Build Files Exist)")))
  466. askClean = true
  467. }
  468. toPrint += "\n"
  469. }
  470. fmt.Print(toPrint)
  471. return askClean
  472. }
  473. func cleanNumberMenu(bases []Base, installed stringset.StringSet, hasClean bool) ([]Base, error) {
  474. toClean := make([]Base, 0)
  475. if !hasClean {
  476. return toClean, nil
  477. }
  478. text.Infoln(gotext.Get("Packages to cleanBuild?"))
  479. text.Infoln(gotext.Get("%s [A]ll [Ab]ort [I]nstalled [No]tInstalled or (1 2 3, 1-3, ^4)", cyan(gotext.Get("[N]one"))))
  480. cleanInput, err := getInput(config.AnswerClean)
  481. if err != nil {
  482. return nil, err
  483. }
  484. cInclude, cExclude, cOtherInclude, cOtherExclude := intrange.ParseNumberMenu(cleanInput)
  485. cIsInclude := len(cExclude) == 0 && len(cOtherExclude) == 0
  486. if cOtherInclude.Get("abort") || cOtherInclude.Get("ab") {
  487. return nil, fmt.Errorf(gotext.Get("aborting due to user"))
  488. }
  489. if !cOtherInclude.Get("n") && !cOtherInclude.Get("none") {
  490. for i, base := range bases {
  491. pkg := base.Pkgbase()
  492. anyInstalled := false
  493. for _, b := range base {
  494. anyInstalled = anyInstalled || installed.Get(b.Name)
  495. }
  496. dir := filepath.Join(config.BuildDir, pkg)
  497. if _, err := os.Stat(dir); os.IsNotExist(err) {
  498. continue
  499. }
  500. if !cIsInclude && cExclude.Get(len(bases)-i) {
  501. continue
  502. }
  503. if anyInstalled && (cOtherInclude.Get("i") || cOtherInclude.Get("installed")) {
  504. toClean = append(toClean, base)
  505. continue
  506. }
  507. if !anyInstalled && (cOtherInclude.Get("no") || cOtherInclude.Get("notinstalled")) {
  508. toClean = append(toClean, base)
  509. continue
  510. }
  511. if cOtherInclude.Get("a") || cOtherInclude.Get("all") {
  512. toClean = append(toClean, base)
  513. continue
  514. }
  515. if cIsInclude && (cInclude.Get(len(bases)-i) || cOtherInclude.Get(pkg)) {
  516. toClean = append(toClean, base)
  517. continue
  518. }
  519. if !cIsInclude && (!cExclude.Get(len(bases)-i) && !cOtherExclude.Get(pkg)) {
  520. toClean = append(toClean, base)
  521. continue
  522. }
  523. }
  524. }
  525. return toClean, nil
  526. }
  527. func editNumberMenu(bases []Base, installed stringset.StringSet) ([]Base, error) {
  528. return editDiffNumberMenu(bases, installed, false)
  529. }
  530. func diffNumberMenu(bases []Base, installed stringset.StringSet) ([]Base, error) {
  531. return editDiffNumberMenu(bases, installed, true)
  532. }
  533. func editDiffNumberMenu(bases []Base, installed stringset.StringSet, diff bool) ([]Base, error) {
  534. toEdit := make([]Base, 0)
  535. var editInput string
  536. var err error
  537. if diff {
  538. text.Infoln(gotext.Get("Diffs to show?"))
  539. text.Infoln(gotext.Get("%s [A]ll [Ab]ort [I]nstalled [No]tInstalled or (1 2 3, 1-3, ^4)", cyan(gotext.Get("[N]one"))))
  540. editInput, err = getInput(config.AnswerDiff)
  541. if err != nil {
  542. return nil, err
  543. }
  544. } else {
  545. text.Infoln(gotext.Get("PKGBUILDs to edit?"))
  546. text.Infoln(gotext.Get("%s [A]ll [Ab]ort [I]nstalled [No]tInstalled or (1 2 3, 1-3, ^4)", cyan(gotext.Get("[N]one"))))
  547. editInput, err = getInput(config.AnswerEdit)
  548. if err != nil {
  549. return nil, err
  550. }
  551. }
  552. eInclude, eExclude, eOtherInclude, eOtherExclude := intrange.ParseNumberMenu(editInput)
  553. eIsInclude := len(eExclude) == 0 && len(eOtherExclude) == 0
  554. if eOtherInclude.Get("abort") || eOtherInclude.Get("ab") {
  555. return nil, fmt.Errorf(gotext.Get("aborting due to user"))
  556. }
  557. if !eOtherInclude.Get("n") && !eOtherInclude.Get("none") {
  558. for i, base := range bases {
  559. pkg := base.Pkgbase()
  560. anyInstalled := false
  561. for _, b := range base {
  562. anyInstalled = anyInstalled || installed.Get(b.Name)
  563. }
  564. if !eIsInclude && eExclude.Get(len(bases)-i) {
  565. continue
  566. }
  567. if anyInstalled && (eOtherInclude.Get("i") || eOtherInclude.Get("installed")) {
  568. toEdit = append(toEdit, base)
  569. continue
  570. }
  571. if !anyInstalled && (eOtherInclude.Get("no") || eOtherInclude.Get("notinstalled")) {
  572. toEdit = append(toEdit, base)
  573. continue
  574. }
  575. if eOtherInclude.Get("a") || eOtherInclude.Get("all") {
  576. toEdit = append(toEdit, base)
  577. continue
  578. }
  579. if eIsInclude && (eInclude.Get(len(bases)-i) || eOtherInclude.Get(pkg)) {
  580. toEdit = append(toEdit, base)
  581. }
  582. if !eIsInclude && (!eExclude.Get(len(bases)-i) && !eOtherExclude.Get(pkg)) {
  583. toEdit = append(toEdit, base)
  584. }
  585. }
  586. }
  587. return toEdit, nil
  588. }
  589. func updatePkgbuildSeenRef(bases []Base) error {
  590. var errMulti multierror.MultiError
  591. for _, base := range bases {
  592. pkg := base.Pkgbase()
  593. err := gitUpdateSeenRef(config.BuildDir, pkg)
  594. if err != nil {
  595. errMulti.Add(err)
  596. }
  597. }
  598. return errMulti.Return()
  599. }
  600. func showPkgbuildDiffs(bases []Base, cloned stringset.StringSet) error {
  601. var errMulti multierror.MultiError
  602. for _, base := range bases {
  603. pkg := base.Pkgbase()
  604. dir := filepath.Join(config.BuildDir, pkg)
  605. start, err := getLastSeenHash(config.BuildDir, pkg)
  606. if err != nil {
  607. errMulti.Add(err)
  608. continue
  609. }
  610. if cloned.Get(pkg) {
  611. start = gitEmptyTree
  612. } else {
  613. hasDiff, err := gitHasDiff(config.BuildDir, pkg)
  614. if err != nil {
  615. errMulti.Add(err)
  616. continue
  617. }
  618. if !hasDiff {
  619. text.Warnln(gotext.Get("%s: No changes -- skipping", cyan(base.String())))
  620. continue
  621. }
  622. }
  623. args := []string{"diff",
  624. start + "..HEAD@{upstream}", "--src-prefix",
  625. dir + "/", "--dst-prefix", dir + "/", "--", ".", ":(exclude).SRCINFO"}
  626. if text.UseColor {
  627. args = append(args, "--color=always")
  628. } else {
  629. args = append(args, "--color=never")
  630. }
  631. _ = show(passToGit(dir, args...))
  632. }
  633. return errMulti.Return()
  634. }
  635. func editPkgbuilds(bases []Base, srcinfos map[string]*gosrc.Srcinfo) error {
  636. pkgbuilds := make([]string, 0, len(bases))
  637. for _, base := range bases {
  638. pkg := base.Pkgbase()
  639. dir := filepath.Join(config.BuildDir, pkg)
  640. pkgbuilds = append(pkgbuilds, filepath.Join(dir, "PKGBUILD"))
  641. for _, splitPkg := range srcinfos[pkg].SplitPackages() {
  642. if splitPkg.Install != "" {
  643. pkgbuilds = append(pkgbuilds, filepath.Join(dir, splitPkg.Install))
  644. }
  645. }
  646. }
  647. if len(pkgbuilds) > 0 {
  648. editor, editorArgs := editor()
  649. editorArgs = append(editorArgs, pkgbuilds...)
  650. editcmd := exec.Command(editor, editorArgs...)
  651. editcmd.Stdin, editcmd.Stdout, editcmd.Stderr = os.Stdin, os.Stdout, os.Stderr
  652. err := editcmd.Run()
  653. if err != nil {
  654. return errors.New(gotext.Get("editor did not exit successfully, aborting: %s", err))
  655. }
  656. }
  657. return nil
  658. }
  659. func parseSrcinfoFiles(bases []Base, errIsFatal bool) (map[string]*gosrc.Srcinfo, error) {
  660. srcinfos := make(map[string]*gosrc.Srcinfo)
  661. for k, base := range bases {
  662. pkg := base.Pkgbase()
  663. dir := filepath.Join(config.BuildDir, pkg)
  664. text.OperationInfoln(gotext.Get("(%d/%d) Parsing SRCINFO: %s", k+1, len(bases), cyan(base.String())))
  665. pkgbuild, err := gosrc.ParseFile(filepath.Join(dir, ".SRCINFO"))
  666. if err != nil {
  667. if !errIsFatal {
  668. text.Warnln(gotext.Get("failed to parse %s -- skipping: %s", base.String(), err))
  669. continue
  670. }
  671. return nil, errors.New(gotext.Get("failed to parse %s: %s", base.String(), err))
  672. }
  673. srcinfos[pkg] = pkgbuild
  674. }
  675. return srcinfos, nil
  676. }
  677. func pkgbuildsToSkip(bases []Base, targets stringset.StringSet) stringset.StringSet {
  678. toSkip := make(stringset.StringSet)
  679. for _, base := range bases {
  680. isTarget := false
  681. for _, pkg := range base {
  682. isTarget = isTarget || targets.Get(pkg.Name)
  683. }
  684. if (config.ReDownload == "yes" && isTarget) || config.ReDownload == "all" {
  685. continue
  686. }
  687. dir := filepath.Join(config.BuildDir, base.Pkgbase(), ".SRCINFO")
  688. pkgbuild, err := gosrc.ParseFile(dir)
  689. if err == nil {
  690. if alpm.VerCmp(pkgbuild.Version(), base.Version()) >= 0 {
  691. toSkip.Set(base.Pkgbase())
  692. }
  693. }
  694. }
  695. return toSkip
  696. }
  697. func mergePkgbuilds(bases []Base) error {
  698. for _, base := range bases {
  699. err := gitMerge(config.BuildDir, base.Pkgbase())
  700. if err != nil {
  701. return err
  702. }
  703. }
  704. return nil
  705. }
  706. func downloadPkgbuilds(bases []Base, toSkip stringset.StringSet, buildDir string) (stringset.StringSet, error) {
  707. cloned := make(stringset.StringSet)
  708. downloaded := 0
  709. var wg sync.WaitGroup
  710. var mux sync.Mutex
  711. var errs multierror.MultiError
  712. download := func(base Base) {
  713. defer wg.Done()
  714. pkg := base.Pkgbase()
  715. if toSkip.Get(pkg) {
  716. mux.Lock()
  717. downloaded++
  718. text.OperationInfoln(
  719. gotext.Get("PKGBUILD up to date, Skipping (%d/%d): %s",
  720. downloaded, len(bases), cyan(base.String())))
  721. mux.Unlock()
  722. return
  723. }
  724. clone, err := gitDownload(config.AURURL+"/"+pkg+".git", buildDir, pkg)
  725. if err != nil {
  726. errs.Add(err)
  727. return
  728. }
  729. if clone {
  730. mux.Lock()
  731. cloned.Set(pkg)
  732. mux.Unlock()
  733. }
  734. mux.Lock()
  735. downloaded++
  736. text.OperationInfoln(gotext.Get("Downloaded PKGBUILD (%d/%d): %s", downloaded, len(bases), cyan(base.String())))
  737. mux.Unlock()
  738. }
  739. count := 0
  740. for _, base := range bases {
  741. wg.Add(1)
  742. go download(base)
  743. count++
  744. if count%25 == 0 {
  745. wg.Wait()
  746. }
  747. }
  748. wg.Wait()
  749. return cloned, errs.Return()
  750. }
  751. func downloadPkgbuildsSources(bases []Base, incompatible stringset.StringSet) (err error) {
  752. for _, base := range bases {
  753. pkg := base.Pkgbase()
  754. dir := filepath.Join(config.BuildDir, pkg)
  755. args := []string{"--verifysource", "-Ccf"}
  756. if incompatible.Get(pkg) {
  757. args = append(args, "--ignorearch")
  758. }
  759. err = show(passToMakepkg(dir, args...))
  760. if err != nil {
  761. return errors.New(gotext.Get("error downloading sources: %s", cyan(base.String())))
  762. }
  763. }
  764. return
  765. }
  766. func buildInstallPkgbuilds(
  767. dp *depPool,
  768. do *depOrder,
  769. srcinfos map[string]*gosrc.Srcinfo,
  770. parser *settings.Arguments,
  771. incompatible stringset.StringSet,
  772. conflicts stringset.MapStringSet,
  773. alpmHandle *alpm.Handle) error {
  774. arguments := parser.Copy()
  775. arguments.ClearTargets()
  776. arguments.Op = "U"
  777. arguments.DelArg("confirm")
  778. arguments.DelArg("noconfirm")
  779. arguments.DelArg("c", "clean")
  780. arguments.DelArg("q", "quiet")
  781. arguments.DelArg("q", "quiet")
  782. arguments.DelArg("y", "refresh")
  783. arguments.DelArg("u", "sysupgrade")
  784. arguments.DelArg("w", "downloadonly")
  785. deps := make([]string, 0)
  786. exp := make([]string, 0)
  787. oldConfirm := config.NoConfirm
  788. config.NoConfirm = true
  789. //remotenames: names of all non repo packages on the system
  790. _, _, localNames, remoteNames, err := query.FilterPackages(alpmHandle)
  791. if err != nil {
  792. return err
  793. }
  794. // cache as a stringset. maybe make it return a string set in the first
  795. // place
  796. remoteNamesCache := stringset.FromSlice(remoteNames)
  797. localNamesCache := stringset.FromSlice(localNames)
  798. doInstall := func() error {
  799. if len(arguments.Targets) == 0 {
  800. return nil
  801. }
  802. if errShow := show(passToPacman(arguments)); errShow != nil {
  803. return errShow
  804. }
  805. err = saveVCSInfo(config.Runtime.VCSPath)
  806. if err != nil {
  807. fmt.Fprintln(os.Stderr, err)
  808. }
  809. if errDeps := asdeps(parser, deps); err != nil {
  810. return errDeps
  811. }
  812. if errExps := asexp(parser, exp); err != nil {
  813. return errExps
  814. }
  815. config.NoConfirm = oldConfirm
  816. arguments.ClearTargets()
  817. deps = make([]string, 0)
  818. exp = make([]string, 0)
  819. config.NoConfirm = true
  820. return nil
  821. }
  822. for _, base := range do.Aur {
  823. pkg := base.Pkgbase()
  824. dir := filepath.Join(config.BuildDir, pkg)
  825. built := true
  826. satisfied := true
  827. all:
  828. for _, pkg := range base {
  829. for _, deps := range [3][]string{pkg.Depends, pkg.MakeDepends, pkg.CheckDepends} {
  830. for _, dep := range deps {
  831. if _, errSatisfier := dp.LocalDB.PkgCache().FindSatisfier(dep); errSatisfier != nil {
  832. satisfied = false
  833. text.Warnln(gotext.Get("%s not satisfied, flushing install queue", dep))
  834. break all
  835. }
  836. }
  837. }
  838. }
  839. if !satisfied || !config.BatchInstall {
  840. err = doInstall()
  841. if err != nil {
  842. return err
  843. }
  844. }
  845. srcinfo := srcinfos[pkg]
  846. args := []string{"--nobuild", "-fC"}
  847. if incompatible.Get(pkg) {
  848. args = append(args, "--ignorearch")
  849. }
  850. // pkgver bump
  851. if err = show(passToMakepkg(dir, args...)); err != nil {
  852. return errors.New(gotext.Get("error making: %s", base.String()))
  853. }
  854. pkgdests, pkgVersion, errList := parsePackageList(dir)
  855. if errList != nil {
  856. return errList
  857. }
  858. isExplicit := false
  859. for _, b := range base {
  860. isExplicit = isExplicit || dp.Explicit.Get(b.Name)
  861. }
  862. if config.ReBuild == "no" || (config.ReBuild == "yes" && !isExplicit) {
  863. for _, split := range base {
  864. pkgdest, ok := pkgdests[split.Name]
  865. if !ok {
  866. return errors.New(gotext.Get("could not find PKGDEST for: %s", split.Name))
  867. }
  868. if _, errStat := os.Stat(pkgdest); os.IsNotExist(errStat) {
  869. built = false
  870. } else if errStat != nil {
  871. return errStat
  872. }
  873. }
  874. } else {
  875. built = false
  876. }
  877. if cmdArgs.ExistsArg("needed") {
  878. installed := true
  879. for _, split := range base {
  880. if alpmpkg := dp.LocalDB.Pkg(split.Name); alpmpkg == nil || alpmpkg.Version() != pkgVersion {
  881. installed = false
  882. }
  883. }
  884. if installed {
  885. err = show(passToMakepkg(dir, "-c", "--nobuild", "--noextract", "--ignorearch"))
  886. if err != nil {
  887. return errors.New(gotext.Get("error making: %s", err))
  888. }
  889. fmt.Fprintln(os.Stdout, gotext.Get("%s is up to date -- skipping", cyan(pkg+"-"+pkgVersion)))
  890. continue
  891. }
  892. }
  893. if built {
  894. err = show(passToMakepkg(dir, "-c", "--nobuild", "--noextract", "--ignorearch"))
  895. if err != nil {
  896. return errors.New(gotext.Get("error making: %s", err))
  897. }
  898. text.Warnln(gotext.Get("%s already made -- skipping build", cyan(pkg+"-"+pkgVersion)))
  899. } else {
  900. args := []string{"-cf", "--noconfirm", "--noextract", "--noprepare", "--holdver"}
  901. if incompatible.Get(pkg) {
  902. args = append(args, "--ignorearch")
  903. }
  904. if errMake := show(passToMakepkg(dir, args...)); errMake != nil {
  905. return errors.New(gotext.Get("error making: %s", base.String))
  906. }
  907. }
  908. // conflicts have been checked so answer y for them
  909. if config.UseAsk && cmdArgs.ExistsArg("ask") {
  910. ask, _ := strconv.Atoi(cmdArgs.Options["ask"].First())
  911. uask := alpm.QuestionType(ask) | alpm.QuestionTypeConflictPkg
  912. cmdArgs.Options["ask"].Set(fmt.Sprint(uask))
  913. } else {
  914. for _, split := range base {
  915. if _, ok := conflicts[split.Name]; ok {
  916. config.NoConfirm = false
  917. break
  918. }
  919. }
  920. }
  921. doAddTarget := func(name string, optional bool) error {
  922. pkgdest, ok := pkgdests[name]
  923. if !ok {
  924. if optional {
  925. return nil
  926. }
  927. return errors.New(gotext.Get("could not find PKGDEST for: %s", name))
  928. }
  929. if _, errStat := os.Stat(pkgdest); os.IsNotExist(errStat) {
  930. if optional {
  931. return nil
  932. }
  933. return errors.New(
  934. gotext.Get(
  935. "the PKGDEST for %s is listed by makepkg but does not exist: %s",
  936. name, pkgdest))
  937. }
  938. arguments.AddTarget(pkgdest)
  939. if parser.ExistsArg("asdeps", "asdep") {
  940. deps = append(deps, name)
  941. } else if parser.ExistsArg("asexplicit", "asexp") {
  942. exp = append(exp, name)
  943. } else if !dp.Explicit.Get(name) && !localNamesCache.Get(name) && !remoteNamesCache.Get(name) {
  944. deps = append(deps, name)
  945. }
  946. return nil
  947. }
  948. for _, split := range base {
  949. if errAdd := doAddTarget(split.Name, false); errAdd != nil {
  950. return errAdd
  951. }
  952. if errAddDebug := doAddTarget(split.Name+"-debug", true); errAddDebug != nil {
  953. return errAddDebug
  954. }
  955. }
  956. var mux sync.Mutex
  957. var wg sync.WaitGroup
  958. for _, pkg := range base {
  959. wg.Add(1)
  960. go updateVCSData(config.Runtime.VCSPath, pkg.Name, srcinfo.Source, &mux, &wg)
  961. }
  962. wg.Wait()
  963. }
  964. err = doInstall()
  965. config.NoConfirm = oldConfirm
  966. return err
  967. }