refactor: refactor types

src/database.ts

@@ -7,6 +7,7 @@ import Schema from './schema';
 import SchemaType from './schematype';
 import WarehouseError from './error';
 import { logger } from 'hexo-log';
+import type { NodeJSLikeCallback } from './types';
 
 const log = logger();
 const pkg = require('../package.json');
@@ -23,7 +24,7 @@ if (typeof writev === 'function') {
   };
 }
 
-async function exportAsync(database: Database, path: string) {
+async function exportAsync(database: Database, path: string): Promise<void> {
   const handle = await open(path, 'w');
 
   try {
@@ -79,7 +80,7 @@ interface DatabaseOptions {
 
 class Database {
   options: DatabaseOptions;
-  _models: any;
+  _models: Record<string, Model>;
   Model: typeof Model;
 
   /**
@@ -117,7 +118,7 @@ class Database {
    * @param {Schema|object} [schema]
    * @return {Model}
    */
-  model(name: string, schema?: any) {
+  model(name: string, schema?: Schema | object): Model {
     if (this._models[name]) {
       return this._models[name];
     }
@@ -133,7 +134,7 @@ class Database {
    * @param {function} [callback]
    * @return {Promise}
    */
-  load(callback?) {
+  load(callback?: NodeJSLikeCallback<any>): Bluebird<any> {
     const { path, onUpgrade, onDowngrade, version: newVersion } = this.options;
 
     if (!path) throw new WarehouseError('options.path is required');
@@ -173,14 +174,14 @@ class Database {
    * @param {function} [callback]
    * @return {Promise}
    */
-  save(callback?) {
+  save(callback?: NodeJSLikeCallback<any>): Bluebird<void> {
     const { path } = this.options;
 
     if (!path) throw new WarehouseError('options.path is required');
     return Bluebird.resolve(exportAsync(this, path)).asCallback(callback);
   }
 
-  toJSON() {
+  toJSON(): { meta: { version: number, warehouse: string }, models: Record<string, Model> } {
     const models = Object.keys(this._models)
       .reduce((obj, key) => {
         const value = this._models[key];

src/document.ts

@@ -1,17 +1,21 @@
 import rfdc from 'rfdc';
+import type Model from './model';
+import type Schema from './schema';
+import type { NodeJSLikeCallback } from './types';
 const cloneDeep = rfdc();
 
 abstract class Document {
-  abstract _model;
-  _id!: any;
-  abstract _schema;
+  abstract _model: Model;
+  _id!: string | number | undefined;
+  abstract _schema: Schema;
+  [key: PropertyKey]: any;
 
   /**
    * Document constructor.
    *
    * @param {object} data
    */
-  constructor(data) {
+  constructor(data?: object) {
     if (data) {
       Object.assign(this, data);
     }
@@ -23,7 +27,7 @@ abstract class Document {
    * @param {function} [callback]
    * @return {Promise}
    */
-  save(callback) {
+  save(callback?: NodeJSLikeCallback<any>): Promise<any> {
     return this._model.save(this, callback);
   }
 
@@ -34,7 +38,7 @@ abstract class Document {
    * @param {function} [callback]
    * @return {Promise}
    */
-  update(data, callback) {
+  update(data: Record<string, any>, callback?: NodeJSLikeCallback<any>): Promise<any> {
     return this._model.updateById(this._id, data, callback);
   }
 
@@ -45,7 +49,7 @@ abstract class Document {
    * @param {function} [callback]
    * @return {Promise}
    */
-  replace(data, callback) {
+  replace(data: object | Document, callback?: NodeJSLikeCallback<any>): Promise<any> {
     return this._model.replaceById(this._id, data, callback);
   }
 
@@ -55,7 +59,7 @@ abstract class Document {
    * @param {function} [callback]
    * @return {Promise}
    */
-  remove(callback) {
+  remove(callback?: NodeJSLikeCallback<any>): Promise<any> {
     return this._model.removeById(this._id, callback);
   }
 
@@ -64,7 +68,7 @@ abstract class Document {
    *
    * @return {object}
    */
-  toObject() {
+  toObject(): object {
     const keys = Object.keys(this);
     const obj = {};
 
@@ -83,7 +87,7 @@ abstract class Document {
    *
    * @return {String}
    */
-  toString() {
+  toString(): string {
     return JSON.stringify(this);
   }
 
@@ -93,13 +97,13 @@ abstract class Document {
    * @param {String|Object} expr
    * @return {Document}
    */
-  populate(expr) {
+  populate(expr: string | any[] | { path?: string; model?: any; [key: PropertyKey]: any }): Document {
     const stack = this._schema._parsePopulate(expr);
     return this._model._populate(this, stack);
   }
 }
 
-function isGetter(obj, key) {
+function isGetter(obj: any, key: PropertyKey): any {
   return Object.getOwnPropertyDescriptor(obj, key).get;
 }
 

src/lib/jsonstream/index.ts

@@ -7,7 +7,7 @@ import Parser from 'jsonparse';
  * @param {*} y
  * @returns {boolean}
  */
-const check = (x, y) => {
+const check = (x, y): boolean => {
   if (typeof x === 'string') {
     return y === x;
   }
@@ -27,7 +27,7 @@ const check = (x, y) => {
   return false;
 };
 
-export function parse(path, map = null) {
+export function parse(path: string | any[], map = null) {
   let header, footer;
 
   const parser = new Parser();