From a83d62691203f5933d670c19fe78883d87512bd3 Mon Sep 17 00:00:00 2001
From: Nate Butler <iamnbutler@gmail.com>
Date: Mon, 20 Nov 2023 17:50:25 -0500
Subject: [PATCH] Make the script a bit more resilient

---
 script/deploy-docs | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/script/deploy-docs b/script/deploy-docs
index ce606e0327..9632215ff4 100755
--- a/script/deploy-docs
+++ b/script/deploy-docs
@@ -1,5 +1,11 @@
 #!/bin/bash
 
+# Check if the script is run from the root of the repository
+if [ ! -f "Cargo.toml" ] || [ ! -d "crates/zed" ]; then
+    echo "Please run the script from the root of the repository."
+    exit 1
+fi
+
 # Set the environment variables
 TARGET_DIR="../zed-docs"
 PUSH_CHANGES=false
@@ -21,17 +27,10 @@ while getopts "pc" opt; do
   esac
 done
 
-if "$CLEAN_FOLDERS"; then
-  echo "Cleaning ./doc and ./debug folders..."
-  rm -rf "$TARGET_DIR/doc"
-  rm -rf "$TARGET_DIR/debug"
-fi
-
 # Check if the target documentation directory exists
 if [ ! -d "$TARGET_DIR" ]; then
     # Prompt the user for input
-    read -p "The zed-docs directory does not exist. Make sure you are running this from the zed repo root." -n 1 -r
-    read -p "Do you want to clone the repository (y/n)? " -n 1 -r
+    read -p "Can't find ../zed-docs. Do you want to clone the repository (y/n)?" -n 1 -r
     echo  # Move to a new line
 
     if [[ $REPLY =~ ^[Yy]$ ]]; then
@@ -49,6 +48,12 @@ else
     popd > /dev/null
 fi
 
+if "$CLEAN_FOLDERS"; then
+  echo "Cleaning ./doc and ./debug folders..."
+  rm -rf "$TARGET_DIR/doc"
+  rm -rf "$TARGET_DIR/debug"
+fi
+
 # Build the documentation
 CARGO_TARGET_DIR="$TARGET_DIR" cargo doc --workspace --no-deps --open \
 --exclude activity_indicator \