From e94a52d00a992f090efbf8db5275364667523fd3 Mon Sep 17 00:00:00 2001
From: Daniel Kluge <daniel-git@c0ntroller.de>
Date: Sat, 7 May 2022 16:40:36 +0200
Subject: [PATCH] Add album chooser

---
 package.json     |  1 +
 src/getAlbums.js | 72 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 73 insertions(+)
 create mode 100644 src/getAlbums.js

diff --git a/package.json b/package.json
index 567be10..f751348 100644
--- a/package.json
+++ b/package.json
@@ -5,6 +5,7 @@
   "main": "src/sync.js",
   "type": "module",
   "scripts": {
+    "albums": "node src/getAlbums.js",
     "token": "node src/getToken.js",
     "test": "echo \"Error: no test specified\" && exit 1"
   },
diff --git a/src/getAlbums.js b/src/getAlbums.js
new file mode 100644
index 0000000..fdfebe1
--- /dev/null
+++ b/src/getAlbums.js
@@ -0,0 +1,72 @@
+import fetch from "node-fetch";
+import readline from "readline";
+import {readFileSync, writeFileSync} from "fs";
+import { exit } from "process";
+import {dirname} from "path";
+import {fileURLToPath} from "url";
+
+// Read and parse JSON. There is no __dirname for modules
+const secretsFile = `${dirname(fileURLToPath(import.meta.url))}/../secrets.json`;
+const secrets = JSON.parse(readFileSync(secretsFile, "utf8"));
+
+if (!secrets.refreshToken) {
+    console.error("No refresh token found. Please run 'npm run getToken' first.");
+    exit();
+}
+
+async function getToken() {
+    const response = await fetch("https://oauth2.googleapis.com/token", {
+        method: "POST",
+        body: JSON.stringify({
+            client_id: secrets.clientId,
+            client_secret: secrets.clientSecret,
+            grant_type: "refresh_token",
+            refresh_token: secrets.refreshToken
+        }),
+        headers: {'Content-Type': 'application/json'}
+    });
+    const data = await response.json();
+    return data.access_token;
+}
+
+const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout
+});
+
+
+(async () => {
+    let nextPageToken;
+    const albums = [];
+    do {
+        const response = await fetch(`https://photoslibrary.googleapis.com/v1/albums?pageSize=50${nextPageToken ? `&${nextPageToken}`: ""}`, {
+            headers: {
+                Authorization: `Bearer ${await getToken()}`,
+                "Content-Type": "application/json"
+            }
+        });
+
+        const data = await response.json();
+        albums.push(...data.albums);
+        nextPageToken = data.nextPageToken;
+    } while (nextPageToken);
+    for(let i = 0; i < albums.length; i++) {
+        console.log(`${i+1}:\t${albums[i].title} (${albums[i].mediaItemsCount} items)`);
+    }
+    console.log("\n");
+    let albumIndex;
+    while(true) {
+        albumIndex = await new Promise((resolve) => rl.question("Please choose an album: ", resolve));
+        if(Number.isNaN(Number.parseInt(albumIndex)) || Number.parseInt(albumIndex) < 1 || Number.parseInt(albumIndex) > albums.length) {
+            console.warn("This is not a valid number, please try again.");
+            continue;
+        } else {
+            break;
+        }
+    }
+    rl.close();
+    secrets.albumId = albums[albumIndex-1].id;
+    console.log("Writeing album ID to secrets...");
+    writeFileSync(secretsFile, JSON.stringify(secrets, null, 2));
+    console.log("Done!");
+})()