From 3fe26f19ffa406d8a020fe4e1383a20156dd361a Mon Sep 17 00:00:00 2001
From: Hariom Balhara <hariombalhara@gmail.com>
Date: Tue, 25 Jun 2024 17:15:33 +0530
Subject: [PATCH] Fix git-setup.sh (#15560)

---
 git-setup.sh | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/git-setup.sh b/git-setup.sh
index f9deccf6d5..3defaecb7e 100755
--- a/git-setup.sh
+++ b/git-setup.sh
@@ -2,7 +2,7 @@
 # If no project name is given
 if [ $# -eq 0 ]; then
   # Display usage and stop
-  echo "Usage: git-setup.sh <console,website>"
+  echo "Usage: git-setup.sh <console,website,platform>"
   exit 1
 fi
 # Get remote url to support either https or ssh
@@ -21,11 +21,18 @@ for module in "$@"; do
     git config -f .gitmodules --unset-all "submodule.apps/$module.branch"
     # Add the submodule
     git submodule add --force $project "apps/$module"
-    # Set the default branch to main
-    git config -f .gitmodules --add "submodule.apps/$module.branch" main
     
-    # Update to the latest from main in that submodule
-    cd apps/$module && git pull origin main && cd ../..
+    # Determine the branch based on module
+    branch="main"
+    if [ "$module" = "website" ]; then
+      branch="production"
+    fi
+
+    # Set the default branch
+    git config -f .gitmodules --add "submodule.apps/$module.branch" ${branch}
+    
+    # Update to the latest of branch in that submodule
+    cd apps/$module && git pull origin ${branch} && cd ../..
 
     # We forcefully added the subdmoule which was in .gitignore, so unstage it.
     git restore --staged apps/$module