defmodule ElixirSense.Core.ParserTest do
  use ExUnit.Case, async: true

  alias ElixirSense.Core.{Metadata, State.Env, State.VarInfo, State.CallInfo, Parser}

  defp parse(source, cursor) do
    metadata = Parser.parse_string(source, true, false, cursor)
    {metadata, Metadata.get_cursor_env(metadata, cursor)}
  end

  test "parse_string creates a Metadata struct" do
    source = """
    defmodule MyModule do
      import List

    end
    """

    assert {%Metadata{
              error: nil,
              mods_funs_to_positions: %{{MyModule, nil, nil} => %{positions: [{1, 1}]}},
              source: "defmodule MyModule" <> _
            }, %Env{functions: functions}} = parse(source, {3, 3})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with syntax error" do
    source = """
    defmodule MyModule do
      import List
      Enum +
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions3, module: MyModule}} = parse(source, {3, 10})

    assert Keyword.has_key?(functions3, List)
  end

  test "parse_string with syntax error (missing param)" do
    source = """
    defmodule MyModule do
      import List
      IO.puts(:stderr, )
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 20})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with missing terminator \")\"" do
    source = """
    defmodule MyModule do
      import List
      func(
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 8})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with missing terminator \"]\"" do
    source = """
    defmodule MyModule do
      import List
      list = [
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 11})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with missing terminator \"}\"" do
    source = """
    defmodule MyModule do
      import List
      tuple = {
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 12})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with missing terminator \"\"\"" do
    source = """
    defmodule MyModule do
      import List
      var = "
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 10})

    if Version.match?(System.version(), ">= 1.15.0") do
      assert Keyword.has_key?(functions, List)
    end
  end

  test "parse_string with missing terminator \"\'\"" do
    source = """
    defmodule MyModule do
      import List
      var = '
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 10})

    if Version.match?(System.version(), ">= 1.15.0") do
      assert Keyword.has_key?(functions, List)
    end
  end

  test "parse_string with missing heredoc terminator" do
    source = """
    defmodule MyModule do
      import List
      var = \"\"\"
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 12})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with missing interpolation terminator in \"\"\"" do
    source = """
    defmodule MyModule do
      import List
      var = "\#{
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 12})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with missing interpolation terminator in \"\'\"" do
    source = """
    defmodule MyModule do
      import List
      var = '\#{
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 12})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with missing interpolation terminator in heredoc" do
    source = """
    defmodule MyModule do
      import List
      var = \"\"\"\#{
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{functions: functions}} = parse(source, {3, 14})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with missing terminator \"end\" attempts to fix it by inserting end at line from error" do
    source = """
    defmodule MyModule do
      []
      |> Enum.map(fn x ->

      a = 5
    end
    """

    {_metadata, env} = parse(source, {3, 23})

    if Version.match?(System.version(), ">= 1.17.0") do
      assert %Env{
               vars: [
                 %VarInfo{name: :x}
               ]
             } = env
    end
  end

  test "parse_string with missing terminator \"end\" attempts to insert `end` at correct indentation" do
    source = """
    defmodule MyModule do

    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{module: MyModule}} = parse(source, {2, 3})

    source = """
    defmodule MyModule do

    defmodule MyModule1 do
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{module: MyModule}} = parse(source, {3, 1})

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{module: MyModule}} = parse(source, {2, 1})

    source = """
    defmodule MyModule do

      defmodule MyModule1 do
      
      end
    """

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{module: MyModule}} = parse(source, {2, 1})

    assert {%Metadata{
              error: {:error, :parse_error}
            }, %Env{module: MyModule.MyModule1}} = parse(source, {4, 5})
  end

  test "parse_string with incomplete key for multiline keyword as argument" do
    source = """
    defmodule MyModule do
      IO.inspect(
        :stderr,
        label: "label",
        limit
      )
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error},
              calls: %{
                2 => [%CallInfo{func: :inspect}, %CallInfo{mod: IO}]
              }
            },
            %Env{module: MyModule}} =
             parse(source, {5, 10})
  end

  test "parse_string with missing value for multiline keyword as argument" do
    source = """
    defmodule MyModule do
      IO.inspect(
        :stderr,
        label: "label",
        limit:
      )
    end
    """

    assert {%Metadata{
              error: {:error, :parse_error},
              calls: %{
                2 => [%CallInfo{func: :inspect}, %CallInfo{mod: IO}]
              }
            },
            %Env{module: MyModule}} =
             parse(source, {5, 12})
  end

  @tag capture_log: true
  test "parse_string ignores non existing modules in `use`" do
    source = """
    defmodule MyModule do
      use EnumFake
      import List
      foo()

    end
    """

    assert {%Metadata{
              error: nil,
              mods_funs_to_positions: %{{MyModule, nil, nil} => %{positions: [{1, 1}]}},
              source: "defmodule MyModule" <> _
            }, %Env{functions: functions}} = parse(source, {5, 3})

    assert Keyword.has_key?(functions, List)
  end

  test "parse_string with malformed `do` expression" do
    source = """
    defmodule MyModule, do
    """

    assert {%ElixirSense.Core.Metadata{
              error: {:error, :parse_error}
            }, %Env{module: MyModule}} = parse(source, {1, 23})
  end

  test "parse_string with literal strings" do
    source = ~S'''
    defmodule MyMod do
      @my_attr """
      Some docs.
      """
      def func do
        1
      end
    end
    '''

    assert {%ElixirSense.Core.Metadata{},
            %ElixirSense.Core.State.Env{
              attributes: [%ElixirSense.Core.State.AttributeInfo{name: :my_attr}]
            }} = parse(source, {6, 6})
  end

  test "parse_string with literal strings in sigils" do
    source = ~S'''
    defmodule MyMod do
      def render(y) do
        x = y
        ~E"""
        <h1><%= x %></h1>
        """
      end
    end
    '''

    assert {%Metadata{},
            %Env{
              vars: vars
            }} = parse(source, {5, 14})

    if Version.match?(System.version(), "< 1.15.0") do
      # container_cursor_to_quoted removes function body
      assert [
               %ElixirSense.Core.State.VarInfo{name: :y}
             ] = Enum.sort(vars)
    else
      assert [
               %ElixirSense.Core.State.VarInfo{name: :x},
               %ElixirSense.Core.State.VarInfo{name: :y}
             ] = Enum.sort(vars)
    end
  end

  test "parse struct" do
    source = """
    defmodule MyModule do
      def func() do
        %{
          data: foo()
        }
      end
    end
    """

    assert {%ElixirSense.Core.Metadata{
              calls: %{
                4 => [%CallInfo{func: :foo}]
              }
            }, %Env{function: {:func, 0}}} = parse(source, {4, 7})
  end

  test "parse struct with missing terminator" do
    source = """
    defmodule MyModule do
      def func() do
        %{
          data: foo()

      end
    end
    """

    assert {%ElixirSense.Core.Metadata{
              calls: %{
                4 => [%{func: :foo}]
              }
            }, %Env{function: {:func, 0}}} = parse(source, {4, 8})
  end
end
