fix(sloth-entity): returns type constructor

compactd · Apr 24, 2018 · f41833d · f41833d
1 parent 4d29c90
commit f41833d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/decorators/SlothEntity.ts b/src/decorators/SlothEntity.ts
@@ -35,11 +35,13 @@ function mapPropsOrDocToDocument({ fields }: ProtoData, data: any) {
 export default function SlothEntity<S extends { _id: string }>(name: string) {
   return <T extends BaseEntity<S>>(constructor: {
     new (factory: PouchFactory<S>, idOrProps: Partial<S> | string): T
-  }) => {
+  }): EntityConstructor<S, T> => {
     const data = getProtoData(constructor.prototype, true)
 
     data.name = name
+
     const BaseEntity = constructor as EntityConstructor<any, any>
+
     return class WrappedEntity extends BaseEntity {
       constructor(factory: PouchFactory<S>, idOrProps: Partial<S> | string) {
         super(factory, idOrProps)
@@ -48,6 +50,6 @@ export default function SlothEntity<S extends { _id: string }>(name: string) {
           idOrProps
         )
       }
-    }
+    } as any
   }
 }