refactor: extract function for remote

Author: necusjz

.devcontainer/setup.sh

@@ -5,24 +5,39 @@ GREEN="\033[0;32m"
 YELLOW="\033[1;33m"
 NC="\033[0m"  # no color
 
+set_or_add_remote() {
+    local REMOTE_NAME=$1
+    local REMOTE_URL=$2
+    local REPO_PATH="/workspaces/$REPO_NAME"
+
+    git -C "$REPO_PATH" remote get-url "$REMOTE_NAME" &>/dev/null || git -C "$REPO_PATH" remote add "$REMOTE_NAME" "$REMOTE_URL"
+    git -C "$REPO_PATH" remote set-url "$REMOTE_NAME" "$REMOTE_URL"
+}
+
 setup_repo() {
     local DIR_NAME="$1"
     local DIR_PATH="/workspaces/$DIR_NAME"
     local REPO="Azure/$DIR_NAME"
 
     if [ -d "$DIR_PATH" ]; then
         echo -e "\n${YELLOW}($DIR_NAME) Pulling the latest changes from upstream...${NC}"
+        gh repo fork "$REPO" --clone=false
     else
         echo -e "\n${GREEN}($DIR_NAME) Forking and cloning the repository...${NC}"
         gh repo fork "$REPO" --clone=true
     fi
 
-    HEAD_BRANCH=$(git -C "$DIR_PATH" remote show upstream | grep "HEAD branch" | awk '{print $NF}')
-    git -C "$DIR_PATH" pull -r upstream "$HEAD_BRANCH"
+    cd "$DIR_PATH"
+    gh repo sync --source "$REPO"
+    cd /workspaces
 }
 
 SECONDS=0
 
+REPO_NAME=$(basename "$GITHUB_REPOSITORY")
+set_or_add_remote origin "https://github.com/$GITHUB_USER/$REPO_NAME.git"
+set_or_add_remote upstream "https://github.com/Azure/$REPO_NAME.git"
+
 pip install aaz-dev
 
 # azdev repositories