import { RelationsTable } from '../pg-models/tables';
import { ContentEntityModelPostprocessor } from './content-entity-model-postprocessor';
import { findEntityModel } from './postprocessing-utils';

export const collectionPostprocessor: ContentEntityModelPostprocessor = (
  mainModel,
  otherModels,
  options,
) => {
  const relatedItemModels = ['movie', 'tvshow', 'season', 'episode'].map((n) =>
    findEntityModel(otherModels, n),
  );

  mainModel.relatedObjects.push(
    new RelationsTable(
      options,
      mainModel.contentEntity,
      relatedItemModels.map((m) => m.contentEntity),
      'items',
      [
        {
          name: 'order_no',
          type: 'INTEGER',
          options: { extraParams: 'NOT NULL DEFAULT 0', indexable: true },
        },
        {
          name: 'relation_type',
          type: 'TEXT',
        },
      ],
    ),
  );
};