exec.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. package main
  2. import (
  3. "bytes"
  4. "fmt"
  5. "os"
  6. "os/exec"
  7. "path/filepath"
  8. "strings"
  9. "time"
  10. )
  11. func show(cmd *exec.Cmd) error {
  12. cmd.Stdin, cmd.Stdout, cmd.Stderr = os.Stdin, os.Stdout, os.Stderr
  13. err := cmd.Run()
  14. if err != nil {
  15. return fmt.Errorf("")
  16. }
  17. return nil
  18. }
  19. func capture(cmd *exec.Cmd) (string, string, error) {
  20. var outbuf, errbuf bytes.Buffer
  21. cmd.Stdout = &outbuf
  22. cmd.Stderr = &errbuf
  23. err := cmd.Run()
  24. stdout := strings.TrimSpace(outbuf.String())
  25. stderr := strings.TrimSpace(errbuf.String())
  26. return stdout, stderr, err
  27. }
  28. func sudoLoopBackground() {
  29. updateSudo()
  30. go sudoLoop()
  31. }
  32. func sudoLoop() {
  33. for {
  34. updateSudo()
  35. time.Sleep(298 * time.Second)
  36. }
  37. }
  38. func updateSudo() {
  39. for {
  40. err := show(exec.Command("sudo", "-v"))
  41. if err != nil {
  42. fmt.Fprintln(os.Stderr, err)
  43. } else {
  44. break
  45. }
  46. }
  47. }
  48. // waitLock will lock yay checking the status of db.lck until it does not exist
  49. func waitLock() {
  50. if _, err := os.Stat(filepath.Join(pacmanConf.DBPath, "db.lck")); err != nil {
  51. return
  52. }
  53. fmt.Println(bold(yellow(smallArrow)), filepath.Join(pacmanConf.DBPath, "db.lck"), "is present.")
  54. fmt.Print(bold(yellow(smallArrow)), " There may be another Pacman instance running. Waiting...")
  55. for {
  56. time.Sleep(3 * time.Second)
  57. if _, err := os.Stat(filepath.Join(pacmanConf.DBPath, "db.lck")); err != nil {
  58. fmt.Println()
  59. return
  60. }
  61. }
  62. }
  63. func passToPacman(args *arguments) *exec.Cmd {
  64. argArr := make([]string, 0)
  65. if args.needRoot() {
  66. argArr = append(argArr, "sudo")
  67. }
  68. argArr = append(argArr, config.PacmanBin)
  69. argArr = append(argArr, cmdArgs.formatGlobals()...)
  70. argArr = append(argArr, args.formatArgs()...)
  71. if config.NoConfirm {
  72. argArr = append(argArr, "--noconfirm")
  73. }
  74. argArr = append(argArr, "--config", config.PacmanConf)
  75. argArr = append(argArr, "--")
  76. argArr = append(argArr, args.targets...)
  77. if args.needRoot() {
  78. waitLock()
  79. }
  80. return exec.Command(argArr[0], argArr[1:]...)
  81. }
  82. func passToMakepkg(dir string, args ...string) *exec.Cmd {
  83. mflags := strings.Fields(config.MFlags)
  84. args = append(args, mflags...)
  85. if config.MakepkgConf != "" {
  86. args = append(args, "--config", config.MakepkgConf)
  87. }
  88. cmd := exec.Command(config.MakepkgBin, args...)
  89. cmd.Dir = dir
  90. return cmd
  91. }
  92. func passToGit(dir string, _args ...string) *exec.Cmd {
  93. gitflags := strings.Fields(config.GitFlags)
  94. args := []string{"-C", dir}
  95. args = append(args, gitflags...)
  96. args = append(args, _args...)
  97. cmd := exec.Command(config.GitBin, args...)
  98. return cmd
  99. }
  100. func isTty() bool {
  101. cmd := exec.Command("test", "-t", "1")
  102. cmd.Stdout = os.Stdout
  103. err := cmd.Run()
  104. return err == nil
  105. }