install.go 28 KB

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