97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log/slog"
|
|
"os"
|
|
|
|
"focs.ji.sjtu.edu.cn/git/FOCS-dev/JOJ3/internal/executors"
|
|
_ "focs.ji.sjtu.edu.cn/git/FOCS-dev/JOJ3/internal/parsers"
|
|
"focs.ji.sjtu.edu.cn/git/FOCS-dev/JOJ3/internal/stage"
|
|
|
|
"github.com/jinzhu/copier"
|
|
)
|
|
|
|
func setupSlog(conf Conf) {
|
|
lvl := new(slog.LevelVar)
|
|
lvl.Set(slog.Level(conf.LogLevel))
|
|
opts := &slog.HandlerOptions{Level: lvl}
|
|
handler := slog.NewTextHandler(os.Stderr, opts)
|
|
logger := slog.New(handler)
|
|
slog.SetDefault(logger)
|
|
}
|
|
|
|
func generateStages(conf Conf) []stage.Stage {
|
|
stages := []stage.Stage{}
|
|
for _, s := range conf.Stages {
|
|
var cmds []stage.Cmd
|
|
defaultCmd := s.Executor.With.Default
|
|
for _, optionalCmd := range s.Executor.With.Cases {
|
|
cmd := s.Executor.With.Default
|
|
err := copier.Copy(&cmd, &optionalCmd)
|
|
if err != nil {
|
|
slog.Error("generate stages", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
// since these 3 values are pointers, copier will always copy
|
|
// them, so we need to check them manually
|
|
if defaultCmd.Stdin != nil && optionalCmd.Stdin == nil {
|
|
cmd.Stdin = defaultCmd.Stdin
|
|
}
|
|
if defaultCmd.Stdout != nil && optionalCmd.Stdout == nil {
|
|
cmd.Stdout = defaultCmd.Stdout
|
|
}
|
|
if defaultCmd.Stderr != nil && optionalCmd.Stderr == nil {
|
|
cmd.Stderr = defaultCmd.Stderr
|
|
}
|
|
cmds = append(cmds, cmd)
|
|
}
|
|
if len(s.Executor.With.Cases) == 0 {
|
|
cmds = []stage.Cmd{defaultCmd}
|
|
}
|
|
slog.Debug("parse stages conf", "cmds", cmds)
|
|
stages = append(stages, stage.Stage{
|
|
Name: s.Name,
|
|
ExecutorName: s.Executor.Name,
|
|
ExecutorCmds: cmds,
|
|
ParserName: s.Parser.Name,
|
|
ParserConf: s.Parser.With,
|
|
})
|
|
}
|
|
return stages
|
|
}
|
|
|
|
func outputResult(conf Conf, results []stage.StageResult) error {
|
|
content, err := json.Marshal(results)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return os.WriteFile(conf.OutputPath,
|
|
append(content, []byte("\n")...), 0o600)
|
|
}
|
|
|
|
func main() {
|
|
retCode := 0
|
|
defer func() {
|
|
os.Exit(retCode)
|
|
}()
|
|
conf, err := commitMsgToConf()
|
|
if err != nil {
|
|
slog.Error("no conf found", "error", err)
|
|
retCode = 1
|
|
return
|
|
}
|
|
setupSlog(conf)
|
|
executors.InitWithConf(conf.SandboxExecServer, conf.SandboxToken)
|
|
stages := generateStages(conf)
|
|
defer stage.Cleanup()
|
|
results, stageErr := stage.Run(stages)
|
|
if err := outputResult(conf, results); err != nil {
|
|
slog.Error("output result", "error", err)
|
|
}
|
|
if stageErr != nil {
|
|
retCode = 1
|
|
return
|
|
}
|
|
}
|