metal-orm 1.1.5 → 1.1.6

package/dist/index.cjs

@@ -114,6 +114,8 @@ __export(index_exports, {
   acos: () => acos,
   add: () => add,
   addDomainEvent: () => addDomainEvent,
+  addEntityRelation: () => addEntityRelation,
+  addRelation: () => addRelation,
   age: () => age,
   aliasRef: () => aliasRef,
   and: () => and,
@@ -461,6 +463,9 @@ var defineTable = (name, columns, relations = {}, hooks, options = {}) => {
 function setRelations(table, relations) {
   table.relations = relations;
 }
+function addRelation(table, name, relation) {
+  table.relations[name] = relation;
+}
 var TABLE_REF_CACHE = /* @__PURE__ */ new WeakMap();
 var withColumnProps = (table) => {
   const cached = TABLE_REF_CACHE.get(table);
@@ -9194,6 +9199,69 @@ var bootstrapEntities = () => {
   }
   return metas.map((meta) => meta.table);
 };
+var resolveSingleRelation = (relationName, relation, rootMeta) => {
+  const tableMap = /* @__PURE__ */ new Map();
+  for (const m of getAllEntityMetadata()) {
+    if (m.table) tableMap.set(m.target, m.table);
+  }
+  switch (relation.kind) {
+    case RelationKinds.HasOne: {
+      const foreignKey = relation.foreignKey ?? `${getPivotKeyBaseFromRoot(rootMeta)}_id`;
+      return hasOne(
+        resolveTableTarget(relation.target, tableMap),
+        foreignKey,
+        relation.localKey,
+        relation.cascade
+      );
+    }
+    case RelationKinds.HasMany: {
+      const foreignKey = relation.foreignKey ?? `${getPivotKeyBaseFromRoot(rootMeta)}_id`;
+      return hasMany(
+        resolveTableTarget(relation.target, tableMap),
+        foreignKey,
+        relation.localKey,
+        relation.cascade
+      );
+    }
+    case RelationKinds.BelongsTo: {
+      return belongsTo(
+        resolveTableTarget(relation.target, tableMap),
+        relation.foreignKey,
+        relation.localKey,
+        relation.cascade
+      );
+    }
+    case RelationKinds.BelongsToMany: {
+      const pivotForeignKeyToRoot = relation.pivotForeignKeyToRoot ?? `${getPivotKeyBaseFromRoot(rootMeta)}_id`;
+      const pivotForeignKeyToTarget = relation.pivotForeignKeyToTarget ?? `${getPivotKeyBaseFromTarget(relation.target)}_id`;
+      return belongsToMany(
+        resolveTableTarget(relation.target, tableMap),
+        resolveTableTarget(relation.pivotTable, tableMap),
+        {
+          pivotForeignKeyToRoot,
+          pivotForeignKeyToTarget,
+          localKey: relation.localKey,
+          targetKey: relation.targetKey,
+          pivotPrimaryKey: relation.pivotPrimaryKey,
+          defaultPivotColumns: relation.defaultPivotColumns,
+          cascade: relation.cascade
+        }
+      );
+    }
+    default:
+      throw new Error(`Unknown relation kind for relation '${relationName}'`);
+  }
+};
+var addEntityRelation = (ctor, name, relation) => {
+  const meta = getEntityMetadata(ctor);
+  if (!meta) {
+    throw new Error(`Entity '${ctor.name}' is not registered. Did you decorate it with @Entity?`);
+  }
+  addRelationMetadata(ctor, name, relation);
+  if (meta.table) {
+    meta.table.relations[name] = resolveSingleRelation(name, relation, meta);
+  }
+};
 var getTableDefFromEntity = (ctor) => {
   const meta = getEntityMetadata(ctor);
   if (!meta) return void 0;
@@ -19815,6 +19883,8 @@ var TagIndex = class {
   acos,
   add,
   addDomainEvent,
+  addEntityRelation,
+  addRelation,
   age,
   aliasRef,
   and,