import fs from 'fs';
import { MongoClient, MongoClientOptions } from 'mongodb';
import { ClientEncryption } from 'mongodb-client-encryption';

process.on('uncaughtException', (ex) => {
  console.error('Uncaught Exception: %s', ex.message);
});

process.on('unhandledRejection', (re) => {
  console.error('Unahandled Rejection: %s', re);
  process.exit(1);
});

(async () => {
  const uri = 'mongodb://test:test@172.17.0.2:27017?poolSize=2&authSource=admin';
  console.log('URI: %s', uri);  
  const masterKey = fs.readFileSync('master-key.txt');
  console.log('Master Key Read');  
  const client = new MongoClient(uri, {
    appname: 'mongenc-plain',
    useUnifiedTopology: true,
    useNewUrlParser: true,
  } as MongoClientOptions);
  console.log('Try to connect');
  await client.connect();
  console.log('Client connected');  
  const dbName = 'test';
  const collName = 'col';
  const keyVaultNamespace = `${dbName}.datakeys`;
  const kmsProviders = {
    local: {
      key: masterKey,
    },
  };
  // Create new ClientEncryption.
  const clientEncryption = new ClientEncryption(client, {
    kmsProviders,
    keyVaultNamespace,
  })  
  // Check for mySpecialKey.
  const db = client.db(dbName);
  const col = db.collection('datakeys');
  const resultDK = await col.find({ keyAltNames: 'mySpecialKey'}).limit(1).toArray();
  const dataKeyId = (typeof resultDK[0] !== 'undefined')
    ? resultDK[0]._id
    : await clientEncryption.createDataKey('local', { keyAltNames: ['mySpecialKey'] });  
  const schemaMap = {
    [`${dbName}.${collName}`]: {
      bsonType: 'object',
      properties: {
        encryptedField: {
          encrypt: {
            keyId: [ dataKeyId ],
            bsonType: 'string',
            algorithm: 'AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic'
          }
        }
      }
    }
  };  
  const encryptedClient = new MongoClient(uri, {
    appname: 'mongenc-secure',
    useUnifiedTopology: true,
    autoEncryption: {
      keyVaultNamespace,
      kmsProviders,
      schemaMap,
      extraOptions: {
        mongocryptdSpawnPath: './bin/mongocryptd',
        mongocryptdSpawnArgs: ['--nounixsocket'],
      },
    },
  } as any);  
  await encryptedClient.connect();
  console.log('Encrypted client connect.');  
  const dbSecure= encryptedClient.db(dbName);
  const collectionSecure = dbSecure.collection(collName);  
  const sampleString = new Date().toISOString();
  console.log('Sample String:', sampleString);  
  await collectionSecure.insertOne({ encryptedField: sampleString });
  const result = await collectionSecure.findOne({
    encryptedField: { $eq: sampleString },
  });
  console.log(result);  

  await encryptedClient.close();
  await client.close();
  console.log('Close client');
})();