medusa-store/packages/modules/link-modules/src/utils/generate-entity.ts

import {
  JoinerRelationship,
  ModuleJoinerConfig,
} from "@medusajs/framework/types"
import {
  composeTableName,
  compressName,
  mikroOrmSoftDeletableFilterOptions,
  simpleHash,
  SoftDeletableFilterKey,
} from "@medusajs/framework/utils"

import { EntitySchema } from "@mikro-orm/core"

function getClass(...properties) {
  return class LinkModel {
    constructor(...values) {
      properties.forEach((name, idx) => {
        this[name] = values[idx]
      })
    }
  }
}

export function generateEntity(
  joinerConfig: ModuleJoinerConfig,
  primary: JoinerRelationship,
  foreign: JoinerRelationship
) {
  const fieldNames = primary.foreignKey.split(",").concat(foreign.foreignKey)

  const tableName =
    joinerConfig.databaseConfig?.tableName ??
    composeTableName(
      primary.serviceName,
      primary.foreignKey,
      foreign.serviceName,
      foreign.foreignKey
    ).toLowerCase()

  const fields = fieldNames.reduce((acc, curr) => {
    acc[curr] = {
      type: "string",
      nullable: false,
      primary: true,
    }
    return acc
  }, {})

  const extraFields = joinerConfig.databaseConfig?.extraFields ?? {}

  for (const column in extraFields) {
    fieldNames.push(column)

    fields[column] = {
      type: extraFields[column].type,
      nullable: !!extraFields[column].nullable,
      defaultRaw: extraFields[column].defaultValue,
      ...(extraFields[column].options ?? {}),
    }
  }

  const hashTableName = simpleHash(tableName)

  return new EntitySchema({
    class: getClass(
      ...fieldNames.concat("created_at", "updated_at", "deleted_at")
    ) as any,
    tableName: compressName(tableName),
    properties: {
      id: {
        type: "string",
        nullable: false,
      },
      ...fields,
      created_at: {
        columnType: "timestamptz",
        type: "date",
        nullable: false,
        defaultRaw: "CURRENT_TIMESTAMP",
      },
      updated_at: {
        columnType: "timestamptz",
        type: "date",
        nullable: false,
        defaultRaw: "CURRENT_TIMESTAMP",
      },
      deleted_at: {
        columnType: "timestamptz",
        type: "date",
        nullable: true,
      },
    } as any,
    filters: {
      [SoftDeletableFilterKey]: mikroOrmSoftDeletableFilterOptions,
    },
    hooks: {
      beforeUpdate: [
        (args) => {
          args.entity.updated_at = new Date()
        },
      ],
    },
    indexes: [
      {
        properties: ["id"],
        name: "IDX_id_" + hashTableName,
      },
      {
        properties: primary.foreignKey.split(",") as any,
        name:
          "IDX_" +
          primary.foreignKey.split(",").join("_") +
          "_" +
          hashTableName,
        expression:
          "CREATE INDEX IF NOT EXISTS " +
          '"IDX_' +
          primary.foreignKey.split(",").join("_") +
          "_" +
          hashTableName +
          '" ON "' +
          compressName(tableName) +
          '" ("' +
          primary.foreignKey.split(",").join(",") +
          '") WHERE deleted_at IS NULL',
      },
      {
        properties: foreign.foreignKey as any,
        name: "IDX_" + foreign.foreignKey + "_" + hashTableName,
        expression:
          "CREATE INDEX IF NOT EXISTS " +
          '"IDX_' +
          foreign.foreignKey +
          "_" +
          hashTableName +
          '" ON "' +
          compressName(tableName) +
          '" ("' +
          foreign.foreignKey +
          '") WHERE deleted_at IS NULL',
      },
      {
        properties: ["deleted_at"],
        name: "IDX_deleted_at_" + hashTableName,
      },
    ],
  })
}