From fe9e0e1574d0592c66a51b246c58ed8a94a75777 Mon Sep 17 00:00:00 2001
From: Michael Jerris <mike@jerris.com>
Date: Tue, 14 Oct 2008 17:14:56 +0000
Subject: [PATCH] (MODLANG-86) add local script directory support to mod_lua

git-svn-id: http://svn.freeswitch.org/svn/freeswitch/trunk@10008 d0543943-73ff-0310-b7d9-9358b9ac24b2
---
 conf/autoload_configs/lua.conf.xml    | 15 +++++++
 src/mod/languages/mod_lua/mod_lua.cpp | 56 ++++++++++++++++++++++++++-
 2 files changed, 69 insertions(+), 2 deletions(-)

diff --git a/conf/autoload_configs/lua.conf.xml b/conf/autoload_configs/lua.conf.xml
index 0dd13d87af..881049e23d 100644
--- a/conf/autoload_configs/lua.conf.xml
+++ b/conf/autoload_configs/lua.conf.xml
@@ -1,5 +1,20 @@
 <configuration name="lua.conf" description="LUA Configuration">
   <settings>
+
+    <!-- 
+    Specify local directories that will be searched for LUA modules
+    These entries will be pre-pended to the LUA_CPATH environment variable
+    -->
+    <!-- <param name="module-directory" value="/usr/lib/lua/5.1/?.so"/> -->
+    <!-- <param name="module-directory" value="/usr/local/lib/lua/5.1/?.so"/> -->
+
+    <!-- 
+    Specify local directories that will be searched for LUA scripts
+    These entries will be pre-pended to the LUA_PATH environment variable
+    -->
+    <!-- <param name="script-directory" value="/usr/local/lua/?.lua"/> -->
+    <!-- <param name="script-directory" value="/usr/local/freeswitch/scripts/?.lua"/> -->
+
     <!--<param name="xml-handler-script" value="/dp.lua"/>-->
     <!--<param name="xml-handler-bindings" value="dialplan"/>-->
 
diff --git a/src/mod/languages/mod_lua/mod_lua.cpp b/src/mod/languages/mod_lua/mod_lua.cpp
index d10f2b9b4e..04b17f75df 100644
--- a/src/mod/languages/mod_lua/mod_lua.cpp
+++ b/src/mod/languages/mod_lua/mod_lua.cpp
@@ -257,12 +257,16 @@ static switch_status_t do_config(void)
 {
 	const char *cf = "lua.conf";
 	switch_xml_t cfg, xml, settings, param;
-
-	if (!(xml = switch_xml_open_cfg(cf, &cfg, NULL))) {
+	switch_stream_handle_t path_stream = {0};
+	switch_stream_handle_t cpath_stream = {0};
+	
+    if (!(xml = switch_xml_open_cfg(cf, &cfg, NULL))) {
 		switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "open of %s failed\n", cf);
 		return SWITCH_STATUS_TERM;
 	}
 
+	SWITCH_STANDARD_STREAM(path_stream);
+	SWITCH_STANDARD_STREAM(cpath_stream);
 	if ((settings = switch_xml_child(cfg, "settings"))) {
 		for (param = switch_xml_child(settings, "param"); param; param = param->next) {
 			char *var = (char *) switch_xml_attr_soft(param, "name");
@@ -279,10 +283,58 @@ static switch_status_t do_config(void)
 				if (val) {
 					lua_thread(val);
 				}
+			} else if (!strcmp(var, "module-directory") && !switch_strlen_zero(val)) {
+				switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "lua: appending module directory: '%s'\n", val);
+				if(cpath_stream.data_len) {
+					cpath_stream.write_function(&cpath_stream, ";");
+				}
+				cpath_stream.write_function(&cpath_stream, "%s", val);
+			} else if (!strcmp(var, "script-directory") && !switch_strlen_zero(val)) {
+				switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "lua: appending script directory: '%s'\n", val);
+				if(path_stream.data_len) {
+					path_stream.write_function(&path_stream, ";");
+				}
+				path_stream.write_function(&path_stream, "%s", val);
 			}
 		}
 	}
 
+	if (cpath_stream.data_len) {
+		char *lua_cpath = NULL;
+		if (lua_cpath = getenv("LUA_CPATH")) {
+			switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "lua: appending LUA_CPATH: '%s'\n", lua_cpath);
+			cpath_stream.write_function(&cpath_stream, ";%s", lua_cpath);
+		}
+#ifdef WIN32
+		if (_putenv_s("LUA_CPATH", (char *)cpath_stream.data) != 0) {
+#else
+		if (setenv("LUA_CPATH", (char *)cpath_stream.data, 1) == ENOMEM) {
+#endif
+			switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "lua: LUA_CPATH unable to be set, out of memory: '%s'\n", (char *)cpath_stream.data);
+		} else {
+			switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "lua: LUA_CPATH set to: '%s'\n", (char *)cpath_stream.data);
+		}
+	}
+	switch_safe_free(cpath_stream.data);
+
+	if (path_stream.data_len) {
+		char *lua_path = NULL;
+		if (lua_path = getenv("LUA_PATH")) {
+			switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "lua: appending LUA_PATH: '%s'\n", lua_path);
+			path_stream.write_function(&path_stream, ";%s", lua_path);
+		}
+#ifdef WIN32
+		if (_putenv_s("LUA_PATH", (char *)path_stream.data) != 0) {
+#else
+		if (setenv("LUA_PATH", (char *)path_stream.data, 1) == ENOMEM) {
+#endif
+			switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "lua: LUA_PATH unable to be set, out of memory: '%s'\n", (char *)path_stream.data);
+		} else {
+			switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "lua: LUA_PATH set to: '%s'\n", (char *)path_stream.data);
+		}
+	}
+	switch_safe_free(path_stream.data);
+    
 	switch_xml_free(xml);
 
 	return SWITCH_STATUS_SUCCESS;