feat: support split_with_parser (#1)

Author: fahchen

lib/sql_parser.ex

@@ -3,7 +3,8 @@ defmodule SQLParser do
   An SQL parser that uses [sqlparser](https://github.com/sqlparser-rs/sqlparser-rs.git) underneath.
   """
 
-  @typep ast() :: map()
+  @type ast() :: map()
+  @type statement() :: binary()
 
   @spec parse(sql :: binary(), dialect :: binary()) :: {:ok, [ast()]} | {:error, binary()}
   def parse(sql, dialect \\ "postgres") when dialect in ["bigquery", "postgres"] do
@@ -20,4 +21,27 @@ defmodule SQLParser do
     |> Jason.encode!()
     |> SQLParser.Native.to_sql()
   end
+
+  @doc """
+  Splits the given SQL string into a list of statements using the pg_query library.
+
+  ## Examples
+
+      iex> SQLParser.split_with_parser("SELECT 1; SELECT 2;")
+      {:ok, ["SELECT 1;", "SELECT 2;"]}
+  """
+  @spec split_with_parser(sql :: binary()) ::
+          {:ok, [statement()]} | {:error, binary()}
+  def split_with_parser(sql) do
+    with {:ok, statements} <- SQLParser.Native.split_with_parser(sql) do
+      statements =
+        Enum.map(statements, fn statement ->
+          statement
+          |> String.trim()
+          |> Kernel.<>(";")
+        end)
+
+      {:ok, statements}
+    end
+  end
 end

lib/sql_parser/native.ex

@@ -17,4 +17,7 @@ defmodule SQLParser.Native do
   def parse(_sql, _dialect), do: :erlang.nif_error(:nif_not_loaded)
 
   def to_sql(_ast), do: :erlang.nif_error(:nif_not_loaded)
+
+  @spec split_with_parser(binary()) :: {:ok, [binary()]} | {:error, binary()}
+  def split_with_parser(_sql), do: :erlang.nif_error(:nif_not_loaded)
 end