install.go 26 KB

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