From 3e4e4ef4a9483cecea52ffe9be02e45cfe7c7010 Mon Sep 17 00:00:00 2001
From: Daniel Kluge <daniel-git@c0ntroller.de>
Date: Sat, 7 May 2022 17:32:49 +0200
Subject: [PATCH] Sync script

---
 .gitignore   |  3 ++-
 package.json |  1 +
 src/sync.js  | 64 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 67 insertions(+), 1 deletion(-)
 create mode 100644 src/sync.js

diff --git a/.gitignore b/.gitignore
index ef51211..6bde204 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 node_modules
-secrets.*
\ No newline at end of file
+secrets.*
+album
\ No newline at end of file
diff --git a/package.json b/package.json
index f751348..88fcf1f 100644
--- a/package.json
+++ b/package.json
@@ -6,6 +6,7 @@
   "type": "module",
   "scripts": {
     "albums": "node src/getAlbums.js",
+    "sync": "node src/sync.js",
     "token": "node src/getToken.js",
     "test": "echo \"Error: no test specified\" && exit 1"
   },
diff --git a/src/sync.js b/src/sync.js
new file mode 100644
index 0000000..f38b241
--- /dev/null
+++ b/src/sync.js
@@ -0,0 +1,64 @@
+import fetch from "node-fetch";
+import { createWriteStream } from "fs";
+import { secrets, getToken } from "./common.js";
+
+if (!secrets.refreshToken) {
+    console.error("No refresh token found. Please run 'npm run token' first.");
+} else if (!secrets.album || !secrets.album.id || !secrets.album.count || !secrets.album.name) {
+    console.error("No valid album found in secrets. Please run 'npm run albums' first.");
+}
+
+// TODO: Check for argument for folder
+// TODO: Create folder and check permissions
+const path = "./album";
+
+let tokenDeath = 0;
+let token = "";
+
+async function syncImage(imageObject) {
+    // TODO: Test if file exists
+    if (imageObject.mimeType !== "image/jpeg" && imageObject.mimeType !== "image/png") return;
+
+    const response = await fetch(`${imageObject.baseUrl}=w${imageObject.mediaMetadata.width}-h${imageObject.mediaMetadata.height}`);
+    const fileStream = createWriteStream(`${path}/${imageObject.filename}`);
+    await new Promise((resolve, reject) => {
+        response.body.pipe(fileStream);
+        response.body.on("error", reject);
+        fileStream.on("finish", resolve);
+    });
+}
+
+async function downloadList(imageList) {
+    const asyncDownloads = imageList.map(syncImage);
+    await Promise.all(asyncDownloads);
+}
+
+(async () => {
+    let nextPageToken;
+    let runs = 0;
+    do {
+        if(tokenDeath < Date.now()) {
+            const [newToken, expiresIn] = await getToken();
+            token = newToken;
+            tokenDeath = Date.now() + (expiresIn - 30) * 1000;
+        }
+
+        const searchOptions = {
+            albumId: secrets.album.id,
+            pageSize: 100
+        }
+        if (nextPageToken) searchOptions.pageToken = nextPageToken;
+        const response = await fetch("https://photoslibrary.googleapis.com/v1/mediaItems:search", {
+            method: "POST",
+            body: JSON.stringify(searchOptions),
+            headers: {
+                Authorization: `Bearer ${token}`,
+                "Content-Type": "application/json"
+            }
+        });
+        const data = await response.json();
+        nextPageToken = data.nextPageToken;
+        console.log(`Downloading images ${runs * 100 + 1} to ${runs * 100 + data.mediaItems.length}... (of ${secrets.album.count})`);
+        await downloadList(data.mediaItems);
+    } while (nextPageToken);
+})()