11package commands
22
33import (
4- "bufio"
54 "bytes"
65 "errors"
76 "fmt"
87 "github.com/chermehdi/egor/config"
8+ "github.com/fatih/color"
9+ "github.com/jedib0t/go-pretty/table"
910 "io/ioutil"
1011 "os"
1112 "os/exec"
2324 WA int8 = 3
2425)
2526
27+ var (
28+ green = color .New (color .FgGreen ).SprintfFunc ()
29+ red = color .New (color .FgRed ).SprintfFunc ()
30+ magenta = color .New (color .FgMagenta ).SprintfFunc ()
31+ yellow = color .New (color .FgYellow ).SprintfFunc ()
32+ )
33+
2634// Checks the output of a given testcase against it's expected output
2735type Checker interface {
2836 // Execute the check (got, expected) and returns
@@ -36,8 +44,8 @@ type DiffChecker struct {
3644
3745func (c * DiffChecker ) Check (got , expected string ) error {
3846 // Compare the trimmed output from both input and output
39- if strings .Trim (got , " \t \n \r " ) != strings .Trim (expected , " \t \n \r " ) {
40- return errors .New (fmt .Sprintf ("Checker failed, expected %s, found %s" , got , expected ))
47+ if strings .TrimRight (got , " \t \n \r " ) != strings .TrimRight (expected , " \t \n \r " ) {
48+ return errors .New (fmt .Sprintf ("Checker failed, expected: \n %s \n found: \n %s" , expected , got ))
4149 }
4250 return nil
4351}
@@ -94,21 +102,56 @@ type Judge interface {
94102
95103// This represents the result of running the testcases of a given task
96104type JudgeReport interface {
97- Add (status CaseStatus )
105+ // Add the pair To the list of executions processed
106+ Add (status CaseStatus , description CaseDescription )
98107
99- Display () string
108+ // Display the current report to os.Stdout
109+ Display ()
100110}
101111
102112type ConsoleJudgeReport struct {
103113 Stats []CaseStatus
114+ Descs []CaseDescription
104115}
105116
106- func (c * ConsoleJudgeReport ) Add (status CaseStatus ) {
117+ func (c * ConsoleJudgeReport ) Add (status CaseStatus , description CaseDescription ) {
107118 c .Stats = append (c .Stats , status )
119+ c .Descs = append (c .Descs , description )
108120}
109121
110- func (c * ConsoleJudgeReport ) Display () string {
111- panic ("implement me" )
122+ func getDisplayStatus (status int8 ) string {
123+ switch status {
124+ case AC :
125+ return green ("AC" )
126+ case RE :
127+ return magenta ("RE" )
128+ case SK :
129+ return yellow ("SK" )
130+ case WA :
131+ return red ("WA" )
132+ }
133+ return "Unknown"
134+ }
135+
136+ func (c * ConsoleJudgeReport ) Display () {
137+ t := table .NewWriter ()
138+ t .SetOutputMirror (os .Stdout )
139+ t .AppendHeader (table.Row {"#" , "Test Name" , "Status" , "Custom" , "Additional infos" })
140+ for i , stat := range c .Stats {
141+ output := "None"
142+ if stat .CheckerError != nil {
143+ output = fmt .Sprintf ("FAILED, %s" , stat .CheckerError .Error ())
144+ }
145+ t .AppendRow ([]interface {}{
146+ i ,
147+ c .Descs [i ].InputFile ,
148+ getDisplayStatus (stat .Status ),
149+ c .Descs [i ].CustomCase ,
150+ output ,
151+ })
152+ }
153+ t .SetStyle (table .StyleLight )
154+ t .Render ()
112155}
113156
114157func newJudgeReport () JudgeReport {
@@ -130,14 +173,18 @@ func (judge *JavaJudge) Setup() error {
130173 return err
131174 }
132175 workDirPath := path .Join (currentDir , "work" )
133- if err = os .Mkdir (workDirPath , 777 ); err != nil {
176+ if err = os .Mkdir (workDirPath , 0777 ); err != nil {
134177 return err
135178 }
136179 //TODO(chermehdi): make the executables path configurable #14
137-
138180 // Compilation for Java
139- cmd := exec .Command ("javac" , judge .Meta .TaskFile , "-d" , workDirPath )
181+ var stderrBuffer bytes.Buffer
182+ cmd := exec .Command ("javac" , judge .Meta .TaskFile , "-d" , "work" )
183+ cmd .Dir = currentDir
184+ fmt .Printf ("javac %s -d \" %s\" \n " , judge .Meta .TaskFile , workDirPath )
185+ cmd .Stderr = & stderrBuffer
140186 if err = cmd .Run (); err != nil {
187+ color .Red ("Could not compile %s, Cause: \n " , stderrBuffer .String ())
141188 return err
142189 }
143190 judge .CurrentWorkDir = workDirPath
@@ -146,8 +193,7 @@ func (judge *JavaJudge) Setup() error {
146193
147194func (judge * JavaJudge ) RunTestCase (desc CaseDescription ) CaseStatus {
148195 // We suppose that all java executables will be called Main
149- execFilePath := path .Join (judge .CurrentWorkDir , "Main" )
150- cmd := exec .Command ("java" , execFilePath )
196+ cmd := exec .Command ("java" , "Main" )
151197 cmd .Dir = judge .CurrentWorkDir
152198 inputFile , err := os .Open (desc .InputFile )
153199 if err != nil {
@@ -169,7 +215,7 @@ func (judge *JavaJudge) RunTestCase(desc CaseDescription) CaseStatus {
169215
170216 var stderrBuffer bytes.Buffer
171217 cmd .Stdin = inputFile
172- cmd .Stdout = bufio . NewWriter ( outputFile )
218+ cmd .Stdout = outputFile
173219 cmd .Stderr = & stderrBuffer
174220 if err = cmd .Run (); err != nil {
175221 return CaseStatus {
@@ -188,6 +234,7 @@ func (judge *JavaJudge) RunTestCase(desc CaseDescription) CaseStatus {
188234 }
189235 }
190236 output , err := ioutil .ReadFile (desc .WorkFile )
237+
191238 if err != nil {
192239 return CaseStatus {
193240 Status : RE ,
@@ -204,7 +251,7 @@ func (judge *JavaJudge) RunTestCase(desc CaseDescription) CaseStatus {
204251 }
205252 return CaseStatus {
206253 Status : AC ,
207- CheckerError : err ,
254+ CheckerError : nil ,
208255 Stderr : stderrBuffer .String (),
209256 }
210257}
@@ -254,7 +301,7 @@ func (judge *PythonJudge) Cleanup() error {
254301func NewJudgeFor (meta config.EgorMeta ) (Judge , error ) {
255302 switch meta .TaskLang {
256303 case "java" :
257- return & JavaJudge {Meta : meta }, nil
304+ return & JavaJudge {Meta : meta , Checker : & DiffChecker {} }, nil
258305 case "cpp" :
259306 case "c" :
260307 return & CppJudge {Meta : meta }, nil
@@ -265,6 +312,7 @@ func NewJudgeFor(meta config.EgorMeta) (Judge, error) {
265312}
266313
267314func RunAction (_ * cli.Context ) error {
315+ color .Green ("in the run action" )
268316 configuration , err := config .LoadDefaultConfiguration ()
269317 if err != nil {
270318 return err
@@ -288,12 +336,15 @@ func RunAction(_ *cli.Context) error {
288336 }
289337 defer judge .Cleanup ()
290338 report := newJudgeReport ()
339+
291340 for i , input := range egorMeta .Inputs {
292341 output := egorMeta .Outputs [i ]
293342 caseDescription := NewCaseDescription (input , output )
294343 status := judge .RunTestCase (* caseDescription )
295- report .Add (status )
344+ report .Add (status , * caseDescription )
296345 }
346+ report .Display ()
347+ return nil
297348}
298349
299350var TestCommand = cli.Command {
0 commit comments