src/services/collection-service/collection-service-files-response.ts

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 import { createCollectionFilesTree, CollectionDirectory, CollectionFile, CollectionFileType, createCollectionDirectory, createCollectionFile } from "../../models/collection-file";
   6 import { Tree, mapTree, getNodeChildren, getNode, TreeNode } from "../../models/tree";
   7 import { getTagValue } from "../../common/xml";
   8
   9 export const parseFilesResponse = (document: Document) => {
  10     const files = extractFilesData(document);
  11     const tree = createCollectionFilesTree(files);
  12     return sortFilesTree(tree);
  13 };
  14
  15 export const sortFilesTree = (tree: Tree<CollectionDirectory | CollectionFile>) => {
  16     return mapTree(node => {
  17         const children = getNodeChildren(node.id)(tree)
  18             .map(id => getNode(id)(tree))
  19             .filter(node => node !== undefined) as TreeNode<CollectionDirectory | CollectionFile>[];
  20
  21         children.sort((a, b) =>
  22             a.value.type !== b.value.type
  23                 ? a.value.type === CollectionFileType.DIRECTORY ? -1 : 1
  24                 : a.value.name.localeCompare(b.value.name)
  25         );
  26         return { ...node, children: children.map(child => child.id) };
  27     })(tree) as Tree<CollectionDirectory | CollectionFile>;
  28 };
  29
  30 export const extractFilesData = (document: Document) => {
  31     const collectionUrlPrefix = /\/c=[0-9a-zA-Z\-]*/;
  32     return Array
  33         .from(document.getElementsByTagName('D:response'))
  34         .slice(1) // omit first element which is collection itself
  35         .map(element => {
  36             const name = getTagValue(element, 'D:displayname', '');
  37             const size = parseInt(getTagValue(element, 'D:getcontentlength', '0'), 10);
  38             const url = getTagValue(element, 'D:href', '');
  39             const nameSuffix = `/${name || ''}`;
  40             const directory = url
  41                 .replace(collectionUrlPrefix, '')
  42                 .replace(nameSuffix, '');
  43
  44             const data = {
  45                 url,
  46                 id: `${directory}/${name}`,
  47                 name,
  48                 path: directory,
  49             };
  50
  51             return getTagValue(element, 'D:resourcetype', '')
  52                 ? createCollectionDirectory(data)
  53                 : createCollectionFile({ ...data, size });
  54
  55         });
  56 };