From 290a4403e4375d6ee63d32f76606d7f89d507bb7 Mon Sep 17 00:00:00 2001
From: Abdul Munif Hanafi <amunifhanafi@gmail.com>
Date: Fri, 20 Dec 2024 21:24:03 +0700
Subject: [PATCH] Refactor run_main_file function

---
 src/lib.rs | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 8f9fe31..4afe312 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -11,7 +11,7 @@ use mlua::{chunk, Error::RuntimeError, Integer, Lua, MultiValue, Table, Value};
 use modules::{apt, base_module, cmd, download, lineinfile, script, template, upload};
 use rustyline::DefaultEditor;
 use ssh::{ElevateMethod, Elevation, SSHAuthMethod, SSHSession};
-use std::{env, path::Path};
+use std::{env, fs, path::Path};
 use util::{
     dprint, filter_hosts, host_display, parse_hosts_json, regex_is_match, set_defaults,
     task_display,
@@ -316,12 +316,14 @@ fn komando(lua: &Lua, (host, task): (Value, Value)) -> mlua::Result<Table> {
 }
 
 pub fn run_main_file(lua: &Lua, main_file: &String) -> anyhow::Result<()> {
-    let main_file = main_file.clone();
-    lua.load(chunk! {
-        dofile($main_file)
-    })
-    .set_name("main")
-    .exec()?;
+    let script = match fs::read_to_string(main_file) {
+        Ok(script) => script,
+        Err(e) => {
+            return Err(anyhow::anyhow!("Failed to read the main file ({}): {}", main_file, e));
+        }
+    };
+
+    lua.load(&script).set_name(main_file).exec()?;
 
     Ok(())
 }