install.go 24 KB

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