105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
package runner
|
|
|
|
import (
|
|
"fmt"
|
|
"git.0x7f.app/WOJ/woj-server/internal/e"
|
|
"git.0x7f.app/WOJ/woj-server/pkg/down"
|
|
"git.0x7f.app/WOJ/woj-server/pkg/file"
|
|
"git.0x7f.app/WOJ/woj-server/pkg/unzip"
|
|
"go.uber.org/zap"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
)
|
|
|
|
func (s *service) download(version uint, url string) e.Status {
|
|
zipPath := filepath.Join(TmpDir, fmt.Sprintf("%d.zip", version))
|
|
problemPath := filepath.Join(ProblemDir, fmt.Sprintf("%d", version))
|
|
|
|
err := down.Down(zipPath, url)
|
|
if err != nil {
|
|
s.log.Error("download problem failed", zap.Error(err))
|
|
return e.RunnerDownloadFailed
|
|
}
|
|
|
|
err = unzip.Unzip(zipPath, problemPath)
|
|
if err != nil {
|
|
s.log.Warn("unzip problem failed", zap.Error(err))
|
|
return e.RunnerUnzipFailed
|
|
}
|
|
|
|
return e.Success
|
|
}
|
|
|
|
func (s *service) prebuild(version uint, force bool) e.Status {
|
|
if !s.ProblemExists(version) {
|
|
return e.RunnerProblemNotExist
|
|
}
|
|
|
|
mark := filepath.Join(ProblemDir, fmt.Sprintf("%d", version), ".mark.prebuild")
|
|
if force {
|
|
_ = os.Remove(mark)
|
|
} else if file.Exist(mark) {
|
|
return e.Success
|
|
}
|
|
|
|
prebuildScript := filepath.Join(ProblemDir, fmt.Sprintf("%d", version), "judge", "prebuild.Makefile")
|
|
if !file.Exist(prebuildScript) {
|
|
s.log.Info("prebuild script not found", zap.String("path", prebuildScript), zap.Uint("version", version))
|
|
return e.Success
|
|
}
|
|
|
|
dataDir := filepath.Join(ProblemDir, fmt.Sprintf("%d", version), "data")
|
|
judgeDir := filepath.Join(ProblemDir, fmt.Sprintf("%d", version), "judge")
|
|
|
|
args := []string{
|
|
"-v", fmt.Sprintf("%s:/woj/problem/data", dataDir),
|
|
"-v", fmt.Sprintf("%s:/woj/problem/judge", judgeDir),
|
|
"-e", "PREFIX=/woj/problem",
|
|
"git.0x7f.app/woj/ubuntu-full:latest",
|
|
"sh", "-c", "cd /woj/problem/judge && make -f prebuild.Makefile prebuild && touch .mark.prebuild",
|
|
}
|
|
runArgs := &podmanArgs{
|
|
executeArgs: executeArgs{
|
|
args: args,
|
|
timeout: 300 * time.Second,
|
|
},
|
|
memory: "1g",
|
|
}
|
|
err := s.podmanRun(runArgs)
|
|
|
|
if err != nil {
|
|
s.log.Warn("prebuild problem failed", zap.Error(err), zap.Uint("version", version))
|
|
return e.RunnerProblemPrebuildFailed
|
|
}
|
|
|
|
return e.Success
|
|
}
|
|
|
|
func (s *service) NewProblem(version uint, url string, force bool) (Config, e.Status) {
|
|
if force {
|
|
problemPath := filepath.Join(ProblemDir, fmt.Sprintf("%d", version))
|
|
_ = os.RemoveAll(problemPath)
|
|
}
|
|
|
|
if !s.ProblemExists(version) {
|
|
status := s.download(version, url)
|
|
if status != e.Success {
|
|
return Config{}, status
|
|
}
|
|
}
|
|
|
|
cfg, err := s.ParseConfig(version, false)
|
|
if err != nil {
|
|
// TODO: err is dropped here, should return to server: add a new column in problem_version table
|
|
return Config{}, e.RunnerProblemParseFailed
|
|
}
|
|
|
|
status := s.prebuild(version, true)
|
|
if status != e.Success {
|
|
return Config{}, status
|
|
}
|
|
|
|
return cfg, e.Success
|
|
}
|