fix: fix hasOne relation with the same table

Signed-off-by: Muhammad Aaqil <aaqilcs102@gmail.com>
loopbackio · Aug 19, 2024 · 8d3a440 · 8d3a440
1 parent 30a2d56
commit 8d3a440
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packages/cli/generators/relation/has-one-relation.generator.js b/packages/cli/generators/relation/has-one-relation.generator.js
@@ -106,7 +106,11 @@ module.exports = class HasOneRelationGenerator extends BaseRelationGenerator {
     );
 
     relationUtils.addProperty(sourceClass, modelProperty);
-    const imports = relationUtils.getRequiredImports(targetModel, relationType);
+    const imports = relationUtils.getRequiredImports(
+      targetModel,
+      relationType,
+      sourceModel,
+    );
 
     relationUtils.addRequiredImports(sourceFile, imports);
     await sourceFile.save();
@@ -159,6 +163,7 @@ module.exports = class HasOneRelationGenerator extends BaseRelationGenerator {
     const importsArray = super._getRepositoryRequiredImports(
       dstModelClassName,
       dstRepositoryClassName,
+      this.artifactInfo.srcModelClass,
     );
     importsArray.push({
       name: 'HasOneRepositoryFactory',