index.ts

import { activeChain, chainDefinitions } from "../chainDefinitions";
import { Model, ModelCtor } from "sequelize-typescript";
import { AddressReference, Block as BaseBlock, Day, Message as BaseMessage, Transaction, Validator, MonitoredValue } from "./base";
import { UserSetting, Template, TemplateFavorite, UserAddressName } from "./user";

function getFilteredBaseModel(): ModelCtor<Model<any, any>>[] {
  let models: ModelCtor<Model<any, any>>[] = baseModels;
  if (activeChain.customBlockModel) {
    models = models.filter((model) => model !== BaseBlock);
  }
  if (activeChain.customMessageModel) {
    models = models.filter((model) => model !== BaseMessage);
  }

  return models;
}

const baseModels: ModelCtor<Model<any, any>>[] = [AddressReference, BaseBlock, Day, BaseMessage, Transaction, Validator, MonitoredValue];

export function getChainModels(chainName: string) {
  let models: ModelCtor<Model<any, any>>[] = baseModels;
  if (chainDefinitions[chainName].customBlockModel) {
    models = models.filter((model) => model !== BaseBlock);
  }
  if (chainDefinitions[chainName].customMessageModel) {
    models = models.filter((model) => model !== BaseMessage);
  }

  return [...models, ...(chainDefinitions[chainName].customModels ?? [])];
}

export const chainModels = [...getFilteredBaseModel(), ...(activeChain.customModels ?? [])];
export const userModels: ModelCtor<Model<any, any>>[] = [UserSetting, Template, TemplateFavorite, UserAddressName];
export const Block = activeChain.customBlockModel || BaseBlock;
export const Message = activeChain.customMessageModel || BaseMessage;