core Data typings

src/core/Data.ts

@@ -1,17 +1,18 @@
-const db = require('./db')
+import { Collection } from 'mongodb'
+import * as db from './db'
 
-const store = new Map()
+const store = new Map<string, Collection>()
 
 export default class Data {
-    static setStore(Class, collection) {
+    static setStore(key: string, collection: Collection) {
-        store.set(Class, collection)
+        store.set(key, collection)
     }
 
-    static getStore(Class) {
+    static getStore(key: string): Collection {
-        return store.get(Class)
+        return store.get(key)
     }
 
-    static getCollection(name) {
+    static getCollection(name: string): Collection {
         return db.collection(name)
     }
 }

src/core/Page.ts

@@ -3,7 +3,7 @@ import { find as storeFind, findOne, total } from '../lib/store'
 const Image = require('./Image')
 const Config = require('./Config')
 
-const store = () => Data.getStore(Page)
+const store = () => Data.getStore('Page')
 
 export default class Page {
     static async find(query, options = {}): Promise<Page[]> {