feat(pinia-orm): Add setModel for Repository

CodeDredd · CodeDredd · commit d0cd5e639ddf · 2023-09-17T10:28:08.000+02:00
This is handy if you want to create a custom `useRepo` composable
diff --git a/packages/pinia-orm/src/repository/Repository.ts b/packages/pinia-orm/src/repository/Repository.ts
@@ -82,6 +82,11 @@ export class Repository<M extends Model = Model> {
     this.hydratedDataCache = hydratedDataCache as Map<string, M>
   }
 
+  static setModel(model: Model) {
+    this.useModel = model
+    return this
+  }
+
   /**
    * Set the global config
    */
diff --git a/packages/pinia-orm/tests/unit/repository/Repository.spec.ts b/packages/pinia-orm/tests/unit/repository/Repository.spec.ts
@@ -88,6 +88,17 @@ describe('unit/repository/Repository', () => {
     expect(postRepo.getModel()).toBeInstanceOf(Post)
   })
 
+  it('can create a new repository from the custom repository without use defined', () => {
+    class PostRepository extends Repository<Post> {
+    }
+
+    const userRepo = useRepo(User)
+
+    const postRepo = userRepo.repo(PostRepository.setModel(Post))
+
+    expect(postRepo.getModel()).toBeInstanceOf(Post)
+  })
+
   it('can trigger cache & hook from the pinia store action', () => {
     const userRepo = useRepo(User)
     expect(userRepo.hydratedDataCache.size).toBe(0)
