dep_graph.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825
  1. package dep
  2. import (
  3. "context"
  4. "fmt"
  5. "strconv"
  6. aurc "github.com/Jguer/aur"
  7. alpm "github.com/Jguer/go-alpm/v2"
  8. gosrc "github.com/Morganamilo/go-srcinfo"
  9. mapset "github.com/deckarep/golang-set/v2"
  10. "github.com/leonelquinteros/gotext"
  11. "github.com/Jguer/yay/v12/pkg/db"
  12. "github.com/Jguer/yay/v12/pkg/intrange"
  13. aur "github.com/Jguer/yay/v12/pkg/query"
  14. "github.com/Jguer/yay/v12/pkg/text"
  15. "github.com/Jguer/yay/v12/pkg/topo"
  16. )
  17. type InstallInfo struct {
  18. Source Source
  19. Reason Reason
  20. Version string
  21. LocalVersion string
  22. SrcinfoPath *string
  23. AURBase *string
  24. SyncDBName *string
  25. IsGroup bool
  26. Upgrade bool
  27. Devel bool
  28. }
  29. func (i *InstallInfo) String() string {
  30. return fmt.Sprintf("InstallInfo{Source: %v, Reason: %v}", i.Source, i.Reason)
  31. }
  32. type (
  33. Reason int
  34. Source int
  35. )
  36. func (r Reason) String() string {
  37. return ReasonNames[r]
  38. }
  39. func (s Source) String() string {
  40. return SourceNames[s]
  41. }
  42. const (
  43. Explicit Reason = iota // 0
  44. Dep // 1
  45. MakeDep // 2
  46. CheckDep // 3
  47. )
  48. var ReasonNames = map[Reason]string{
  49. Explicit: gotext.Get("Explicit"),
  50. Dep: gotext.Get("Dependency"),
  51. MakeDep: gotext.Get("Make Dependency"),
  52. CheckDep: gotext.Get("Check Dependency"),
  53. }
  54. const (
  55. AUR Source = iota
  56. Sync
  57. Local
  58. SrcInfo
  59. Missing
  60. )
  61. var SourceNames = map[Source]string{
  62. AUR: gotext.Get("AUR"),
  63. Sync: gotext.Get("Sync"),
  64. Local: gotext.Get("Local"),
  65. SrcInfo: gotext.Get("SRCINFO"),
  66. Missing: gotext.Get("Missing"),
  67. }
  68. var bgColorMap = map[Source]string{
  69. AUR: "lightblue",
  70. Sync: "lemonchiffon",
  71. Local: "darkolivegreen1",
  72. Missing: "tomato",
  73. }
  74. var colorMap = map[Reason]string{
  75. Explicit: "black",
  76. Dep: "deeppink",
  77. MakeDep: "navyblue",
  78. CheckDep: "forestgreen",
  79. }
  80. type Grapher struct {
  81. logger *text.Logger
  82. providerCache map[string][]aur.Pkg
  83. dbExecutor db.Executor
  84. aurClient aurc.QueryClient
  85. fullGraph bool // If true, the graph will include all dependencies including already installed ones or repo
  86. noConfirm bool // If true, the graph will not prompt for confirmation
  87. noDeps bool // If true, the graph will not include dependencies
  88. noCheckDeps bool // If true, the graph will not include check dependencies
  89. needed bool // If true, the graph will only include packages that are not installed
  90. }
  91. func NewGrapher(dbExecutor db.Executor, aurCache aurc.QueryClient,
  92. fullGraph, noConfirm, noDeps, noCheckDeps, needed bool,
  93. logger *text.Logger,
  94. ) *Grapher {
  95. return &Grapher{
  96. dbExecutor: dbExecutor,
  97. aurClient: aurCache,
  98. fullGraph: fullGraph,
  99. noConfirm: noConfirm,
  100. noDeps: noDeps,
  101. noCheckDeps: noCheckDeps,
  102. needed: needed,
  103. providerCache: make(map[string][]aurc.Pkg, 5),
  104. logger: logger,
  105. }
  106. }
  107. func (g *Grapher) GraphFromTargets(ctx context.Context,
  108. graph *topo.Graph[string, *InstallInfo], targets []string,
  109. ) (*topo.Graph[string, *InstallInfo], error) {
  110. if graph == nil {
  111. graph = topo.New[string, *InstallInfo]()
  112. }
  113. aurTargets := make([]string, 0, len(targets))
  114. for _, targetString := range targets {
  115. target := ToTarget(targetString)
  116. switch target.DB {
  117. case "": // unspecified db
  118. if pkg := g.dbExecutor.SyncSatisfier(target.Name); pkg != nil {
  119. dbName := pkg.DB().Name()
  120. reason := Explicit
  121. if localPkg := g.dbExecutor.LocalPackage(pkg.Name()); localPkg != nil {
  122. reason = Reason(localPkg.Reason())
  123. }
  124. g.GraphSyncPkg(ctx, graph, pkg, &InstallInfo{
  125. Source: Sync,
  126. Reason: reason,
  127. Version: pkg.Version(),
  128. SyncDBName: &dbName,
  129. })
  130. continue
  131. }
  132. groupPackages := g.dbExecutor.PackagesFromGroup(target.Name)
  133. if len(groupPackages) > 0 {
  134. dbName := groupPackages[0].DB().Name()
  135. graph.AddNode(target.Name)
  136. g.ValidateAndSetNodeInfo(graph, target.Name, &topo.NodeInfo[*InstallInfo]{
  137. Color: colorMap[Explicit],
  138. Background: bgColorMap[Sync],
  139. Value: &InstallInfo{
  140. Source: Sync,
  141. Reason: Explicit,
  142. Version: "",
  143. SyncDBName: &dbName,
  144. IsGroup: true,
  145. },
  146. })
  147. continue
  148. }
  149. fallthrough
  150. case "aur":
  151. aurTargets = append(aurTargets, target.Name)
  152. default:
  153. reason := Explicit
  154. if pkg := g.dbExecutor.LocalPackage(target.Name); pkg != nil {
  155. reason = Reason(pkg.Reason())
  156. }
  157. graph.AddNode(target.Name)
  158. g.ValidateAndSetNodeInfo(graph, target.Name, &topo.NodeInfo[*InstallInfo]{
  159. Color: colorMap[reason],
  160. Background: bgColorMap[Sync],
  161. Value: &InstallInfo{
  162. Source: Sync,
  163. Reason: reason,
  164. Version: target.Version,
  165. SyncDBName: &target.DB,
  166. },
  167. })
  168. }
  169. }
  170. var errA error
  171. graph, errA = g.GraphFromAUR(ctx, graph, aurTargets)
  172. if errA != nil {
  173. return nil, errA
  174. }
  175. return graph, nil
  176. }
  177. func (g *Grapher) pickSrcInfoPkgs(pkgs []*aurc.Pkg) ([]*aurc.Pkg, error) {
  178. final := make([]*aurc.Pkg, 0, len(pkgs))
  179. for i := range pkgs {
  180. g.logger.Println(text.Magenta(strconv.Itoa(i+1)+" ") + text.Bold(pkgs[i].Name) +
  181. " " + text.Cyan(pkgs[i].Version))
  182. g.logger.Println(" " + pkgs[i].Description)
  183. }
  184. g.logger.Infoln(gotext.Get("Packages to exclude") + " (eg: \"1 2 3\", \"1-3\", \"^4\"):")
  185. numberBuf, err := g.logger.GetInput("", g.noConfirm)
  186. if err != nil {
  187. return nil, err
  188. }
  189. include, exclude, _, otherExclude := intrange.ParseNumberMenu(numberBuf)
  190. isInclude := len(exclude) == 0 && len(otherExclude) == 0
  191. for i := 1; i <= len(pkgs); i++ {
  192. target := i - 1
  193. if isInclude && !include.Get(i) {
  194. final = append(final, pkgs[target])
  195. }
  196. if !isInclude && (exclude.Get(i)) {
  197. final = append(final, pkgs[target])
  198. }
  199. }
  200. return final, nil
  201. }
  202. func (g *Grapher) addAurPkgProvides(pkg *aurc.Pkg, graph *topo.Graph[string, *InstallInfo]) {
  203. for i := range pkg.Provides {
  204. depName, mod, version := splitDep(pkg.Provides[i])
  205. g.logger.Debugln(pkg.String() + " provides: " + depName)
  206. graph.Provides(depName, &alpm.Depend{
  207. Name: depName,
  208. Version: version,
  209. Mod: aurDepModToAlpmDep(mod),
  210. }, pkg.Name)
  211. }
  212. }
  213. func (g *Grapher) GraphFromSrcInfos(ctx context.Context, graph *topo.Graph[string, *InstallInfo],
  214. srcInfos map[string]*gosrc.Srcinfo,
  215. ) (*topo.Graph[string, *InstallInfo], error) {
  216. if graph == nil {
  217. graph = topo.New[string, *InstallInfo]()
  218. }
  219. aurPkgsAdded := []*aurc.Pkg{}
  220. for pkgBuildDir, pkgbuild := range srcInfos {
  221. pkgBuildDir := pkgBuildDir
  222. aurPkgs, err := makeAURPKGFromSrcinfo(g.dbExecutor, pkgbuild)
  223. if err != nil {
  224. return nil, err
  225. }
  226. if len(aurPkgs) > 1 {
  227. var errPick error
  228. aurPkgs, errPick = g.pickSrcInfoPkgs(aurPkgs)
  229. if errPick != nil {
  230. return nil, errPick
  231. }
  232. }
  233. for _, pkg := range aurPkgs {
  234. pkg := pkg
  235. reason := Explicit
  236. if pkg := g.dbExecutor.LocalPackage(pkg.Name); pkg != nil {
  237. reason = Reason(pkg.Reason())
  238. }
  239. graph.AddNode(pkg.Name)
  240. g.addAurPkgProvides(pkg, graph)
  241. g.ValidateAndSetNodeInfo(graph, pkg.Name, &topo.NodeInfo[*InstallInfo]{
  242. Color: colorMap[reason],
  243. Background: bgColorMap[AUR],
  244. Value: &InstallInfo{
  245. Source: SrcInfo,
  246. Reason: reason,
  247. SrcinfoPath: &pkgBuildDir,
  248. AURBase: &pkg.PackageBase,
  249. Version: pkg.Version,
  250. },
  251. })
  252. }
  253. aurPkgsAdded = append(aurPkgsAdded, aurPkgs...)
  254. }
  255. g.AddDepsForPkgs(ctx, aurPkgsAdded, graph)
  256. return graph, nil
  257. }
  258. func (g *Grapher) AddDepsForPkgs(ctx context.Context, pkgs []*aur.Pkg, graph *topo.Graph[string, *InstallInfo]) {
  259. for _, pkg := range pkgs {
  260. g.addDepNodes(ctx, pkg, graph)
  261. }
  262. }
  263. func (g *Grapher) addDepNodes(ctx context.Context, pkg *aur.Pkg, graph *topo.Graph[string, *InstallInfo]) {
  264. if len(pkg.MakeDepends) > 0 {
  265. g.addNodes(ctx, graph, pkg.Name, pkg.MakeDepends, MakeDep)
  266. }
  267. if !g.noDeps && len(pkg.Depends) > 0 {
  268. g.addNodes(ctx, graph, pkg.Name, pkg.Depends, Dep)
  269. }
  270. if !g.noCheckDeps && !g.noDeps && len(pkg.CheckDepends) > 0 {
  271. g.addNodes(ctx, graph, pkg.Name, pkg.CheckDepends, CheckDep)
  272. }
  273. }
  274. func (g *Grapher) GraphSyncPkg(ctx context.Context,
  275. graph *topo.Graph[string, *InstallInfo],
  276. pkg alpm.IPackage, instalInfo *InstallInfo,
  277. ) *topo.Graph[string, *InstallInfo] {
  278. if graph == nil {
  279. graph = topo.New[string, *InstallInfo]()
  280. }
  281. graph.AddNode(pkg.Name())
  282. _ = pkg.Provides().ForEach(func(p *alpm.Depend) error {
  283. g.logger.Debugln(pkg.Name() + " provides: " + p.String())
  284. graph.Provides(p.Name, p, pkg.Name())
  285. return nil
  286. })
  287. g.ValidateAndSetNodeInfo(graph, pkg.Name(), &topo.NodeInfo[*InstallInfo]{
  288. Color: colorMap[Reason(pkg.Reason())],
  289. Background: bgColorMap[Sync],
  290. Value: instalInfo,
  291. })
  292. return graph
  293. }
  294. func (g *Grapher) GraphAURTarget(ctx context.Context,
  295. graph *topo.Graph[string, *InstallInfo],
  296. pkg *aurc.Pkg, instalInfo *InstallInfo,
  297. ) *topo.Graph[string, *InstallInfo] {
  298. if graph == nil {
  299. graph = topo.New[string, *InstallInfo]()
  300. }
  301. graph.AddNode(pkg.Name)
  302. g.addAurPkgProvides(pkg, graph)
  303. g.ValidateAndSetNodeInfo(graph, pkg.Name, &topo.NodeInfo[*InstallInfo]{
  304. Color: colorMap[instalInfo.Reason],
  305. Background: bgColorMap[AUR],
  306. Value: instalInfo,
  307. })
  308. return graph
  309. }
  310. func (g *Grapher) GraphFromAUR(ctx context.Context,
  311. graph *topo.Graph[string, *InstallInfo],
  312. targets []string,
  313. ) (*topo.Graph[string, *InstallInfo], error) {
  314. if graph == nil {
  315. graph = topo.New[string, *InstallInfo]()
  316. }
  317. if len(targets) == 0 {
  318. return graph, nil
  319. }
  320. aurPkgs, errCache := g.aurClient.Get(ctx, &aurc.Query{By: aurc.Name, Needles: targets})
  321. if errCache != nil {
  322. g.logger.Errorln(errCache)
  323. }
  324. for i := range aurPkgs {
  325. pkg := &aurPkgs[i]
  326. if _, ok := g.providerCache[pkg.Name]; !ok {
  327. g.providerCache[pkg.Name] = []aurc.Pkg{*pkg}
  328. }
  329. }
  330. aurPkgsAdded := []*aurc.Pkg{}
  331. for _, target := range targets {
  332. if cachedProvidePkg, ok := g.providerCache[target]; ok {
  333. aurPkgs = cachedProvidePkg
  334. } else {
  335. var errA error
  336. aurPkgs, errA = g.aurClient.Get(ctx, &aurc.Query{By: aurc.Provides, Needles: []string{target}, Contains: true})
  337. if errA != nil {
  338. g.logger.Errorln(gotext.Get("Failed to find AUR package for"), " ", target, ":", errA)
  339. }
  340. }
  341. if len(aurPkgs) == 0 {
  342. g.logger.Errorln(gotext.Get("No AUR package found for"), " ", target)
  343. continue
  344. }
  345. aurPkg := &aurPkgs[0]
  346. if len(aurPkgs) > 1 {
  347. chosen := g.provideMenu(target, aurPkgs)
  348. aurPkg = chosen
  349. g.providerCache[target] = []aurc.Pkg{*aurPkg}
  350. }
  351. reason := Explicit
  352. if pkg := g.dbExecutor.LocalPackage(aurPkg.Name); pkg != nil {
  353. reason = Reason(pkg.Reason())
  354. if g.needed {
  355. if db.VerCmp(pkg.Version(), aurPkg.Version) >= 0 {
  356. g.logger.Warnln(gotext.Get("%s is up to date -- skipping", text.Cyan(pkg.Name()+"-"+pkg.Version())))
  357. continue
  358. }
  359. }
  360. }
  361. graph = g.GraphAURTarget(ctx, graph, aurPkg, &InstallInfo{
  362. AURBase: &aurPkg.PackageBase,
  363. Reason: reason,
  364. Source: AUR,
  365. Version: aurPkg.Version,
  366. })
  367. aurPkgsAdded = append(aurPkgsAdded, aurPkg)
  368. }
  369. g.AddDepsForPkgs(ctx, aurPkgsAdded, graph)
  370. return graph, nil
  371. }
  372. // Removes found deps from the deps mapset and returns the found deps.
  373. func (g *Grapher) findDepsFromAUR(ctx context.Context,
  374. deps mapset.Set[string],
  375. ) []aurc.Pkg {
  376. pkgsToAdd := make([]aurc.Pkg, 0, deps.Cardinality())
  377. if deps.Cardinality() == 0 {
  378. return []aurc.Pkg{}
  379. }
  380. missingNeedles := make([]string, 0, deps.Cardinality())
  381. for _, depString := range deps.ToSlice() {
  382. if _, ok := g.providerCache[depString]; !ok {
  383. depName, _, _ := splitDep(depString)
  384. missingNeedles = append(missingNeedles, depName)
  385. }
  386. }
  387. if len(missingNeedles) != 0 {
  388. g.logger.Debugln("deps to find", missingNeedles)
  389. // provider search is more demanding than a simple search
  390. // try to find name match if possible and then try to find provides.
  391. aurPkgs, errCache := g.aurClient.Get(ctx, &aurc.Query{
  392. By: aurc.Name, Needles: missingNeedles, Contains: false,
  393. })
  394. if errCache != nil {
  395. g.logger.Errorln(errCache)
  396. }
  397. for i := range aurPkgs {
  398. pkg := &aurPkgs[i]
  399. if deps.Contains(pkg.Name) {
  400. g.providerCache[pkg.Name] = append(g.providerCache[pkg.Name], *pkg)
  401. }
  402. for _, val := range pkg.Provides {
  403. if val == pkg.Name {
  404. continue
  405. }
  406. if deps.Contains(val) {
  407. g.providerCache[val] = append(g.providerCache[val], *pkg)
  408. }
  409. }
  410. }
  411. }
  412. for _, depString := range deps.ToSlice() {
  413. var aurPkgs []aurc.Pkg
  414. depName, _, _ := splitDep(depString)
  415. if cachedProvidePkg, ok := g.providerCache[depString]; ok {
  416. aurPkgs = cachedProvidePkg
  417. } else {
  418. var errA error
  419. aurPkgs, errA = g.aurClient.Get(ctx, &aurc.Query{By: aurc.Provides, Needles: []string{depName}, Contains: true})
  420. if errA != nil {
  421. g.logger.Errorln(gotext.Get("Failed to find AUR package for"), depString, ":", errA)
  422. }
  423. }
  424. // remove packages that don't satisfy the dependency
  425. for i := 0; i < len(aurPkgs); i++ {
  426. if !satisfiesAur(depString, &aurPkgs[i]) {
  427. aurPkgs = append(aurPkgs[:i], aurPkgs[i+1:]...)
  428. i--
  429. }
  430. }
  431. if len(aurPkgs) == 0 {
  432. g.logger.Errorln(gotext.Get("No AUR package found for"), " ", depString)
  433. continue
  434. }
  435. pkg := aurPkgs[0]
  436. if len(aurPkgs) > 1 {
  437. chosen := g.provideMenu(depString, aurPkgs)
  438. pkg = *chosen
  439. }
  440. g.providerCache[depString] = []aurc.Pkg{pkg}
  441. deps.Remove(depString)
  442. pkgsToAdd = append(pkgsToAdd, pkg)
  443. }
  444. return pkgsToAdd
  445. }
  446. func (g *Grapher) ValidateAndSetNodeInfo(graph *topo.Graph[string, *InstallInfo],
  447. node string, nodeInfo *topo.NodeInfo[*InstallInfo],
  448. ) {
  449. info := graph.GetNodeInfo(node)
  450. if info != nil && info.Value != nil {
  451. if info.Value.Reason < nodeInfo.Value.Reason {
  452. return // refuse to downgrade reason
  453. }
  454. if info.Value.Upgrade {
  455. return // refuse to overwrite an upgrade
  456. }
  457. }
  458. graph.SetNodeInfo(node, nodeInfo)
  459. }
  460. func (g *Grapher) addNodes(
  461. ctx context.Context,
  462. graph *topo.Graph[string, *InstallInfo],
  463. parentPkgName string,
  464. deps []string,
  465. depType Reason,
  466. ) {
  467. targetsToFind := mapset.NewThreadUnsafeSet(deps...)
  468. // Check if in graph already
  469. for _, depString := range targetsToFind.ToSlice() {
  470. depName, _, _ := splitDep(depString)
  471. if !graph.Exists(depName) && !graph.ProvidesExists(depName) {
  472. continue
  473. }
  474. if graph.Exists(depName) {
  475. if err := graph.DependOn(depName, parentPkgName); err != nil {
  476. g.logger.Warnln(depString, parentPkgName, err)
  477. }
  478. targetsToFind.Remove(depString)
  479. }
  480. if p := graph.GetProviderNode(depName); p != nil {
  481. if provideSatisfies(p.String(), depString, p.Version) {
  482. if err := graph.DependOn(p.Provider, parentPkgName); err != nil {
  483. g.logger.Warnln(p.Provider, parentPkgName, err)
  484. }
  485. targetsToFind.Remove(depString)
  486. }
  487. }
  488. }
  489. // Check installed
  490. for _, depString := range targetsToFind.ToSlice() {
  491. depName, _, _ := splitDep(depString)
  492. if !g.dbExecutor.LocalSatisfierExists(depString) {
  493. continue
  494. }
  495. if g.fullGraph {
  496. g.ValidateAndSetNodeInfo(
  497. graph,
  498. depName,
  499. &topo.NodeInfo[*InstallInfo]{Color: colorMap[depType], Background: bgColorMap[Local]})
  500. if err := graph.DependOn(depName, parentPkgName); err != nil {
  501. g.logger.Warnln(depName, parentPkgName, err)
  502. }
  503. }
  504. targetsToFind.Remove(depString)
  505. }
  506. // Check Sync
  507. for _, depString := range targetsToFind.ToSlice() {
  508. alpmPkg := g.dbExecutor.SyncSatisfier(depString)
  509. if alpmPkg == nil {
  510. continue
  511. }
  512. if err := graph.DependOn(alpmPkg.Name(), parentPkgName); err != nil {
  513. g.logger.Warnln("repo dep warn:", depString, parentPkgName, err)
  514. }
  515. dbName := alpmPkg.DB().Name()
  516. g.ValidateAndSetNodeInfo(
  517. graph,
  518. alpmPkg.Name(),
  519. &topo.NodeInfo[*InstallInfo]{
  520. Color: colorMap[depType],
  521. Background: bgColorMap[Sync],
  522. Value: &InstallInfo{
  523. Source: Sync,
  524. Reason: depType,
  525. Version: alpmPkg.Version(),
  526. SyncDBName: &dbName,
  527. },
  528. })
  529. if newDeps := alpmPkg.Depends().Slice(); len(newDeps) != 0 && g.fullGraph {
  530. newDepsSlice := make([]string, 0, len(newDeps))
  531. for _, newDep := range newDeps {
  532. newDepsSlice = append(newDepsSlice, newDep.Name)
  533. }
  534. g.addNodes(ctx, graph, alpmPkg.Name(), newDepsSlice, Dep)
  535. }
  536. targetsToFind.Remove(depString)
  537. }
  538. // Check AUR
  539. pkgsToAdd := g.findDepsFromAUR(ctx, targetsToFind)
  540. for i := range pkgsToAdd {
  541. aurPkg := &pkgsToAdd[i]
  542. if err := graph.DependOn(aurPkg.Name, parentPkgName); err != nil {
  543. g.logger.Warnln("aur dep warn:", aurPkg.Name, parentPkgName, err)
  544. }
  545. graph.SetNodeInfo(
  546. aurPkg.Name,
  547. &topo.NodeInfo[*InstallInfo]{
  548. Color: colorMap[depType],
  549. Background: bgColorMap[AUR],
  550. Value: &InstallInfo{
  551. Source: AUR,
  552. Reason: depType,
  553. AURBase: &aurPkg.PackageBase,
  554. Version: aurPkg.Version,
  555. },
  556. })
  557. g.addDepNodes(ctx, aurPkg, graph)
  558. }
  559. // Add missing to graph
  560. for _, depString := range targetsToFind.ToSlice() {
  561. depName, mod, ver := splitDep(depString)
  562. // no dep found. add as missing
  563. if err := graph.DependOn(depName, parentPkgName); err != nil {
  564. g.logger.Warnln("missing dep warn:", depString, parentPkgName, err)
  565. }
  566. graph.SetNodeInfo(depName, &topo.NodeInfo[*InstallInfo]{
  567. Color: colorMap[depType],
  568. Background: bgColorMap[Missing],
  569. Value: &InstallInfo{
  570. Source: Missing,
  571. Reason: depType,
  572. Version: fmt.Sprintf("%s%s", mod, ver),
  573. },
  574. })
  575. }
  576. }
  577. func (g *Grapher) provideMenu(dep string, options []aur.Pkg) *aur.Pkg {
  578. size := len(options)
  579. if size == 1 {
  580. return &options[0]
  581. }
  582. str := text.Bold(gotext.Get("There are %d providers available for %s:", size, dep))
  583. str += "\n"
  584. size = 1
  585. str += g.logger.SprintOperationInfo(gotext.Get("Repository AUR"), "\n ")
  586. for i := range options {
  587. str += fmt.Sprintf("%d) %s ", size, options[i].Name)
  588. size++
  589. }
  590. g.logger.OperationInfoln(str)
  591. for {
  592. g.logger.Println(gotext.Get("\nEnter a number (default=1): "))
  593. if g.noConfirm {
  594. g.logger.Println("1")
  595. return &options[0]
  596. }
  597. numberBuf, err := g.logger.GetInput("", false)
  598. if err != nil {
  599. g.logger.Errorln(err)
  600. break
  601. }
  602. if numberBuf == "" {
  603. return &options[0]
  604. }
  605. num, err := strconv.Atoi(numberBuf)
  606. if err != nil {
  607. g.logger.Errorln(gotext.Get("invalid number: %s", numberBuf))
  608. continue
  609. }
  610. if num < 1 || num >= size {
  611. g.logger.Errorln(gotext.Get("invalid value: %d is not between %d and %d",
  612. num, 1, size-1))
  613. continue
  614. }
  615. return &options[num-1]
  616. }
  617. return nil
  618. }
  619. func makeAURPKGFromSrcinfo(dbExecutor db.Executor, srcInfo *gosrc.Srcinfo) ([]*aur.Pkg, error) {
  620. pkgs := make([]*aur.Pkg, 0, 1)
  621. alpmArch, err := dbExecutor.AlpmArchitectures()
  622. if err != nil {
  623. return nil, err
  624. }
  625. alpmArch = append(alpmArch, "") // srcinfo assumes no value as ""
  626. getDesc := func(pkg *gosrc.Package) string {
  627. if pkg.Pkgdesc != "" {
  628. return pkg.Pkgdesc
  629. }
  630. return srcInfo.Pkgdesc
  631. }
  632. for i := range srcInfo.Packages {
  633. pkg := &srcInfo.Packages[i]
  634. pkgs = append(pkgs, &aur.Pkg{
  635. ID: 0,
  636. Name: pkg.Pkgname,
  637. PackageBaseID: 0,
  638. PackageBase: srcInfo.Pkgbase,
  639. Version: srcInfo.Version(),
  640. Description: getDesc(pkg),
  641. URL: pkg.URL,
  642. Depends: append(archStringToString(alpmArch, pkg.Depends),
  643. archStringToString(alpmArch, srcInfo.Package.Depends)...),
  644. MakeDepends: archStringToString(alpmArch, srcInfo.PackageBase.MakeDepends),
  645. CheckDepends: archStringToString(alpmArch, srcInfo.PackageBase.CheckDepends),
  646. Conflicts: append(archStringToString(alpmArch, pkg.Conflicts),
  647. archStringToString(alpmArch, srcInfo.Package.Conflicts)...),
  648. Provides: append(archStringToString(alpmArch, pkg.Provides),
  649. archStringToString(alpmArch, srcInfo.Package.Provides)...),
  650. Replaces: append(archStringToString(alpmArch, pkg.Replaces),
  651. archStringToString(alpmArch, srcInfo.Package.Replaces)...),
  652. OptDepends: []string{},
  653. Groups: pkg.Groups,
  654. License: pkg.License,
  655. Keywords: []string{},
  656. })
  657. }
  658. return pkgs, nil
  659. }
  660. func archStringToString(alpmArches []string, archString []gosrc.ArchString) []string {
  661. pkgs := make([]string, 0, len(archString))
  662. for _, arch := range archString {
  663. if db.ArchIsSupported(alpmArches, arch.Arch) {
  664. pkgs = append(pkgs, arch.Value)
  665. }
  666. }
  667. return pkgs
  668. }
  669. func aurDepModToAlpmDep(mod string) alpm.DepMod {
  670. switch mod {
  671. case "=":
  672. return alpm.DepModEq
  673. case ">=":
  674. return alpm.DepModGE
  675. case "<=":
  676. return alpm.DepModLE
  677. case ">":
  678. return alpm.DepModGT
  679. case "<":
  680. return alpm.DepModLT
  681. }
  682. return alpm.DepModAny
  683. }