深入理解Go语言HTTP客户端的Cookie管理与会话保持

type Jar struct {
    lk      sync.Mutex
    cookies map[string][]*http.Cookie
}

// SetCookies handles the receipt of the cookies in a reply for the
// given URL.  It may or may not choose to save the cookies, depending
// on the jar's policy and implementation.
func (jar *Jar) SetCookies(u *url.URL, cookies []*http.Cookie) {
    jar.lk.Lock()
    jar.cookies[u.Host] = cookies // 问题：直接覆盖，不考虑路径、有效期等
    jar.lk.Unlock()
}

// Cookies returns the cookies to send in a request for the given URL.
// It is up to the implementation to honor the standard cookie use
// restrictions such as in RFC 6265.
func (jar *Jar) Cookies(u *url.URL) []*http.Cookie {
    return jar.cookies[u.Host] // 问题：只按主机名检索，不考虑路径、域匹配
}

import (
    "net/http"
    "net/http/cookiejar"
    "net/url"
    // ... 其他导入
)

func NewClient() *http.Client {
    // 创建一个默认的Cookie Jar
    jar, err := cookiejar.New(nil) // nil表示使用默认的公共后缀列表
    if err != nil {
        panic(err) // 实际应用中应进行更优雅的错误处理
    }

    client := &http.Client{
        Jar: jar, // 将标准库的Cookie Jar赋值给Client
        // CheckRedirect: nil, // 默认行为是自动处理重定向
        // Transport: tr, // 如果需要自定义TLS配置，可以保留
    }
    return client
}

    // 以下代码不再需要，Client.Jar会代劳
    // cookies := CookieJar.Cookies(uri)
    // for i := 0; i < len(cookies); i++ {
    //     fmt.Printf("Cookie[%d]: %s", i, cookies[i])
    //     req.AddCookie(cookies[i])
    // }

package main

import (
    "crypto/tls"
    "fmt"
    "io/ioutil"
    "net/http"
    "net/http/cookiejar" // 引入标准库的cookiejar
    "net/url"
    "strings"
    // "sync" // 不再需要自定义Jar的锁
)

// 假设 username 和 password 已定义
var username = "your_username"
var password = "your_password"

// NewClient 负责创建并配置一个带有标准库CookieJar的http.Client
func NewClient() *http.Client {
    // 配置TLS，如果需要跳过证书验证（不推荐用于生产环境）
    tr := &http.Transport{
        TLSClientConfig: &tls.Config{InsecureSkipVerify: false},
    }

    // 创建一个标准的Cookie Jar
    jar, err := cookiejar.New(nil)
    if err != nil {
        panic(fmt.Errorf("failed to create cookie jar: %v", err))
    }

    client := &http.Client{
        Transport: tr,
        Jar:       jar, // 将标准库的Jar赋值给Client
        // CheckRedirect: nil, // 默认行为是自动处理重定向，通常不需要修改
    }
    return client
}

func Login() {
    client := NewClient() // 获取配置好的Client

    api := "https://www.statuscake.com/App/"
    // uri, _ := url.Parse("https://www.statuscake.com") // 这个uri在原始代码中似乎没有被直接使用，保留以防万一

    values := url.Values{}
    values.Add("username", username)
    values.Add("password", password)
    values.Add("Login", "yes")
    values.Add("redirect", "")
    str := values.Encode()

    req, err := http.NewRequest("POST", api, strings.NewReader(str))
    if err != nil {
        panic(fmt.Errorf("failed to create request: %v", err))
    }

    req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
    req.Header.Set("Accept", "text/html")
    req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_8_4) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/29.0.1547.65 Safari/537.36")

    // 注意：这里不再需要手动添加Cookie，Client.Jar会处理
    // for i := 0; i < len(cookies); i++ {
    //     req.AddCookie(cookies[i])
    // }

    resp, err := client.Do(req)
    if err != nil {
        panic(fmt.Errorf("failed to perform request: %v", err))
    }
    defer resp.Body.Close()

    fmt.Printf("Response Status: %v\n", resp.Status)

    // Client.Jar已经自动从resp中提取并存储了Cookie
    // 所以这里不需要手动调用 resp.Cookies() 或 client.Jar.SetCookies()
    fmt.Printf("Response.Cookies (from response header): %v\n", resp.Cookies())

    // 可以通过 client.Jar 访问当前存储的Cookie，但通常不需要直接操作
    // 例如，获取登录目标URL的Cookie
    loginURL, _ := url.Parse(api)
    currentCookies := client.Jar.Cookies(loginURL)
    fmt.Printf("Cookies in Jar for %s: %v\n", loginURL.Host, currentCookies)


    if resp.StatusCode == http.StatusOK {
        fmt.Printf("\n\n-----\n")
        fmt.Println("HTTP Code: ", resp.StatusCode)
        // 再次获取响应中的Cookie，确认是否有设置
        fmt.Println("Response Cookies: ", resp.Cookies())
        fmt.Println("Request Headers: ", req.Header)
        // 打印Client发送请求时实际携带的Cookie
        // 注意：req.Cookies() 仅包含手动添加到请求的Cookie，
        // 如果依赖Client.Jar，则此处可能为空，但Client实际发送了Cookie。
        // 要查看Client实际发送的Cookie，需要通过其他方式（如抓包）
        fmt.Println("Response Headers: ", resp.Header)

        bodyBytes, _ := ioutil.ReadAll(resp.Body)
        fmt.Printf("Response Body (first 500 chars):\n%s\n", string(bodyBytes[:min(len(bodyBytes), 500)]))
        fmt.Printf("-----\n\n")
    } else {
        fmt.Printf("Login failed with status code: %d\n", resp.StatusCode)
        bodyBytes, _ := ioutil.ReadAll(resp.Body)
        fmt.Printf("Response Body:\n%s\n", string(bodyBytes))
    }
}

func min(a, b int) int {
    if a < b {
        return a
    }
    return b
}

func main() {
    // 假设 username 和 password 在这里被设置或从环境变量读取
    // 例如: username = os.Getenv("MY_USERNAME")
    // password = os.Getenv("MY_PASSWORD")
    Login()
}