1 files changed, 14 insertions, 0 deletions
diff --git a/common.go b/common.go
index fb17679..a9bc736 100644
--- a/common.go
+++ b/common.go
@@ -32,3 +32,17 @@ func (repo *Repo) SetGoPrimitive(b bool) {
 	}
 	repo.GoInfo.GoPrimitive = b
 }
+
+func (repo *Repo) IsUserBranch() bool {
+	if repo.GetCurrentBranchName() == repo.GetUserBranchName() {
+		return true
+	}
+	return false
+}
+
+func (repo *Repo) IsMasterBranch() bool {
+	if repo.GetCurrentBranchName() == repo.GetMasterBranchName() {
+		return true
+	}
+	return false
+}