schedule_converter/auth/login.go

package auth

import (
	"context"
	"fmt"
	"io"
	"log/slog"
	"net/http"
	"regexp"
	"strings"

	"schedule_converter/client"
	"schedule_converter/config"
)

var nameRegexp = regexp.MustCompile(`(\w+)同学`)

// LoginWithClient 使用指定的 HTTP 客户端和凭据登录。
func LoginWithClient(ctx context.Context, httpClient *http.Client, captchaCode, username, password string) (bool, error) {
	log := slog.With("username", username)
	log.Debug("执行登录")

	data := fmt.Sprintf("usercode=%s&password=%s&code=%s", username, password, captchaCode)
	req, err := http.NewRequestWithContext(ctx, "POST", config.Get().BaseURL+"/login", strings.NewReader(data))
	if err != nil {
		return false, err
	}

	req.Header = client.GetHeadersWithContentType("application/x-www-form-urlencoded")

	resp, err := httpClient.Do(req)
	if err != nil {
		return false, err
	}
	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return false, err
	}

	bodyStr := string(body)
	if client.CheckLoginSuccess(bodyStr) {
		if match := nameRegexp.FindStringSubmatch(bodyStr); len(match) > 1 {
			log.Info("登录成功", "name", match[1])
		} else {
			log.Info("登录成功")
		}
		return true, nil
	}

	if client.IsCaptchaError(bodyStr) {
		log.Warn("登录失败：验证码错误")
	} else {
		log.Warn("登录失败")
	}
	return false, nil
}