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))
|
|||
|
|
}
|