Code actions

analysis/src/Cli.ml

@@ -80,6 +80,9 @@ let main () =
   | [_; "documentSymbol"; path] -> DocumentSymbol.command ~path
   | [_; "hover"; path; line; col] ->
     Commands.hover ~path ~line:(int_of_string line) ~col:(int_of_string col)
+  | [_; "codeAction"; path; line; col; currentFile] ->
+    Commands.codeAction ~path ~line:(int_of_string line)
+      ~col:(int_of_string col) ~currentFile
   | [_; "references"; path; line; col] ->
     Commands.references ~path ~line:(int_of_string line)
       ~col:(int_of_string col)

analysis/src/CodeActions.ml

@@ -0,0 +1,17 @@
+(* This is the return that's expected when resolving code actions *)
+type result = Protocol.codeAction list
+
+let stringifyCodeActions codeActions =
+  Printf.sprintf {|%s|}
+    (codeActions |> List.map Protocol.stringifyCodeAction |> Protocol.array)
+
+let make ~title ~kind ~uri ~newText ~range =
+  {
+    Protocol.title;
+    codeActionKind = kind;
+    edit =
+      {
+        documentChanges =
+          [{textDocument = {version = None; uri}; edits = [{newText; range}]}];
+      };
+  }

analysis/src/Commands.ml

@@ -63,6 +63,10 @@ let hover ~path ~line ~col =
   in
   print_endline result
 
+let codeAction ~path ~line ~col ~currentFile =
+  Xform.extractCodeActions ~path ~pos:(line, col) ~currentFile
+  |> CodeActions.stringifyCodeActions |> print_endline
+
 let definition ~path ~line ~col =
   let locationOpt =
     match Cmt.fromPath ~path with
@@ -325,6 +329,26 @@ let test ~path =
               dir ++ parent_dir_name ++ "lib" ++ "bs" ++ "src" ++ name
             in
             Printf.printf "%s" (CreateInterface.command ~path ~cmiFile)
+          | "xfm" ->
+            print_endline
+              ("Xform " ^ path ^ " " ^ string_of_int line ^ ":"
+             ^ string_of_int col);
+            let codeActions =
+              Xform.extractCodeActions ~path ~pos:(line, col) ~currentFile:path
+            in
+            codeActions
+            |> List.iter (fun {Protocol.title; edit = {documentChanges}} ->
+                   Printf.printf "Hit: %s\n" title;
+                   documentChanges
+                   |> List.iter (fun {Protocol.edits} ->
+                          edits
+                          |> List.iter (fun {Protocol.range; newText} ->
+                                 let indent =
+                                   String.make range.start.character ' '
+                                 in
+                                 Printf.printf "%s\nnewText:\n%s<--here\n%s%s\n"
+                                   (Protocol.stringifyRange range)
+                                   indent indent newText)))
           | _ -> ());
           print_newline ())
     in

analysis/src/Protocol.ml

@@ -1,7 +1,5 @@
 type position = {line : int; character : int}
-
 type range = {start : position; end_ : position}
-
 type markupContent = {kind : string; value : string}
 
 type completionItem = {
@@ -13,13 +11,9 @@ type completionItem = {
 }
 
 type hover = {contents : string}
-
 type location = {uri : string; range : range}
-
 type documentSymbolItem = {name : string; kind : int; location : location}
-
 type renameFile = {oldUri : string; newUri : string}
-
 type textEdit = {range : range; newText : string}
 
 type optionalVersionedTextDocumentIdentifier = {
@@ -32,8 +26,16 @@ type textDocumentEdit = {
   edits : textEdit list;
 }
 
-let null = "null"
+type codeActionEdit = {documentChanges : textDocumentEdit list}
+type codeActionKind = RefactorRewrite
 
+type codeAction = {
+  title : string;
+  codeActionKind : codeActionKind;
+  edit : codeActionEdit;
+}
+
+let null = "null"
 let array l = "[" ^ String.concat ", " l ^ "]"
 
 let stringifyPosition p =
@@ -110,3 +112,15 @@ let stringifyTextDocumentEdit tde =
   }|}
     (stringifyoptionalVersionedTextDocumentIdentifier tde.textDocument)
     (tde.edits |> List.map stringifyTextEdit |> array)
+
+let codeActionKindToString kind =
+  match kind with RefactorRewrite -> "refactor.rewrite"
+
+let stringifyCodeActionEdit cae =
+  Printf.sprintf {|{"documentChanges": %s}|}
+    (cae.documentChanges |> List.map stringifyTextDocumentEdit |> array)
+
+let stringifyCodeAction ca =
+  Printf.sprintf {|{"title": "%s", "kind": "%s", "edit": %s}|} ca.title
+    (codeActionKindToString ca.codeActionKind)
+    (ca.edit |> stringifyCodeActionEdit)