Implemented Lua.load() and Lua.load!().

dryex · Aug 4, 2016 · 7bc9777 · 7bc9777
1 parent 3dd4f18
commit 7bc9777
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/lib/lua.ex b/lib/lua.ex
@@ -33,6 +33,27 @@ defmodule Lua do
     end
   end
 
+  @doc "Compiles a Lua code snippet."
+  @spec load(Lua.State.t, binary) :: {:ok, Lua.State.t, Lua.Chunk.t} | {:error, any, any}
+  def load(%State{luerl: state}, code) do
+    case :luerl.load(code, state) do
+      {:ok, function, state} ->
+        {:ok, %State{luerl: state}, %Chunk{luerl: function}}
+      error -> error
+    end
+  end
+
+  @doc "Compiles a Lua code snippet."
+  @spec load!(Lua.State.t, binary) :: {Lua.State.t, Lua.Chunk.t}
+  def load!(%State{luerl: state}, code) do
+    case :luerl.load(code, state) do
+      {:ok, function, state} ->
+        {%State{luerl: state}, %Chunk{luerl: function}}
+      {:error, reason, _} ->
+        raise Error, reason: reason, message: inspect(reason)
+    end
+  end
+
   @doc "Compiles a Lua source file."
   @spec load_file(Lua.State.t, binary) :: {:ok, Lua.State.t, Lua.Chunk.t} | {:error, any, any}
   def load_file(%State{luerl: state}, filepath) do

diff --git a/test/lua_test.exs b/test/lua_test.exs
@@ -25,6 +25,16 @@ defmodule LuaTest do
     assert_raise Lua.Error, fn -> Lua.eval_file!(Lua.State.new, "test/scripts/_enoent.lua") end
   end
 
+  test "Lua.load/2" do
+    assert {:ok, %Lua.State{}, %Lua.Chunk{}} = Lua.load(Lua.State.new, "return 6 * 7")
+    assert {:error, _, _} = Lua.load(Lua.State.new, "foobar")
+  end
+
+  test "Lua.load!/2" do
+    assert {%Lua.State{}, %Lua.Chunk{}} = Lua.load!(Lua.State.new, "return 6 * 7")
+    assert_raise Lua.Error, fn -> Lua.load!(Lua.State.new, "foobar") end
+  end
+
   test "Lua.load_file/2" do
     assert {:ok, %Lua.State{}, %Lua.Chunk{}} = Lua.load_file(Lua.State.new, "test/scripts/hello.lua")
     assert {:error, _, _} = Lua.load_file(Lua.State.new, "test/scripts/_enoent.lua")