ai_wht_wechat/go_backend/service/xhs_service.go

package service

import (
	"ai_xhs/config"
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"log"
	"net/http"
	"time"
)

type XHSService struct{}

// SendCodeRequest 发送验证码请求
type SendCodeRequest struct {
	Phone       string `json:"phone" binding:"required"`
	CountryCode string `json:"country_code"`
}

// SendCodeResponse 发送验证码响应
type SendCodeResponse struct {
	Code    int                    `json:"code"`
	Message string                 `json:"message"`
	Data    map[string]interface{} `json:"data"`
}

// LoginRequest 登录请求
type LoginRequest struct {
	Phone       string `json:"phone" binding:"required"`
	Code        string `json:"code" binding:"required"`
	CountryCode string `json:"country_code"`
}

// LoginResponse 登录响应
type LoginResponse struct {
	Code    int                    `json:"code"`
	Message string                 `json:"message"`
	Data    map[string]interface{} `json:"data"`
}

// SendVerificationCode 调用Python HTTP API发送验证码
func (s *XHSService) SendVerificationCode(phone, countryCode string) (*SendCodeResponse, error) {
	// 如果没有传国家码，默认使用+86
	if countryCode == "" {
		countryCode = "+86"
	}

	// 获取Python服务地址
	pythonURL := config.GetPythonServiceURL()
	apiURL := fmt.Sprintf("%s/api/xhs/send-code", pythonURL)

	// 构造请求体
	reqData := map[string]interface{}{
		"phone":        phone,
		"country_code": countryCode,
	}
	reqBody, _ := json.Marshal(reqData)

	// 发送HTTP POST请求
	client := &http.Client{
		Timeout: 60 * time.Second, // 60秒超时
	}
	resp, err := client.Post(apiURL, "application/json", bytes.NewBuffer(reqBody))
	if err != nil {
		return nil, fmt.Errorf("调用Python服务失败: %w", err)
	}
	defer resp.Body.Close()

	// 读取响应
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("读取响应失败: %w", err)
	}

	log.Printf("[Python API-发送验证码] 响应: %s", string(body))

	// 解析JSON响应
	var result SendCodeResponse
	if err := json.Unmarshal(body, &result); err != nil {
		return nil, fmt.Errorf("解析响应失败: %w, body: %s", err, string(body))
	}

	return &result, nil
}

// VerifyLogin 调用Python HTTP API验证登录
func (s *XHSService) VerifyLogin(phone, code, countryCode string) (*LoginResponse, error) {
	// 如果没有传国家码，默认使用+86
	if countryCode == "" {
		countryCode = "+86"
	}

	// 获取Python服务地址
	pythonURL := config.GetPythonServiceURL()
	apiURL := fmt.Sprintf("%s/api/xhs/login", pythonURL)

	// 构造请求体
	reqData := map[string]interface{}{
		"phone":        phone,
		"code":         code,
		"country_code": countryCode,
	}
	reqBody, _ := json.Marshal(reqData)

	// 发送HTTP POST请求
	client := &http.Client{
		Timeout: 120 * time.Second, // 120秒超时（登录可能较慢）
	}
	resp, err := client.Post(apiURL, "application/json", bytes.NewBuffer(reqBody))
	if err != nil {
		return nil, fmt.Errorf("调用Python服务失败: %w", err)
	}
	defer resp.Body.Close()

	// 读取响应
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("读取响应失败: %w", err)
	}

	log.Printf("[Python API-验证登录] 响应: %s", string(body))

	// 解析JSON响应
	var result LoginResponse
	if err := json.Unmarshal(body, &result); err != nil {
		return nil, fmt.Errorf("解析响应失败: %w, body: %s", err, string(body))
	}

	return &result, nil
}