Go App Engine中整合goauth2与urlfetch的认证请求实践

package myapp

import (
    "appengine"
    "appengine/urlfetch"
    "code.google.com/p/goauth2/oauth" // 导入 goauth2 包
    "net/http"
    "log"
)

// handleOAuthCallback 模拟一个处理OAuth回调的HTTP处理器
func handleOAuthCallback(w http.ResponseWriter, r *http.Request) {
    // 1. 获取App Engine请求上下文
    // 所有的urlfetch操作都需要一个 appengine.Context
    c := appengine.NewContext(r)

    // 2. 定义OAuth配置
    // 这是一个示例配置，实际应用中你需要替换为你的OAuth客户端ID、密钥等信息
    oauthConf := &oauth.Config{
        ClientId:     "YOUR_CLIENT_ID.apps.googleusercontent.com",
        ClientSecret: "YOUR_CLIENT_SECRET",
        RedirectURL:  "https://your-app-id.appspot.com/oauth2callback", // 你的回调URL
        Scope:        "https://www.googleapis.com/auth/userinfo.email",  // 请求的权限范围
        AuthURL:      "https://accounts.google.com/o/oauth2/auth",       // 授权服务器URL
        TokenURL:     "https://accounts.google.com/o/oauth2/token",      // 令牌交换URL
    }

    // 3. 关键步骤：创建并配置 oauth.Transport
    // 将 urlfetch.Transport 实例作为 oauth.Transport 的底层传输机制
    t := &oauth.Transport{
        Config: oauthConf,
        // 此处是核心：将 App Engine 的 urlfetch.Transport 注入
        Transport: &urlfetch.Transport{Context: c},
    }

    // 4. 模拟OAuth流程的后续步骤（例如，交换授权码获取令牌）
    // 在实际应用中，授权码 'code' 会从请求参数中获取
    // 例如：code := r.FormValue("code")
    // 这里我们为了示例目的，假设我们有一个授权码
    authCode := r.URL.Query().Get("code") // 从URL参数中获取授权码

    if authCode == "" {
        // 如果没有授权码，重定向用户到授权URL
        url := t.Config.AuthCodeURL("state-token") // "state-token" 用于防止CSRF
        http.Redirect(w, r, url, http.StatusFound)
        return
    }

    // 使用获取到的授权码交换Access Token和Refresh Token
    token, err := t.Exchange(authCode)
    if err != nil {
        c.Errorf("Error exchanging token: %v", err)
        http.Error(w, "Failed to exchange token", http.StatusInternalServerError)
        return
    }

    // 此时，token中包含了Access Token、Refresh Token等信息
    // 你可以将 token 存储起来（例如在Datastore或Memcache中）供后续使用
    log.Printf(c, "Successfully exchanged token: %+v", token)

    // 5. 使用认证后的客户端发起请求（例如，获取用户信息）
    // t.Client() 返回一个 *http.Client，它会使用我们之前配置的 urlfetch.Transport
    client := t.Client()
    resp, err := client.Get("https://www.googleapis.com/oauth2/v1/userinfo")
    if err != nil {
        c.Errorf("Error fetching user info: %v", err)
        http.Error(w, "Failed to fetch user info", http.StatusInternalServerError)
        return
    }
    defer resp.Body.Close()

    // 读取并处理响应
    // body, _ := ioutil.ReadAll(resp.Body)
    // log.Printf(c, "User info: %s", string(body))

    w.WriteHeader(http.StatusOK)
    w.Write([]byte("OAuth process completed successfully with urlfetch!"))
}

// init 函数注册HTTP处理器
func init() {
    http.HandleFunc("/oauth2callback", handleOAuthCallback)
    // 也可以添加一个初始的登录/授权触发点
    http.HandleFunc("/login", func(w http.ResponseWriter, r *http.Request) {
        c := appengine.NewContext(r)
        oauthConf := &oauth.Config{
            ClientId:     "YOUR_CLIENT_ID.apps.googleusercontent.com",
            ClientSecret: "YOUR_CLIENT_SECRET",
            RedirectURL:  "https://your-app-id.appspot.com/oauth2callback",
            Scope:        "https://www.googleapis.com/auth/userinfo.email",
            AuthURL:      "https://accounts.google.com/o/oauth2/auth",
            TokenURL:     "https://accounts.google.com/o/oauth2/token",
        }
        t := &oauth.Transport{
            Config:    oauthConf,
            Transport: &urlfetch.Transport{Context: c},
        }
        url := t.Config.AuthCodeURL("state-token")
        http.Redirect(w, r, url, http.StatusFound)
    })
}