/
submit.go
66 lines (52 loc) · 1.13 KB
/
submit.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package client
import (
"fmt"
"net/url"
"github.com/patelpreet422/at-tool/util"
"github.com/fatih/color"
)
// Submit submit (block while pending)
func (c *Client) Submit(info Info, langID, source string) (err error) {
color.Cyan("Submit " + info.Hint())
URL, err := info.SubmitURL(c.host)
if err != nil {
return
}
body, err := util.GetBody(c.client, URL)
if err != nil {
return
}
handle, err := findUsername(body)
if err != nil {
return
}
fmt.Printf("Current user: %v\n", handle)
csrf, err := findCsrf(body)
if err != nil {
return
}
body, err = util.PostBody(c.client, URL, url.Values{
"data.TaskScreenName": {info.ProblemID},
"data.LanguageId": {langID},
"csrf_token": {csrf},
})
if err != nil {
return
}
// msg, err := findMessage(body)
// if err != nil {
// return errors.New("Submit failed")
// }
// if !strings.Contains(msg, "AC") {
// return errors.New(msg)
// }
color.Green("Submitted")
submissions, err := c.WatchSubmission(info, 1, true)
if err != nil {
return
}
info.SubmissionID = submissions[0].id
c.Username = handle
c.LastSubmission = &info
return c.save()
}