pre release

Author: limstash

app/contest.go

@@ -14,11 +14,11 @@ import (
 )
 
 // FindContest will find the latest contest in the contests array
-func FindContest(contests []contest.Contest) (int, error) {
+func FindContest(contests []contest.Contest) (contest.Contest, error) {
 	contestsSiz := len(contests)
 
 	if contestsSiz <= 0 {
-		return 0, errors.New("[Info] Contests is an empty field")
+		return contest.Contest{}, errors.New("[Info] Contests is an empty field")
 	}
 
 	var lastContestTime int64 = 100000000
@@ -39,30 +39,30 @@ func FindContest(contests []contest.Contest) (int, error) {
 	}
 
 	if lastContestIndex == -1 {
-		return 0, errors.New("[Info] No vaild contest")
+		return contest.Contest{}, errors.New("[Info] No vaild contest")
 	}
 
 	fmt.Println("[Info] The current contest is " + contests[lastContestIndex].Name)
 
 	openHackingPhase := currentTime - contests[lastContestIndex].StartTimeSeconds - contests[lastContestIndex].DurationSeconds
 
-	if openHackingPhase > 12*3600 {
-		return 0, errors.New("[Info] Open hacking phase finished")
+	if openHackingPhase < 12*3600 {
+		return contest.Contest{}, errors.New("[Info] Open hacking phase finished")
 	}
 
 	fmt.Println("[Info] Open hacking phase running")
 
-	return contests[lastContestIndex].ID, nil
+	return contests[lastContestIndex], nil
 }
 
 // ChooseProblem will fetch the problem in the contest and read user's input from stdin
-func ChooseProblem(ContestID int, cookie *[]*http.Cookie) (int, error) {
+func ChooseProblem(ContestID int, cookie *[]*http.Cookie) (string, error) {
 	fmt.Println("[Info] Fetching Problems...")
 
 	problems, e := contest.GetProblems(ContestID, cookie)
 
 	if e != nil {
-		return 0, e
+		return "", e
 	}
 
 	fmt.Println("")
@@ -83,7 +83,7 @@ func ChooseProblem(ContestID int, cookie *[]*http.Cookie) (int, error) {
 	content, e := myReader.ReadString('\n')
 
 	if e != nil {
-		return 0, e
+		return "", e
 	}
 
 	fields := strings.Fields(content)
@@ -102,7 +102,7 @@ func ChooseProblem(ContestID int, cookie *[]*http.Cookie) (int, error) {
 		content, e = myReader.ReadString('\n')
 
 		if e != nil {
-			return 0, e
+			return "", e
 		}
 
 		fields = strings.Fields(content)
@@ -115,5 +115,5 @@ func ChooseProblem(ContestID int, cookie *[]*http.Cookie) (int, error) {
 		choose, e = strconv.Atoi(fields[0])
 	}
 
-	return choose, nil
+	return problems[choose-1].Index, nil
 }

app/contest_test.go

@@ -32,11 +32,11 @@ func Test_FindContest(t *testing.T) {
 	res = contest.Contests{}
 	e = json.Unmarshal([]byte(data2Contest), &res)
 
-	id, e := FindContest(res.Result)
+	resp, e := FindContest(res.Result)
 
 	if e != nil {
 		t.Error(e)
-	} else if id != 1148 {
+	} else if resp.ID != 1148 {
 		t.Error("App Contest failed in finding contest")
 	}
 

app/judge.go

@@ -24,7 +24,7 @@ func runCode(SubmissionID int, Language string, customDiff bool) (bool, error) {
 	}
 
 	if e.Error() == "Compile Error" {
-		fmt.Println("[Warning] Code " + strconv.Itoa(SubmissionID) + " Compile Failed")
+		fmt.Println("[Info] Code " + strconv.Itoa(SubmissionID) + " Compile Failed")
 	}
 
 	if e.Error() == "Runtime Error" {

app/submission.go

@@ -0,0 +1,13 @@
+package app
+
+import (
+	"net/http"
+
+	"github.com/hytzongxuan/Codeforces-Hacker/module/contest"
+	"github.com/hytzongxuan/Codeforces-Hacker/module/submission"
+)
+
+func GetSubmission(cookie *[]*http.Cookie, contestInfo contest.Contest, problem string) []submission.Submission {
+	data, _ := submission.GetSubmissionArray(contestInfo.ID, cookie, contestInfo.StartTimeSeconds, contestInfo.DurationSeconds, problem)
+	return data
+}

main.go

@@ -3,6 +3,8 @@ package main
 import (
 	"fmt"
 	"net/http"
+	"os"
+	"strconv"
 	"time"
 
 	"github.com/hytzongxuan/Codeforces-Hacker/app"
@@ -17,6 +19,7 @@ var Cookie []*http.Cookie
 
 func finish() {
 	time.Sleep(time.Millisecond * 5000)
+	os.Exit(0)
 }
 
 func main() {
@@ -29,14 +32,14 @@ func main() {
 		finish()
 	}
 
-	contestID, e := app.FindContest(Contest)
+	contestInfo, e := app.FindContest(Contest)
 
 	if e != nil {
 		fmt.Println(e)
 		finish()
 	}
 
-	id, e := app.ChooseProblem(contestID, &Cookie)
+	choose, e := app.ChooseProblem(contestInfo.ID, &Cookie)
 
 	if e != nil {
 		fmt.Println(e)
@@ -50,19 +53,43 @@ func main() {
 		finish()
 	}
 
+	var hackChoose bool
+
 	if loginChoose == true {
-		hackChoose, e := app.QueryHackChoose()
+		hackChoose, e = app.QueryHackChoose()
 
 		if e != nil {
 			fmt.Println(e)
 			finish()
 		}
 
 		app.Login(&Cookie, CSRF)
-		fmt.Println(hackChoose)
 	}
 
 	app.SaveData()
-	fmt.Println(id)
 
+	fmt.Println("[Info] Fetching submissions...")
+	submit := app.GetSubmission(&Cookie, contestInfo, choose)
+
+	fmt.Println(" ")
+
+	if hackChoose == true {
+		fmt.Println("[Info] Not Support auto hack")
+	}
+
+	for i := 0; i < len(submit); i++ {
+		flag, _ := app.TestCode(submit[i].SubmissionID, submit[i].Language, false, &Cookie, CSRF)
+
+		if flag == true {
+			fmt.Println("[Info] Submission " + strconv.Itoa(submit[i].SubmissionID) + " Accepted")
+		}
+
+		if e != nil {
+			if hackChoose == true {
+
+			} else {
+
+			}
+		}
+	}
 }

module/submission/submission.go

@@ -0,0 +1,61 @@
+package submission
+
+import (
+	"errors"
+	"net/http"
+	"strconv"
+
+	"github.com/bitly/go-simplejson"
+	"github.com/hytzongxuan/Codeforces-Hacker/module/conn"
+)
+
+func querySubmission(contestID int, cookie *[]*http.Cookie) ([]byte, error) {
+	res, err := conn.HTTPGetByte("http://codeforces.com/api/contest.status?contestId="+strconv.Itoa(contestID)+"&from=1&count=100000000", cookie, map[string]string{"HOST": "codeforces.com"})
+
+	if err != nil {
+		return nil, err
+	}
+
+	return res, nil
+}
+
+type Submission struct {
+	SubmissionID int
+	Language     string
+}
+
+func GetSubmissionArray(contestID int, cookie *[]*http.Cookie, startTime int64, duringTime int64, problem string) ([]Submission, error) {
+	resp, e := querySubmission(contestID, cookie)
+
+	if e != nil {
+		return nil, e
+	}
+
+	js, e := simplejson.NewJson(resp)
+
+	status, e := js.Get("status").String()
+
+	if status != "OK" {
+		return nil, errors.New("Codeforces Return Error Response")
+	}
+
+	submissions, e := js.Get("result").Array()
+
+	data := []Submission{}
+
+	for i := 0; i < len(submissions); i++ {
+		submission := js.Get("result").GetIndex(i)
+
+		verdict := submission.Get("verdict").MustString()
+		submissionID := submission.Get("id").MustInt()
+		language := submission.Get("programmingLanguage").MustString()
+		problemIndex := submission.Get("problem").Get("index").MustString()
+		submitTime := submission.Get("creationTimeSeconds").MustInt64()
+
+		if verdict == "OK" && problemIndex == problem && submitTime > startTime && submitTime <= startTime+duringTime {
+			data = append(data, Submission{submissionID, language})
+		}
+	}
+
+	return data, nil
+}