defmodule Loupe.Language do
  @moduledoc """
  Loupe's compiler entrypoint.
  """
  alias Loupe.Language.Ast

  alias Loupe.Errors.LexerError
  alias Loupe.Errors.ParserError

  @type compile_error :: ParserError.t() | LexerError.t()

  @doc "Compiles a query to AST"
  @spec compile(String.t() | charlist()) :: {:ok, Ast.t()} | compile_error()
  def compile(string) when is_binary(string) do
    string
    |> String.to_charlist()
    |> compile()
  end

  def compile(charlist) do
    with {:ok, tokens, _} <- :lexer.string(charlist),
         {:ok, ast} <- :parser.parse(tokens) do
      {:ok, new_ast(ast)}
    else
      {:error, {line, :parser, messages}} -> %ParserError{line: line, message: messages}
      {:error, {line, :lexer, messages}} -> %LexerError{line: line, message: messages}
    end
  rescue
    error ->
      {:error, error}
  end

  defp new_ast({action, quantifier, schema, predicates}) do
    Ast.new(action, schema, quantifier, predicates)
  end
end