feat: migrate from Oh-My-Zsh to ZInit

2024-08-14 13:49:10 +02:00
parent 4a8dcfba3d
commit adc26f9860
7 changed files with 212 additions and 68 deletions
--- a/zsh/prompt.sh
+++ b/zsh/prompt.sh
@@ -0,0 +1,131 @@
+# Custom prompt
+update_prompt() {
+	ret_code=$?
+	PROMPT=""
+
+	# Git
+	if [[ $(git_repo_name) ]]; then
+		# PROMPT+="%{$fg[yellow]%}$(git remote show | head -n 1)"
+		# PROMPT+="%{$fg_bold[magenta] $(git_repo_name)"
+		# PROMPT+="%{$fg_bold[blue]($(git_current_branch))%{$fg[white]%}:"
+		PROMPT+="%{$fg[yellow]%}$(git remote show | head -n 1) %{$fg_bold[magenta]%}$(git_repo_name)%{$fg_bold[blue]%}($(git_current_branch))%{$fg[white]%}:"
+
+	fi
+
+	# Path
+	PROMPT+="%{$fg[cyan]%}%c%{$reset_color%} "
+
+	# Arrow
+	if [[ started -eq 0 ]] || [[ $ret_code -eq 0 ]]; then
+		PROMPT+="%{$fg_bold[green]%}%1{➜%} "
+	else
+		PROMPT+="%{$fg[red]%}%? %{$fg_bold[red]%}%1{➜%} "
+	fi
+	# PROMPT+="%(?:%{$fg_bold[green]%}%1{➜%} :%{$fg[red]%}(%?%) %{$fg_bold[red]%}%1{➜%} )"
+	# PROMPT+="%(?:%{${started:+$fg_bold[green]}${started:=$fg[green]}%}%1{➜%} :%{${started:+$fg_bold[green]%)"
+	
+	return $?
+}
+
+update_prompt
+
+# Define a map of error codes to messages
+declare -A ERROR_MESSAGES=(
+	[64]="USAGE"
+	[65]="FORMAT"
+	[68]="HOSTNAME"
+	[69]="UNAVAIL"
+	[70]="SOFTWARE"
+	[71]="OSERR"
+	[72]="OSFILE"
+	[73]="CREATEFAIL"
+	[74]="IOERR"
+	[75]="TEMP"
+	[76]="PROTOCOL"
+	[77]="DENIED"
+    [78]="CONFERR"   # Configuration error
+	[126]="EXECFAIL"
+    [127]="COMMAND"   # Command not found
+	[130]="SIGINT"
+	[137]="SIGKILL"
+    # Add more error codes and messages as needed
+)
+
+# === FUNCTIONS FOR COMMAND TIMING ===
+
+# Function to capture the start time of the command in milliseconds
+preexec() {
+	started=1
+    timer_start=$(( $(date +%s%3N) ))  # Captures current time in milliseconds
+}
+
+# Function to capture the end time of the command and calculate the elapsed time
+precmd() {
+	update_prompt
+	if [[ started -eq 0 ]]; then
+		RPROMPT=""
+		return
+	fi
+	started=0
+
+    local timer_end=$(( $(date +%s%3N) ))  # Captures current time in milliseconds
+    local elapsed=$((timer_end - timer_start))
+
+    # Reset RPROMPT
+    RPROMPT=""
+
+    # Handle error codes: Prepend the error message if the command failed
+	if [[ ret_code -ne 0 ]]; then
+        local error_msg=${ERROR_MESSAGES[$ret_code]}
+        if [[ -n $error_msg ]]; then
+            RPROMPT="%{$fg_bold[red]%}(${error_msg})%{$reset_color%}$RPROMPT"
+        fi
+    fi
+
+    # Do not display the time if it is above 100 hours (360000000 ms)
+    if (( elapsed > 360000000 )); then
+        return
+    fi
+
+    if (( elapsed > 0 )); then
+        local color time_str
+        local seconds=$((elapsed / 1000))
+        local milliseconds=$((elapsed % 1000))
+
+        if (( seconds >= 60 )); then
+            local minutes=$((seconds / 60))
+            seconds=$((seconds % 60))
+            if (( minutes >= 60 )); then
+                local hours=$((minutes / 60))
+                minutes=$((minutes % 60))
+                time_str="${hours}h ${minutes}m"
+                if (( seconds > 0 )); then
+                    time_str="${time_str} ${seconds}s"
+                fi
+            else
+                time_str="${minutes}m"
+                if (( seconds > 0 )); then
+                    time_str="${time_str} ${seconds}s"
+                fi
+            fi
+            color="%{$fg[red]%}"
+        elif (( seconds >= 10 )); then
+            time_str="${seconds}s"
+            color="%{$fg[orange]%}"
+        elif (( seconds >= 5 )); then
+            time_str="${seconds}s"
+            color="%{$fg[yellow]%}"
+        elif (( seconds >= 1 )); then
+            time_str="${seconds}s ${milliseconds}ms"
+            color="%{$fg_bold[green]%}"
+        else
+            time_str="${milliseconds}ms"
+            color="%{$fg_bold[green]%}"
+        fi
+
+        # Append the time string to RPROMPT
+        if [[ -n $time_str ]]; then
+            RPROMPT="${RPROMPT} ${color}${time_str}%{$reset_color%}"
+        fi
+    fi
+}