import { PgColumn } from '../abstractions';
import { ContentEntityModel } from '../content-entity-model';

export function findEntityModel(
  models: ContentEntityModel[],
  entityName: string,
): ContentEntityModel {
  const entityModel = models.find((m) => m.contentEntity.name === entityName);
  if (!entityModel) {
    throw Error(`Content entity model for ${entityName} not found.`);
  }
  return entityModel;
}

export function removeColumnByName(columns: PgColumn[], name: string): void {
  columns.forEach((c, i) => {
    if (c.name === name) {
      columns.splice(i, 1);
    }
  });
}