Merge branch 'master' into feat/releasetag

Co-authored-by: zzjc1234 <2359047351@qq.com>
Co-authored-by: Boming Zhang <bomingzh@sjtu.edu.cn>
Reviewed-on: #56
Reviewed-by: 张泊明518370910136 <bomingzh@sjtu.edu.cn>
Co-authored-by: 周赵嘉程521432910016 <zzjc123@sjtu.edu.cn>
Co-committed-by: 周赵嘉程521432910016 <zzjc123@sjtu.edu.cn>

Makefile

@@ -3,7 +3,9 @@
 BUILD_DIR = ./build
 TMP_DIR = ./tmp
 APPS := $(notdir $(wildcard ./cmd/*))
-VERSION := $(shell git rev-parse --short HEAD)
+COMMIT_HASH := $(shell git rev-parse --short HEAD)
+DATE := $(shell date +"%Y%m%d-%H%M%S")
+VERSION := $(COMMIT_HASH)-$(DATE)
 FLAGS := "-s -w -X main.Version=$(VERSION)"
 
 all: build
@@ -30,4 +32,4 @@ test:
 ci-test:
 	./scripts/prepare_test_repos.sh $(TMP_DIR)
 	./scripts/run_foreach_test_repos.sh $(TMP_DIR) "sed -i '2i \ \ \"sandboxExecServer\": \"172.17.0.1:5051\",' conf.json"
-	go test -coverprofile cover.out -v ./...
+	GITHUB_ACTIONS="test" go test -coverprofile cover.out -v ./...

README.md

@@ -4,12 +4,16 @@
 
 ## Quick Start
 
-1. Make sure you are in a Unix-like OS (Linux, MacOS). For Windows, use [WSL 2](https://learn.microsoft.com/en-us/windows/wsl/install).
+1. Clone this repo in a Linux computer. For Windows, use [WSL 2](https://learn.microsoft.com/en-us/windows/wsl/install).
+
+```bash
+$ git clone ssh://git@focs.ji.sjtu.edu.cn:2222/JOJ/JOJ3.git
+```
 
 2. Install [Go](https://go.dev/doc/install). Also, make sure `make` and `git` are installed and all 3 programs are presented in `$PATH`.
    - If you have problem on connecting to the Go website and Go packages, download Go from [studygolang](https://studygolang.com/dl) and run `go env -w GOPROXY=https://goproxy.io,direct` to set the Go modules mirror proxy after installing Go.
 
-3. Enable cgroup v2 for your OS. Check [here](https://stackoverflow.com/a/73376219/13724598). So that you do not need root permission to run `go-judge`.
+3. Enable cgroup v2 for your OS. For WSL2, check [here](https://stackoverflow.com/a/73376219/13724598). So that you do not need root permission to run `go-judge`.
 
 4. Clone [go-judge](https://github.com/criyle/go-judge).
 

cmd/joj3/conf/conf.go

@@ -13,16 +13,48 @@ import (
 	"github.com/koding/multiconfig"
 )
 
+type ConfStage struct {
+	Name     string
+	Group    string
+	Executor struct {
+		Name string
+		With struct {
+			Default stage.Cmd
+			Cases   []OptionalCmd
+		}
+	}
+	Parsers []struct {
+		Name string
+		With interface{}
+	}
+}
+
 type Conf struct {
+	Name    string `default:"unknown"`
+	LogPath string `default:""`
+	Stage   struct {
+		SandboxExecServer string `default:"localhost:5051"`
+		SandboxToken      string `default:""`
+		OutputPath        string `default:"joj3_result.json"`
+		Stages            []ConfStage
+	}
+	Teapot struct {
+		LogPath         string `default:"/home/tt/.cache/joint-teapot-debug.log"`
+		ScoreboardPath  string `default:"scoreboard.csv"`
+		FailedTablePath string `default:"failed-table.md"`
+		GradingRepoName string `default:""`
+		SkipIssue       bool   `default:"false"`
+		SkipScoreboard  bool   `default:"false"`
+		SkipFailedTable bool   `default:"false"`
+	}
+	// TODO: remove the following backward compatibility fields
 	SandboxExecServer string `default:"localhost:5051"`
 	SandboxToken      string `default:""`
-	LogPath           string `default:""`
 	OutputPath        string `default:"joj3_result.json"`
 	GradingRepoName   string `default:""`
 	SkipTeapot        bool   `default:"true"`
 	ScoreboardPath    string `default:"scoreboard.csv"`
 	FailedTablePath   string `default:"failed-table.md"`
-	Name              string `default:"unknown"`
 	Stages            []struct {
 		Name     string
 		Group    string
@@ -112,25 +144,55 @@ func parseConventionalCommit(commit string) (*ConventionalCommit, error) {
 	return cc, nil
 }
 
-func parseConfFile(path string) (conf Conf, err error) {
+func ParseConfFile(path string) (conf *Conf, err error) {
+	conf = new(Conf)
 	d := &multiconfig.DefaultLoader{}
 	d.Loader = multiconfig.MultiLoader(
 		&multiconfig.TagLoader{},
 		&multiconfig.JSONLoader{Path: path},
 	)
 	d.Validator = multiconfig.MultiValidator(&multiconfig.RequiredValidator{})
-	if err = d.Load(&conf); err != nil {
+	if err = d.Load(conf); err != nil {
 		slog.Error("parse stages conf", "error", err)
 		return
 	}
-	if err = d.Validate(&conf); err != nil {
+	if err = d.Validate(conf); err != nil {
 		slog.Error("validate stages conf", "error", err)
 		return
 	}
+	// TODO: remove the following backward compatibility codes
+	if len(conf.Stage.Stages) == 0 {
+		conf.Stage.SandboxExecServer = conf.SandboxExecServer
+		conf.Stage.SandboxToken = conf.SandboxToken
+		conf.Stage.OutputPath = conf.OutputPath
+		conf.Stage.Stages = make([]ConfStage, len(conf.Stages))
+		for i, stage := range conf.Stages {
+			conf.Stage.Stages[i].Name = stage.Name
+			conf.Stage.Stages[i].Group = stage.Group
+			conf.Stage.Stages[i].Executor = stage.Executor
+			conf.Stage.Stages[i].Parsers = []struct {
+				Name string
+				With interface{}
+			}{
+				{
+					Name: stage.Parser.Name,
+					With: stage.Parser.With,
+				},
+			}
+		}
+		conf.Teapot.GradingRepoName = conf.GradingRepoName
+		conf.Teapot.ScoreboardPath = conf.ScoreboardPath
+		conf.Teapot.FailedTablePath = conf.FailedTablePath
+		if conf.SkipTeapot {
+			conf.Teapot.SkipScoreboard = true
+			conf.Teapot.SkipFailedTable = true
+			conf.Teapot.SkipIssue = true
+		}
+	}
 	return
 }
 
-func ParseMsg(confRoot, confName, msg string) (conf Conf, group string, err error) {
+func ParseMsg(confRoot, confName, msg string) (confPath, group string, err error) {
 	slog.Info("parse msg", "msg", msg)
 	conventionalCommit, err := parseConventionalCommit(msg)
 	if err != nil {
@@ -138,7 +200,7 @@ func ParseMsg(confRoot, confName, msg string) (conf Conf, group string, err erro
 	}
 	slog.Info("conventional commit", "commit", conventionalCommit)
 	confRoot = filepath.Clean(confRoot)
-	confPath := filepath.Clean(fmt.Sprintf("%s/%s/%s",
+	confPath = filepath.Clean(fmt.Sprintf("%s/%s/%s",
 		confRoot, conventionalCommit.Scope, confName))
 	relPath, err := filepath.Rel(confRoot, confPath)
 	if err != nil {
@@ -148,11 +210,6 @@ func ParseMsg(confRoot, confName, msg string) (conf Conf, group string, err erro
 		err = fmt.Errorf("invalid scope as path: %s", conventionalCommit.Scope)
 		return
 	}
-	slog.Info("try to load conf", "path", confPath)
-	conf, err = parseConfFile(confPath)
-	if err != nil {
-		return
-	}
 	groupKeywords := []string{"joj"}
 	for _, groupKeyword := range groupKeywords {
 		if strings.Contains(
@@ -161,7 +218,6 @@ func ParseMsg(confRoot, confName, msg string) (conf Conf, group string, err erro
 			break
 		}
 	}
-	slog.Debug("conf loaded", "conf", conf)
 	return
 }
 

cmd/joj3/main.go

@@ -45,7 +45,7 @@ func mainImpl() error {
 			return err
 		}
 	}
-	confObj, group, err := conf.ParseMsg(confRoot, confName, msg)
+	confPath, group, err := conf.ParseMsg(confRoot, confName, msg)
 	if err != nil {
 		slog.Error("parse msg", "error", err)
 		validScopes, scopeErr := conf.ListValidScopes(
@@ -58,6 +58,13 @@ func mainImpl() error {
 			"valid scopes", validScopes)
 		return err
 	}
+	slog.Info("try to load conf", "path", confPath)
+	confObj, err := conf.ParseConfFile(confPath)
+	if err != nil {
+		slog.Error("parse conf", "error", err)
+		return err
+	}
+	slog.Debug("conf loaded", "conf", confObj)
 	if err := setupSlog(confObj.LogPath); err != nil { // after conf is loaded
 		slog.Error("setup slog", "error", err)
 		return err

cmd/joj3/stage/main.go

@@ -13,10 +13,10 @@ import (
 	"github.com/jinzhu/copier"
 )
 
-func generateStages(conf conf.Conf, group string) ([]stage.Stage, error) {
+func generateStages(conf *conf.Conf, group string) ([]stage.Stage, error) {
 	stages := []stage.Stage{}
 	existNames := map[string]bool{}
-	for _, s := range conf.Stages {
+	for _, s := range conf.Stage.Stages {
 		if s.Group != "" && group != s.Group {
 			continue
 		}
@@ -50,12 +50,20 @@ func generateStages(conf conf.Conf, group string) ([]stage.Stage, error) {
 		if len(s.Executor.With.Cases) == 0 {
 			cmds = []stage.Cmd{defaultCmd}
 		}
+		parsers := []stage.StageParser{}
+		for _, p := range s.Parsers {
+			parsers = append(parsers, stage.StageParser{
+				Name: p.Name,
+				Conf: p.With,
+			})
+		}
 		stages = append(stages, stage.Stage{
-			Name:         s.Name,
-			ExecutorName: s.Executor.Name,
-			ExecutorCmds: cmds,
-			ParserName:   s.Parser.Name,
-			ParserConf:   s.Parser.With,
+			Name: s.Name,
+			Executor: stage.StageExecutor{
+				Name: s.Executor.Name,
+				Cmds: cmds,
+			},
+			Parsers: parsers,
 		})
 	}
 	slog.Debug("stages generated", "stages", stages)
@@ -73,8 +81,11 @@ func outputResult(outputPath string, results []stage.StageResult) error {
 		append(content, []byte("\n")...), 0o600)
 }
 
-func Run(conf conf.Conf, group string) error {
-	executors.InitWithConf(conf.SandboxExecServer, conf.SandboxToken)
+func Run(conf *conf.Conf, group string) error {
+	executors.InitWithConf(
+		conf.Stage.SandboxExecServer,
+		conf.Stage.SandboxToken,
+	)
 	stages, err := generateStages(conf, group)
 	if err != nil {
 		slog.Error("generate stages", "error", err)
@@ -86,7 +97,7 @@ func Run(conf conf.Conf, group string) error {
 		slog.Error("run stages", "error", err)
 		return err
 	}
-	if err := outputResult(conf.OutputPath, results); err != nil {
+	if err := outputResult(conf.Stage.OutputPath, results); err != nil {
 		slog.Error("output result", "error", err)
 		return err
 	}

cmd/joj3/teapot/main.go

@@ -7,17 +7,22 @@ import (
 	"os/exec"
 	"regexp"
 	"strings"
+	"sync"
 
 	"github.com/joint-online-judge/JOJ3/cmd/joj3/conf"
 )
 
-func Run(conf conf.Conf) error {
-	if conf.SkipTeapot {
+func Run(conf *conf.Conf) error {
+	actions := os.Getenv("GITHUB_ACTIONS")
+	if actions != "true" {
+		slog.Info("teapot exit", "GITHUB_ACTIONS", actions)
 		return nil
 	}
-	os.Setenv("LOG_FILE_PATH", "/home/tt/.cache/joint-teapot-debug.log")
+	os.Setenv("LOG_FILE_PATH", conf.Teapot.LogPath)
 	os.Setenv("_TYPER_STANDARD_TRACEBACK", "1")
 	envFilePath := "/home/tt/.config/teapot/teapot.env"
+	// TODO: pass sha to joint-teapot
+	// sha := os.Getenv("GITHUB_SHA")
 	actor := os.Getenv("GITHUB_ACTOR")
 	repository := os.Getenv("GITHUB_REPOSITORY")
 	runNumber := os.Getenv("GITHUB_RUN_NUMBER")
@@ -29,49 +34,61 @@ func Run(conf conf.Conf) error {
 	repoParts := strings.Split(repository, "/")
 	repoName := repoParts[1]
 	re := regexp.MustCompile(`\x1b\[[0-9;]*[a-zA-Z]`)
-	cmd := exec.Command("joint-teapot", "joj3-scoreboard",
-		envFilePath, conf.OutputPath, actor, conf.GradingRepoName, repoName,
-		runNumber, conf.ScoreboardPath, conf.Name) // #nosec G204
-	outputBytes, err := cmd.CombinedOutput()
-	output := re.ReplaceAllString(string(outputBytes), "")
-	for _, line := range strings.Split(output, "\n") {
-		if line == "" {
-			continue
+	execCommand := func(name string, cmdArgs []string) error {
+		cmd := exec.Command(name, cmdArgs...) // #nosec G204
+		outputBytes, err := cmd.CombinedOutput()
+		output := re.ReplaceAllString(string(outputBytes), "")
+		for _, line := range strings.Split(output, "\n") {
+			if line == "" {
+				continue
+			}
+			slog.Info(fmt.Sprintf("%s %s", name, cmdArgs[0]), "output", line)
 		}
-		slog.Info("joint-teapot joj3-scoreboard", "output", line)
-	}
-	if err != nil {
-		slog.Error("joint-teapot joj3-scoreboard", "err", err)
 		return err
 	}
-	cmd = exec.Command("joint-teapot", "joj3-failed-table",
-		envFilePath, conf.OutputPath, actor, conf.GradingRepoName, repoName,
-		runNumber, conf.FailedTablePath, conf.Name) // #nosec G204
-	outputBytes, err = cmd.CombinedOutput()
-	output = re.ReplaceAllString(string(outputBytes), "")
-	for _, line := range strings.Split(output, "\n") {
-		if line == "" {
-			continue
+	var wg sync.WaitGroup
+	var scoreboardErr, failedTableErr, issueErr error
+	wg.Add(2)
+	go func() {
+		defer wg.Done()
+		if !conf.Teapot.SkipScoreboard {
+			err := execCommand("joint-teapot", []string{
+				"joj3-scoreboard", envFilePath, conf.Stage.OutputPath, actor,
+				conf.Teapot.GradingRepoName, repoName, runNumber,
+				conf.Teapot.ScoreboardPath, conf.Name,
+			})
+			if err != nil {
+				scoreboardErr = err
+			}
 		}
-		slog.Info("joint-teapot joj3-failed-table", "output", line)
-	}
-	if err != nil {
-		slog.Error("joint-teapot joj3-failed-table", "err", err)
-		return err
-	}
-	cmd = exec.Command("joint-teapot", "joj3-create-result-issue",
-		envFilePath, conf.OutputPath, repoName, runNumber, conf.Name) // #nosec G204
-	outputBytes, err = cmd.CombinedOutput()
-	output = re.ReplaceAllString(string(outputBytes), "")
-	for _, line := range strings.Split(output, "\n") {
-		if line == "" {
-			continue
+		if !conf.Teapot.SkipFailedTable {
+			err := execCommand("joint-teapot", []string{
+				"joj3-failed-table", envFilePath, conf.Stage.OutputPath, actor,
+				conf.Teapot.GradingRepoName, repoName, runNumber,
+				conf.Teapot.FailedTablePath, conf.Name,
+			})
+			if err != nil {
+				failedTableErr = err
+			}
 		}
-		slog.Info("joint-teapot joj3-create-result-issue", "output", line)
-	}
-	if err != nil {
-		slog.Error("joint-teapot joj3-create-result-issue", "err", err)
-		return err
+	}()
+	go func() {
+		defer wg.Done()
+		if !conf.Teapot.SkipIssue {
+			err := execCommand("joint-teapot", []string{
+				"joj3-create-result-issue", envFilePath, conf.Stage.OutputPath,
+				repoName, runNumber, conf.Name,
+			})
+			if err != nil {
+				issueErr = err
+			}
+		}
+	}()
+	wg.Wait()
+	if scoreboardErr != nil || failedTableErr != nil || issueErr != nil {
+		slog.Error("teapot exit", "scoreboardErr", scoreboardErr,
+			"failedTableErr", failedTableErr, "issueErr", issueErr)
+		return fmt.Errorf("teapot exit")
 	}
 	return nil
 }

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.Float64("reposize", 2, "size of the repo")
 	localList := flag.String("localList", "", "")
 	droneBranch := flag.String("droneBranch", "", "")
 	releaseCategories := flag.String("releaseCategories", "", "")
@@ -59,7 +60,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())
 	}

internal/parser/diff/parser.go

@@ -21,7 +21,9 @@ const (
 )
 
 type Conf struct {
-	Cases []struct {
+	PassComment string `default:"🥳Passed!\n"`
+	FailComment string `default:"🧐Failed...\n"`
+	Cases       []struct {
 		IgnoreResultStatus bool
 		Outputs            []struct {
 			Score        int
@@ -72,9 +74,9 @@ func (*Diff) Run(results []stage.ExecutorResult, confAny any) (
 				if compareChars(string(answer), result.Files[output.FileName],
 					output.CompareSpace) {
 					score += output.Score
-					comment += "Pass!\n"
+					comment += conf.PassComment
 				} else {
-					comment += "Fail!\n"
+					comment += conf.FailComment
 					comment += fmt.Sprintf("Difference found in `%s`.\n",
 						output.FileName)
 					if !output.AlwaysHide {

internal/parser/resultdetail/parser.go

@@ -29,27 +29,27 @@ func (*ResultDetail) Run(results []stage.ExecutorResult, confAny any) (
 	var res []stage.ParserResult
 	for _, result := range results {
 		comment := ""
-		// TODO: more human readable units
 		if conf.ShowExitStatus {
-			comment += fmt.Sprintf("Exit Status: %d\n", result.ExitStatus)
+			comment += fmt.Sprintf("Exit Status: `%d`\n", result.ExitStatus)
 		}
 		if conf.ShowError {
-			comment += fmt.Sprintf("Error: %s\n", result.Error)
+			comment += fmt.Sprintf("Error: `%s`\n", result.Error)
 		}
 		if conf.ShowTime {
-			comment += fmt.Sprintf("Time: %d\n", result.Time)
+			comment += fmt.Sprintf("Time: `%d ms`\n", result.Time/1e9)
 		}
 		if conf.ShowMemory {
-			comment += fmt.Sprintf("Memory: %d\n", result.Memory)
+			comment += fmt.Sprintf("Memory: `%.2f MiB`\n",
+				float64(result.Memory)/(1024*1024))
 		}
 		if conf.ShowRunTime {
-			comment += fmt.Sprintf("RunTime: %d\n", result.RunTime)
+			comment += fmt.Sprintf("RunTime: `%d ms`\n", result.RunTime/1e9)
 		}
 		for _, file := range conf.ShowFiles {
 			content, ok := result.Files[file]
-			comment += fmt.Sprintf("File: `%s`.\n", file)
+			comment += fmt.Sprintf("File `%s`:\n", file)
 			if ok {
-				comment += fmt.Sprintf("```%s```\n", content)
+				comment += fmt.Sprintf("```\n%s\n```\n", content)
 			} else {
 				comment += "Not found.\n"
 			}

internal/stage/model.go

@@ -150,12 +150,19 @@ func (r ExecutorResult) String() string {
 	return fmt.Sprintf("%+v", d)
 }
 
+type StageExecutor struct {
+	Name string
+	Cmds []Cmd
+}
+type StageParser struct {
+	Name string
+	Conf any
+}
+
 type Stage struct {
-	Name         string
-	ExecutorName string
-	ExecutorCmds []Cmd
-	ParserName   string
-	ParserConf   any
+	Name     string
+	Executor StageExecutor
+	Parsers  []StageParser
 }
 
 type ParserResult struct {

internal/stage/run.go

@@ -8,51 +8,68 @@ import (
 func Run(stages []Stage) (stageResults []StageResult, err error) {
 	var executorResults []ExecutorResult
 	var parserResults []ParserResult
+	var tmpParserResults []ParserResult
 	var forceQuit bool
 	slog.Info("stage run start")
 	for _, stage := range stages {
 		slog.Info("stage start", "name", stage.Name)
-		slog.Info("executor run start", "name", stage.ExecutorName)
-		slog.Debug("executor run start", "name", stage.ExecutorName,
-			"cmds", stage.ExecutorCmds)
-		executor, ok := executorMap[stage.ExecutorName]
+		slog.Info("executor run start", "name", stage.Executor.Name)
+		slog.Debug("executor run start", "name", stage.Executor.Name,
+			"cmds", stage.Executor.Cmds)
+		executor, ok := executorMap[stage.Executor.Name]
 		if !ok {
-			slog.Error("executor not found", "name", stage.ExecutorName)
-			err = fmt.Errorf("executor not found: %s", stage.ExecutorName)
+			slog.Error("executor not found", "name", stage.Executor.Name)
+			err = fmt.Errorf("executor not found: %s", stage.Executor.Name)
 			return
 		}
-		executorResults, err = executor.Run(stage.ExecutorCmds)
+		executorResults, err = executor.Run(stage.Executor.Cmds)
 		if err != nil {
-			slog.Error("executor run error", "name", stage.ExecutorName, "error", err)
+			slog.Error("executor run error", "name", stage.Executor.Name, "error", err)
 			return
 		}
 		slog.Debug("executor run done", "results", executorResults)
 		for _, executorResult := range executorResults {
 			slog.Debug("executor run done", "result.Files", executorResult.Files)
 		}
-		slog.Info("parser run start", "name", stage.ParserName)
-		slog.Debug("parser run start", "name", stage.ParserName,
-			"conf", stage.ParserConf)
-		parser, ok := parserMap[stage.ParserName]
-		if !ok {
-			slog.Error("parser not found", "name", stage.ParserName)
-			err = fmt.Errorf("parser not found: %s", stage.ParserName)
-			return
+		parserResults = []ParserResult{}
+		stageForceQuit := false
+		for _, stageParser := range stage.Parsers {
+			slog.Info("parser run start", "name", stageParser.Name)
+			slog.Debug("parser run start", "name", stageParser.Name,
+				"conf", stageParser.Conf)
+			parser, ok := parserMap[stageParser.Name]
+			if !ok {
+				slog.Error("parser not found", "name", stageParser.Name)
+				err = fmt.Errorf("parser not found: %s", stageParser.Name)
+				return
+			}
+			tmpParserResults, forceQuit, err = parser.Run(
+				executorResults, stageParser.Conf)
+			if err != nil {
+				slog.Error("parser run error", "name", stageParser.Name, "error", err)
+				return
+			}
+			stageForceQuit = stageForceQuit || forceQuit
+			slog.Debug("parser run done", "results", tmpParserResults)
+			if len(parserResults) == 0 {
+				parserResults = tmpParserResults
+			} else {
+				for i := range len(parserResults) {
+					parserResults[i].Score += tmpParserResults[i].Score
+					parserResults[i].Comment += tmpParserResults[i].Comment
+				}
+			}
+			if forceQuit {
+				slog.Error("parser force quit", "name", stageParser.Name)
+			}
 		}
-		parserResults, forceQuit, err = parser.Run(executorResults, stage.ParserConf)
-		if err != nil {
-			slog.Error("parser run error", "name", stage.ParserName, "error", err)
-			return
-		}
-		slog.Debug("parser run done", "results", parserResults)
 		stageResults = append(stageResults, StageResult{
 			Name:      stage.Name,
 			Results:   parserResults,
-			ForceQuit: forceQuit,
+			ForceQuit: stageForceQuit,
 		})
-		if forceQuit {
-			slog.Error("parser force quit", "name", stage.ParserName)
-			return
+		if stageForceQuit {
+			break
 		}
 	}
 	return

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(confSize float64) 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 > int(confSize*1024) {
+		return fmt.Errorf("Repository larger than %.1f MiB. Please clean up or contact the teaching team.", confSize)
 	}
 	return nil
 }