Handle no-specify, no-ex-check and no-gradualizer-check options

Author: Premwoik

lib/gradient.ex

@@ -5,6 +5,9 @@ defmodule Gradient do
   Options:
   - `app_path` - Path to the app that contains file with code (for umbrella apps).
   - `code_path` - Path to a file with code (e.g. when beam was compiled without project).
+  - `no_gradualizer_check` - Skip Gradualizer checks if true.
+  - `no_ex_check` - Skip Elixir checks if true.
+  - `no_specify` - Skip AST specifying if true.
   """
 
   alias Gradient.ElixirFileUtils
@@ -21,12 +24,9 @@ defmodule Gradient do
     opts = Keyword.put(opts, :return_errors, true)
 
     with {:ok, forms} <- ElixirFileUtils.get_forms(file) do
-      forms =
-        forms
-        |> put_code_path(opts)
-        |> AstSpecifier.specify()
+      forms = maybe_specify_forms(forms, opts)
 
-      case ElixirChecker.check(forms, opts) ++ :gradualizer.type_check_forms(forms, opts) do
+      case maybe_gradient_check(forms, opts) ++ maybe_gradualizer_check(forms, opts) do
         [] ->
           :ok
 
@@ -42,6 +42,32 @@ defmodule Gradient do
     end
   end
 
+  defp maybe_gradualizer_check(forms, opts) do
+    unless opts[:no_gradualizer_check] do
+      :gradualizer.type_check_forms(forms, opts)
+    else
+      []
+    end
+  end
+
+  defp maybe_gradient_check(forms, opts) do
+    unless opts[:no_ex_check] do
+      ElixirChecker.check(forms, opts)
+    else
+      []
+    end
+  end
+
+  defp maybe_specify_forms(forms, opts) do
+    unless opts[:no_specify] do
+      forms
+      |> put_code_path(opts)
+      |> AstSpecifier.specify()
+    else
+      forms
+    end
+  end
+
   defp put_code_path(forms, opts) do
     case opts[:code_path] do
       nil ->

lib/gradient/doublespec.ex

@@ -0,0 +1,17 @@
+defmodule Doublespec do
+  @spec convert(integer()) :: float()
+  @spec convert(atom()) :: binary()
+  def convert(int) when is_integer(int), do: int / 1
+  def convert(atom) when is_atom(atom), do: to_string(atom)
+
+  @spec convert(atom()) :: binary()
+  def last_two(list) do
+    [last, penultimate | _tail] = Enum.reverse(list)
+    [penultimate, last]
+  end
+
+  @spec last_two(atom()) :: atom()
+  def last_three(:ok) do
+    :ok
+  end
+end

lib/mix/tasks/gradient.ex

@@ -8,8 +8,11 @@ defmodule Mix.Tasks.Gradient do
     * `--no-compile` - do not compile even if needed
     * `--no-ex-check` - do not perform checks specyfic for Elixir
       (from ElixirChecker module)
-    * `--no-gradualizer-check` - do not perform Gradualizer checks
+    * `--no-gradualizer-check` - do not perform the Gradualizer checks
+    * `--no-specify` - do not specify missing lines in AST what can
+      result in less precise error messages
 
+    * `--no-deps` - do not import dependencies to the Gradualizer
     * `--stop_on_first_error` - stop type checking at the first error
     * `--infer` - infer type information from literals and other language
       constructs,