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 { CommonResourceService } from "../../common/api/common-resource-service";
   6 import { CollectionResource } from "../../models/collection";
   7 import { AxiosInstance } from "axios";
   8 import { CollectionFile, CollectionDirectory } from "../../models/collection-file";
   9 import { WebDAV } from "../../common/webdav";
  10 import { AuthService } from "../auth-service/auth-service";
  11 import { mapTreeValues } from "../../models/tree";
  12 import { parseFilesResponse } from "./collection-service-files-response";
  13
  14 export type UploadProgress = (fileId: number, loaded: number, total: number, currentTime: number) => void;
  15
  16 export class CollectionService extends CommonResourceService<CollectionResource> {
  17     constructor(serverApi: AxiosInstance, private webdavClient: WebDAV, private authService: AuthService) {
  18         super(serverApi, "collections");
  19     }
  20
  21     async files(uuid: string) {
  22         const request = await this.webdavClient.propfind(`/c=${uuid}`);
  23         if (request.responseXML != null) {
  24             const filesTree = parseFilesResponse(request.responseXML);
  25             return mapTreeValues(this.extendFileURL)(filesTree);
  26         }
  27         return Promise.reject();
  28     }
  29
  30     async deleteFile(collectionUuid: string, filePath: string) {
  31         return this.webdavClient.delete(`/c=${collectionUuid}${filePath}`);
  32     }
  33
  34     async uploadFiles(collectionUuid: string, files: File[], onProgress?: UploadProgress) {
  35         for (let idx = 0; idx < files.length; idx++) {
  36             await this.uploadFile(collectionUuid, files[idx], idx, onProgress);
  37         }
  38     }
  39
  40     private extendFileURL = (file: CollectionDirectory | CollectionFile) => ({
  41         ...file,
  42         url: this.webdavClient.defaults.baseURL + file.url + '?api_token=' + this.authService.getApiToken()
  43     })
  44
  45     private uploadFile(collectionUuid: string, file: File, fileId: number, onProgress: UploadProgress = () => { return; }) {
  46         return this.readFile(file).then(content =>
  47             this.webdavClient.put(`/c=${collectionUuid}/${file.name}`, content, {
  48                 headers: {
  49                     'Content-Type': 'text/octet-stream'
  50                 },
  51                 onUploadProgress: (e: ProgressEvent) => {
  52                     onProgress(fileId, e.loaded, e.total, Date.now());
  53                 }
  54             })
  55         );
  56     }
  57
  58     private readFile(file: File): Promise<ArrayBuffer> {
  59         return new Promise<ArrayBuffer>(resolve => {
  60             const reader = new FileReader();
  61             reader.onload = () => {
  62                 resolve(reader.result as ArrayBuffer);
  63             };
  64             reader.readAsArrayBuffer(file);
  65         });
  66     }
  67
  68 }