From 8fb707e07daaccf8604e5a4ea0a7df553c76bc99 Mon Sep 17 00:00:00 2001
From: zzjc1234 <2359047351@qq.com>
Date: Fri, 11 Oct 2024 18:51:50 +0800
Subject: [PATCH] feat(healthcheck/reposize): customize repo size

---
 cmd/repo-health-checker/main.go | 3 ++-
 pkg/healthcheck/reposize.go     | 6 +++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/cmd/repo-health-checker/main.go b/cmd/repo-health-checker/main.go
index 160b587..8a75261 100644
--- a/cmd/repo-health-checker/main.go
+++ b/cmd/repo-health-checker/main.go
@@ -44,6 +44,7 @@ func main() {
 	checkRelease := flag.Bool("checkRelease", true, "trigger release check")
 	rootDir := flag.String("root", "", "")
 	repo := flag.String("repo", "", "")
+	size := flag.Int("reposize", 2, "size of the repo")
 	localList := flag.String("localList", "", "")
 	droneBranch := flag.String("droneBranch", "", "")
 	checkFileNameList := flag.String("checkFileNameList", "", "Comma-separated list of files to check.")
@@ -57,7 +58,7 @@ func main() {
 	}
 	setupSlog()
 	var err error
-	err = healthcheck.RepoSize()
+	err = healthcheck.RepoSize(*size)
 	if err != nil {
 		fmt.Printf("### Repo Size Check Failed:\n%s\n", err.Error())
 	}
diff --git a/pkg/healthcheck/reposize.go b/pkg/healthcheck/reposize.go
index dd17d06..4740a5a 100644
--- a/pkg/healthcheck/reposize.go
+++ b/pkg/healthcheck/reposize.go
@@ -10,7 +10,7 @@ import (
 
 // RepoSize checks the size of the repository to determine if it is oversized.
 // It executes the 'git count-objects -v' command to obtain the size information,
-func RepoSize() error {
+func RepoSize(conf_size int) error {
 	// TODO: reimplement here when go-git is available
 	// https://github.com/go-git/go-git/blob/master/COMPATIBILITY.md
 	cmd := exec.Command("git", "count-objects", "-v")
@@ -33,8 +33,8 @@ func RepoSize() error {
 			sum += size
 		}
 	}
-	if sum > 2048 {
-		return fmt.Errorf("Repository larger than 2MB. Please clean up or contact the teaching team.")
+	if sum > conf_size {
+		return fmt.Errorf("Repository larger than %d M. Please clean up or contact the teaching team.", conf_size)
 	}
 	return nil
 }