54 lines
1.7 KiB
Go
54 lines
1.7 KiB
Go
// Copyright 2018 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package packagesdriver fetches type sizes for go/packages and go/analysis.
|
|
package packagesdriver
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"golang.org/x/tools/internal/gocommand"
|
|
)
|
|
|
|
func GetSizesForArgsGolist(ctx context.Context, inv gocommand.Invocation, gocmdRunner *gocommand.Runner) (string, string, error) {
|
|
inv.Verb = "list"
|
|
inv.Args = []string{"-f", "{{context.GOARCH}} {{context.Compiler}}", "--", "unsafe"}
|
|
stdout, stderr, friendlyErr, rawErr := gocmdRunner.RunRaw(ctx, inv)
|
|
var goarch, compiler string
|
|
if rawErr != nil {
|
|
rawErrMsg := rawErr.Error()
|
|
if strings.Contains(rawErrMsg, "cannot find main module") ||
|
|
strings.Contains(rawErrMsg, "go.mod file not found") {
|
|
// User's running outside of a module.
|
|
// All bets are off. Get GOARCH and guess compiler is gc.
|
|
// TODO(matloob): Is this a problem in practice?
|
|
inv.Verb = "env"
|
|
inv.Args = []string{"GOARCH"}
|
|
envout, enverr := gocmdRunner.Run(ctx, inv)
|
|
if enverr != nil {
|
|
return "", "", enverr
|
|
}
|
|
goarch = strings.TrimSpace(envout.String())
|
|
compiler = "gc"
|
|
} else if friendlyErr != nil {
|
|
return "", "", friendlyErr
|
|
} else {
|
|
// This should be unreachable, but be defensive
|
|
// in case RunRaw's error results are inconsistent.
|
|
return "", "", rawErr
|
|
}
|
|
} else {
|
|
fields := strings.Fields(stdout.String())
|
|
if len(fields) < 2 {
|
|
return "", "", fmt.Errorf("could not parse GOARCH and Go compiler in format \"<GOARCH> <compiler>\":\nstdout: <<%s>>\nstderr: <<%s>>",
|
|
stdout.String(), stderr.String())
|
|
}
|
|
goarch = fields[0]
|
|
compiler = fields[1]
|
|
}
|
|
return compiler, goarch, nil
|
|
}
|