open-vault/ui/app/serializers/clients/monthly.js

import ApplicationSerializer from '../application';
import { formatISO } from 'date-fns';

export default class MonthlySerializer extends ApplicationSerializer {
  flattenDataset(object) {
    let flattenedObject = {};
    Object.keys(object['counts']).forEach((key) => (flattenedObject[key] = object['counts'][key]));
    return this.homogenizeClientNaming(flattenedObject);
  }

  formatByNamespace(namespaceArray) {
    return namespaceArray?.map((ns) => {
      // 'namespace_path' is an empty string for root
      if (ns['namespace_id'] === 'root') ns['namespace_path'] = 'root';
      let label = ns['namespace_path'];
      let flattenedNs = this.flattenDataset(ns);
      // if no mounts, mounts will be an empty array
      flattenedNs.mounts = [];
      if (ns?.mounts && ns.mounts.length > 0) {
        flattenedNs.mounts = ns.mounts.map((mount) => {
          return {
            label: mount['mount_path'],
            ...this.flattenDataset(mount),
          };
        });
      }
      return {
        label,
        ...flattenedNs,
      };
    });
  }

  // In 1.10 'distinct_entities' changed to 'entity_clients' and
  // 'non_entity_tokens' to 'non_entity_clients'
  homogenizeClientNaming(object) {
    // if new key names exist, only return those key/value pairs
    if (Object.keys(object).includes('entity_clients')) {
      let { clients, entity_clients, non_entity_clients } = object;
      return {
        clients,
        entity_clients,
        non_entity_clients,
      };
    }
    // if object only has outdated key names, update naming
    if (Object.keys(object).includes('distinct_entities')) {
      let { clients, distinct_entities, non_entity_tokens } = object;
      return {
        clients,
        entity_clients: distinct_entities,
        non_entity_clients: non_entity_tokens,
      };
    }
    return object;
  }

  normalizeResponse(store, primaryModelClass, payload, id, requestType) {
    if (payload.id === 'no-data') {
      return super.normalizeResponse(store, primaryModelClass, payload, id, requestType);
    }
    let response_timestamp = formatISO(new Date());
    // TODO CMB: the following is assumed, need to confirm
    // the months array will always include a single object: a timestamp of the current month and new/total count data, if available
    let newClientsData = payload.data.months[0]?.new_clients || null;
    let by_namespace_new_clients, new_clients;
    if (newClientsData) {
      by_namespace_new_clients = this.formatByNamespace(newClientsData.namespaces);
      new_clients = this.homogenizeClientNaming(newClientsData.counts);
    } else {
      by_namespace_new_clients = [];
      new_clients = [];
    }
    let transformedPayload = {
      ...payload,
      response_timestamp,
      by_namespace_total_clients: this.formatByNamespace(payload.data.by_namespace),
      by_namespace_new_clients,
      // nest within 'total' object to mimic /activity response shape
      total: this.homogenizeClientNaming(payload.data),
      new: new_clients,
    };
    delete payload.data.by_namespace;
    delete payload.data.months;
    delete payload.data.total;
    return super.normalizeResponse(store, primaryModelClass, transformedPayload, id, requestType);
  }
}