Skip to content

Commit

Permalink
fix: errno type error
Browse files Browse the repository at this point in the history
  • Loading branch information
jiuxia211 committed Dec 7, 2024
1 parent 86066db commit 0493ecd
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
8 changes: 4 additions & 4 deletions jwch.go
Original file line number Diff line number Diff line change
Expand Up @@ -79,12 +79,12 @@ func (s *Student) NewRequest() *resty.Request {
func (s *Student) GetWithIdentifier(url string) (*html.Node, error) {
resp, err := s.NewRequest().SetHeader("Referer", constants.JwchReferer).SetQueryParam("id", s.Identifier).Get(url)
if err != nil {
return nil, errno.SSOLoginFailedError.WithErr(err)
return nil, errno.CookieError.WithErr(err)
}

// 会话过期 TODO: 判断条件有点简陋
if strings.Contains(string(resp.Body()), "重新登录") {
return nil, errno.SSOLoginFailedError
return nil, errno.CookieError
}

return htmlquery.Parse(bytes.NewReader(resp.Body()))
Expand All @@ -97,12 +97,12 @@ func (s *Student) PostWithIdentifier(url string, formData map[string]string) (*h
s.NewRequest().EnableTrace()

if err != nil {
return nil, errno.SSOLoginFailedError.WithErr(err)
return nil, errno.CookieError.WithErr(err)
}

// Identifier缺失 TODO: 判断条件有点简陋
if strings.Contains(string(resp.Body()), "处理URL失败") {
return nil, errno.SSOLoginFailedError
return nil, errno.CookieError
}

return htmlquery.Parse(strings.NewReader(strings.TrimSpace(string(resp.Body()))))
Expand Down
6 changes: 3 additions & 3 deletions user.go
Original file line number Diff line number Diff line change
Expand Up @@ -127,13 +127,13 @@ func (s *Student) Login() error {

// 这里是err == nil 因为禁止了重定向,正常登录是会出现异常的
if err == nil {
return errno.SSOLoginFailedError
return errno.CookieError
}

data := regexp.MustCompile(`id=(.*?)&`).FindStringSubmatch(err.Error())

if len(data) < 1 {
return errno.SSOLoginFailedError
return errno.CookieError
}

s.SetIdentifier(data[1])
Expand Down Expand Up @@ -167,7 +167,7 @@ func (s *Student) CheckSession() error {
res := htmlquery.FindOne(resp, `//*[@id="ContentPlaceHolder1_LB_xh"]`)

if res == nil {
return errno.SSOLoginFailedError.WithErr(err)
return errno.CookieError.WithErr(err)
}

if htmlquery.OutputHTML(res, false) != s.ID {
Expand Down

0 comments on commit 0493ecd

Please sign in to comment.