Fix scripts/serve-docs.sh

RELNOTES: None.
PiperOrigin-RevId: 157463020
diff --git a/scripts/serve-docs.sh b/scripts/serve-docs.sh
index 27d1577..edbe413 100755
--- a/scripts/serve-docs.sh
+++ b/scripts/serve-docs.sh
@@ -15,6 +15,12 @@
 
 set -eu
 
+readonly WORKING_DIR=$(mktemp -d)
+HOST=localhost
+PORT=12345
+TARGET=
+SERVING_PREFIX=
+
 usage() {
   cat <<EOF
 Usage: $0 [--port 12345] [--target DIR [PREFIX]] [--share]
@@ -33,45 +39,6 @@
 EOF
 }
 
-HOST=localhost
-PORT=12345
-TARGET=
-SERVING_PREFIX=
-while [[ $# -gt 0 ]]
-do
-  key="$1"
-  case $key in
-    --port)
-      PORT="$2"
-      shift
-      ;;
-    --share)
-      HOST="$HOSTNAME"
-      ;;
-    --target)
-      TARGET="$2"
-      shift
-      SERVING_PREFIX="${2:-}"
-      build_static
-      exit 0
-      ;;
-    --help|help)
-      usage
-      exit 0
-      ;;
-    *)
-      usage
-      exit 1
-  esac
-  shift
-done
-
-readonly WORKING_DIR=$(mktemp -d)
-
-check() {
-  which $1 > /dev/null || (echo "$1 not installed. Please install $1."; exit 1)
-}
-
 build_tree() {
   bazel build //site:jekyll-tree.tar
   rm -rf $WORKING_DIR/*
@@ -96,6 +63,20 @@
   jekyll serve --host "$HOST" --detach --quiet --port "$PORT" --source "$WORKING_DIR"
 }
 
+check() {
+  which $1 > /dev/null || (echo "$1 not installed. Please install $1."; exit 1)
+}
+
+kill_jekyll() {
+  pid="$(lsof "-tiTCP:$PORT" -sTCP:LISTEN)" || true
+  if [ ! -z "$pid" ]; then
+     kill "$pid"
+  fi
+  # I found I got bind errors sometimes if I didn't wait a second for the server to
+  # actually shut down.
+  sleep 2
+}
+
 main() {
   check jekyll
 
@@ -125,15 +106,34 @@
   done
 }
 
-kill_jekyll() {
-  pid="$(lsof "-tiTCP:$PORT" -sTCP:LISTEN)" || true
-  if [ ! -z "$pid" ]; then
-     kill "$pid"
-  fi
-  # I found I got bind errors sometimes if I didn't wait a second for the server to
-  # actually shut down.
-  sleep 2
-}
+while [[ $# -gt 0 ]]
+do
+  key="$1"
+  case $key in
+    --port)
+      PORT="$2"
+      shift
+      ;;
+    --share)
+      HOST="$HOSTNAME"
+      ;;
+    --target)
+      TARGET="$2"
+      shift
+      SERVING_PREFIX="${2:-}"
+      build_static
+      exit 0
+      ;;
+    --help|help)
+      usage
+      exit 0
+      ;;
+    *)
+      usage
+      exit 1
+  esac
+  shift
+done
 
 cleanup() {
   rm -rf $WORKING_DIR