barend-erasmus/web-push-service

import { IClientRepository } from '../interfaces/client-repository';
import { Client } from '../models/client';
import * as mongodb from 'mongodb';

export class MongoClientRepository implements IClientRepository {
  protected static client: mongodb.MongoClient = null;

  protected static database: mongodb.Collection = null;

  constructor(protected host: string) {
    if (!MongoClientRepository.database) {
      this.initialize();
    }
  }

  public close(): void {
    MongoClientRepository.client.close();
  }

  public async find(key: string): Promise<Client> {
    const document: any = await MongoClientRepository.database.findOne({ key });

    if (!document) {
      return null;
    }

    return new Client(document.id, document.key, document.publicKey, document.privateKey, document.endpoint);
  }

  public async findById(id: string): Promise<Client> {
    const document: any = await MongoClientRepository.database.findOne({ id });

    if (!document) {
      return null;
    }

    return new Client(document.id, document.key, document.publicKey, document.privateKey, document.endpoint);
  }

  public async findByPublicKey(publicKey: string): Promise<Client> {
    const document: any = await MongoClientRepository.database.findOne({ publicKey });

    if (!document) {
      return null;
    }

    return new Client(document.id, document.key, document.publicKey, document.privateKey, document.endpoint);
  }

  public async insert(client: Client): Promise<void> {
    await MongoClientRepository.database.insert(client);
  }

  public wait(): Promise<void> {
    return new Promise((resolve: () => void) => {
      setTimeout(resolve, 1100);
    });
  }

  protected async initialize(): Promise<void> {
    MongoClientRepository.client = await mongodb.connect(
      this.host,
      { useNewUrlParser: true },
    );

    const database: mongodb.Db = MongoClientRepository.client.db('web-push-service');

    MongoClientRepository.database = database.collection('clients');
  }
}