update: performance installer and installer

Author: Esonhugh

Makefile

@@ -9,5 +9,10 @@ install: build
 	WEAPON_LOCATION=$(shell pwd)/weapon_test ./installer.sh
 	@echo "Installation complete."
 
+install-trace: build
+	@echo "Preparing the environment with trace..."
+	WEAPON_LOCATION=$(shell pwd)/weapon_test ./installer.sh --trace-mode
+	@echo "Installation with trace complete."
+
 clean:
 	rm -rf ./hosts ./users ./weapon_test ./createhackenv.sh ./zsh_history

installer.sh

@@ -29,6 +29,27 @@ function check_requirements() {
     check_command "uv" "uv is optional but recommended for better performance, install it with 'pip install uv'. Their script is useful"
 }
 
+export INSTALL_IDENTIFIER="WeaponizedVSCode installation script. The following commands are generated by WeaponizedVSCode installer."
+
+function args() {
+    while [[ "$#" -gt 0 ]]; do
+        case "$1" in
+            (--trace-mode)
+                export WEAPON_TRACE_MODE=true
+                shift
+                ;;
+            (--force-install)
+                export WEAPON_FORCE_INSTALL=true
+                shift
+                ;;
+            *)
+                echo "Unknown option: $1"
+                exit 1
+                ;;
+        esac
+    done
+}
+
 function main() {
     check_requirements
     if [[ $? -ne 0 && "$FORCE_INSTALL" != "true" ]]; then
@@ -45,8 +66,16 @@ function main() {
 
     if [[ -d "$WEAPON_LOCATION" ]]; then
         echo "Weapon already installed."
+        if [[ "$WEAPON_FORCE_INSTALL" == "true" ]]; then
+            echo "Force install mode enabled, removing existing Weapon directory at $WEAPON_LOCATION."
+            rm -rf "$WEAPON_LOCATION"
+        else
+            echo "If you want to reinstall Weapon, please remove the existing directory at $WEAPON_LOCATION or use --force-install option."
+            exit 0
+        fi
         exit 0
     fi
+
     export LOCATION="$WEAPON_LOCATION"
     echo "Weapon will be installed to $LOCATION, Press Enter key to continue or Ctrl+C to cancel..."
     read
@@ -55,15 +84,37 @@ function main() {
     cat ./createhackenv.sh |sed -e "s#__REPLACE__#$LOCATION#g" > $LOCATION/createhackenv.sh && echo "createhackenv.sh copied to $LOCATION."
     cp -f ./zsh_history $LOCATION && echo "zsh_history copied to $LOCATION."
 
-    grep -zq "source $LOCATION/createhackenv.sh" ~/.zshrc
+    if [[ "$WEAPON_TRACE_MODE" == "true" ]];then
+        cp -f ./zsh_preformance_tracer.sh $LOCATION && echo "zsh_performance_tracer is copied to $LOCATION."
+    fi
+
+    grep -zq "${INSTALL_IDENTIFIER}" ~/.zshrc
     if [[ $? -eq 0 ]]; then
         echo "createhackenv.sh already sourced in ~/.zshrc. sktpping..."
     else
         echo "Sourcing createhackenv.sh in ~/.zshrc..."
-        echo "source $LOCATION/createhackenv.sh" >>~/.zshrc
+        cp ~/.zshrc ~/.zshrc.pre-weaponized-vscode && echo "Backup ~/.zshrc to ~/.zshrc.pre-weaponized-vscode"
+
+        if [[ "$WEAPON_TRACE_MODE" == "true" ]]; then
+cat<<EOF >> ~/.zshrc
+
+# ${INSTALL_IDENTIFIER}
+[[ -f $LOCATION/zsh_preformance_tracer.sh ]] && source $LOCATION/zsh_preformance_tracer.sh && zsh_trace_start
+[[ -f $LOCATION/createhackenv.sh ]] && source $LOCATION/createhackenv.sh
+[[ -f $LOCATION/zsh_performance_tracer ]] && zsh_trace_end
+EOF
+        else
+cat<<EOF >> ~/.zshrc
+
+# ${INSTALL_IDENTIFIER}
+[[ -f $LOCATION/createhackenv.sh ]] && source $LOCATION/createhackenv.sh
+EOF
+        fi
+
     fi
 
     echo "Weaponized VSCode installation completed."
 }
 
-main "$@"
+args "$@"
+main 

zsh_preformance_tracer.sh

@@ -1,3 +1,5 @@
+export WEAPON_TRACE_LOG=/tmp/weapon_trace_log.$$
+
 zsh_trace_start() {
     echo "starting zsh tracing"
     # set the trace prompt to include seconds, nanoseconds, script name and line number
@@ -11,7 +13,7 @@ zsh_trace_start() {
     fi
     # save file stderr to file descriptor 3 and redirect stderr (including trace 
     # output) to a file with the script's PID as an extension
-    exec 3>&2 2>/tmp/startlog.$$
+    exec 3>&2 2>$WEAPON_TRACE_LOG
     # set options to turn on tracing and expansion of commands contained in the prompt
     setopt xtrace prompt_subst
     trap 'setopt xtrace' EXIT
@@ -22,7 +24,7 @@ zsh_trace_end() {
     unsetopt xtrace
     # restore stderr to the value saved in FD 3
     exec 2>&3 3>&-
-    echo "zsh tracing done. See /tmp/startlog.$$"
+    echo "zsh tracing done. See $WEAPON_TRACE_LOG for details."
     trap 'unsetopt xtrace' EXIT
 }