version.go 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. // Copyright 2020 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package gocommand
  5. import (
  6. "context"
  7. "fmt"
  8. "strings"
  9. )
  10. // GoVersion checks the go version by running "go list" with modules off.
  11. // It returns the X in Go 1.X.
  12. func GoVersion(ctx context.Context, inv Invocation, r *Runner) (int, error) {
  13. inv.Verb = "list"
  14. inv.Args = []string{"-e", "-f", `{{context.ReleaseTags}}`}
  15. inv.Env = append(append([]string{}, inv.Env...), "GO111MODULE=off")
  16. // Unset any unneeded flags, and remove them from BuildFlags, if they're
  17. // present.
  18. inv.ModFile = ""
  19. inv.ModFlag = ""
  20. var buildFlags []string
  21. for _, flag := range inv.BuildFlags {
  22. // Flags can be prefixed by one or two dashes.
  23. f := strings.TrimPrefix(strings.TrimPrefix(flag, "-"), "-")
  24. if strings.HasPrefix(f, "mod=") || strings.HasPrefix(f, "modfile=") {
  25. continue
  26. }
  27. buildFlags = append(buildFlags, flag)
  28. }
  29. inv.BuildFlags = buildFlags
  30. stdoutBytes, err := r.Run(ctx, inv)
  31. if err != nil {
  32. return 0, err
  33. }
  34. stdout := stdoutBytes.String()
  35. if len(stdout) < 3 {
  36. return 0, fmt.Errorf("bad ReleaseTags output: %q", stdout)
  37. }
  38. // Split up "[go1.1 go1.15]"
  39. tags := strings.Fields(stdout[1 : len(stdout)-2])
  40. for i := len(tags) - 1; i >= 0; i-- {
  41. var version int
  42. if _, err := fmt.Sscanf(tags[i], "go1.%d", &version); err != nil {
  43. continue
  44. }
  45. return version, nil
  46. }
  47. return 0, fmt.Errorf("no parseable ReleaseTags in %v", tags)
  48. }