@cumulus/pvl/lib/models.js

'use strict';

const isString = require('lodash/isString');
const { areIDsSame } = require('./utils');

class PVLAggregate {
  constructor() {
    this.store = [];
  }

  add(key, value) {
    this.store.push([key, value]);
    return this;
  }

  get(key) {
    return this.store.find((item) => item[0] === key)
      ? this.store.find((item) => item[0] === key)[1]
      : null;
  }

  getAll(key) {
    return this.store.filter((item) => item[0] === key).map((item) => item[1]);
  }

  removeAll(key) {
    this.store = this.store.filter((item) => item[0] !== key);
  }

  // Since OBJECT and GROUP are reserved keywords, this won't collide with attribute keys
  addAggregate(aggregate) {
    this.store.push([aggregate.type, aggregate]);
    return this;
  }

  objects(key) {
    return this.getAll('OBJECT').filter((o) => (key ? areIDsSame(o.identifier, key) : true));
  }

  groups(key) {
    return this.getAll('GROUP').filter((g) => (key ? areIDsSame(g.identifier, key) : true));
  }

  aggregates(key) {
    return this.objects(key).concat(this.groups(key));
  }

  toPVL() {
    return this.store.reduce(
      (last, curr) => last.concat(`${curr[0]} = ${curr[1].toPVL()};\n`),
      `${this.identifier};\n`
    ).concat(`END_${this.type} = ${this.identifier}`);
  }
}

class PVLRoot extends PVLAggregate {
  constructor() {
    super();
    this.type = 'ROOT';
    this.depth = 0;
  }

  toPVL() {
    return this.store.reduce((last, curr) => last.concat(`${curr[0]} = ${curr[1].toPVL()};\n`), '');
  }
}

class PVLObject extends PVLAggregate {
  constructor(identifier) {
    super();
    this.identifier = identifier;
    this.type = 'OBJECT';
  }
}

class PVLGroup extends PVLAggregate {
  constructor(identifier) {
    super();
    this.identifier = identifier;
    this.type = 'GROUP';
  }
}

class PVLValue {}

class PVLScalar extends PVLValue {
  constructor(value) {
    super(value);
    this.value = value;
  }
}

class PVLNumeric extends PVLScalar {
  constructor(value, units) {
    super(value);
    this.value = Number(this.value);
    if (isString(units)) {
      this.units = units.toUpperCase();
    }
    this.type = 'numeric';
  }

  toPVL() {
    return this.units ? `${this.value} <${this.units}>` : `${this.value}`;
  }
}

class PVLDateTime extends PVLScalar {
  constructor(value) {
    super(value);
    this.value = new Date(this.value);
    this.type = 'date time';
  }

  toPVL() {
    return this.value.toISOString();
  }
}

class PVLTextString extends PVLScalar {
  constructor(value) {
    super(value);
    this.type = 'text string';
  }

  toPVL() {
    return this.value.includes('"') ? `'${this.value}'` : `"${this.value}"`;
  }
}

module.exports = {
  PVLAggregate,
  PVLRoot,
  PVLObject,
  PVLGroup,
  PVLNumeric,
  PVLDateTime,
  PVLTextString
};