76 lines
2.2 KiB
TypeScript
76 lines
2.2 KiB
TypeScript
import { createDBClient } from '../db/createClient';
|
|
import { Maybe } from '@/util/helpers';
|
|
import { ParseParams, SafeParseReturnType } from 'zod';
|
|
import { MongoClient, WithId, Filter, InsertOneResult } from 'mongodb';
|
|
|
|
type FullParserType<T extends { [key: string]: any }> = (data: any, params?: Partial<ParseParams> | undefined) => SafeParseReturnType<any, T>
|
|
|
|
type ControllerOptions<T extends { [key: string]: any }> = {
|
|
tableName: string
|
|
parser?: FullParserType<T>
|
|
}
|
|
|
|
export default abstract class BaseController<T extends { _id?: any, [key: string]: any }> {
|
|
protected client: MongoClient
|
|
collectionName: string
|
|
parser?: FullParserType<T>
|
|
|
|
constructor(options: ControllerOptions<T>) {
|
|
this.collectionName = options.tableName;
|
|
this.client = createDBClient();
|
|
this.parser = options.parser;
|
|
}
|
|
|
|
async getAll() {
|
|
let result: Maybe<WithId<T>[]>;
|
|
|
|
try {
|
|
// we'll enable cache here later
|
|
await this.client.connect();
|
|
result = await this.client.db().collection<T>(this.collectionName)
|
|
.find()
|
|
.toArray();
|
|
|
|
return result;
|
|
} catch (error) {
|
|
console.log({ error });
|
|
result = null;
|
|
} finally {
|
|
await this.client.close();
|
|
return result;
|
|
}
|
|
}
|
|
|
|
async getByID(id: string): Promise<Maybe<WithId<T>>> {
|
|
let result: Maybe<WithId<T>>;
|
|
|
|
try {
|
|
await this.client.connect();
|
|
result = await this.client.db().collection<T>(this.collectionName)
|
|
.findOne({ where: { _id: id }});
|
|
} catch (error) {
|
|
console.log({ error });
|
|
result = null;
|
|
} finally {
|
|
await this.client.close();
|
|
return result;
|
|
}
|
|
}
|
|
|
|
async post(data: T) {
|
|
let result: Maybe<InsertOneResult<T>>;
|
|
|
|
try {
|
|
await this.client.connect();
|
|
result = await this.client.db().collection<T>(this.collectionName)
|
|
.insertOne(data as any);
|
|
} catch(error) {
|
|
console.log({ error });
|
|
result = null;
|
|
} finally {
|
|
await this.client.close();
|
|
return result;
|
|
}
|
|
}
|
|
}
|