add: GetValidateCode

jwch.go

@@ -3,6 +3,7 @@ package jwch
 import (
 	"bytes"
 	"crypto/tls"
+	"encoding/json"
 	"net/http"
 	"strings"
 
@@ -88,3 +89,24 @@ func (s *Student) PostWithIdentifier(url string, formData map[string]string) (*h
 
 	return htmlquery.Parse(strings.NewReader(strings.TrimSpace(string(resp.Body()))))
 }
+
+// 获取验证码
+func GetValidateCode(image string) (string, error) {
+	// 请求西二服务器，自动识别验证码
+	code := verifyCodeResponse{}
+
+	s := NewStudent()
+	resp, err := s.NewRequest().SetFormData(map[string]string{
+		"validateCode": image,
+	}).Post("https://statistics.fzuhelper.w2fzu.com/api/login/validateCode?validateCode")
+	if err != nil {
+		return "", errno.HTTPQueryError.WithMessage("automatic code identification failed")
+	}
+
+	err = json.Unmarshal(resp.Body(), &code)
+
+	if err != nil {
+		return "", errno.HTTPQueryError.WithErr(err)
+	}
+	return code.Message, nil
+}

jwch_test.go

@@ -36,6 +36,20 @@ func login() error {
 	return nil
 }
 
+func Test_GetValidateCode(t *testing.T) {
+	// 获取验证码图片
+	s := NewStudent()
+	resp, err := s.NewRequest().Get("https://jwcjwxt1.fzu.edu.cn/plus/verifycode.asp")
+	if err != nil {
+		t.Error(err)
+	}
+	code, err := GetValidateCode(utils.Base64EncodeHTTPImage(resp.Body()))
+	if err != nil {
+		t.Error(err)
+	}
+	fmt.Println(code)
+}
+
 func Test_GetIdentifierAndCookies(t *testing.T) {
 	Identifier, cookies := stu.GetIdentifierAndCookies()
 	fmt.Println(Identifier)
@@ -124,9 +138,9 @@ func Test_GetQiShanEmptyRoom(t *testing.T) {
 
 	rooms, err := stu.GetQiShanEmptyRoom(EmptyRoomReq{
 		Campus: "旗山校区",
-		Time:   "2024-09-19",
+		Time:   "2024-09-26",
 		Start:  "1",
-		End:    "2",
+		End:    "8",
 	})
 	if err != nil {
 		t.Error(err)