From 9ebac97bc6891b5f4f4a13e730584c45da5d333f Mon Sep 17 00:00:00 2001
From: Daniel Gibbs <me@danielgibbs.co.uk>
Date: Wed, 5 Apr 2023 00:59:59 +0100
Subject: [PATCH] feat(core): if github branch doesnt exist default to master

---
 linuxgsm.sh | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/linuxgsm.sh b/linuxgsm.sh
index dd7d51bb3..472617cd7 100755
--- a/linuxgsm.sh
+++ b/linuxgsm.sh
@@ -320,6 +320,18 @@ fn_install_file() {
 	exit
 }
 
+# Check that git branch exists.
+if [ "${remotereponame}" == "GitHub" ]; then
+	branchexistscheck=$(curl -s -o /dev/null -w "%{http_code}" "https://raw.githubusercontent.com/${githubuser}/${githubrepo}/${githubbranch}/linuxgsm.sh" 1> /dev/null)
+else
+	branchexistscheck=$(curl -s -o /dev/null -w "%{http_code}" "https://bitbucket.org/${githubuser}/${githubrepo}/raw/${githubbranch}/linuxgsm.sh" 1> /dev/null)
+fi
+
+if [ "${branchexistscheck}" != "200" ]; then
+	echo -e "${githubbranch} branch does not exist. Defaulting to master branch."
+	githubbranch="master"
+fi
+
 # Prevent LinuxGSM from running as root. Except if doing a dependency install.
 if [ "$(whoami)" == "root" ] && [ ! -f /.dockerenv ]; then
 	if [ "${userinput}" == "install" ] || [ "${userinput}" == "auto-install" ] || [ "${userinput}" == "i" ] || [ "${userinput}" == "ai" ]; then