feat: Implemented logins and bundle importers (only Humble Bundles for now)

Author: simeonradivoev

package-lock.json

@@ -123,7 +123,6 @@
     "electron-log": "^5.4.3",
     "electron-store": "^8.2.0",
     "electron-updater": "^6.7.3",
-    "file-icons-js": "github:websemantics/file-icons-js",
     "idb": "^8.0.3",
     "lodash-es": "^4.17.21",
     "lokijs": "^1.5.12",
@@ -140,6 +139,7 @@
     "react": "^18.3.1",
     "react-audio-player": "^0.17.0",
     "react-dom": "^18.2.0",
+    "react-icons": "^5.5.0",
     "react-markdown": "^10.1.0",
     "react-router-dom": "^7.12.0",
     "react-split": "^2.0.14",
@@ -303,14 +303,18 @@
       "target": [
         "nsis"
       ],
-      "publish": ["github"]
+      "publish": [
+        "github"
+      ]
     },
     "linux": {
       "target": [
         "AppImage"
       ],
       "category": "Development",
-      "publish": ["github"]
+      "publish": [
+        "github"
+      ]
     },
     "directories": {
       "app": "release/app",

package.json

@@ -0,0 +1,19 @@
+import ElectronStore from 'electron-store';
+import { MainIpcGetter, StoreSchema } from 'shared/constants';
+import InstallHumbleImporter from './bundles/humble-importer';
+
+export default function InitializeAccounts(
+  store: ElectronStore<StoreSchema>,
+  database: Loki,
+  api: MainIpcGetter,
+) {
+  const importers = {
+    humble: InstallHumbleImporter(store, database),
+  };
+
+  api.checkLogin = async (provider) => importers[provider].isLoggedIn();
+  api.logout = async (provider) => {};
+  api.login = async (provider) => importers[provider].login();
+
+  return importers;
+}

src/main/api/accounts.ts

@@ -1,16 +1,28 @@
+/* eslint-disable no-await-in-loop */
 import { spawn } from 'child_process';
 import { createWriteStream } from 'fs';
 import { writeFile, readFile, mkdir, rename, stat, rm } from 'fs/promises';
 import path, { normalize } from 'path';
-import { dialog, BrowserWindow, shell, Notification } from 'electron';
-import log from 'electron-log/main';
+import { setTimeout } from 'timers/promises';
+import { uuid } from '@tanstack/react-form';
+import { load } from 'cheerio';
+import { dialog, BrowserWindow, shell, Notification, session, net } from 'electron';
+import log from 'electron-log';
 import Store from 'electron-store';
 import JSZip from 'jszip';
 import Loki from 'lokijs';
 import StreamZip from 'node-stream-zip';
-import { BundleMetaFile, StoreSchema, MainIpcGetter, previewTypes } from '../../shared/constants';
+import {
+  BundleMetaFile,
+  StoreSchema,
+  MainIpcGetter,
+  previewTypes,
+  HUMBLE_PARTITION,
+  ImportType,
+  LoginProvider,
+} from '../../shared/constants';
 import { addTask } from '../managers/task-manager';
-import { FilePath, getProjectDir, getRandom, ignoredFilesMatch } from '../util';
+import { FilePath, foreachAsync, getProjectDir, getRandom, ignoredFilesMatch } from '../util';
 import {
   findBundlePath,
   forAllAssetsInProject,
@@ -219,21 +231,24 @@ export async function getVirtualBundles(
         bundle: b,
         isVirtual: true,
         date: new Date(b.date),
+        sourceType: b.sourceType,
       }) satisfies BundleInfo as BundleInfo,
   );
 }
 
 export async function getBundles(
   store: Store<StoreSchema>,
-  virtualBundles: Collection<VirtualBundle>,
+  virtualBundles?: Collection<VirtualBundle>,
 ): Promise<BundleInfo[]> {
   const bundles: BundleInfo[] = [];
   const projectDir = getProjectDir(store);
   if (projectDir) {
     await findChildrenBundles(new FilePath(projectDir, ''), bundles);
   }
 
-  bundles.push(...(await getVirtualBundles(virtualBundles)));
+  if (virtualBundles) {
+    bundles.push(...(await getVirtualBundles(virtualBundles)));
+  }
 
   return bundles.sort((a, b) => b.date?.getTime() - a.date?.getTime());
 }
@@ -369,9 +384,18 @@ async function exportBundle(
   }
 }
 
+async function getHumbleCookieHeader(): Promise<string> {
+  const cookies = await session.defaultSession.cookies.get({
+    domain: '.humblebundle.com',
+  });
+
+  return cookies.map((c) => `${c.name}=${c.value}`).join('; ');
+}
+
 /* ----------------------------- Initialize API  ----------------------------- */
 
 export default function InitializeBundlesApi(
+  importers: Record<LoginProvider, BundleImporter>,
   api: MainIpcGetter,
   store: Store<StoreSchema>,
   db: Loki,
@@ -447,6 +471,7 @@ export default function InitializeBundlesApi(
         bundle: virtualBundle as Bundle,
         previewUrl: virtualBundle.previewUrl,
         name: virtualBundle.name,
+        sourceType: virtualBundle.sourceType,
       } as BundleInfo;
     }
     return tryGetBundleEntryFromFolderPath(FilePath.fromStore(store, id));
@@ -502,6 +527,15 @@ export default function InitializeBundlesApi(
   }
 
   api.updateBundle = updateBundle;
+  api.importBundles = async (type) => {
+    const importer = importers[type];
+    if (importer) {
+      return addTask('Importing Bundles', async (abort, progress) =>
+        importer.import(abort, progress),
+      );
+    }
+    throw new Error(`No Importer of type ${type}`);
+  };
   api.getBundles = () => getBundles(store, virtualBundles);
   api.getHomeBundles = getHomeBundles;
   api.createVirtualBundle = createVirtualBundle;