Add batch create

Author: chermehdi

commands/batch.go

@@ -0,0 +1,139 @@
+package commands
+
+import (
+	"os"
+	"path"
+	template2 "text/template"
+
+	"github.com/chermehdi/egor/config"
+	"github.com/chermehdi/egor/templates"
+	"github.com/fatih/color"
+	"io/ioutil"
+
+	"github.com/urfave/cli/v2"
+)
+
+type Batcher struct {
+	judge Judge
+}
+
+func (b *Batcher) Setup() error {
+	b.judge.Setup()
+	return nil
+}
+
+func (b *Batcher) Run() error {
+	return nil
+}
+
+func (b *Batcher) Cleanup() error {
+	b.judge.Cleanup()
+	return nil
+}
+
+func CreateBatch(context *cli.Context) error {
+	configuration, err := config.LoadDefaultConfiguration()
+	if err != nil {
+		return err
+	}
+	cwd, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+	egorMetaFile := path.Join(cwd, configuration.ConfigFileName)
+	egorMeta, err := config.LoadMetaFromPath(egorMetaFile)
+	if err != nil {
+		return err
+	}
+
+	if egorMeta.HasBatch() {
+		color.Red("The task already has a batch file, Aborting...")
+		return nil
+	}
+
+	genPath := path.Join(cwd, "gen.cpp")
+	egorMeta.BatchFile = genPath
+	// Create the generator file
+
+	genTemp := template2.New("Solution template")
+	compiledTemplate, err := genTemp.Parse(templates.GeneratorTemplate)
+	if err != nil {
+		return err
+	}
+	genFile, err := config.CreateFile(genPath)
+	if err != nil {
+		return nil
+	}
+	// TODO(chermehdi): Move this outside, it smells of code repetition
+	if err = compiledTemplate.Execute(genFile, configuration); err != nil {
+		return err
+	}
+	randH := path.Join(cwd, "rand.h")
+	// Create the rand.h file
+	if err = ioutil.WriteFile(randH, []byte(templates.RandH), 0755); err != nil {
+		return err
+	}
+	// Update the metadata
+	fileName := path.Join(cwd, configuration.ConfigFileName)
+	if err = egorMeta.SaveToFile(fileName); err != nil {
+		return err
+	}
+	color.Green("Batch creation completed successfully")
+	return nil
+}
+
+func RunBatch(context *cli.Context) error {
+	configuration, err := config.LoadDefaultConfiguration()
+	if err != nil {
+		return err
+	}
+	cwd, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+	egorMetaFile := path.Join(cwd, configuration.ConfigFileName)
+	egorMeta, err := config.LoadMetaFromPath(egorMetaFile)
+	if err != nil {
+		return err
+	}
+
+	judge, err := NewJudgeFor(egorMeta, configuration, getChecker(context.String("checker")))
+	if err != nil {
+		return err
+	}
+	if err := judge.Setup(); err != nil {
+		return err
+	}
+	defer judge.Cleanup()
+	report := newJudgeReport()
+
+	for i, input := range egorMeta.Inputs {
+		output := egorMeta.Outputs[i]
+		caseDescription := NewCaseDescription(input, output, egorMeta.TimeLimit)
+		status := judge.RunTestCase(*caseDescription)
+		report.Add(status, *caseDescription)
+	}
+	report.Display()
+	return nil
+}
+
+var BatchCommand = cli.Command{
+	Name:      "batch",
+	Aliases:   []string{"b"},
+	Usage:     "Create and Run bach tests",
+	UsageText: "egor batch (create | run)",
+	Subcommands: []*cli.Command{
+		{
+			Name:    "create",
+			Aliases: []string{"c"},
+			Usage:   "Create the template for the batch test",
+			Action:  CreateBatch,
+		},
+		{
+			Name:    "run",
+			Aliases: []string{"r"},
+			Usage:   "Run the batch test",
+			Action:  RunBatch,
+		},
+	},
+}

config/meta.go

@@ -46,6 +46,9 @@ type EgorMeta struct {
 	Outputs   []IoFile
 	TaskFile  string
 	TimeLimit float64
+	// Path to the file containing the batch generator.
+	// Not setting this value implies that the task does not contain a Batch file.
+	BatchFile string
 }
 
 // Resolves the task file given the default language.
@@ -82,6 +85,7 @@ func NewEgorMeta(task Task, config Config) EgorMeta {
 		Outputs:   outputs,
 		TaskFile:  taskFile,
 		TimeLimit: task.TimeLimit,
+		BatchFile: "",
 	}
 }
 
@@ -91,6 +95,17 @@ func (egor *EgorMeta) CountTestCases() int {
 	return len(egor.Inputs)
 }
 
+func (egor *EgorMeta) HasBatch() bool {
+	if egor.BatchFile == "" {
+		return false
+	}
+	// If the path points to some deleted file, it's considered none existing.
+	if _, err := os.Stat(egor.BatchFile); os.IsNotExist(err) {
+		return false
+	}
+	return true
+}
+
 func (egor *EgorMeta) toJson() (string, error) {
 	var buffer bytes.Buffer
 	encoder := json2.NewEncoder(&buffer)

main.go

@@ -49,6 +49,7 @@ func main() {
 			&commands.PrintCaseCommand,
 			&commands.TestCommand,
 			&commands.CreateTaskCommand,
+			&commands.BatchCommand,
 		},
 	}
 

templates/resolver.go

@@ -7,6 +7,34 @@ import (
 	"io/ioutil"
 )
 
+// This is embeded as a literal string for easy shipping with the binary.
+// We could consider using some new Go feature to embed it as a static resource.
+// At the time of creation of this, this is not a priority.
+const RandH = `
+class Rand {
+public:
+	Rand(int argc, char** argv) {
+	}
+};
+`
+
+const GeneratorTemplate = `
+//
+// Created by egor https://github.com/chermehdi/egor
+// {{if .Author }}
+// @author {{ .Author }}
+{{end}}
+#include <iostream>
+#include <vector>
+#include "rand.h"
+
+using namespace std;
+
+int main(int argc, char** argv) {
+	// Do not remove this line
+	Rand rand(argc, argv);	
+}
+`
 const CppTemplate = `
 //
 // Created by egor https://github.com/chermehdi/egor