Added textDocument/codeAction

Author: spoenemann

handler/handler.go

@@ -262,6 +262,11 @@ func (handler *InoHandler) transformClangdResult(method string, uri lsp.Document
 	case "textDocument/completion":
 		r := result.(*lsp.CompletionList)
 		handler.cpp2inoCompletionList(r, uri)
+	case "textDocument/codeAction":
+		r := result.(*[]CodeAction)
+		for index := range *r {
+			handler.cpp2inoCodeAction(&(*r)[index], uri)
+		}
 	case "textDocument/hover":
 		r := result.(*Hover)
 		handler.cpp2inoHover(r, uri)
@@ -297,6 +302,36 @@ func (handler *InoHandler) cpp2inoCompletionList(list *lsp.CompletionList, uri l
 	}
 }
 
+func (handler *InoHandler) cpp2inoCodeAction(codeAction *CodeAction, uri lsp.DocumentURI) {
+	newEdit := lsp.WorkspaceEdit{Changes: make(map[string][]lsp.TextEdit)}
+	for uri, edit := range codeAction.Edit.Changes {
+		if data, ok := handler.data[lsp.DocumentURI(uri)]; ok {
+			newValue := make([]lsp.TextEdit, len(edit))
+			for index := range edit {
+				r := edit[index].Range
+				newValue[index] = lsp.TextEdit{
+					NewText: edit[index].NewText,
+					Range: lsp.Range{
+						Start: lsp.Position{Line: data.sourceLineMap[r.Start.Line], Character: r.Start.Character},
+						End:   lsp.Position{Line: data.sourceLineMap[r.End.Line], Character: r.End.Character},
+					},
+				}
+			}
+			newEdit.Changes[string(data.sourceURI)] = newValue
+		} else {
+			newEdit.Changes[uri] = edit
+		}
+	}
+	codeAction.Edit = &newEdit
+	if data, ok := handler.data[uri]; ok {
+		for index := range codeAction.Diagnostics {
+			r := &codeAction.Diagnostics[index].Range
+			r.Start.Line = data.sourceLineMap[r.Start.Line]
+			r.End.Line = data.sourceLineMap[r.End.Line]
+		}
+	}
+}
+
 func (handler *InoHandler) cpp2inoHover(hover *Hover, uri lsp.DocumentURI) {
 	if data, ok := handler.data[uri]; ok {
 		r := hover.Range

handler/protocol.go

@@ -68,6 +68,10 @@ func sendRequest(ctx context.Context, conn *jsonrpc2.Conn, method string, params
 		result := new(lsp.CompletionList)
 		err := conn.Call(ctx, method, params, result)
 		return result, err
+	case "textDocument/codeAction":
+		result := new([]CodeAction)
+		err := conn.Call(ctx, method, params, result)
+		return result, err
 	case "completionItem/resolve":
 		result := new(lsp.CompletionItem)
 		err := conn.Call(ctx, method, params, result)
@@ -100,11 +104,22 @@ func sendRequest(ctx context.Context, conn *jsonrpc2.Conn, method string, params
 	return result, err
 }
 
+// CodeAction structure according to LSP
+type CodeAction struct {
+	Title       string             `json:"title"`
+	Kind        string             `json:"kind,omitempty"`
+	Diagnostics []lsp.Diagnostic   `json:"diagnostics,omitempty"`
+	Edit        *lsp.WorkspaceEdit `json:"edit,omitempty"`
+	Command     *lsp.Command       `json:"command,omitempty"`
+}
+
+// Hover structure according to LSP
 type Hover struct {
 	Contents MarkupContent `json:"contents"`
 	Range    *lsp.Range    `json:"range,omitempty"`
 }
 
+// MarkupContent structure according to LSP
 type MarkupContent struct {
 	Kind  string `json:"kind"`
 	Value string `json:"value"`

handler/stream.go

@@ -12,6 +12,7 @@ type StreamReadWrite struct {
 	outLog    io.Writer
 }
 
+// Read reads from the stream.
 func (srw StreamReadWrite) Read(p []byte) (int, error) {
 	n, err := srw.inStream.Read(p)
 	if n > 0 && srw.inLog != nil {
@@ -20,13 +21,15 @@ func (srw StreamReadWrite) Read(p []byte) (int, error) {
 	return n, err
 }
 
+// Write writes to the stream.
 func (srw StreamReadWrite) Write(p []byte) (int, error) {
 	if srw.outLog != nil {
 		srw.outLog.Write(p)
 	}
 	return srw.outStream.Write(p)
 }
 
+// Close closes the stream.
 func (srw StreamReadWrite) Close() error {
 	err1 := srw.inStream.Close()
 	err2 := srw.outStream.Close()