@protontech/drive-sdk 0.4.0 → 0.4.1

package/src/protonDrivePhotosClient.ts

@@ -8,20 +8,27 @@ import {
     FileUploader,
     SDKEvent,
     MaybeNode,
+    ThumbnailType,
+    ThumbnailResult,
 } from './interface';
 import { getConfig } from './config';
 import { DriveCrypto } from './crypto';
 import { Telemetry } from './telemetry';
-import { convertInternalMissingNodeIterator, convertInternalNodeIterator, getUid, getUids } from './transformers';
+import {
+    convertInternalMissingNodeIterator,
+    convertInternalNodeIterator,
+    convertInternalNodePromise,
+    getUid,
+    getUids,
+} from './transformers';
 import { DriveAPIService } from './internal/apiService';
 import { initDownloadModule } from './internal/download';
 import { DriveEventsService, DriveListener, EventSubscription } from './internal/events';
 import { initNodesModule } from './internal/nodes';
-import { initPhotoSharesModule, initPhotosModule } from './internal/photos';
+import { initPhotosModule, initPhotoSharesModule, initPhotoUploadModule } from './internal/photos';
 import { SDKEvents } from './internal/sdkEvents';
 import { initSharesModule } from './internal/shares';
 import { initSharingModule } from './internal/sharing';
-import { initUploadModule } from './internal/upload';
 
 /**
  * ProtonDrivePhotosClient is the interface to access Photos functionality.
@@ -39,7 +46,7 @@ export class ProtonDrivePhotosClient {
     private nodes: ReturnType<typeof initNodesModule>;
     private sharing: ReturnType<typeof initSharingModule>;
     private download: ReturnType<typeof initDownloadModule>;
-    private upload: ReturnType<typeof initUploadModule>;
+    private upload: ReturnType<typeof initPhotoUploadModule>;
     private photos: ReturnType<typeof initPhotosModule>;
 
     public experimental: {
@@ -115,7 +122,7 @@ export class ProtonDrivePhotosClient {
             this.nodes.access,
             this.nodes.revisions,
         );
-        this.upload = initUploadModule(
+        this.upload = initPhotoUploadModule(
             telemetry,
             apiService,
             cryptoModule,
@@ -208,6 +215,16 @@ export class ProtonDrivePhotosClient {
         yield* convertInternalMissingNodeIterator(this.nodes.access.iterateNodes(getUids(nodeUids), signal));
     }
 
+    /**
+     * Get the node by its UID.
+     *
+     * See `ProtonDriveClient.getNode` for more information.
+     */
+    async getNode(nodeUid: NodeOrUid): Promise<MaybeNode> {
+        this.logger.info(`Getting node ${getUid(nodeUid)}`);
+        return convertInternalNodePromise(this.nodes.access.getNode(getUid(nodeUid)));
+    }
+
     /**
      * Iterates the albums.
      *
@@ -229,12 +246,35 @@ export class ProtonDrivePhotosClient {
         return this.download.getFileDownloader(getUid(nodeUid), signal);
     }
 
+    /**
+     * Iterates the thumbnails of the given nodes.
+     *
+     * See `ProtonDriveClient.iterateThumbnails` for more information.
+     */
+    async *iterateThumbnails(
+        nodeUids: NodeOrUid[],
+        thumbnailType?: ThumbnailType,
+        signal?: AbortSignal,
+    ): AsyncGenerator<ThumbnailResult> {
+        this.logger.info(`Iterating ${nodeUids.length} thumbnails`);
+        yield* this.download.iterateThumbnails(getUids(nodeUids), thumbnailType, signal);
+    }
+
     /**
      * Get the file uploader to upload a new file.
      *
      * See `ProtonDriveClient.getFileUploader` for more information.
      */
-    async getFileUploader(name: string, metadata: UploadMetadata, signal?: AbortSignal): Promise<FileUploader> {
+    async getFileUploader(
+        name: string,
+        metadata: UploadMetadata & {
+            captureTime?: Date;
+            mainPhotoLinkID?: string;
+            // TODO: handle tags enum in the SDK
+            tags?: (0 | 3 | 1 | 2 | 7 | 4 | 5 | 6 | 8 | 9)[];
+        },
+        signal?: AbortSignal,
+    ): Promise<FileUploader> {
         this.logger.info(`Getting file uploader`);
         const parentFolderUid = await this.nodes.access.getVolumeRootFolder();
         return this.upload.getFileUploader(getUid(parentFolderUid), name, metadata, signal);

package/src/transformers.ts

@@ -23,6 +23,7 @@ type InternalPartialNode = Pick<
     | 'type'
     | 'mediaType'
     | 'isShared'
+    | 'isSharedPublicly'
     | 'creationTime'
     | 'trashTime'
     | 'activeRevision'
@@ -90,6 +91,7 @@ export function convertInternalNode(node: InternalPartialNode): PublicMaybeNode
         type: node.type,
         mediaType: node.mediaType,
         isShared: node.isShared,
+        isSharedPublicly: node.isSharedPublicly,
         creationTime: node.creationTime,
         trashTime: node.trashTime,
         totalStorageSize: node.totalStorageSize,