From 6a03dd24361ff189d0683f124af5bfa1f2cf4fca Mon Sep 17 00:00:00 2001
From: hadrienmp <public@hadrienmp.fr>
Date: Thu, 11 Jun 2020 13:25:57 +0200
Subject: [PATCH] Find the lib dir when in an escript

---
 .gitignore             |  1 +
 src/encurses.erl       |  6 +-----
 src/paths.erl          | 17 +++++++++++++++++
 test/nif_path_test.erl | 18 ++++++++++++++++++
 4 files changed, 37 insertions(+), 5 deletions(-)
 create mode 100644 src/paths.erl
 create mode 100644 test/nif_path_test.erl

diff --git a/.gitignore b/.gitignore
index 4af379f..0360de4 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,3 +3,4 @@ c_src/*.o
 ebin/
 _build/
 .rebar
+*.swp
diff --git a/src/encurses.erl b/src/encurses.erl
index 0b5e8ec..3494f42 100644
--- a/src/encurses.erl
+++ b/src/encurses.erl
@@ -90,11 +90,7 @@
 %% =============================================================================
 
 load_nif() ->
-    Dir = case code:priv_dir(encurses) of
-        {error, bad_name} ->
-            filename:dirname(code:which(?MODULE)) ++ "/../priv";
-        OtherDir -> OtherDir
-    end,
+    Dir = paths:nif_dir(),
     erlang:load_nif(Dir ++ "/encurses", 0).
 
 %% =============================================================================
diff --git a/src/paths.erl b/src/paths.erl
new file mode 100644
index 0000000..6e5391f
--- /dev/null
+++ b/src/paths.erl
@@ -0,0 +1,17 @@
+-module(paths).
+-export([nif_dir/0, nif_dir/2]).
+
+nif_dir() -> 
+    Cpd = code:priv_dir(encurses),
+    MoCpd = filename:dirname(code:which(?MODULE)) ++ "/../priv",
+    nif_dir(Cpd, MoCpd).
+
+
+nif_dir({error, _}, ModulePrivDir) -> handle_escript_format(ModulePrivDir);
+nif_dir(Dir, _) -> handle_escript_format(Dir).
+
+handle_escript_format(Dir) -> 
+    Handled = re:replace(Dir, 
+               "(_build/[^/]+)/bin/([^/]+)/encurses/priv",
+               "\\1/lib/encurses/priv"),
+    binary_to_list(iolist_to_binary(Handled)).
diff --git a/test/nif_path_test.erl b/test/nif_path_test.erl
new file mode 100644
index 0000000..4d20ca2
--- /dev/null
+++ b/test/nif_path_test.erl
@@ -0,0 +1,18 @@
+-module(nif_path_test).
+-include_lib("eunit/include/eunit.hrl").
+
+'use the module priv directory when code priv isnt known_test'() ->
+    NifDir = paths:nif_dir({error, bad_name}, "module dir"),
+    ?assertEqual("module dir", NifDir).
+
+'use the code priv directory_test'() -> 
+    NifDir = paths:nif_dir("./priv", "module dir"),
+    ?assertEqual("./priv", NifDir).
+
+'find the lib dir when in a bin escript directory_test'() ->
+    NifDir = paths:nif_dir("/dir/_build/default/bin/app/encurses/priv", "module dir"),
+    ?assertEqual("/dir/_build/default/lib/encurses/priv", NifDir).
+
+'find the lib dir when in a bin escript directory when code priv was not found_test'() ->
+    NifDir = paths:nif_dir({error, bad_name}, "/dir/_build/default/bin/app/encurses/priv"),
+    ?assertEqual("/dir/_build/default/lib/encurses/priv", NifDir).