From 3fd6e86574f932b0597d0ffeb521e1336925a155 Mon Sep 17 00:00:00 2001
From: HlmsDeep <first12last100@gmail.com>
Date: Thu, 20 Mar 2025 10:40:28 +0000
Subject: [PATCH] now

---
 cpp.sh | 32 +++++++++++++++++++-------------
 1 file changed, 19 insertions(+), 13 deletions(-)

diff --git a/cpp.sh b/cpp.sh
index c5387ab..6be54c7 100644
--- a/cpp.sh
+++ b/cpp.sh
@@ -2,6 +2,9 @@
 
 # Path to the code file to compile
 # /usr/src/app/cpp-engine/app/main
+
+set -e
+
 if [ -z "$1" ]; then
     echo "No language specified! Use 'c' or 'cpp'."
     exit 1
@@ -10,38 +13,41 @@ fi
 FILE_EXTENSION="$1"
 CODE_FILE="/usr/src/app/cpp-engine/app/main.${FILE_EXTENSION}"
 
+# Validate file extension
+if [[ "$FILE_EXTENSION" != "c" && "$FILE_EXTENSION" != "cpp" ]]; then
+    echo "Invalid file extension.  Must be 'c' or 'cpp'."
+    exit 1
+fi
+
+# Check if the code file exists
+if [ ! -f "$CODE_FILE" ]; then
+    echo "Error: Code file '$CODE_FILE' not found."
+    exit 1
+fi
 
 if [ "$FILE_EXTENSION" == "c" ]; then
     clang -o output_program "$CODE_FILE" 2> compile_error.txt
 elif [ "$FILE_EXTENSION" == "cpp" ]; then
     clang++ -o output_program "$CODE_FILE" 2> compile_error.txt
 fi
+
 # Check if there were any compilation errors
 if [ $? -ne 0 ]; then
     echo -e "Compilation failed!\n"
     cat compile_error.txt
     exit 1
 fi
-# Run the compiled program and capture runtime errors
-./output_program > output.txt 2> runtime_error.txt
-
-# Check if there were any runtime errors
-if [ $? -ne 0 ]; then
-    echo -e "Runtime error!\n"
-    cat runtime_error.txt
-    exit 1
 
-fi
-# start
+# Determine input redirection and run the program only once
 if [ -f "/usr/src/app/cpp-engine/app/input.txt" ]; then
-    # Run the compiled program with input redirection
     ./output_program < /usr/src/app/cpp-engine/app/input.txt > output.txt 2> runtime_error.txt
 else
-    # Run the compiled program without input redirection
     ./output_program > output.txt 2> runtime_error.txt
 fi
-# end
+
+# Check if there were any runtime errors
 if [ $? -ne 0 ]; then
+    echo -e "Runtime error!\n"
     cat runtime_error.txt
     exit 1
 fi