@protontech/drive-sdk 0.7.0 → 0.7.1

package/src/transformers.ts

@@ -1,15 +1,19 @@
 import {
     MaybeNode as PublicMaybeNode,
     MaybeMissingNode as PublicMaybeMissingNode,
-    NodeEntity as PublicNodeEntity,
     DegradedNode as PublicDegradedNode,
     Revision as PublicRevision,
     Result,
     resultOk,
     resultError,
     MissingNode,
+    MaybePhotoNode as PublicMaybePhotoNode,
+    MaybeMissingPhotoNode as PublicMaybeMissingPhotoNode,
+    PhotoNode as PublicPhotoNode,
+    DegradedPhotoNode as PublicDegradedPhotoNode,
 } from './interface';
 import { DecryptedNode as InternalNode, DecryptedRevision as InternalRevision } from './internal/nodes';
+import { DecryptedPhotoNode as InternalPartialPhotoNode } from './internal/photos';
 
 type InternalPartialNode = Pick<
     InternalNode,
@@ -25,6 +29,7 @@ type InternalPartialNode = Pick<
     | 'isShared'
     | 'isSharedPublicly'
     | 'creationTime'
+    | 'modificationTime'
     | 'trashTime'
     | 'activeRevision'
     | 'folder'
@@ -93,6 +98,7 @@ export function convertInternalNode(node: InternalPartialNode): PublicMaybeNode
         isShared: node.isShared,
         isSharedPublicly: node.isSharedPublicly,
         creationTime: node.creationTime,
+        modificationTime: node.modificationTime,
         trashTime: node.trashTime,
         totalStorageSize: node.totalStorageSize,
         folder: node.folder,
@@ -118,7 +124,48 @@ export function convertInternalNode(node: InternalPartialNode): PublicMaybeNode
         ...baseNodeMetadata,
         name: name.value,
         activeRevision: activeRevision?.ok ? convertInternalRevision(activeRevision.value) : undefined,
-    } as PublicNodeEntity);
+    });
+}
+
+export async function* convertInternalPhotoNodeIterator(
+    photoNodeIterator: AsyncGenerator<InternalPartialPhotoNode>,
+): AsyncGenerator<PublicMaybePhotoNode> {
+    for await (const photoNode of photoNodeIterator) {
+        yield convertInternalPhotoNode(photoNode);
+    }
+}
+
+export async function* convertInternalMissingPhotoNodeIterator(
+    photoNodeIterator: AsyncGenerator<InternalPartialPhotoNode | MissingNode>,
+): AsyncGenerator<PublicMaybeMissingPhotoNode> {
+    for await (const photoNode of photoNodeIterator) {
+        if ('missingUid' in photoNode) {
+            yield resultError(photoNode);
+        } else {
+            yield convertInternalPhotoNode(photoNode);
+        }
+    }
+}
+
+export async function convertInternalPhotoNodePromise(
+    photoNodePromise: Promise<InternalPartialPhotoNode>,
+): Promise<PublicMaybePhotoNode> {
+    const photoNode = await photoNodePromise;
+    return convertInternalPhotoNode(photoNode);
+}
+
+export function convertInternalPhotoNode(photoNode: InternalPartialPhotoNode): PublicMaybePhotoNode {
+    const node = convertInternalNode(photoNode);
+    if (node.ok) {
+        return resultOk({
+            ...node.value,
+            photo: photoNode.photo,
+        } as PublicPhotoNode);
+    }
+    return resultError({
+        ...node.error,
+        photo: photoNode.photo,
+    } as PublicDegradedPhotoNode);
 }
 
 export async function* convertInternalRevisionIterator(