medusa-store/packages/link-modules/src/repositories/link.ts

import { Context, ModuleJoinerConfig } from "@medusajs/types"
import { EntitySchema } from "@mikro-orm/core"

import {
  generateEntityId,
  mikroOrmBaseRepositoryFactory,
} from "@medusajs/utils"
import { SqlEntityManager } from "@mikro-orm/postgresql"

export function getLinkRepository(model: EntitySchema) {
  return class LinkRepository extends mikroOrmBaseRepositoryFactory(model) {
    readonly joinerConfig_: ModuleJoinerConfig

    constructor({ joinerConfig }: { joinerConfig: ModuleJoinerConfig }) {
      // @ts-ignore
      super(...arguments)
      this.joinerConfig_ = joinerConfig
    }

    async delete(data: any, context: Context = {}): Promise<void> {
      const filter = {}
      for (const key in data) {
        filter[key] = {
          $in: Array.isArray(data[key]) ? data[key] : [data[key]],
        }
      }

      const manager = this.getActiveManager<SqlEntityManager>(context)
      await manager.nativeDelete(model, data, {})
    }

    async create(data: object[], context: Context = {}): Promise<object[]> {
      const manager = this.getActiveManager<SqlEntityManager>(context)

      const links = data.map((link: any) => {
        link.id = generateEntityId(
          link.id,
          this.joinerConfig_.databaseConfig?.idPrefix ?? "link"
        )

        return manager.create(model, link)
      })

      await manager.upsertMany(model, links)

      return links
    }
  }
}