Encapsulated sourceror

apps/language_server/lib/language_server/experimental/code_mod/ast.ex

@@ -1,19 +1,28 @@
 defmodule ElixirLS.LanguageServer.Experimental.CodeMod.Ast do
   alias ElixirLS.LanguageServer.Experimental.SourceFile
 
-  @type t :: any()
+  @type source :: SourceFile.t() | String.t()
+  @type t ::
+          atom()
+          | binary()
+          | [any()]
+          | number()
+          | {any(), any()}
+          | {atom() | {any(), [any()], atom() | [any()]}, Keyword.t(), atom() | [any()]}
 
+  @spec from(source) :: t
   def from(%SourceFile{} = source_file) do
     source_file
     |> SourceFile.to_string()
     |> from()
   end
 
   def from(s) when is_binary(s) do
-    parse(s)
+    ElixirSense.string_to_quoted(s, 1, 6, token_metadata: true)
   end
 
-  defp parse(s) when is_binary(s) do
-    ElixirSense.string_to_quoted(s, 1, 6, token_metadata: true)
+  @spec to_string(t()) :: String.t()
+  def to_string(ast) do
+    Sourceror.to_string(ast)
   end
 end

apps/language_server/lib/language_server/experimental/code_mod/replace_with_underscore.ex

@@ -37,7 +37,7 @@ defmodule ElixirLS.LanguageServer.Experimental.CodeMod.ReplaceWithUnderscore do
       other ->
         other
     end)
-    |> Sourceror.to_string()
+    |> Ast.to_string()
     # We're dealing with a single error on a single line.
     # If the line doesn't compile (like it has a do with no end), ElixirSense
     # adds additional lines do documents with errors, so take the first line, as it's