15840: Project panel uses collection attributes for file count and size
[arvados-workbench2.git] / src / services / collection-service / collection-service.ts
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: AGPL-3.0
4
5 import { CollectionResource } from "~/models/collection";
6 import { AxiosInstance } from "axios";
7 import { CollectionFile, CollectionDirectory } from "~/models/collection-file";
8 import { WebDAV } from "~/common/webdav";
9 import { AuthService } from "../auth-service/auth-service";
10 import { extractFilesData } from "./collection-service-files-response";
11 import { TrashableResourceService } from "~/services/common-service/trashable-resource-service";
12 import { ApiActions } from "~/services/api/api-actions";
13
14 export type UploadProgress = (fileId: number, loaded: number, total: number, currentTime: number) => void;
15
16 export class CollectionService extends TrashableResourceService<CollectionResource> {
17     constructor(serverApi: AxiosInstance, private webdavClient: WebDAV, private authService: AuthService, actions: ApiActions) {
18         super(serverApi, "collections", actions);
19     }
20
21     async files(uuid: string) {
22         const request = await this.webdavClient.propfind(`c=${uuid}`);
23         if (request.responseXML != null) {
24             return extractFilesData(request.responseXML);
25         }
26         return Promise.reject();
27     }
28
29     async deleteFiles(collectionUuid: string, filePaths: string[]) {
30         for (const path of filePaths) {
31             const splittedPath = path.split('/');
32             if (collectionUuid) {
33                 await this.webdavClient.delete(`c=${collectionUuid}/${splittedPath[1]}`);
34             } else {
35                 await this.webdavClient.delete(`c=${collectionUuid}${path}`);
36             }
37         }
38     }
39
40     async uploadFiles(collectionUuid: string, files: File[], onProgress?: UploadProgress) {
41         // files have to be uploaded sequentially
42         for (let idx = 0; idx < files.length; idx++) {
43             await this.uploadFile(collectionUuid, files[idx], idx, onProgress);
44         }
45     }
46
47     moveFile(collectionUuid: string, oldPath: string, newPath: string) {
48         return this.webdavClient.move(
49             `c=${collectionUuid}${oldPath}`,
50             `c=${collectionUuid}${encodeURI(newPath)}`
51         );
52     }
53
54     extendFileURL = (file: CollectionDirectory | CollectionFile) => {
55         const baseUrl = this.webdavClient.defaults.baseURL.endsWith('/')
56             ? this.webdavClient.defaults.baseURL.slice(0, -1)
57             : this.webdavClient.defaults.baseURL;
58         const apiToken = this.authService.getApiToken();
59         const splittedApiToken = apiToken ? apiToken.split('/') : [];
60         const userApiToken = `/t=${splittedApiToken[2]}/`;
61         const splittedPrevFileUrl = file.url.split('/');
62         const url = `${baseUrl}/${splittedPrevFileUrl[1]}${userApiToken}${splittedPrevFileUrl[2]}`;
63         return {
64             ...file,
65             url
66         };
67     }
68
69     private async uploadFile(collectionUuid: string, file: File, fileId: number, onProgress: UploadProgress = () => { return; }) {
70         const fileURL = `c=${collectionUuid}/${file.name}`;
71         const requestConfig = {
72             headers: {
73                 'Content-Type': 'text/octet-stream'
74             },
75             onUploadProgress: (e: ProgressEvent) => {
76                 onProgress(fileId, e.loaded, e.total, Date.now());
77             }
78         };
79         return this.webdavClient.upload(fileURL, [file], requestConfig);
80     }
81 }