From 948cc926a5d1a1e1bf0948f4ca3812a995f58698 Mon Sep 17 00:00:00 2001
From: Noel Kwan <47273164+kwannoel@users.noreply.github.com>
Date: Wed, 7 Feb 2024 22:43:08 +0800
Subject: [PATCH] fix(scripts): update the install script to be POSIX
 compliant, support linux-aarch64, pin brew install version (#15028)

---
 scripts/install/install-risingwave.sh | 109 ++++++++++++++++++++------
 1 file changed, 87 insertions(+), 22 deletions(-)

diff --git a/scripts/install/install-risingwave.sh b/scripts/install/install-risingwave.sh
index 7547f574ebe4c..9bdb5a99fa9cd 100755
--- a/scripts/install/install-risingwave.sh
+++ b/scripts/install/install-risingwave.sh
@@ -1,23 +1,46 @@
 #!/bin/sh -e
 
-OS=$(uname -s)
-ARCH=$(uname -m)
-VERSION=$(curl -s https://api.github.com/repos/risingwavelabs/risingwave/releases/latest | grep '.tag_name' | sed -E -n 's/.*(v[0-9]+.[0-9]+.[0-9])\",/\1/p')
+if [ -z "${OS}" ]; then
+  OS=$(uname -s)
+fi
+if [ -z "${ARCH}" ]; then
+  ARCH=$(uname -m)
+fi
+STATE_STORE_PATH="${HOME}/.risingwave/state_store"
+META_STORE_PATH="${HOME}/.risingwave/meta_store"
+
+VERSION="v1.7.0-single-node-2"
+# TODO(kwannoel): re-enable it once we have stable release in latest for single node mode.
+#VERSION=$(curl -s https://api.github.com/repos/risingwavelabs/risingwave/releases/latest \
+# | grep '.tag_name' \
+# | sed -E -n 's/.*(v[0-9]+.[0-9]+.[0-9])\",/\1/p')
 BASE_URL="https://github.com/risingwavelabs/risingwave/releases/download"
 
-# TODO(kwannoel): Add support for other OS and architectures.
-if [[ "${OS}" == "Linux" && "${ARCH}" == "x86_64" || "${ARCH}" == "amd64" ]]; then
-  BASE_ARCHIVE_NAME="risingwave-${VERSION}-x86_64-unknown-linux-all-in-one"
-  ARCHIVE_NAME="${BASE_ARCHIVE_NAME}.tar.gz"
-  URL="${BASE_URL}/${VERSION}/${ARCHIVE_NAME}"
-  USE_BREW=0
-elif [[ "${OS}" == "Darwin" ]] && [[ "${ARCH}" == "x86_64" || "${ARCH}" == "amd64" || "${ARCH}" == "aarch64" || "${ARCH}" == "arm64" ]]; then
-  USE_BREW=1
-else
+if [ "${OS}" = "Linux" ]; then
+  if [ "${ARCH}" = "x86_64" ] || [ "${ARCH}" = "amd64" ]; then
+    BASE_ARCHIVE_NAME="risingwave-${VERSION}-x86_64-unknown-linux-all-in-one"
+    ARCHIVE_NAME="${BASE_ARCHIVE_NAME}.tar.gz"
+    URL="${BASE_URL}/${VERSION}/${ARCHIVE_NAME}"
+    USE_BREW=0
+  elif [ "${ARCH}" = "arm64" ] || [ "${ARCH}" = "aarch64" ]; then
+    BASE_ARCHIVE_NAME="risingwave-${VERSION}-aarch64-unknown-linux-all-in-one"
+    ARCHIVE_NAME="${BASE_ARCHIVE_NAME}.tar.gz"
+    URL="${BASE_URL}/${VERSION}/${ARCHIVE_NAME}"
+    USE_BREW=0
+  fi
+elif [ "${OS}" = "Darwin" ]; then
+  if [ "${ARCH}" = "x86_64" ] || [ "${ARCH}" = "amd64" ] || [ "${ARCH}" = "aarch64" ] || [ "${ARCH}" = "arm64" ]; then
+    echo "Brew installation is not supported yet."
+    # USE_BREW=1
+  fi
+fi
+
+if [ -z "$USE_BREW" ]; then
   echo
   echo "Unsupported OS or Architecture: ${OS}-${ARCH}"
   echo
-  echo "Supported OSes: Linux, macOS"
+  echo "Supported OSes: Linux"
+  # echo "Supported OSes: Linux, macOS"
   echo "Supported architectures: x86_64"
   echo
   echo "Please open an issue at <https://github.com/risingwavelabs/risingwave/issues/new/choose>,"
@@ -27,28 +50,70 @@ else
 fi
 
 ############# Setup data directories
+echo
 echo "Setting up data directories."
-mkdir -p "${HOME}/.risingwave/data/state_store"
-mkdir -p "${HOME}/.risingwave/data/meta_store"
+echo "- ${STATE_STORE_PATH}"
+echo "- ${META_STORE_PATH}"
+mkdir -p "${STATE_STORE_PATH}"
+mkdir -p "${META_STORE_PATH}"
+echo
 
 ############# BREW INSTALL
-if [[ "${USE_BREW}" -eq 1 ]]; then
-  echo "Installing RisingWave using Homebrew."
+if [ "${USE_BREW}" -eq 1 ]; then
+  echo "Installing RisingWave@${VERSION} using Homebrew."
   brew tap risingwavelabs/risingwave
-  brew install risingwave
-  echo "Successfully installed RisingWave using Homebrew."
+  brew install risingwave@${VERSION}
+  echo "Successfully installed RisingWave@${VERSION} using Homebrew."
   echo
   echo "You can run it as:"
-  echo "  risingwave standalone"
+  echo
+  echo "  risingwave >risingwave.log 2>&1 &"
+  echo
+  echo
+  echo "In a separate terminal, you can attach a psql client to the standalone server using:"
+  echo
+  echo "  psql -h localhost -p 4566 -d dev -U root"
+  echo
+  echo
+  echo "To start a fresh cluster, you can just delete the data directory contents:"
+  echo
+  echo "  rm -r ~/.risingwave/state_store/*"
+  echo "  rm -r ~/.risingwave/meta_store/*"
+  echo
+  echo
+  echo "To view available options, run:"
+  echo
+  echo "  ./risingwave single-node --help"
+  echo
+  echo
   exit 0
 fi
 
 ############# BINARY INSTALL
 echo
-echo "Downloading ${URL} into ${PWD}."
+echo "Downloading RisingWave@${VERSION} from ${URL} into ${PWD}."
 echo
 curl -L "${URL}" | tar -zx || exit 1
 chmod +x risingwave
 echo
 echo "Successfully downloaded the RisingWave binary, you can run it as:"
-echo "  ./risingwave standalone"
\ No newline at end of file
+echo
+echo "  ./risingwave >risingwave.log 2>&1 &"
+echo
+echo
+echo "In a separate terminal, you can connect a psql client to the standalone server using:"
+echo
+echo "  psql -h localhost -p 4566 -d dev -U root"
+echo
+echo
+echo "To start a fresh cluster, you can just delete the data directory contents:"
+echo
+echo "  rm -r ~/.risingwave/state_store/*"
+echo "  rm -r ~/.risingwave/meta_store/*"
+echo
+echo
+echo "To view other available options, run:"
+echo
+echo "  ./risingwave single-node --help"
+echo
+# TODO(kwannoel): Include link to our docs.
\ No newline at end of file