73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
package captcha
|
||
|
||
import (
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"os"
|
||
"strings"
|
||
|
||
"jwts/client"
|
||
"jwts/config"
|
||
|
||
"github.com/PuerkitoBio/goquery"
|
||
)
|
||
|
||
// 保存验证码图片
|
||
func SaveCaptchaImage(httpClient *http.Client) ([]byte, error) {
|
||
fmt.Println("[2] 获取验证码...")
|
||
|
||
req, err := http.NewRequest("GET", config.BaseURL+"/captchaImage", nil)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
req.Header = client.GetHeaders()
|
||
|
||
resp, err := httpClient.Do(req)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
body, err := io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
if err := os.WriteFile("captcha.png", body, 0644); err != nil {
|
||
fmt.Printf(" 警告: 无法保存验证码图片: %v\n", err)
|
||
} else {
|
||
fmt.Println(" 验证码已保存到: captcha.png")
|
||
}
|
||
|
||
return body, nil
|
||
}
|
||
|
||
// 识别验证码 - 使用固定验证码用于测试
|
||
func Recognize(imagePath string) (string, error) {
|
||
fmt.Println("[3] 使用固定验证码...")
|
||
// 固定验证码用于测试
|
||
return "....", nil
|
||
}
|
||
|
||
// 获取并识别验证码
|
||
func GetAndRecognize(client *http.Client) (string, error) {
|
||
_, err := SaveCaptchaImage(client)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
captchaCode, err := Recognize("captcha.png")
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
fmt.Printf(" 识别结果: %s\n", captchaCode)
|
||
return captchaCode, nil
|
||
}
|
||
|
||
// 用于解析HTML（登录页面）
|
||
func ParseHTML(html string) (*goquery.Document, error) {
|
||
return goquery.NewDocumentFromReader(strings.NewReader(html))
|
||
}
|