🗃 Enable pgcrypto extension

Signed-off-by: SAMI BETTAYEB <[email protected]>

Author: SAMIBETTAYEB

lib/migration.js

@@ -351,7 +351,8 @@ function mixinMigration(PostgreSQL) {
     });
     // default extension
     if (!createExtensions) {
-      createExtensions = 'CREATE EXTENSION IF NOT EXISTS "uuid-ossp";';
+      createExtensions = `CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
+      CREATE EXTENSION IF NOT EXISTS "pgcrypto";`;
     }
 
     // Please note IF NOT EXISTS is introduced in postgresql v9.3

lib/postgresql.js

@@ -112,6 +112,7 @@ PostgreSQL.prototype.connect = function(callback) {
     self.client = client;
     process.nextTick(releaseCb);
     callback && callback(err, client);
+    if (!err) self.execute('CREATE EXTENSION IF NOT EXISTS pgcrypto', function(createExtensionError) {});
   });
 };
 
@@ -589,15 +590,15 @@ PostgreSQL.prototype.buildWhere = function(model, where) {
 };
 
 PostgreSQL.prototype.getEncryptionFields = function(modelDefinition) {
-  if(modelDefinition 
-    && modelDefinition.settings 
-    && modelDefinition.settings.mixins 
-    && modelDefinition.settings.mixins.Encryption 
-    && modelDefinition.settings.mixins.Encryption.fields){
-      return modelDefinition.settings.mixins.Encryption.fields
-    }
-  return []
-}
+  if (modelDefinition
+    && modelDefinition.settings
+    && modelDefinition.settings.mixins
+    && modelDefinition.settings.mixins.Encryption
+    && modelDefinition.settings.mixins.Encryption.fields) {
+    return modelDefinition.settings.mixins.Encryption.fields;
+  }
+  return [];
+};
 
 /**
  * @private
@@ -617,7 +618,7 @@ PostgreSQL.prototype._buildWhere = function(model, where) {
   const self = this;
   const props = self.getModelDefinition(model).properties;
 
-  const encryptedFields = this.getEncryptionFields(this.getModelDefinition(model))
+  const encryptedFields = this.getEncryptionFields(this.getModelDefinition(model));
   const whereStmts = [];
   for (const key in where) {
     const stmt = new ParameterizedSQL('', []);
@@ -659,16 +660,16 @@ PostgreSQL.prototype._buildWhere = function(model, where) {
     // eslint-disable one-var
     let expression = where[key];
     let columnName = self.columnEscaped(model, key);
-    if(encryptedFields.includes(key)){
+    if (encryptedFields.includes(key)) {
       columnName = `convert_from(
         decrypt_iv(
           DECODE(${key},'hex')::bytea,
           decode('${process.env.ENCRYPTION_HEX_KEY}','hex')::bytea,
-          decode('${process.env.ENCRYPTION_HEX_IV}','hex')::bytea, 
+          decode('${process.env.ENCRYPTION_HEX_IV}','hex')::bytea,
           'aes'
         ),
         'utf8'
-      )`
+      )`;
     }
     // eslint-enable one-var
     if (expression === null || expression === undefined) {