chore: improve install.sh script

Author: Barbapapazes

src/public/install.sh

@@ -7,6 +7,26 @@ GH_REPO="Barbapapazes-Sponsors/vue-ui"
 GH_TOKEN=$GITHUB_TOKEN
 RELATIONS_URL="https://vue-ui.barbapapazes.dev/relations.json"
 
+validate_dependencies() {
+  for cmd in curl jq sed; do
+    if ! command -v $cmd &>/dev/null; then
+      echo "Error: $cmd is not installed. Please install it and try again."
+      exit 1
+    fi
+  done
+
+  if [ -z "$GH_TOKEN" ]; then
+    echo "Error: GH_TOKEN environment variable is not set."
+    exit 1
+  fi
+}
+
+log_debug() {
+  if [ "$DEBUG" = true ]; then
+    echo "[DEBUG] $(date +'%Y-%m-%d %H:%M:%S') $1"
+  fi
+}
+
 fetch_directory_contents() {
   local repo_path=$1
   curl -sL \
@@ -19,8 +39,10 @@ fetch_directory_contents() {
 rewrite_path() {
   local local_path=$1
   local name=$2
-  # If it's a component, we need to remove the subfolder references (components/ComponentName/ComponentName.vue -> components/ComponentName.vue)
-  if [[ "$local_path" == *"/components/"* ]]; then
+
+  # If it's a component, remove the subfolder references (components/ComponentName/ComponentName.vue to components/ComponentName.vue or Components/ComponentName/ComponentName.vue to Components/ComponentName.vue)
+
+  if [[ "$local_path" == *"/components/"* || "$local_path" == *"/Components/"* ]]; then
     echo "${local_path%/*}/$name"
   else
     echo "$local_path/$name"
@@ -30,15 +52,17 @@ rewrite_path() {
 rewrite_imports() {
   local final_path=$1
   local mode=$2
-  sed -i 's|@/ui/components\/\([^/]*\)\/\1.vue|@/ui/components/\1.vue|g' "$final_path"
+  local case_option=$3
+
+  sed -i 's|@/ui/components/\([^/]*\)/\1.vue|@/ui/components/\1.vue|g' "$final_path"
+
   if [[ "$mode" == "inertia" ]]; then
-    sed -i 's|@/ui/components|@/UI/Components|g' "$final_path"
-    sed -i 's|@/ui/composables|@/UI/Composables|g' "$final_path"
-    sed -i 's|@/ui/theme|@/UI/Theme|g' "$final_path"
-    sed -i 's|@/ui/utils|@/UI/Utils|g' "$final_path"
-    sed -i 's|@/ui/types|@/UI/Types|g' "$final_path"
-    sed -i 's|@/ui/icons|@/UI/Icons|g' "$final_path"
-    sed -i 's|@/ui/keys|@/UI/Keys|g' "$final_path"
+    if [[ "$case_option" == "capitalize" ]]; then
+      for folder in components composables theme utils types icons keys; do
+        capitalized_folder=$(echo "$folder" | sed -E 's/^(.)/\U\1/')
+        sed -i "s|@/ui/$folder|@/UI/$capitalized_folder|g" "$final_path"
+      done
+    fi
   fi
 }
 
@@ -47,10 +71,13 @@ download_files() {
   local repo_path=$1
   local local_path=$2
   local mode=$3
+  local case_option=$4
+  local max_jobs=5  # Limit the number of parallel jobs
+  local job_count=0
 
   response=$(fetch_directory_contents "$repo_path")
 
-  [ "$DEBUG" = true ] && echo "DEBUG: Processing $(echo "$response" | jq -r '. | length') items"
+  log_debug "Processing $(echo "$response" | jq -r '. | length') items"
 
   # Loop through each item in the directory
   for item in $(echo "$response" | jq -r '.[] | @base64'); do
@@ -63,19 +90,31 @@ download_files() {
     type=$(_jq '.type')
     download_url=$(_jq '.download_url')
 
+    # Skip the public directory
+    if [[ "$path" == *"/public"* ]]; then
+      log_debug "Skipping public directory: $path"
+      continue
+    fi
+
     if [ "$type" == "dir" ]; then
       if [[ "$name" == "__snapshots__" ]]; then
         continue
       fi
 
-      if [[ "$local_path" != *"/components"* ]]; then
+      if [[ "$case_option" == "capitalize" ]]; then
+        # Capitalize the folder name
+        capitalized_name=$(echo "$name" | sed -E 's/^(.)/\U\1/')
+        mkdir -p "$local_path/$capitalized_name"
+        next_local_path="$local_path/$capitalized_name"
+      else
         mkdir -p "$local_path/$name"
+        next_local_path="$local_path/$name"
       fi
 
-      [ "$DEBUG" = true ] && echo "DEBUG: Processing directory $name"
+      log_debug "Processing directory $name"
 
       # Recursively download the contents of the directory
-      download_files "$path" "$local_path/$name" "$mode"
+      download_files "$path" "$next_local_path" "$mode" "$case_option" &
     else
       if [[ "$name" == *.md ]]; then
         continue
@@ -85,30 +124,40 @@ download_files() {
         continue
       fi
 
-      [ "$DEBUG" = true ] && echo "DEBUG: Processing file $name"
+      log_debug "Processing file $name"
+
+      # Ensure the directory exists before downloading the file
+      mkdir -p "$local_path"
 
       # Determine the final path by calling the new rewrite_path function
       final_path=$(rewrite_path "$local_path" "$name")
 
-      echo "Relations: ${relations[@]}" "$name"
-
+      echo "Processing file $name to $final_path"
       # for each relation, we check if the name of the file starts with the relation (only if the relations is not empty)
       if [ ${#relations[@]} -gt 0 ]; then
         for relation in "${relations[@]}"; do
           if [[ "$name" == "$relation"* ]]; then
-            [ "$DEBUG" = true ] && echo "DEBUG: Downloading file $name to $final_path"
-            download_file "$download_url" "$final_path"
-            rewrite_imports "$final_path" "$mode"
+            log_debug "Downloading file $name to $final_path"
+            download_file "$download_url" "$final_path" && rewrite_imports "$final_path" "$mode" "$case_option" &
           fi
 
         done
       else
-        [ "$DEBUG" = true ] && echo "DEBUG: Downloading file $name to $final_path"
-        download_file "$download_url" "$final_path"
-        rewrite_imports "$final_path" "$mode"
+        log_debug "Downloading file $name to $final_path"
+        download_file "$download_url" "$final_path" && rewrite_imports "$final_path" "$mode" "$case_option" &
       fi
     fi
+
+    # Manage the job queue
+    ((job_count++))
+    if (( job_count >= max_jobs )); then
+      wait -n  # Wait for at least one job to finish before continuing
+      ((job_count--))
+    fi
   done
+
+  # Wait for all background processes to complete
+  wait
 }
 
 download_file() {
@@ -173,6 +222,7 @@ install_component() {
   local component=$1
   local local_path=$2
   local mode=$3
+  local case_option=$4
   local repo_path="src"
 
   all_relations=$(fetch_relations)
@@ -181,24 +231,26 @@ install_component() {
   relations=("$component")
   search_relations "components" "$component" "$all_relations"
 
-  download_files "$repo_path" "$local_path" "$mode"
+  download_files "$repo_path" "$local_path" "$mode" "$case_option"
 }
 
 install_components() {
   local local_path=$1
   local mode=$2
+  local case_option=$3
   local repo_path="src"
 
-  download_files "$repo_path" "$local_path" "$mode"
+  download_files "$repo_path" "$local_path" "$mode" "$case_option"
 }
 
 parse_options() {
-  while getopts "s:d:m:c:" opt; do
+  while getopts "s:d:m:c:C:" opt; do
     case "$opt" in
     s) REPO_PATH="$OPTARG" ;;
     d) LOCAL_PATH="$OPTARG" ;;
     m) MODE="$OPTARG" ;;
     c) COMPONENT="$OPTARG" ;;
+    C) CASE_OPTION="$OPTARG" ;;
     *) usage ;;
     esac
   done
@@ -217,24 +269,46 @@ validate_options() {
     echo "Error: mode must be either 'vite' or 'inertia'."
     usage
   fi
+  # Set default case option if not provided
+  if [ -z "$CASE_OPTION" ]; then
+    CASE_OPTION="lower"
+  fi
+  # Validate case option value
+  if [ "$CASE_OPTION" != "lower" ] && [ "$CASE_OPTION" != "capitalize" ]; then
+    echo "Error: case must be either 'lower' or 'capitalize'."
+    usage
+  fi
 }
 
 usage() {
-  echo "Usage: $0 -s <repo_path> -d <local_path> [-m <mode>] [-c <component>]"
+  echo "Usage: $0 -s <repo_path> -d <local_path> [-m <mode>] [-c <component>] [-C <case_option>]"
   echo "       mode: vite (default) or inertia"
   echo "       component: name of the component to install (optional)"
+  echo "       case_option: lower (default) or capitalize"
+  echo "       --help: display this help message"
   exit 1
 }
 
 main() {
+  if [[ "$1" == "--help" ]]; then
+    usage
+  fi
+
+  validate_dependencies
   parse_options "$@"
   validate_options
 
   if [ -n "$COMPONENT" ]; then
-    install_component "$COMPONENT" "$LOCAL_PATH" "$MODE"
+    install_component "$COMPONENT" "$LOCAL_PATH" "$MODE" "$CASE_OPTION"
   else
-    install_components "$LOCAL_PATH" "$MODE"
+    install_components "$LOCAL_PATH" "$MODE" "$CASE_OPTION"
   fi
+
+  # Remove empty directories
+  find "$LOCAL_PATH" -type d -empty -delete
+
+  echo "Installation completed successfully."
+  echo "You can now use the components in your project."
 }
 
 main "$@"