install.go 29 KB

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