install.go 28 KB

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