| // Copyright 2019 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 cmd |
| |
| import ( |
| "context" |
| "flag" |
| "fmt" |
| "go/token" |
| "io/ioutil" |
| |
| "golang.org/x/tools/internal/lsp/protocol" |
| "golang.org/x/tools/internal/span" |
| ) |
| |
| // definition implements the definition noun for the query command. |
| type check struct { |
| app *Application |
| } |
| |
| type checkClient struct { |
| baseClient |
| diagnostics chan entry |
| } |
| |
| type entry struct { |
| uri span.URI |
| diagnostics []protocol.Diagnostic |
| } |
| |
| func (c *check) Name() string { return "check" } |
| func (c *check) Usage() string { return "<filename>" } |
| func (c *check) ShortHelp() string { return "show diagnostic results for the specified file" } |
| func (c *check) DetailedHelp(f *flag.FlagSet) { |
| fmt.Fprint(f.Output(), ` |
| Example: show the diagnostic results of this file: |
| |
| $ gopls check internal/lsp/cmd/check.go |
| |
| gopls check flags are: |
| `) |
| f.PrintDefaults() |
| } |
| |
| // Run performs the check on the files specified by args and prints the |
| // results to stdout. |
| func (c *check) Run(ctx context.Context, args ...string) error { |
| if len(args) == 0 { |
| // no files, so no results |
| return nil |
| } |
| client := &checkClient{ |
| diagnostics: make(chan entry), |
| } |
| client.app = c.app |
| checking := map[span.URI][]byte{} |
| // now we ready to kick things off |
| server, err := c.app.connect(ctx, client) |
| if err != nil { |
| return err |
| } |
| for _, arg := range args { |
| uri := span.FileURI(arg) |
| content, err := ioutil.ReadFile(arg) |
| if err != nil { |
| return err |
| } |
| checking[uri] = content |
| p := &protocol.DidOpenTextDocumentParams{} |
| p.TextDocument.URI = string(uri) |
| p.TextDocument.Text = string(content) |
| if err := server.DidOpen(ctx, p); err != nil { |
| return err |
| } |
| } |
| // now wait for results |
| for entry := range client.diagnostics { |
| //TODO:timeout? |
| content, found := checking[entry.uri] |
| if !found { |
| continue |
| } |
| fset := token.NewFileSet() |
| f := fset.AddFile(string(entry.uri), -1, len(content)) |
| f.SetLinesForContent(content) |
| m := protocol.NewColumnMapper(entry.uri, fset, f, content) |
| for _, d := range entry.diagnostics { |
| spn, err := m.RangeSpan(d.Range) |
| if err != nil { |
| return fmt.Errorf("Could not convert position %v for %q", d.Range, d.Message) |
| } |
| fmt.Printf("%v: %v\n", spn, d.Message) |
| } |
| delete(checking, entry.uri) |
| if len(checking) == 0 { |
| return nil |
| } |
| } |
| return fmt.Errorf("did not get all results") |
| } |
| |
| func (c *checkClient) PublishDiagnostics(ctx context.Context, p *protocol.PublishDiagnosticsParams) error { |
| c.diagnostics <- entry{ |
| uri: span.URI(p.URI), |
| diagnostics: p.Diagnostics, |
| } |
| return nil |
| } |