Adding BASE_DIR to the files being shebang aware

ind9 · Aug 8, 2016 · 8a6e04d · 8a6e04d
1 parent d2093d8
commit 8a6e04d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mesos-fix b/mesos-fix
@@ -13,7 +13,10 @@ APP_PATH=${APP_PATH:-/app/}
 add_base_path_on_top_in() {
   FILE=$1
   if [ -f $FILE ]; then
-    sed -i '1i export BASE_PATH=$(cd `dirname $BASH_SOURCE`; pwd -P)/../' $FILE
+    # Add after the shebang in the file
+    sed -i '/#\!/a export BASE_PATH=$(cd `dirname $BASH_SOURCE`; pwd -P)/../' $FILE
+    # Add to the BOF if shebang is not present
+    egrep "#\!" $FILE || sed -i '1i export BASE_PATH=$(cd `dirname $BASH_SOURCE`; pwd -P)/../' $FILE
   fi
 }