SUSE/stratos

import { Schema, schema } from '../../../store/src/normalizr/normalizr';

import { EntityCatalogHelpers } from '../entity-catalog/entity-catalog.helper';
import { EntityCatalogEntityConfig } from '../entity-catalog/entity-catalog.types';

function wrapSchema(definition: Schema) {
  return {
    metadata: {},
    entity: definition
  };
}

export class StratosEntitySchema extends schema.Entity {
  constructor(
    public definition?: Schema
  ) {
    super('stratosWrappedEntity', wrapSchema(definition));
  }
}


/**
 * Mostly a wrapper around schema.Entity. Allows a lot of uniformity of types through console. Includes some minor per entity type config
 *
 * @export
 * @extends {schema.Entity}
 */
export class EntitySchema extends schema.Entity implements EntityCatalogEntityConfig {
  schema: Schema;
  schemaKey: string;
  public getId: (input, parent?, key?) => string;
  /**
   * @param entityKey As per schema.Entity ctor
   * @param [definition] As per schema.Entity ctor
   * @param [options] As per schema.Entity ctor
   * @param [relationKey] Allows multiple children of the same type within a single parent entity. For instance user with developer
   * spaces, manager spaces, auditor space, etc
   */
  constructor(
    public entityType: string,
    public endpointType: string,
    public definition?: Schema,
    private options?: schema.EntityOptions,
    public relationKey?: string,
    schemaKey?: string,
    public excludeFromRecursiveDelete: string[] = []
  ) {
    // Note - The core schema.Entity needs to be an entityKey or denormalize will fail
    // Note - Replacing `buildEntityKey` with `entityCatalog.getEntityKey` will cause circular dependency
    super(endpointType ? EntityCatalogHelpers.buildEntityKey(entityType, endpointType) : entityType, definition, options);
    this.schema = definition || {};
    // Normally the entityType === schemaKey. Sometimes we can override that (space entity and space entity with spaceWithOrg schema)
    this.schemaKey = schemaKey;
  }
  public withEmptyDefinition() {
    return new EntitySchema(
      this.entityType,
      this.endpointType,
      {},
      this.options,
      this.relationKey
    );
  }
  public clone() {
    return new EntitySchema(
      this.entityType,
      this.endpointType,
      this.definition,
      this.options,
      this.relationKey,
      this.schemaKey
    );
  }
}