func (r *run) importAlbums(ctx context.Context) error {
albums, err := picago.GetAlbums(ctxutil.Client(ctx), "default")
if err != nil {
return fmt.Errorf("importAlbums: error listing albums: %v", err)
}
albumsNode, err := r.getTopLevelNode("albums", "Albums")
for _, album := range albums {
select {
case <-ctx.Done():
return ctx.Err()
default:
}
if err := r.importAlbum(ctx, albumsNode, album); err != nil {
return fmt.Errorf("picasa importer: error importing album %s: %v", album, err)
}
}
return nil
}
// See https://developers.google.com/accounts/docs/OAuth2InstalledApp .
func main() {
flagID := flag.String("id", os.Getenv("CLIENT_ID"), "application client ID")
flagSecret := flag.String("secret", os.Getenv("CLIENT_SECRET"), "application client secret")
flagCode := flag.String("code", os.Getenv("AUTH_CODE"), "authorization code")
flagTokenCache := flag.String("cache", "token-cache.json", "token cache filename")
flagDir := flag.String("dir", "", "directory to download images to")
flagDebugDir := flag.String("debug", "", "set to a valid path to save the response XMLs there")
flag.Parse()
picago.DebugDir = *flagDebugDir
userid := flag.Arg(0)
client, err := picago.NewClient(*flagID, *flagSecret, *flagCode, *flagTokenCache)
if err != nil {
log.Fatalf("error with authorization: %v", err)
}
user, err := picago.GetUser(client, "")
log.Printf("user=%#v err=%v", user, err)
albums, err := picago.GetAlbums(client, userid)
if err != nil {
log.Fatalf("error listing albums: %v", err)
}
log.Printf("user %s has %d albums.", userid, len(albums))
download := *flagDir != ""
dir, fn := "", ""
for _, album := range albums {
albumJ, err := json.Marshal(album)
if err != nil {
log.Fatalf("error marshaling %#v: %v", album, err)
}
if download {
dir = filepath.Join(*flagDir, album.Name)
if err = os.MkdirAll(dir, 0750); err != nil {
log.Fatalf("cannot create directory %s: %v", dir, err)
}
fn = filepath.Join(dir, "album-"+album.Name+".json")
if err = ioutil.WriteFile(fn, albumJ, 0750); err != nil {
log.Fatalf("error writing %s: %v", fn, err)
}
}
log.Printf("downloading album %s.", albumJ)
photos, err := picago.GetPhotos(client, userid, album.ID)
if err != nil {
log.Printf("error listing photos of %s: %v", album.ID, err)
continue
}
log.Printf("album %s contains %d photos.", album.ID, len(photos))
for _, photo := range photos {
photoJ, err := json.Marshal(photo)
if err != nil {
log.Fatalf("error marshaling %#v: %v", photo, err)
}
log.Printf("Photo: %s", photoJ)
if !download {
continue
}
fn = filepath.Join(dir, photo.Filename)
if err = ioutil.WriteFile(fn+".json", photoJ, 0750); err != nil {
log.Fatalf("error writing %s.json: %v", fn, err)
}
if err = downloadTo(fn, client, photo.URL); err != nil {
log.Fatalf("downloading %s: %v", photo.URL, err)
}
}
}
}