From a2e303781b9a64e670e3e9ef3824edb5a66801b8 Mon Sep 17 00:00:00 2001
From: mpl <mathieu.lonjaret@gmail.com>
Date: Wed, 4 Sep 2019 23:33:04 +0200
Subject: [PATCH] handle end of gallery properly

---
 main.go | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 17d6b16..cac4ae7 100644
--- a/main.go
+++ b/main.go
@@ -28,6 +28,9 @@ var (
 	runFlag    = flag.String("run", "", "the program to run on each downloaded item, right after it is dowloaded. It is also the responsibility of that program to remove the downloaded item, if desired.")
 )
 
+// TODO(mpl): in general everywhere, do not rely so much on sleeps. We need
+// better ways to wait for things to be loaded/ready.
+
 func main() {
 	flag.Parse()
 	if *nItemsFlag == 0 {
@@ -114,7 +117,6 @@ func main() {
 			if err := chromedp.Location(&location).Do(ctx); err != nil {
 				return err
 			}
-			println(location, " VS ", prevLocation)
 			if location == prevLocation {
 				break
 			}
@@ -280,12 +282,16 @@ func main() {
 			if N == 0 {
 				return nil
 			}
-			var location string
+			var location, prevLocation string
+
 			for {
 				if err := chromedp.Location(&location).Do(ctx); err != nil {
 					return err
 				}
-				// TODO(mpl): deal with getting the very last photo to properly exit that loop when N < 0.
+				if location == prevLocation {
+					break
+				}
+				prevLocation = location
 				filePath, err := dlAndMove(ctx, location)
 				if err != nil {
 					return err