--- /dev/null
+// Copyright 2020 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 cache
+
+import (
+ "context"
+ "fmt"
+ "go/ast"
+ "io/ioutil"
+ "os"
+ "path/filepath"
+ "sort"
+ "strconv"
+ "strings"
+
+ "golang.org/x/mod/modfile"
+ "golang.org/x/tools/internal/event"
+ "golang.org/x/tools/internal/gocommand"
+ "golang.org/x/tools/internal/lsp/command"
+ "golang.org/x/tools/internal/lsp/debug/tag"
+ "golang.org/x/tools/internal/lsp/diff"
+ "golang.org/x/tools/internal/lsp/protocol"
+ "golang.org/x/tools/internal/lsp/source"
+ "golang.org/x/tools/internal/memoize"
+ "golang.org/x/tools/internal/span"
+)
+
+type modTidyKey struct {
+ sessionID string
+ env string
+ gomod source.FileIdentity
+ imports string
+ unsavedOverlays string
+ view string
+}
+
+type modTidyHandle struct {
+ handle *memoize.Handle
+}
+
+type modTidyData struct {
+ tidied *source.TidiedModule
+ err error
+}
+
+func (mth *modTidyHandle) tidy(ctx context.Context, snapshot *snapshot) (*source.TidiedModule, error) {
+ v, err := mth.handle.Get(ctx, snapshot.generation, snapshot)
+ if err != nil {
+ return nil, err
+ }
+ data := v.(*modTidyData)
+ return data.tidied, data.err
+}
+
+func (s *snapshot) ModTidy(ctx context.Context, pm *source.ParsedModule) (*source.TidiedModule, error) {
+ if pm.File == nil {
+ return nil, fmt.Errorf("cannot tidy unparseable go.mod file: %v", pm.URI)
+ }
+ if handle := s.getModTidyHandle(pm.URI); handle != nil {
+ return handle.tidy(ctx, s)
+ }
+ fh, err := s.GetFile(ctx, pm.URI)
+ if err != nil {
+ return nil, err
+ }
+ // If the file handle is an overlay, it may not be written to disk.
+ // The go.mod file has to be on disk for `go mod tidy` to work.
+ if _, ok := fh.(*overlay); ok {
+ if info, _ := os.Stat(fh.URI().Filename()); info == nil {
+ return nil, source.ErrNoModOnDisk
+ }
+ }
+ if criticalErr := s.GetCriticalError(ctx); criticalErr != nil {
+ return &source.TidiedModule{
+ Diagnostics: criticalErr.DiagList,
+ }, nil
+ }
+ workspacePkgs, err := s.WorkspacePackages(ctx)
+ if err != nil {
+ return nil, err
+ }
+ importHash, err := hashImports(ctx, workspacePkgs)
+ if err != nil {
+ return nil, err
+ }
+
+ s.mu.Lock()
+ overlayHash := hashUnsavedOverlays(s.files)
+ s.mu.Unlock()
+
+ key := modTidyKey{
+ sessionID: s.view.session.id,
+ view: s.view.folder.Filename(),
+ imports: importHash,
+ unsavedOverlays: overlayHash,
+ gomod: fh.FileIdentity(),
+ env: hashEnv(s),
+ }
+ h := s.generation.Bind(key, func(ctx context.Context, arg memoize.Arg) interface{} {
+ ctx, done := event.Start(ctx, "cache.ModTidyHandle", tag.URI.Of(fh.URI()))
+ defer done()
+
+ snapshot := arg.(*snapshot)
+ inv := &gocommand.Invocation{
+ Verb: "mod",
+ Args: []string{"tidy"},
+ WorkingDir: filepath.Dir(fh.URI().Filename()),
+ }
+ tmpURI, inv, cleanup, err := snapshot.goCommandInvocation(ctx, source.WriteTemporaryModFile, inv)
+ if err != nil {
+ return &modTidyData{err: err}
+ }
+ // Keep the temporary go.mod file around long enough to parse it.
+ defer cleanup()
+
+ if _, err := s.view.session.gocmdRunner.Run(ctx, *inv); err != nil {
+ return &modTidyData{err: err}
+ }
+ // Go directly to disk to get the temporary mod file, since it is
+ // always on disk.
+ tempContents, err := ioutil.ReadFile(tmpURI.Filename())
+ if err != nil {
+ return &modTidyData{err: err}
+ }
+ ideal, err := modfile.Parse(tmpURI.Filename(), tempContents, nil)
+ if err != nil {
+ // We do not need to worry about the temporary file's parse errors
+ // since it has been "tidied".
+ return &modTidyData{err: err}
+ }
+ // Compare the original and tidied go.mod files to compute errors and
+ // suggested fixes.
+ diagnostics, err := modTidyDiagnostics(ctx, snapshot, pm, ideal, workspacePkgs)
+ if err != nil {
+ return &modTidyData{err: err}
+ }
+ return &modTidyData{
+ tidied: &source.TidiedModule{
+ Diagnostics: diagnostics,
+ TidiedContent: tempContents,
+ },
+ }
+ }, nil)
+
+ mth := &modTidyHandle{handle: h}
+ s.mu.Lock()
+ s.modTidyHandles[fh.URI()] = mth
+ s.mu.Unlock()
+
+ return mth.tidy(ctx, s)
+}
+
+func (s *snapshot) uriToModDecl(ctx context.Context, uri span.URI) (protocol.Range, error) {
+ fh, err := s.GetFile(ctx, uri)
+ if err != nil {
+ return protocol.Range{}, nil
+ }
+ pmf, err := s.ParseMod(ctx, fh)
+ if err != nil {
+ return protocol.Range{}, nil
+ }
+ if pmf.File.Module == nil || pmf.File.Module.Syntax == nil {
+ return protocol.Range{}, nil
+ }
+ return rangeFromPositions(pmf.Mapper, pmf.File.Module.Syntax.Start, pmf.File.Module.Syntax.End)
+}
+
+func hashImports(ctx context.Context, wsPackages []source.Package) (string, error) {
+ results := make(map[string]bool)
+ var imports []string
+ for _, pkg := range wsPackages {
+ for _, path := range pkg.Imports() {
+ imp := path.PkgPath()
+ if _, ok := results[imp]; !ok {
+ results[imp] = true
+ imports = append(imports, imp)
+ }
+ }
+ imports = append(imports, pkg.MissingDependencies()...)
+ }
+ sort.Strings(imports)
+ hashed := strings.Join(imports, ",")
+ return hashContents([]byte(hashed)), nil
+}
+
+// modTidyDiagnostics computes the differences between the original and tidied
+// go.mod files to produce diagnostic and suggested fixes. Some diagnostics
+// may appear on the Go files that import packages from missing modules.
+func modTidyDiagnostics(ctx context.Context, snapshot source.Snapshot, pm *source.ParsedModule, ideal *modfile.File, workspacePkgs []source.Package) (diagnostics []*source.Diagnostic, err error) {
+ // First, determine which modules are unused and which are missing from the
+ // original go.mod file.
+ var (
+ unused = make(map[string]*modfile.Require, len(pm.File.Require))
+ missing = make(map[string]*modfile.Require, len(ideal.Require))
+ wrongDirectness = make(map[string]*modfile.Require, len(pm.File.Require))
+ )
+ for _, req := range pm.File.Require {
+ unused[req.Mod.Path] = req
+ }
+ for _, req := range ideal.Require {
+ origReq := unused[req.Mod.Path]
+ if origReq == nil {
+ missing[req.Mod.Path] = req
+ continue
+ } else if origReq.Indirect != req.Indirect {
+ wrongDirectness[req.Mod.Path] = origReq
+ }
+ delete(unused, req.Mod.Path)
+ }
+ for _, req := range wrongDirectness {
+ // Handle dependencies that are incorrectly labeled indirect and
+ // vice versa.
+ srcDiag, err := directnessDiagnostic(pm.Mapper, req, snapshot.View().Options().ComputeEdits)
+ if err != nil {
+ return nil, err
+ }
+ diagnostics = append(diagnostics, srcDiag)
+ }
+ // Next, compute any diagnostics for modules that are missing from the
+ // go.mod file. The fixes will be for the go.mod file, but the
+ // diagnostics should also appear in both the go.mod file and the import
+ // statements in the Go files in which the dependencies are used.
+ missingModuleFixes := map[*modfile.Require][]source.SuggestedFix{}
+ for _, req := range missing {
+ srcDiag, err := missingModuleDiagnostic(snapshot, pm, req)
+ if err != nil {
+ return nil, err
+ }
+ missingModuleFixes[req] = srcDiag.SuggestedFixes
+ diagnostics = append(diagnostics, srcDiag)
+ }
+ // Add diagnostics for missing modules anywhere they are imported in the
+ // workspace.
+ for _, pkg := range workspacePkgs {
+ missingImports := map[string]*modfile.Require{}
+ var importedPkgs []string
+
+ // If -mod=readonly is not set we may have successfully imported
+ // packages from missing modules. Otherwise they'll be in
+ // MissingDependencies. Combine both.
+ for _, imp := range pkg.Imports() {
+ importedPkgs = append(importedPkgs, imp.PkgPath())
+ }
+ importedPkgs = append(importedPkgs, pkg.MissingDependencies()...)
+
+ for _, imp := range importedPkgs {
+ if req, ok := missing[imp]; ok {
+ missingImports[imp] = req
+ break
+ }
+ // If the import is a package of the dependency, then add the
+ // package to the map, this will eliminate the need to do this
+ // prefix package search on each import for each file.
+ // Example:
+ //
+ // import (
+ // "golang.org/x/tools/go/expect"
+ // "golang.org/x/tools/go/packages"
+ // )
+ // They both are related to the same module: "golang.org/x/tools".
+ var match string
+ for _, req := range ideal.Require {
+ if strings.HasPrefix(imp, req.Mod.Path) && len(req.Mod.Path) > len(match) {
+ match = req.Mod.Path
+ }
+ }
+ if req, ok := missing[match]; ok {
+ missingImports[imp] = req
+ }
+ }
+ // None of this package's imports are from missing modules.
+ if len(missingImports) == 0 {
+ continue
+ }
+ for _, pgf := range pkg.CompiledGoFiles() {
+ file, m := pgf.File, pgf.Mapper
+ if file == nil || m == nil {
+ continue
+ }
+ imports := make(map[string]*ast.ImportSpec)
+ for _, imp := range file.Imports {
+ if imp.Path == nil {
+ continue
+ }
+ if target, err := strconv.Unquote(imp.Path.Value); err == nil {
+ imports[target] = imp
+ }
+ }
+ if len(imports) == 0 {
+ continue
+ }
+ for importPath, req := range missingImports {
+ imp, ok := imports[importPath]
+ if !ok {
+ continue
+ }
+ fixes, ok := missingModuleFixes[req]
+ if !ok {
+ return nil, fmt.Errorf("no missing module fix for %q (%q)", importPath, req.Mod.Path)
+ }
+ srcErr, err := missingModuleForImport(snapshot, m, imp, req, fixes)
+ if err != nil {
+ return nil, err
+ }
+ diagnostics = append(diagnostics, srcErr)
+ }
+ }
+ }
+ // Finally, add errors for any unused dependencies.
+ onlyDiagnostic := len(diagnostics) == 0 && len(unused) == 1
+ for _, req := range unused {
+ srcErr, err := unusedDiagnostic(pm.Mapper, req, onlyDiagnostic, snapshot.View().Options().ComputeEdits)
+ if err != nil {
+ return nil, err
+ }
+ diagnostics = append(diagnostics, srcErr)
+ }
+ return diagnostics, nil
+}
+
+// unusedDiagnostic returns a source.Diagnostic for an unused require.
+func unusedDiagnostic(m *protocol.ColumnMapper, req *modfile.Require, onlyDiagnostic bool, computeEdits diff.ComputeEdits) (*source.Diagnostic, error) {
+ rng, err := rangeFromPositions(m, req.Syntax.Start, req.Syntax.End)
+ if err != nil {
+ return nil, err
+ }
+ title := fmt.Sprintf("Remove dependency: %s", req.Mod.Path)
+ cmd, err := command.NewRemoveDependencyCommand(title, command.RemoveDependencyArgs{
+ URI: protocol.URIFromSpanURI(m.URI),
+ OnlyDiagnostic: onlyDiagnostic,
+ ModulePath: req.Mod.Path,
+ })
+ if err != nil {
+ return nil, err
+ }
+ return &source.Diagnostic{
+ URI: m.URI,
+ Range: rng,
+ Severity: protocol.SeverityWarning,
+ Source: source.ModTidyError,
+ Message: fmt.Sprintf("%s is not used in this module", req.Mod.Path),
+ SuggestedFixes: []source.SuggestedFix{source.SuggestedFixFromCommand(cmd)},
+ }, nil
+}
+
+// directnessDiagnostic extracts errors when a dependency is labeled indirect when
+// it should be direct and vice versa.
+func directnessDiagnostic(m *protocol.ColumnMapper, req *modfile.Require, computeEdits diff.ComputeEdits) (*source.Diagnostic, error) {
+ rng, err := rangeFromPositions(m, req.Syntax.Start, req.Syntax.End)
+ if err != nil {
+ return nil, err
+ }
+ direction := "indirect"
+ if req.Indirect {
+ direction = "direct"
+
+ // If the dependency should be direct, just highlight the // indirect.
+ if comments := req.Syntax.Comment(); comments != nil && len(comments.Suffix) > 0 {
+ end := comments.Suffix[0].Start
+ end.LineRune += len(comments.Suffix[0].Token)
+ end.Byte += len([]byte(comments.Suffix[0].Token))
+ rng, err = rangeFromPositions(m, comments.Suffix[0].Start, end)
+ if err != nil {
+ return nil, err
+ }
+ }
+ }
+ // If the dependency should be indirect, add the // indirect.
+ edits, err := switchDirectness(req, m, computeEdits)
+ if err != nil {
+ return nil, err
+ }
+ return &source.Diagnostic{
+ URI: m.URI,
+ Range: rng,
+ Severity: protocol.SeverityWarning,
+ Source: source.ModTidyError,
+ Message: fmt.Sprintf("%s should be %s", req.Mod.Path, direction),
+ SuggestedFixes: []source.SuggestedFix{{
+ Title: fmt.Sprintf("Change %s to %s", req.Mod.Path, direction),
+ Edits: map[span.URI][]protocol.TextEdit{
+ m.URI: edits,
+ },
+ }},
+ }, nil
+}
+
+func missingModuleDiagnostic(snapshot source.Snapshot, pm *source.ParsedModule, req *modfile.Require) (*source.Diagnostic, error) {
+ var rng protocol.Range
+ // Default to the start of the file if there is no module declaration.
+ if pm.File != nil && pm.File.Module != nil && pm.File.Module.Syntax != nil {
+ start, end := pm.File.Module.Syntax.Span()
+ var err error
+ rng, err = rangeFromPositions(pm.Mapper, start, end)
+ if err != nil {
+ return nil, err
+ }
+ }
+ title := fmt.Sprintf("Add %s to your go.mod file", req.Mod.Path)
+ cmd, err := command.NewAddDependencyCommand(title, command.DependencyArgs{
+ URI: protocol.URIFromSpanURI(pm.Mapper.URI),
+ AddRequire: !req.Indirect,
+ GoCmdArgs: []string{req.Mod.Path + "@" + req.Mod.Version},
+ })
+ if err != nil {
+ return nil, err
+ }
+ return &source.Diagnostic{
+ URI: pm.Mapper.URI,
+ Range: rng,
+ Severity: protocol.SeverityError,
+ Source: source.ModTidyError,
+ Message: fmt.Sprintf("%s is not in your go.mod file", req.Mod.Path),
+ SuggestedFixes: []source.SuggestedFix{source.SuggestedFixFromCommand(cmd)},
+ }, nil
+}
+
+// switchDirectness gets the edits needed to change an indirect dependency to
+// direct and vice versa.
+func switchDirectness(req *modfile.Require, m *protocol.ColumnMapper, computeEdits diff.ComputeEdits) ([]protocol.TextEdit, error) {
+ // We need a private copy of the parsed go.mod file, since we're going to
+ // modify it.
+ copied, err := modfile.Parse("", m.Content, nil)
+ if err != nil {
+ return nil, err
+ }
+ // Change the directness in the matching require statement. To avoid
+ // reordering the require statements, rewrite all of them.
+ var requires []*modfile.Require
+ for _, r := range copied.Require {
+ if r.Mod.Path == req.Mod.Path {
+ requires = append(requires, &modfile.Require{
+ Mod: r.Mod,
+ Syntax: r.Syntax,
+ Indirect: !r.Indirect,
+ })
+ continue
+ }
+ requires = append(requires, r)
+ }
+ copied.SetRequire(requires)
+ newContent, err := copied.Format()
+ if err != nil {
+ return nil, err
+ }
+ // Calculate the edits to be made due to the change.
+ diff, err := computeEdits(m.URI, string(m.Content), string(newContent))
+ if err != nil {
+ return nil, err
+ }
+ return source.ToProtocolEdits(m, diff)
+}
+
+// missingModuleForImport creates an error for a given import path that comes
+// from a missing module.
+func missingModuleForImport(snapshot source.Snapshot, m *protocol.ColumnMapper, imp *ast.ImportSpec, req *modfile.Require, fixes []source.SuggestedFix) (*source.Diagnostic, error) {
+ if req.Syntax == nil {
+ return nil, fmt.Errorf("no syntax for %v", req)
+ }
+ spn, err := span.NewRange(snapshot.FileSet(), imp.Path.Pos(), imp.Path.End()).Span()
+ if err != nil {
+ return nil, err
+ }
+ rng, err := m.Range(spn)
+ if err != nil {
+ return nil, err
+ }
+ return &source.Diagnostic{
+ URI: m.URI,
+ Range: rng,
+ Severity: protocol.SeverityError,
+ Source: source.ModTidyError,
+ Message: fmt.Sprintf("%s is not in your go.mod file", req.Mod.Path),
+ SuggestedFixes: fixes,
+ }, nil
+}
+
+func rangeFromPositions(m *protocol.ColumnMapper, s, e modfile.Position) (protocol.Range, error) {
+ spn, err := spanFromPositions(m, s, e)
+ if err != nil {
+ return protocol.Range{}, err
+ }
+ return m.Range(spn)
+}
+
+func spanFromPositions(m *protocol.ColumnMapper, s, e modfile.Position) (span.Span, error) {
+ toPoint := func(offset int) (span.Point, error) {
+ l, c, err := m.Converter.ToPosition(offset)
+ if err != nil {
+ return span.Point{}, err
+ }
+ return span.NewPoint(l, c, offset), nil
+ }
+ start, err := toPoint(s.Byte)
+ if err != nil {
+ return span.Span{}, err
+ }
+ end, err := toPoint(e.Byte)
+ if err != nil {
+ return span.Span{}, err
+ }
+ return span.New(m.URI, start, end), nil
+}