解决Go语言中Google Drive API旧版包安装错误及现代化实践

go get google.golang.org/api/drive/v3
go get golang.org/x/oauth2

package main

import (
    "context"
    "encoding/json"
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "os"

    "golang.org/x/oauth2"
    "golang.org/x/oauth2/google"
    "google.golang.org/api/drive/v3"
    "google.golang.org/api/option"
)

// getClient 使用凭证文件配置OAuth2客户端，并获取或刷新令牌
func getClient(config *oauth2.Config) *http.Client {
    // 尝试从本地文件加载已保存的令牌
    tokFile := "token.json"
    tok, err := tokenFromFile(tokFile)
    if err != nil {
        tok = getTokenFromWeb(config) // 如果没有，则通过网页获取
        saveToken(tokFile, tok)       // 保存新获取的令牌
    }
    return config.Client(context.Background(), tok)
}

// getTokenFromWeb 通过浏览器获取用户授权，并返回令牌
func getTokenFromWeb(config *oauth2.Config) *oauth2.Token {
    authURL := config.AuthCodeURL("state-token", oauth2.AccessTypeOffline)
    fmt.Printf("请在浏览器中打开以下URL进行授权：\n%v\n", authURL)

    fmt.Print("将授权码粘贴到此处：")
    var authCode string
    if _, err := fmt.Scan(&authCode); err != nil {
        log.Fatalf("无法读取授权码: %v", err)
    }

    tok, err := config.Exchange(context.TODO(), authCode)
    if err != nil {
        log.Fatalf("无法从授权码获取令牌: %v", err)
    }
    return tok
}

// tokenFromFile 从文件加载令牌
func tokenFromFile(file string) (*oauth2.Token, error) {
    f, err := os.Open(file)
    if err != nil {
        return nil, err
    }
    defer f.Close()
    tok := &oauth2.Token{}
    err = json.NewDecoder(f).Decode(tok)
    return tok, err
}

// saveToken 将令牌保存到文件
func saveToken(path string, token *oauth2.Token) {
    fmt.Printf("将令牌保存到文件: %s\n", path)
    f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0600)
    if err != nil {
        log.Fatalf("无法保存令牌到文件: %v", err)
    }
    defer f.Close()
    json.NewEncoder(f).Encode(token)
}

func main() {
    ctx := context.Background()

    // 1. 加载凭证文件 (credentials.json)
    // 将您从Google Cloud Console下载的凭证文件命名为 credentials.json 放在项目根目录
    b, err := ioutil.ReadFile("credentials.json")
    if err != nil {
        log.Fatalf("无法读取凭证文件: %v", err)
    }

    // 2. 配置OAuth2客户端
    // drive.DriveScope 授予对用户所有Google Drive文件的读写权限
    // drive.DriveReadonlyScope 仅授予只读权限
    config, err := google.ConfigFromJSON(b, drive.DriveScope)
    if err != nil {
        log.Fatalf("无法解析凭证文件: %v", err)
    }

    // 3. 获取HTTP客户端（包含认证信息）
    client := getClient(config)

    // 4. 创建Drive服务客户端
    srv, err := drive.NewService(ctx, option.WithHTTPClient(client))
    if err != nil {
        log.Fatalf("无法创建Drive服务客户端: %v", err)
    }

    // 5. 执行一个简单的API操作：列出文件
    fmt.Println("正在列出Google Drive中的文件...")
    r, err := srv.Files.List().PageSize(10).
        Fields("nextPageToken, files(id, name)").Do()
    if err != nil {
        log.Fatalf("无法列出文件: %v", err)
    }
    if len(r.Files) == 0 {
        fmt.Println("未找到任何文件。")
    } else {
        fmt.Println("文件列表:")
        for _, i := range r.Files {
            fmt.Printf("%s (%s)\n", i.Name, i.Id)
        }
    }
}