searches .lush/scripts for lua files

2 files changed, 27 insertions, 6 deletions

diff --git a/README.md b/README.md
index 96ba0c8..ea1813a 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,7 @@
 
 Lunar Shell (lush) is an open source unix shell with a single goal in mind. That goal is to offer the ability to write shell scripts for your operating system entirely in Lua. The Lua scripting language has many powerful features that allow for more control in the hands of the user to automate tasks on their machine.
 
-## Compiling
+## Compiling/Installation
 
 Clone the repo and run the install script.
 
@@ -20,10 +20,6 @@ sh install.sh
 
 For future compiles just run ```make``` from the root directory. If you change the premake5.lua file rebuild the makefiles with ```premake5 gmake```.
 
-## Releases
-
-Binary builds are available from [releases](https://github.com/BanceDev/lush/releases).
-
 ## Contributing
 
 - For bug reports and feature suggestions please use [issues](https://github.com/BanceDev/lush/issues).
diff --git a/src/lua_api.c b/src/lua_api.c
index becd2b7..5a92b58 100644
--- a/src/lua_api.c
+++ b/src/lua_api.c
@@ -22,12 +22,37 @@ IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 #include <lualib.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <unistd.h>
 
 // -- script execution --
 void lua_load_script(lua_State *L, const char *script) {
-	if (luaL_dofile(L, script) != LUA_OK) {
+	char script_path[512];
+	// check if script is in the current directory
+	if (access(script, F_OK) == 0) {
+		snprintf(script_path, sizeof(script_path), "%s", script);
+	} else {
+		const char *home_dir = getenv("HOME");
+		if (home_dir != NULL) {
+			snprintf(script_path, sizeof(script_path), "%s/.lush/scripts/%s",
+					 home_dir, script);
+
+			if (access(script_path, F_OK) != 0) {
+				// script not in either location
+				fprintf(stderr, "[C] Script not found: %s\n", script);
+				return;
+			}
+		} else {
+			// HOME not set
+			fprintf(stderr, "[C] HOME directory is not set.\n");
+			return;
+		}
+	}
+	// if we got here the file exists
+	if (luaL_dofile(L, script_path) != LUA_OK) {
 		printf("[C] Error reading script\n");
 		luaL_error(L, "Error: %s\n", lua_tostring(L, -1));
+		// remove error from stack
+		lua_pop(L, 1);
 	}
 }