张泊明518370910136
f5b6b3cc9a
All checks were successful
continuous-integration/drone/push Build is passing
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"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/pelletier/go-toml/v2" may panic on some error
|
|
"github.com/BurntSushi/toml"
|
|
"github.com/jinzhu/copier"
|
|
)
|
|
|
|
func parseConfFile(tomlPath *string) Conf {
|
|
tomlConfig, err := os.ReadFile(*tomlPath)
|
|
if err != nil {
|
|
slog.Error("read toml config", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
// fill in default value of config file
|
|
conf := Conf{
|
|
LogLevel: 0,
|
|
OutputPath: "joj3_result.json",
|
|
}
|
|
err = toml.Unmarshal(tomlConfig, &conf)
|
|
if err != nil {
|
|
slog.Error("parse stages config", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
return conf
|
|
}
|
|
|
|
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
|
|
for _, optionalCmd := range s.Executor.With.Cases {
|
|
cmd := s.Executor.With.Default
|
|
copier.Copy(&cmd, &optionalCmd)
|
|
cmds = append(cmds, cmd)
|
|
}
|
|
if len(s.Executor.With.Cases) == 0 {
|
|
cmds = append(cmds, s.Executor.With.Default)
|
|
}
|
|
slog.Info("parse stages config", "cmds", cmds)
|
|
stages = append(stages, stage.Stage{
|
|
Name: s.Name,
|
|
ExecutorName: s.Executor.Name,
|
|
ExecutorCmds: cmds,
|
|
ParserName: s.Parser.Name,
|
|
ParserConfig: 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")...), 0o666)
|
|
}
|
|
|
|
func main() {
|
|
tomlPath := flag.String("c", "conf.toml", "file path of the toml config")
|
|
flag.Parse()
|
|
conf := parseConfFile(tomlPath)
|
|
setupSlog(conf)
|
|
stages := generateStages(conf)
|
|
defer stage.Cleanup()
|
|
results := stage.Run(stages)
|
|
err := outputResult(conf, results)
|
|
if err != nil {
|
|
slog.Error("output result", "error", err)
|
|
}
|
|
}
|