Add auth

2 files changed, 25 insertions, 4 deletions

diff --git a/lib/boltex.ex b/lib/boltex.ex
index 9de7d3b..af1a770 100644
--- a/lib/boltex.ex
+++ b/lib/boltex.ex
@@ -1,13 +1,13 @@
 defmodule Boltex do
   alias Boltex.Bolt
 
-  def test(host, port, query) do
+  def test(host, port, query, params \\ %{}) do
     {:ok, p}   = :gen_tcp.connect host, port, [active: false, mode: :binary, packet: :raw]
 
     :ok        = Bolt.handshake :gen_tcp, p
-    :ok        = Bolt.init :gen_tcp, p
+    :ok        = Bolt.init :gen_tcp, p, params
 
-    Enum.map Bolt.run_statement(:gen_tcp, p, query), &IO.inspect/1
+    IO.inspect Bolt.run_statement(:gen_tcp, p, query, params)
   end
 
 end
diff --git a/lib/boltex/bolt.ex b/lib/boltex/bolt.ex
index 2e7f6b5..8bbf3e7 100644
--- a/lib/boltex/bolt.ex
+++ b/lib/boltex/bolt.ex
@@ -48,8 +48,20 @@ defmodule Boltex.Bolt do
 
   @doc """
   Initialises the connection.
+
+  Expects a transport module (i.e. `gen_tcp`) and a `Port`. Accepts
+  authorisation params in the form of {username, password}.
+
+  ## Examples
+
+      iex> Boltex.Bolt.init :gen_tcp, port
+      :ok
+
+      iex> Boltex.Bolt.init :gen_tcp, port, {"username", "password"}
+      :ok
   """
-  def init(transport, port, params \\ %{}) do
+  def init(transport, port, auth \\ nil) do
+    params = auth_params auth
     send_messages transport, port, [{[@user_agent, params], @sig_init}]
 
     case receive_data(transport, port) do
@@ -62,6 +74,15 @@ defmodule Boltex.Bolt do
     end
   end
 
+  defp auth_params(nil), do: %{}
+  defp auth_params({username, password}) do
+    %{
+      scheme: "basic",
+      principal: username,
+      credentials: password
+    }
+  end
+
   @doc """
   Sends a list of messages using the Bolt protocol and PackStream encoding.