feat(generators/with-mongo/templates): change implementation of model creattion, get model from mongoose in repository for usage

Author: phatpham9

generators/with-mongo/templates/<%= compNameParamCasePlural %>/model.ts

@@ -4,17 +4,18 @@ import {
   Model as MongooseModel,
 } from 'mongoose';
 
-import mongo from '../../db/mongo';
 import { <%= compNamePascalCase %> } from './types';
 import { ENTITY } from './constants';
 
 interface Document extends MongooseDocument, Omit<<%= compNamePascalCase %>, 'id'> {}
 interface Model extends MongooseModel<Document> {}
 
+const name = ENTITY;
+
 const schema = new Schema<Document>({
   name: String,
   // TODO: add more fields
 });
 
-export default mongo.createModel<Document, Model>(ENTITY, schema);
-export { Model, Document };
+export default { name, schema };
+export { Document, Model };

generators/with-mongo/templates/<%= compNameParamCasePlural %>/repository.ts

@@ -1,36 +1,39 @@
+import mongoose from 'mongoose';
 import { MyError } from '@boringcodes/utils/error';
 
 import { <%= compNamePascalCase %> } from './types';
-import Model, { Document } from './model';
+import model, { Document, Model } from './model';
 
 const list = async (): Promise<<%= compNamePascalCase %>[]> => {
   // list documents
-  const documents = await Model.find();
+  const documents = await getModel().find();
 
   return documents.map(transform);
 };
 
 const create = async (object: Omit<<%= compNamePascalCase %>, 'id'>): Promise<<%= compNamePascalCase %>> => {
   // create document
-  const document = new Model(object);
-  await document.save();
+  const document = await getModel().create(object);
 
   return transform(document);
 };
 
 const get = async (id: string): Promise<<%= compNamePascalCase %>> => {
   // get document
-  const document = await Model.findById(id).exec();
+  const document = await getModel().findById(id).exec();
   if (document === null) {
     throw new MyError('Document not found');
   }
 
   return transform(document);
 };
 
-const update = async (id: string, object: Omit<<%= compNamePascalCase %>, 'id'>): Promise<<%= compNamePascalCase %>> => {
+const update = async (
+  id: string,
+  object: Omit<<%= compNamePascalCase %>, 'id'>,
+): Promise<<%= compNamePascalCase %>> => {
   // get document
-  const document = await Model.findById(id).exec();
+  const document = await getModel().findById(id).exec();
   if (document === null) {
     throw new MyError('Document not found');
   }
@@ -44,7 +47,7 @@ const update = async (id: string, object: Omit<<%= compNamePascalCase %>, 'id'>)
 
 const del = async (id: string): Promise<<%= compNamePascalCase %>> => {
   // get document
-  const document = await Model.findById(id).exec();
+  const document = await getModel().findById(id).exec();
   if (document === null) {
     throw new MyError('Document not found');
   }
@@ -55,6 +58,11 @@ const del = async (id: string): Promise<<%= compNamePascalCase %>> => {
   return transform(document);
 };
 
+// get model
+const getModel = (): Model => {
+  return mongoose.models[model.name];
+};
+
 // transform document to <%= compNamePascalCase %>
 const transform = (document: Document): <%= compNamePascalCase %> => {
   const { _id: id, ...restObject } = document.toObject({