@protontech/drive-sdk 0.0.10

package/dist/internal/upload/manager.d.ts

@@ -0,0 +1,31 @@
+import { ProtonDriveTelemetry, UploadMetadata } from "../../interface";
+import { UploadAPIService } from "./apiService";
+import { UploadCryptoService } from "./cryptoService";
+import { NodeRevisionDraft, NodesService, NodesEvents } from "./interface";
+/**
+ * UploadManager is responsible for creating and deleting draft nodes
+ * on the server. It handles the creation of draft nodes, including
+ * generating the necessary cryptographic keys and metadata.
+ */
+export declare class UploadManager {
+    private apiService;
+    private cryptoService;
+    private nodesService;
+    private nodesEvents;
+    private logger;
+    constructor(telemetry: ProtonDriveTelemetry, apiService: UploadAPIService, cryptoService: UploadCryptoService, nodesService: NodesService, nodesEvents: NodesEvents);
+    createDraftNode(parentFolderUid: string, name: string, metadata: UploadMetadata): Promise<NodeRevisionDraft>;
+    private createDraftOnAPI;
+    private findAvailableName;
+    deleteDraftNode(nodeUid: string): Promise<void>;
+    createDraftRevision(nodeUid: string, metadata: UploadMetadata): Promise<NodeRevisionDraft>;
+    deleteDraftRevision(nodeRevisionUid: string): Promise<void>;
+    commitDraft(nodeRevisionDraft: NodeRevisionDraft, manifest: Uint8Array, metadata: UploadMetadata, extendedAttributes: {
+        modificationTime?: Date;
+        size?: number;
+        blockSizes?: number[];
+        digests?: {
+            sha1?: string;
+        };
+    }, encryptedSize: number): Promise<void>;
+}

package/dist/internal/upload/manager.js

@@ -0,0 +1,250 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UploadManager = void 0;
+const ttag_1 = require("ttag");
+const interface_1 = require("../../interface");
+const errors_1 = require("../../errors");
+const nodes_1 = require("../nodes");
+/**
+ * UploadManager is responsible for creating and deleting draft nodes
+ * on the server. It handles the creation of draft nodes, including
+ * generating the necessary cryptographic keys and metadata.
+ */
+class UploadManager {
+    apiService;
+    cryptoService;
+    nodesService;
+    nodesEvents;
+    logger;
+    constructor(telemetry, apiService, cryptoService, nodesService, nodesEvents) {
+        this.apiService = apiService;
+        this.cryptoService = cryptoService;
+        this.nodesService = nodesService;
+        this.nodesEvents = nodesEvents;
+        this.logger = telemetry.getLogger('upload');
+        this.apiService = apiService;
+        this.cryptoService = cryptoService;
+        this.nodesService = nodesService;
+    }
+    async createDraftNode(parentFolderUid, name, metadata) {
+        const parentKeys = await this.nodesService.getNodeKeys(parentFolderUid);
+        if (!parentKeys.hashKey) {
+            throw new errors_1.ValidationError((0, ttag_1.c)('Error').t `Creating files in non-folders is not allowed`);
+        }
+        const generatedNodeCrypto = await this.cryptoService.generateFileCrypto(parentFolderUid, { key: parentKeys.key, hashKey: parentKeys.hashKey }, name);
+        const { nodeUid, nodeRevisionUid } = await this.createDraftOnAPI(parentFolderUid, parentKeys.hashKey, name, metadata, generatedNodeCrypto);
+        return {
+            nodeUid,
+            nodeRevisionUid,
+            nodeKeys: {
+                key: generatedNodeCrypto.nodeKeys.decrypted.key,
+                contentKeyPacketSessionKey: generatedNodeCrypto.contentKey.decrypted.contentKeyPacketSessionKey,
+                signatureAddress: generatedNodeCrypto.signatureAddress,
+            },
+            newNodeInfo: {
+                parentUid: parentFolderUid,
+                name,
+                encryptedName: generatedNodeCrypto.encryptedNode.encryptedName,
+                hash: generatedNodeCrypto.encryptedNode.hash,
+            },
+        };
+    }
+    async createDraftOnAPI(parentFolderUid, parentHashKey, name, metadata, generatedNodeCrypto) {
+        try {
+            const result = await this.apiService.createDraft(parentFolderUid, {
+                armoredEncryptedName: generatedNodeCrypto.encryptedNode.encryptedName,
+                hash: generatedNodeCrypto.encryptedNode.hash,
+                mediaType: metadata.mediaType,
+                intendedUploadSize: metadata.expectedSize,
+                armoredNodeKey: generatedNodeCrypto.nodeKeys.encrypted.armoredKey,
+                armoredNodePassphrase: generatedNodeCrypto.nodeKeys.encrypted.armoredPassphrase,
+                armoredNodePassphraseSignature: generatedNodeCrypto.nodeKeys.encrypted.armoredPassphraseSignature,
+                base64ContentKeyPacket: generatedNodeCrypto.contentKey.encrypted.base64ContentKeyPacket,
+                armoredContentKeyPacketSignature: generatedNodeCrypto.contentKey.encrypted.armoredContentKeyPacketSignature,
+                signatureEmail: generatedNodeCrypto.signatureAddress.email,
+                // FIXME: client UID
+            });
+            return result;
+        }
+        catch (error) {
+            if (error instanceof errors_1.ValidationError) {
+                if (error.code === 2500 /* ErrorCode.ALREADY_EXISTS */) {
+                    this.logger.info(`Node with given name already exists`);
+                    const availableName = await this.findAvailableName(parentFolderUid, parentHashKey, name, generatedNodeCrypto.encryptedNode.hash);
+                    // If there is existing draft created by this client,
+                    // automatically delete it and try to create a new one
+                    // with the same name again.
+                    if (availableName.existingDraftNodeUid) {
+                        let deleteFailed = false;
+                        try {
+                            this.logger.warn(`Deleting existing draft node ${availableName.existingDraftNodeUid}`);
+                            await this.apiService.deleteDraft(availableName.existingDraftNodeUid);
+                        }
+                        catch (deleteDraftError) {
+                            // Do not throw, let return the next available name to the client.
+                            deleteFailed = true;
+                            this.logger.error('Failed to delete existing draft node', deleteDraftError);
+                        }
+                        if (!deleteFailed) {
+                            return this.createDraftOnAPI(parentFolderUid, parentHashKey, name, metadata, generatedNodeCrypto);
+                        }
+                    }
+                    // If there is existing node, return special error
+                    // that includes the available name the client can use.
+                    throw new errors_1.NodeAlreadyExistsValidationError(error.message, error.code, availableName.availableName);
+                }
+            }
+            throw error;
+        }
+    }
+    async findAvailableName(parentFolderUid, parentHashKey, name, nameHash) {
+        const [namePart, extension] = splitExtension(name);
+        const batchSize = 10;
+        let startIndex = 1;
+        while (true) {
+            const namesToCheck = [];
+            for (let i = startIndex; i < startIndex + batchSize; i++) {
+                namesToCheck.push(joinNameAndExtension(namePart, i, extension));
+            }
+            const hashesToCheck = await this.cryptoService.generateNameHashes(parentHashKey, namesToCheck);
+            const { pendingHashes, availalbleHashes } = await this.apiService.checkAvailableHashes(parentFolderUid, [
+                ...hashesToCheck.map(({ hash }) => hash),
+                // Adding the current name hash to get the existing draft
+                // node UID if it exists.
+                ...startIndex ? [nameHash] : [],
+            ]);
+            if (!availalbleHashes.length) {
+                startIndex += batchSize;
+                continue;
+            }
+            const availableHash = hashesToCheck.find(({ hash }) => hash === availalbleHashes[0]);
+            if (!availableHash) {
+                throw Error('Backend returned unexpected hash');
+            }
+            // FIXME: use client UID to ensure its own pending draft
+            const ownPendingHash = pendingHashes.find(({ hash }) => hash === nameHash);
+            return {
+                availableName: availableHash.name,
+                existingDraftNodeUid: ownPendingHash?.nodeUid,
+            };
+        }
+    }
+    async deleteDraftNode(nodeUid) {
+        try {
+            await this.apiService.deleteDraft(nodeUid);
+        }
+        catch (error) {
+            // Only log the error but do not fail the operation as we are
+            // deleting draft only when somethign fails and original error
+            // will bubble up.
+            this.logger.error('Failed to delete draft node', error);
+        }
+    }
+    async createDraftRevision(nodeUid, metadata) {
+        const node = await this.nodesService.getNode(nodeUid);
+        const nodeKeys = await this.nodesService.getNodeKeys(nodeUid);
+        if (!node.activeRevision?.ok || !nodeKeys.contentKeyPacketSessionKey) {
+            throw new errors_1.ValidationError((0, ttag_1.c)('Error').t `Creating revisions in non-files is not allowed`);
+        }
+        const signatureAddress = await this.nodesService.getRootNodeEmailKey(nodeUid);
+        const { nodeRevisionUid } = await this.apiService.createDraftRevision(nodeUid, {
+            currentRevisionUid: node.activeRevision.value.uid,
+            intendedUploadSize: metadata.expectedSize,
+        });
+        return {
+            nodeUid,
+            nodeRevisionUid,
+            nodeKeys: {
+                key: nodeKeys.key,
+                contentKeyPacketSessionKey: nodeKeys.contentKeyPacketSessionKey,
+                signatureAddress: signatureAddress,
+            },
+        };
+    }
+    async deleteDraftRevision(nodeRevisionUid) {
+        try {
+            await this.apiService.deleteDraftRevision(nodeRevisionUid);
+        }
+        catch (error) {
+            // Only log the error but do not fail the operation as we are
+            // deleting draft only when somethign fails and original error
+            // will bubble up.
+            this.logger.error('Failed to delete draft node revision', error);
+        }
+    }
+    async commitDraft(nodeRevisionDraft, manifest, metadata, extendedAttributes, encryptedSize) {
+        const generatedExtendedAttributes = (0, nodes_1.generateFileExtendedAttributes)(extendedAttributes);
+        const nodeCommitCrypto = await this.cryptoService.commitFile(nodeRevisionDraft.nodeKeys, manifest, generatedExtendedAttributes);
+        await this.apiService.commitDraftRevision(nodeRevisionDraft.nodeRevisionUid, nodeCommitCrypto);
+        const activeRevision = (0, interface_1.resultOk)({
+            uid: nodeRevisionDraft.nodeRevisionUid,
+            state: interface_1.RevisionState.Active,
+            creationTime: new Date(),
+            storageSize: encryptedSize,
+            contentAuthor: (0, interface_1.resultOk)(nodeCommitCrypto.signatureEmail),
+            claimedSize: metadata.expectedSize,
+            claimedModificationTime: extendedAttributes.modificationTime,
+            claimedDigests: {
+                sha1: extendedAttributes.digests?.sha1,
+            },
+        });
+        if (nodeRevisionDraft.newNodeInfo) {
+            const node = {
+                // Internal metadata
+                hash: nodeRevisionDraft.newNodeInfo.hash,
+                encryptedName: nodeRevisionDraft.newNodeInfo.encryptedName,
+                // Basic node metadata
+                uid: nodeRevisionDraft.nodeUid,
+                parentUid: nodeRevisionDraft.newNodeInfo.parentUid,
+                type: interface_1.NodeType.File,
+                mediaType: metadata.mediaType,
+                creationTime: new Date(),
+                totalStorageSize: encryptedSize,
+                // Share node metadata
+                isShared: false,
+                directMemberRole: interface_1.MemberRole.Inherited,
+                // Decrypted metadata
+                isStale: false,
+                keyAuthor: (0, interface_1.resultOk)(nodeRevisionDraft.nodeKeys.signatureAddress.email),
+                nameAuthor: (0, interface_1.resultOk)(nodeRevisionDraft.nodeKeys.signatureAddress.email),
+                name: (0, interface_1.resultOk)(nodeRevisionDraft.newNodeInfo.name),
+                activeRevision,
+            };
+            await this.nodesEvents.nodeCreated(node);
+        }
+        else {
+            await this.nodesEvents.nodeUpdated({
+                uid: nodeRevisionDraft.nodeUid,
+                activeRevision,
+            });
+        }
+    }
+}
+exports.UploadManager = UploadManager;
+/**
+ * Split a filename into `[name, extension]`
+ */
+function splitExtension(filename = '') {
+    const endIdx = filename.lastIndexOf('.');
+    if (endIdx === -1 || endIdx === filename.length - 1) {
+        return [filename, ''];
+    }
+    return [filename.slice(0, endIdx), filename.slice(endIdx + 1)];
+}
+;
+/**
+ * Join a filename into `name (index).extension`
+ */
+function joinNameAndExtension(name, index, extension) {
+    if (!name && !extension) {
+        return `(${index})`;
+    }
+    if (!name) {
+        return `(${index}).${extension}`;
+    }
+    if (!extension) {
+        return `${name} (${index})`;
+    }
+    return `${name} (${index}).${extension}`;
+}
+//# sourceMappingURL=manager.js.map

package/dist/internal/upload/manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../src/internal/upload/manager.ts"],"names":[],"mappings":";;;AAAA,+BAAyB;AAEzB,+CAAwI;AACxI,yCAAiF;AAEjF,oCAAyE;AAKzE;;;;GAIG;AACH,MAAa,aAAa;IAKV;IACA;IACA;IACA;IAPJ,MAAM,CAAS;IAEvB,YACI,SAA+B,EACvB,UAA4B,EAC5B,aAAkC,EAClC,YAA0B,EAC1B,WAAwB;QAHxB,eAAU,GAAV,UAAU,CAAkB;QAC5B,kBAAa,GAAb,aAAa,CAAqB;QAClC,iBAAY,GAAZ,YAAY,CAAc;QAC1B,gBAAW,GAAX,WAAW,CAAa;QAEhC,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,eAAuB,EAAE,IAAY,EAAE,QAAwB;QACjF,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACxE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACtB,MAAM,IAAI,wBAAe,CAAC,IAAA,QAAC,EAAC,OAAO,CAAC,CAAC,CAAC,CAAA,8CAA8C,CAAC,CAAC;QAC1F,CAAC;QAED,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,kBAAkB,CACnE,eAAe,EACf,EAAE,GAAG,EAAE,UAAU,CAAC,GAAG,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,EACpD,IAAI,CACP,CAAC;QAEF,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC5D,eAAe,EACf,UAAU,CAAC,OAAO,EAClB,IAAI,EACJ,QAAQ,EACR,mBAAmB,CACtB,CAAC;QAEF,OAAO;YACH,OAAO;YACP,eAAe;YACf,QAAQ,EAAE;gBACN,GAAG,EAAE,mBAAmB,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG;gBAC/C,0BAA0B,EAAE,mBAAmB,CAAC,UAAU,CAAC,SAAS,CAAC,0BAA0B;gBAC/F,gBAAgB,EAAE,mBAAmB,CAAC,gBAAgB;aACzD;YACD,WAAW,EAAE;gBACT,SAAS,EAAE,eAAe;gBAC1B,IAAI;gBACJ,aAAa,EAAE,mBAAmB,CAAC,aAAa,CAAC,aAAa;gBAC9D,IAAI,EAAE,mBAAmB,CAAC,aAAa,CAAC,IAAI;aAC/C;SACJ,CAAC;IACN,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC1B,eAAuB,EACvB,aAAyB,EACzB,IAAY,EACZ,QAAwB,EACxB,mBAA+B;QAK/B,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,eAAe,EAAE;gBAC9D,oBAAoB,EAAE,mBAAmB,CAAC,aAAa,CAAC,aAAa;gBACrE,IAAI,EAAE,mBAAmB,CAAC,aAAa,CAAC,IAAI;gBAC5C,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,kBAAkB,EAAE,QAAQ,CAAC,YAAY;gBACzC,cAAc,EAAE,mBAAmB,CAAC,QAAQ,CAAC,SAAS,CAAC,UAAU;gBACjE,qBAAqB,EAAE,mBAAmB,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB;gBAC/E,8BAA8B,EAAE,mBAAmB,CAAC,QAAQ,CAAC,SAAS,CAAC,0BAA0B;gBACjG,sBAAsB,EAAE,mBAAmB,CAAC,UAAU,CAAC,SAAS,CAAC,sBAAsB;gBACvF,gCAAgC,EAAE,mBAAmB,CAAC,UAAU,CAAC,SAAS,CAAC,gCAAgC;gBAC3G,cAAc,EAAE,mBAAmB,CAAC,gBAAgB,CAAC,KAAK;gBAC1D,oBAAoB;aACvB,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,KAAK,YAAY,wBAAe,EAAE,CAAC;gBACnC,IAAI,KAAK,CAAC,IAAI,wCAA6B,EAAE,CAAC;oBAC1C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;oBACxD,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAC9C,eAAe,EACf,aAAa,EACb,IAAI,EACJ,mBAAmB,CAAC,aAAa,CAAC,IAAI,CACzC,CAAC;oBAEF,qDAAqD;oBACrD,sDAAsD;oBACtD,4BAA4B;oBAC5B,IAAI,aAAa,CAAC,oBAAoB,EAAE,CAAC;wBACrC,IAAI,YAAY,GAAG,KAAK,CAAC;wBACzB,IAAI,CAAC;4BACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gCAAgC,aAAa,CAAC,oBAAoB,EAAE,CAAC,CAAC;4BACvF,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,aAAa,CAAC,oBAAoB,CAAC,CAAC;wBAC1E,CAAC;wBAAC,OAAO,gBAAyB,EAAE,CAAC;4BACjC,kEAAkE;4BAClE,YAAY,GAAG,IAAI,CAAC;4BACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE,gBAAgB,CAAC,CAAC;wBAChF,CAAC;wBACD,IAAI,CAAC,YAAY,EAAE,CAAC;4BAChB,OAAO,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE,aAAa,EAAE,IAAI,EAAE,QAAQ,EAAE,mBAAmB,CAAC,CAAC;wBACtG,CAAC;oBACL,CAAC;oBAED,kDAAkD;oBAClD,uDAAuD;oBACvD,MAAM,IAAI,yCAAgC,CACtC,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,IAAI,EACV,aAAa,CAAC,aAAa,CAC9B,CAAC;gBACN,CAAC;YACL,CAAC;YACD,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,eAAuB,EAAE,aAAyB,EAAE,IAAY,EAAE,QAAgB;QAI9G,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QAEnD,MAAM,SAAS,GAAG,EAAE,CAAC;QACrB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,OAAO,IAAI,EAAE,CAAC;YACV,MAAM,YAAY,GAAG,EAAE,CAAC;YACxB,KAAK,IAAI,CAAC,GAAG,UAAU,EAAE,CAAC,GAAG,UAAU,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvD,YAAY,CAAC,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC;YACpE,CAAC;YAED,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAE/F,MAAM,EAAE,aAAa,EAAE,gBAAgB,EAAE,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAClF,eAAe,EACf;gBACI,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC;gBACxC,yDAAyD;gBACzD,yBAAyB;gBACzB,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE;aAClC,CACJ,CAAC;YAEF,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC;gBAC3B,UAAU,IAAI,SAAS,CAAC;gBACxB,SAAS;YACb,CAAC;YAED,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC;YACrF,IAAI,CAAC,aAAa,EAAE,CAAC;gBACjB,MAAM,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACpD,CAAC;YAED,wDAAwD;YACxD,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YAC3E,OAAO;gBACH,aAAa,EAAE,aAAa,CAAC,IAAI;gBACjC,oBAAoB,EAAE,cAAc,EAAE,OAAO;aAChD,CAAA;QACL,CAAC;IACL,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAe;QACjC,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,6DAA6D;YAC7D,8DAA8D;YAC9D,kBAAkB;YAClB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC;IACL,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,OAAe,EAAE,QAAwB;QAC/D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACtD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAE9D,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,0BAA0B,EAAE,CAAC;YACnE,MAAM,IAAI,wBAAe,CAAC,IAAA,QAAC,EAAC,OAAO,CAAC,CAAC,CAAC,CAAA,gDAAgD,CAAC,CAAC;QAC5F,CAAC;QAED,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAE9E,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,OAAO,EAAE;YAC3E,kBAAkB,EAAE,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG;YACjD,kBAAkB,EAAE,QAAQ,CAAC,YAAY;SAC5C,CAAC,CAAC;QAEH,OAAO;YACH,OAAO;YACP,eAAe;YACf,QAAQ,EAAE;gBACN,GAAG,EAAE,QAAQ,CAAC,GAAG;gBACjB,0BAA0B,EAAE,QAAQ,CAAC,0BAA0B;gBAC/D,gBAAgB,EAAE,gBAAgB;aACrC;SACJ,CAAA;IACL,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,eAAuB;QAC7C,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,6DAA6D;YAC7D,8DAA8D;YAC9D,kBAAkB;YAClB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAK,CAAC,CAAC;QACrE,CAAC;IACL,CAAC;IAED,KAAK,CAAC,WAAW,CACb,iBAAoC,EACpC,QAAoB,EACpB,QAAwB,EACxB,kBAOC,EACD,aAAqB;QAErB,MAAM,2BAA2B,GAAG,IAAA,sCAA8B,EAAC,kBAAkB,CAAC,CAAC;QACvF,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,EAAE,2BAA2B,CAAC,CAAC;QAChI,MAAM,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;QAE/F,MAAM,cAAc,GAAG,IAAA,oBAAQ,EAAkB;YAC7C,GAAG,EAAE,iBAAiB,CAAC,eAAe;YACtC,KAAK,EAAE,yBAAa,CAAC,MAAM;YAC3B,YAAY,EAAE,IAAI,IAAI,EAAE;YACxB,WAAW,EAAE,aAAa;YAC1B,aAAa,EAAE,IAAA,oBAAQ,EAAC,gBAAgB,CAAC,cAAc,CAAC;YACxD,WAAW,EAAE,QAAQ,CAAC,YAAY;YAClC,uBAAuB,EAAE,kBAAkB,CAAC,gBAAgB;YAC5D,cAAc,EAAE;gBACZ,IAAI,EAAE,kBAAkB,CAAC,OAAO,EAAE,IAAI;aACzC;SACJ,CAAC,CAAC;QACH,IAAI,iBAAiB,CAAC,WAAW,EAAE,CAAC;YAChC,MAAM,IAAI,GAAkB;gBACxB,oBAAoB;gBACpB,IAAI,EAAE,iBAAiB,CAAC,WAAW,CAAC,IAAI;gBACxC,aAAa,EAAE,iBAAiB,CAAC,WAAW,CAAC,aAAa;gBAE1D,sBAAsB;gBACtB,GAAG,EAAE,iBAAiB,CAAC,OAAO;gBAC9B,SAAS,EAAE,iBAAiB,CAAC,WAAW,CAAC,SAAS;gBAClD,IAAI,EAAE,oBAAQ,CAAC,IAAI;gBACnB,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,YAAY,EAAE,IAAI,IAAI,EAAE;gBACxB,gBAAgB,EAAE,aAAa;gBAE/B,sBAAsB;gBACtB,QAAQ,EAAE,KAAK;gBACf,gBAAgB,EAAE,sBAAU,CAAC,SAAS;gBAEtC,qBAAqB;gBACrB,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,IAAA,oBAAQ,EAAC,iBAAiB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC;gBACtE,UAAU,EAAE,IAAA,oBAAQ,EAAC,iBAAiB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC;gBACvE,IAAI,EAAE,IAAA,oBAAQ,EAAC,iBAAiB,CAAC,WAAW,CAAC,IAAI,CAAC;gBAElD,cAAc;aACjB,CAAA;YACD,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACJ,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC;gBAC/B,GAAG,EAAE,iBAAiB,CAAC,OAAO;gBAC9B,cAAc;aACjB,CAAC,CAAC;QACP,CAAC;IACL,CAAC;CACJ;AArRD,sCAqRC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,QAAQ,GAAG,EAAE;IACjC,MAAM,MAAM,GAAG,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACzC,IAAI,MAAM,KAAK,CAAC,CAAC,IAAI,MAAM,KAAK,QAAQ,CAAC,MAAM,GAAC,CAAC,EAAE,CAAC;QAChD,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC1B,CAAC;IACD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;AACnE,CAAC;AAAA,CAAC;AAEF;;GAEG;AACH,SAAS,oBAAoB,CAAC,IAAY,EAAE,KAAa,EAAE,SAAiB;IACxE,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACtB,OAAO,IAAI,KAAK,GAAG,CAAC;IACxB,CAAC;IACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACR,OAAO,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;IACrC,CAAC;IACD,IAAI,CAAC,SAAS,EAAE,CAAC;QACb,OAAO,GAAG,IAAI,KAAK,KAAK,GAAG,CAAC;IAChC,CAAC;IACD,OAAO,GAAG,IAAI,KAAK,KAAK,KAAK,SAAS,EAAE,CAAC;AAC7C,CAAC"}

package/dist/internal/upload/manager.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/internal/upload/manager.test.js

@@ -0,0 +1,349 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const errors_1 = require("../../errors");
+const interface_1 = require("../../interface");
+const telemetry_1 = require("../../tests/telemetry");
+const manager_1 = require("./manager");
+describe("UploadManager", () => {
+    let telemetry;
+    let apiService;
+    let cryptoService;
+    let nodesService;
+    let nodesEvents;
+    let manager;
+    beforeEach(() => {
+        telemetry = (0, telemetry_1.getMockTelemetry)();
+        // @ts-expect-error No need to implement all methods for mocking
+        apiService = {
+            createDraft: jest.fn().mockResolvedValue({
+                nodeUid: "newNode:nodeUid",
+                nodeRevisionUid: "newNode:nodeRevisionUid",
+            }),
+            deleteDraft: jest.fn(),
+            checkAvailableHashes: jest.fn().mockResolvedValue({
+                availalbleHashes: ["name1Hash"],
+                pendingHashes: [],
+            }),
+            commitDraftRevision: jest.fn(),
+        };
+        // @ts-expect-error No need to implement all methods for mocking
+        cryptoService = {
+            generateFileCrypto: jest.fn().mockResolvedValue({
+                nodeKeys: {
+                    decrypted: { key: 'newNode:key' },
+                    encrypted: {
+                        armoredKey: 'newNode:armoredKey',
+                        armoredPassphrase: 'newNode:armoredPassphrase',
+                        armoredPassphraseSignature: 'newNode:armoredPassphraseSignature',
+                    },
+                },
+                contentKey: {
+                    decrypted: { contentKeyPacketSessionKey: 'newNode:ContentKeyPacketSessionKey' },
+                    encrypted: {
+                        base64ContentKeyPacket: 'newNode:base64ContentKeyPacket',
+                        armoredContentKeyPacketSignature: 'newNode:armoredContentKeyPacketSignature',
+                    },
+                },
+                encryptedNode: {
+                    encryptedName: "newNode:encryptedName",
+                    hash: "newNode:hash",
+                },
+                signatureAddress: {
+                    email: "signatureEmail",
+                },
+            }),
+            generateNameHashes: jest.fn().mockResolvedValue([{
+                    name: "name1",
+                    hash: "name1Hash",
+                }, {
+                    name: "name2",
+                    hash: "name2Hash",
+                }, {
+                    name: "name3",
+                    hash: "name3Hash",
+                }]),
+            commitFile: jest.fn().mockResolvedValue({
+                armoredManifestSignature: "newNode:armoredManifestSignature",
+                signatureEmail: "signatureEmail",
+                armoredExtendedAttributes: "newNode:armoredExtendedAttributes",
+            }),
+        };
+        // @ts-expect-error No need to implement all methods for mocking
+        nodesService = {
+            getNodeKeys: jest.fn().mockResolvedValue({
+                hashKey: 'parentNode:hashKey',
+                key: 'parentNode:nodekey',
+            }),
+            getRootNodeEmailKey: jest.fn().mockResolvedValue({
+                email: "signatureEmail",
+                addressId: "addressId",
+            }),
+        };
+        nodesEvents = {
+            nodeCreated: jest.fn(),
+            nodeUpdated: jest.fn(),
+        };
+        manager = new manager_1.UploadManager(telemetry, apiService, cryptoService, nodesService, nodesEvents);
+    });
+    describe("createDraftNode", () => {
+        it("should fail to create node in non-folder parent", async () => {
+            nodesService.getNodeKeys = jest.fn().mockResolvedValue({ hashKey: undefined });
+            const result = manager.createDraftNode("parentUid", "name", {});
+            await expect(result).rejects.toThrow("Creating files in non-folders is not allowed");
+        });
+        it("should create draft node", async () => {
+            const result = await manager.createDraftNode("parentUid", "name", {
+                mediaType: "myMimeType",
+                expectedSize: 123456,
+            });
+            expect(result).toEqual({
+                nodeUid: "newNode:nodeUid",
+                nodeRevisionUid: "newNode:nodeRevisionUid",
+                nodeKeys: {
+                    key: "newNode:key",
+                    contentKeyPacketSessionKey: "newNode:ContentKeyPacketSessionKey",
+                    signatureAddress: {
+                        email: "signatureEmail",
+                    },
+                },
+                newNodeInfo: {
+                    parentUid: "parentUid",
+                    name: "name",
+                    encryptedName: "newNode:encryptedName",
+                    hash: "newNode:hash",
+                },
+            });
+            expect(apiService.createDraft).toHaveBeenCalledWith("parentUid", {
+                armoredEncryptedName: "newNode:encryptedName",
+                hash: "newNode:hash",
+                mediaType: "myMimeType",
+                intendedUploadSize: 123456,
+                armoredNodeKey: "newNode:armoredKey",
+                armoredNodePassphrase: "newNode:armoredPassphrase",
+                armoredNodePassphraseSignature: "newNode:armoredPassphraseSignature",
+                base64ContentKeyPacket: "newNode:base64ContentKeyPacket",
+                armoredContentKeyPacketSignature: "newNode:armoredContentKeyPacketSignature",
+                signatureEmail: "signatureEmail",
+            });
+            expect(apiService.checkAvailableHashes).not.toHaveBeenCalled();
+        });
+        it("should handle existing draft by deleting and trying again", async () => {
+            let hashChecked = false;
+            apiService.createDraft = jest.fn().mockImplementation(() => {
+                if (!hashChecked) {
+                    throw new errors_1.ValidationError("Draft already exists", 2500 /* ErrorCode.ALREADY_EXISTS */);
+                }
+                return {
+                    nodeUid: "newNode:nodeUid",
+                    nodeRevisionUid: "newNode:nodeRevisionUid",
+                };
+            });
+            apiService.checkAvailableHashes = jest.fn().mockImplementation(() => {
+                if (!hashChecked) {
+                    hashChecked = true;
+                    return {
+                        availalbleHashes: ["name1Hash"],
+                        pendingHashes: [{
+                                hash: "newNode:hash",
+                                nodeUid: "nodeUidToDelete"
+                            }],
+                    };
+                }
+                return {
+                    availalbleHashes: ["name1Hash"],
+                    pendingHashes: [],
+                };
+            });
+            const result = await manager.createDraftNode("parentUid", "name", {});
+            expect(apiService.checkAvailableHashes).toHaveBeenCalledTimes(1);
+            expect(apiService.deleteDraft).toHaveBeenCalledTimes(1);
+            expect(result).toEqual({
+                nodeUid: "newNode:nodeUid",
+                nodeRevisionUid: "newNode:nodeRevisionUid",
+                nodeKeys: {
+                    key: "newNode:key",
+                    contentKeyPacketSessionKey: "newNode:ContentKeyPacketSessionKey",
+                    signatureAddress: {
+                        email: "signatureEmail",
+                    },
+                },
+                newNodeInfo: {
+                    parentUid: "parentUid",
+                    name: "name",
+                    encryptedName: "newNode:encryptedName",
+                    hash: "newNode:hash",
+                },
+            });
+            expect(apiService.deleteDraft).toHaveBeenCalledWith("nodeUidToDelete");
+        });
+        it("should handle error when deleting existing draft", async () => {
+            let hashChecked = false;
+            apiService.createDraft = jest.fn().mockImplementation(() => {
+                if (!hashChecked) {
+                    throw new errors_1.ValidationError("Draft already exists", 2500 /* ErrorCode.ALREADY_EXISTS */);
+                }
+                return {
+                    nodeUid: "newNode:nodeUid",
+                    nodeRevisionUid: "newNode:nodeRevisionUid",
+                };
+            });
+            apiService.deleteDraft = jest.fn().mockImplementation(() => {
+                throw new Error("Failed to delete draft");
+            });
+            apiService.checkAvailableHashes = jest.fn().mockImplementation(() => {
+                if (!hashChecked) {
+                    hashChecked = true;
+                    return {
+                        availalbleHashes: ["name1Hash"],
+                        pendingHashes: [{
+                                hash: "newNode:hash",
+                                nodeUid: "nodeUidToDelete"
+                            }],
+                    };
+                }
+                return {
+                    availalbleHashes: ["name1Hash"],
+                    pendingHashes: [],
+                };
+            });
+            const result = manager.createDraftNode("parentUid", "name", {});
+            await expect(result).rejects.toThrow("Draft already exists");
+            expect(apiService.checkAvailableHashes).toHaveBeenCalledTimes(1);
+            expect(apiService.deleteDraft).toHaveBeenCalledTimes(1);
+        });
+        it("should handle existing name by providing available name", async () => {
+            let count = 0;
+            apiService.createDraft = jest.fn().mockImplementation(() => {
+                if (count === 0) {
+                    count++;
+                    throw new errors_1.ValidationError("Draft already exists", 2500 /* ErrorCode.ALREADY_EXISTS */);
+                }
+                return {
+                    nodeUid: "newNode:nodeUid",
+                    nodeRevisionUid: "newNode:nodeRevisionUid",
+                };
+            });
+            const result = manager.createDraftNode("parentUid", "name", {});
+            await expect(result).rejects.toThrow("Draft already exists");
+            expect(apiService.checkAvailableHashes).toHaveBeenCalledTimes(1);
+            try {
+                await result;
+            }
+            catch (error) {
+                expect(error.availableName).toBe("name1");
+            }
+        });
+        it("should handle existing name by providing available name when there is too many conflicts", async () => {
+            let hashChecked = false;
+            apiService.createDraft = jest.fn().mockImplementation(() => {
+                if (!hashChecked) {
+                    throw new errors_1.ValidationError("Draft already exists", 2500 /* ErrorCode.ALREADY_EXISTS */);
+                }
+                return {
+                    nodeUid: "newNode:nodeUid",
+                    nodeRevisionUid: "newNode:nodeRevisionUid",
+                };
+            });
+            apiService.checkAvailableHashes = jest.fn().mockImplementation(() => {
+                if (!hashChecked) {
+                    hashChecked = true;
+                    return {
+                        // First page has no available hashes
+                        availalbleHashes: [],
+                        pendingHashes: [],
+                    };
+                }
+                return {
+                    availalbleHashes: ["name3Hash"],
+                    pendingHashes: [],
+                };
+            });
+            const result = manager.createDraftNode("parentUid", "name", {});
+            await expect(result).rejects.toThrow("Draft already exists");
+            expect(apiService.checkAvailableHashes).toHaveBeenCalledTimes(2);
+            try {
+                await result;
+            }
+            catch (error) {
+                expect(error.availableName).toBe("name3");
+            }
+        });
+    });
+    describe("commit draft", () => {
+        const nodeRevisionDraft = {
+            nodeUid: "newNode:nodeUid",
+            nodeRevisionUid: "newNode:nodeRevisionUid",
+            nodeKeys: {
+                key: { _idx: 32321 },
+                contentKeyPacketSessionKey: "newNode:contentKeyPacketSessionKey",
+                signatureAddress: {
+                    email: "signatureEmail",
+                    addressId: "addressId",
+                    addressKey: "addressKey",
+                },
+            },
+        };
+        const manifest = new Uint8Array([1, 2, 3]);
+        const metadata = {
+            mediaType: "myMimeType",
+            expectedSize: 123456,
+        };
+        const extendedAttributes = {
+            modificationTime: new Date(),
+            digests: {
+                sha1: "sha1",
+            }
+        };
+        it("should commit revision draft", async () => {
+            await manager.commitDraft(nodeRevisionDraft, manifest, metadata, extendedAttributes, 1234567);
+            expect(cryptoService.commitFile).toHaveBeenCalledWith(nodeRevisionDraft.nodeKeys, manifest, expect.anything());
+            expect(apiService.commitDraftRevision).toHaveBeenCalledWith(nodeRevisionDraft.nodeRevisionUid, expect.anything());
+            expect(nodesEvents.nodeUpdated).toHaveBeenCalledWith({
+                uid: "newNode:nodeUid",
+                activeRevision: {
+                    ok: true,
+                    value: {
+                        uid: "newNode:nodeRevisionUid",
+                        state: interface_1.RevisionState.Active,
+                        creationTime: expect.any(Date),
+                        contentAuthor: { ok: true, value: "signatureEmail" },
+                        storageSize: 1234567,
+                        claimedSize: 123456,
+                        claimedModificationTime: extendedAttributes.modificationTime,
+                        claimedDigests: {
+                            sha1: "sha1",
+                        },
+                    },
+                },
+            });
+        });
+        it("should commit node draft", async () => {
+            const nodeRevisionDraftWithNewNodeInfo = {
+                ...nodeRevisionDraft,
+                newNodeInfo: {
+                    parentUid: "parentUid",
+                    name: "newNode:name",
+                    encryptedName: "newNode:encryptedName",
+                    hash: "newNode:hash",
+                }
+            };
+            await manager.commitDraft(nodeRevisionDraftWithNewNodeInfo, manifest, metadata, extendedAttributes, 1234567);
+            expect(cryptoService.commitFile).toHaveBeenCalledWith(nodeRevisionDraft.nodeKeys, manifest, expect.anything());
+            expect(apiService.commitDraftRevision).toHaveBeenCalledWith(nodeRevisionDraft.nodeRevisionUid, expect.anything());
+            expect(nodesEvents.nodeCreated).toHaveBeenCalledWith(expect.objectContaining({
+                uid: "newNode:nodeUid",
+                parentUid: "parentUid",
+                type: interface_1.NodeType.File,
+                totalStorageSize: 1234567,
+                activeRevision: {
+                    ok: true,
+                    value: expect.objectContaining({
+                        uid: "newNode:nodeRevisionUid",
+                        storageSize: 1234567,
+                    }),
+                },
+            }));
+        });
+    });
+});
+//# sourceMappingURL=manager.test.js.map