Created
October 10, 2015 08:54
-
-
Save stanwu/4687edd903e5f16cf5ab to your computer and use it in GitHub Desktop.
google drive list all of files with MD5 checksum
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| package main | |
| import ( | |
| "encoding/json" | |
| "fmt" | |
| "io/ioutil" | |
| "log" | |
| "net/http" | |
| "net/url" | |
| "os" | |
| "os/user" | |
| "path/filepath" | |
| "golang.org/x/net/context" | |
| "golang.org/x/oauth2" | |
| "golang.org/x/oauth2/google" | |
| "google.golang.org/api/drive/v2" | |
| ) | |
| // getClient uses a Context and Config to retrieve a Token | |
| // then generate a Client. It returns the generated Client. | |
| func getClient(ctx context.Context, config *oauth2.Config) *http.Client { | |
| cacheFile, err := tokenCacheFile() | |
| if err != nil { | |
| log.Fatalf("Unable to get path to cached credential file. %v", err) | |
| } | |
| tok, err := tokenFromFile(cacheFile) | |
| if err != nil { | |
| tok = getTokenFromWeb(config) | |
| saveToken(cacheFile, tok) | |
| } | |
| return config.Client(ctx, tok) | |
| } | |
| // getTokenFromWeb uses Config to request a Token. | |
| // It returns the retrieved Token. | |
| func getTokenFromWeb(config *oauth2.Config) *oauth2.Token { | |
| authURL := config.AuthCodeURL("state-token", oauth2.AccessTypeOffline) | |
| fmt.Printf("Go to the following link in your browser then type the "+ | |
| "authorization code: \n%v\n", authURL) | |
| var code string | |
| if _, err := fmt.Scan(&code); err != nil { | |
| log.Fatalf("Unable to read authorization code %v", err) | |
| } | |
| tok, err := config.Exchange(oauth2.NoContext, code) | |
| if err != nil { | |
| log.Fatalf("Unable to retrieve token from web %v", err) | |
| } | |
| return tok | |
| } | |
| // tokenCacheFile generates credential file path/filename. | |
| // It returns the generated credential path/filename. | |
| func tokenCacheFile() (string, error) { | |
| usr, err := user.Current() | |
| if err != nil { | |
| return "", err | |
| } | |
| tokenCacheDir := filepath.Join(usr.HomeDir, ".credentials") | |
| os.MkdirAll(tokenCacheDir, 0700) | |
| return filepath.Join(tokenCacheDir, | |
| url.QueryEscape("drive-go-quickstart.json")), err | |
| } | |
| // tokenFromFile retrieves a Token from a given file path. | |
| // It returns the retrieved Token and any read error encountered. | |
| func tokenFromFile(file string) (*oauth2.Token, error) { | |
| f, err := os.Open(file) | |
| if err != nil { | |
| return nil, err | |
| } | |
| t := &oauth2.Token{} | |
| err = json.NewDecoder(f).Decode(t) | |
| defer f.Close() | |
| return t, err | |
| } | |
| // saveToken uses a file path to create a file and store the | |
| // token in it. | |
| func saveToken(file string, token *oauth2.Token) { | |
| fmt.Printf("Saving credential file to: %s\n", file) | |
| f, err := os.Create(file) | |
| if err != nil { | |
| log.Fatalf("Unable to cache oauth token: %v", err) | |
| } | |
| defer f.Close() | |
| json.NewEncoder(f).Encode(token) | |
| } | |
| func main() { | |
| ctx := context.Background() | |
| b, err := ioutil.ReadFile("client_secret.json") | |
| if err != nil { | |
| log.Fatalf("Unable to read client secret file: %v", err) | |
| } | |
| config, err := google.ConfigFromJSON(b, drive.DriveMetadataReadonlyScope) | |
| if err != nil { | |
| log.Fatalf("Unable to parse client secret file to config: %v", err) | |
| } | |
| client := getClient(ctx, config) | |
| srv, err := drive.New(client) | |
| if err != nil { | |
| log.Fatalf("Unable to retrieve drive Client %v", err) | |
| } | |
| //below: add by stanwu 2015/10/10 | |
| pageToken := "" | |
| for { | |
| q := srv.Files.List() | |
| // If we have a pageToken set, apply it to the query | |
| if pageToken != "" { | |
| q = q.PageToken(pageToken) | |
| } | |
| r, err := q.Do() | |
| if err != nil { | |
| fmt.Printf("An error occurred: %v\n", err) | |
| os.Exit(3) | |
| } | |
| //fs = append(fs, r.Items...) | |
| if len(r.Items) > 0 { | |
| for _, i := range r.Items { | |
| fmt.Printf("\"%s\",\"%s\"\n", i.Title, i.Md5Checksum) | |
| } | |
| } else { | |
| fmt.Print("No files found.") | |
| } | |
| pageToken = r.NextPageToken | |
| if pageToken == "" { | |
| break | |
| } | |
| } | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment